import { single } from "./initial_states/single.js";
import { two } from "./initial_states/two.js";
import { three } from "./initial_states/three.js";
import { four } from "./initial_states/four.js";
import { five } from "./initial_states/five.js";
import { head_on } from "./initial_states/head_on.js";
import { cross_two } from "./initial_states/cross_two.js";
import { cross_three } from "./initial_states/cross_three.js";
import {
  get_moves,
  block_is_movable,
  move_block,
  move_state_block,
  arrays_are_equal,
} from "./state.js";
import { DIRECTIONS, LEFT, RIGHT, UP, DOWN } from "../../constants.js";
import { HORIZONTAL, VERTICAL } from "../../constants.js";
import { key_of } from "../../main.js";

const initial_states = [
  single,
  two,
  three,
  four,
  // five,
  head_on,
  cross_two,
  cross_three,
];

const generate = (states, initial_state, graph, previous_move) => {
  const stack = [[initial_state, previous_move]];

  let last_print = 0;

  let data = graph.graphData();

  while (stack.length > 0) {
    const [current_state, prev_move] = stack.pop();

    const moves = get_moves(prev_move, current_state);
    for (const m of moves) {
      const [block, direction] = m;

      const states_before = states.length;
      const [new_state, new_block] = move(
        states,
        data,
        current_state,
        block,
        direction,
      );
      const states_after = states.length;

      if (states_after - last_print > 250) {
        console.log(`Generating: Found ${states_after} states...`);
        last_print = states_after;
      }

      if (states_after > states_before) {
        stack.push([new_state, m]);
      }
    }
  }

  graph.graphData(data);
};

const select = (state, offsetX, offsetY) => {
  const canvas = document.getElementById("model_canvas");

  const square_width = canvas.width / state.width;
  const square_height = canvas.height / state.height;

  const x = Math.floor(offsetX / square_width);
  const y = Math.floor(offsetY / square_height);

  for (let i = 0; i < state.board.length; i++) {
    const [id, dir, x0, y0, x1, y1] = state.board[i];

    if (x >= x0 && x <= x1 && y >= y0 && y <= y1) {
      return [id, dir, x0, y0, x1, y1];
    }
  }

  return null;
};

const move = (states, data, state, block, direction) => {
  if (
    block[1] === HORIZONTAL &&
    (arrays_are_equal(direction, DIRECTIONS[UP]) ||
      arrays_are_equal(direction, DIRECTIONS[DOWN]))
  ) {
    console.log("Can't move block horizontally");
    return [null, null];
  }

  if (
    block[1] === VERTICAL &&
    (arrays_are_equal(direction, DIRECTIONS[LEFT]) ||
      arrays_are_equal(direction, DIRECTIONS[RIGHT]))
  ) {
    console.log("Can't move block vertically");
    return [null, null];
  }

  if (!block_is_movable(state, block, direction)) {
    return [null, null];
  }

  let new_state = structuredClone(state);
  delete new_state.name;
  let new_block = move_block(block, direction);
  move_state_block(new_state, block, direction);

  const new_link = {
    source: key_of(state), // We're coming from this state...
    target: key_of(new_state), // ...and ended up here, at a previous state.
  };

  let new_node = null;
  if (!states.has(new_state)) {
    states.add(new_state);
    new_node = {
      id: key_of(new_state),
    };
  }

  data.links.push(new_link);
  if (new_node !== null) {
    data.nodes.push(new_node);
  }

  return [new_state, new_block];
};

const rect = (context, x0, y0, x1, y1, square_width, square_height, color) => {
  const x = x0 * square_width;
  const y = y0 * square_height;
  const width = (x1 - x0 + 1) * square_width;
  const height = (y1 - y0 + 1) * square_height;

  context.fillStyle = color;
  context.fillRect(x, y, width, height);

  context.strokeStyle = "#000000";
  context.lineWidth = 1;
  context.strokeRect(x, y, width, height);
};

const visualize = (state) => {
  const canvas = document.getElementById("model_canvas");
  const context = canvas.getContext("2d");

  const square_width = canvas.width / state.width;
  const square_height = canvas.height / state.height;

  // console.log(`Canvas: (${canvas.width}x${canvas.height})`);
  // console.log(`Klotski 1x1 Size: (${square_width}x${square_height})`);

  for (let i = 0; i < state.board.length; i++) {
    const [id, dir, x0, y0, x1, y1] = state.board[i];

    rect(context, x0, y0, x1, y1, square_width, square_height, "#555555");
  }
};

const highlight = (state, block) => {
  const [id, dir, x0, y0, x1, y1] = block;

  const canvas = document.getElementById("model_canvas");
  const context = canvas.getContext("2d");

  const square_width = canvas.width / state.width;
  const square_height = canvas.height / state.height;

  rect(context, x0, y0, x1, y1, square_width, square_height, "#AAAAAA");
};

const visualize_path = (from_state, to_state) => {
  // Find path (general graph helper function)
  // For each state in path: visualize(state)
};

const add_block = (state, x0, y0, x1, y1) => {
  let dir;
  if (x1 - x0 > y1 - y0) {
    dir = HORIZONTAL;
  } else if (x1 - x0 < y1 - y0) {
    dir = VERTICAL;
  } else {
    // No square blocks for this model
    return null;
  }

  let new_block = [state.board.length, dir, x0, y0, x1, y1];
  state.board.push(new_block);
  return new_block;
};

export const restricted_sliding_blocks_model = {
  name: "Restricted Sliding Blocks",
  generate,
  visualize,
  highlight,
  visualize_path,
  initial_states,
  select,
  move,
  empty_state: {
    name: "Custom",
    width: 5,
    height: 5,
    board: [],
  },
  add_block,
};