import { thoriumContext } from "@thorium/utils/.server/context";
import { createMockDataContext } from "@thorium/utils/.server/createMockDataContext";
import { testDataStoreProps } from "@thorium/utils/.server/db-fs/testDataStoreProps";
import {
	calculateShipMapPath,
	createShipMapGraph,
} from "@thorium/utils/.server/ship/shipMapPathfinder";
import { ECS, Entity } from "@thorium/utils/ecs";
import { aroundEach, beforeEach, describe, expect, it } from "vitest";

import { PassengerMovementSystem } from "../PassengerMovementSystem";

aroundEach(async (runTest) => {
	await thoriumContext.run(testDataStoreProps, async () => {
		await runTest();
	});
});

describe("PassengerMovementSystem", () => {
	let ecs: ECS;
	let ship: Entity;
	beforeEach(() => {
		const mockDataContext = createMockDataContext();

		ecs = new ECS(mockDataContext.server);
		ecs.addSystem(new PassengerMovementSystem());
		ship = new Entity();
		ship.addComponent("shipMap", {
			decks: [
				{
					name: "Deck 1",
					backgroundUrl: "",
				},
				{
					name: "Deck 2",
					backgroundUrl: "",
				},
			],
			deckNodes: [
				{
					id: 1,
					deckIndex: 0,
					x: 0,
					y: 0,
					contents: {},
					volume: 12000,
					flags: [],
					systems: [],
				},
				{
					id: 2,
					deckIndex: 0,
					x: 1,
					y: 1,
					contents: {},
					volume: 12000,
					flags: [],
					systems: [],
				},
				{
					id: 3,
					deckIndex: 0,
					x: 2,
					y: 1,
					contents: {},
					volume: 12000,
					flags: [],
					systems: [],
				},
				{
					id: 4,
					deckIndex: 1,
					x: 2,
					y: 1,
					contents: {},
					volume: 12000,
					flags: [],
					systems: [],
				},
				{
					id: 5,
					deckIndex: 1,
					x: 1,
					y: 0,
					contents: {},
					volume: 12000,
					flags: [],
					systems: [],
				},
			],
			deckEdges: [
				{
					id: 1,
					from: 1,
					to: 2,
				},
				{
					id: 2,
					from: 2,
					to: 3,
				},
				{
					id: 2,
					from: 3,
					to: 4,
				},
				{
					id: 2,
					from: 4,
					to: 5,
				},
			],
		});
		ecs.addEntity(ship);
	});
	it("should move a passenger within the same deck", () => {
		let updateTime = 0;
		function update(updateCount = 1) {
			for (let i = 20 * updateCount; i > 0; i--) {
				ecs.update(16);
				updateTime += 16;
			}
		}
		const passenger = new Entity();
		passenger.addComponent("position", {
			x: 0,
			y: 0,
			z: 0,
			parentId: ship.id,
			type: "ship",
		});
		passenger.addComponent("passengerMovement");
		ecs.addEntity(passenger);
		ecs.update(16);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 1,
        "type": "ship",
        "x": 0,
        "y": 0,
        "z": 0,
      }
    `);
		const graph = createShipMapGraph(ship.components.shipMap?.deckEdges || []);
		const nodePath = calculateShipMapPath(graph, 1, 3);
		if (!nodePath) throw new Error("unable to calculate node path");
		passenger.updateComponent("passengerMovement", {
			nodePath,
			nextNodeIndex: 0,
		});
		update(2);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 1,
        "type": "ship",
        "x": 0.44215339876352006,
        "y": 0.44215339876352006,
        "z": 0,
      }
    `);
		expect(passenger.components.passengerMovement?.nextNodeIndex).toBe(1);
		update(7);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 1,
        "type": "ship",
        "x": 1.0995775896953266,
        "y": 0.9459775896953266,
        "z": 0,
      }
    `);
		expect(passenger.components.passengerMovement?.nextNodeIndex).toBe(2);
		update(8);
		const oldX = passenger.components.position?.x;
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 1,
        "type": "ship",
        "x": 2,
        "y": 1,
        "z": 0,
      }
    `);
		expect(passenger.components.passengerMovement?.nextNodeIndex).toBe(0);
		expect(updateTime).toMatchInlineSnapshot(`5440`);
		update();
		expect(passenger.components.position?.x).toBe(oldX);
	});
	it("should move a passenger between decks", () => {
		let updateTime = 0;
		function update(updateCount = 1) {
			for (let i = 20 * updateCount; i > 0; i--) {
				ecs.update(16);
				updateTime += 16;
			}
		}

		const passenger = new Entity();
		passenger.addComponent("position", {
			x: 2,
			y: 1,
			z: 0,
			parentId: ship.id,
			type: "ship",
		});
		passenger.addComponent("passengerMovement");
		ecs.addEntity(passenger);
		ecs.update(16);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 3,
        "type": "ship",
        "x": 2,
        "y": 1,
        "z": 0,
      }
    `);
		const graph = createShipMapGraph(ship.components.shipMap?.deckEdges || []);
		const nodePath = calculateShipMapPath(graph, 3, 5);
		if (!nodePath) throw new Error("unable to calculate node path");
		passenger.updateComponent("passengerMovement", {
			nodePath,
			nextNodeIndex: 0,
		});
		update(2);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 3,
        "type": "ship",
        "x": 2,
        "y": 1,
        "z": 0.16799999999999998,
      }
    `);
		expect(passenger.components.passengerMovement?.nextNodeIndex).toBe(1);
		update(22);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 3,
        "type": "ship",
        "x": 1,
        "y": 0,
        "z": 1,
      }
    `);
		update(4);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 3,
        "type": "ship",
        "x": 1,
        "y": 0,
        "z": 1,
      }
    `);
		update(4);
		expect(passenger.components.position).toMatchInlineSnapshot(`
      {
        "parentId": 3,
        "type": "ship",
        "x": 1,
        "y": 0,
        "z": 1,
      }
    `);
		expect(updateTime).toMatchInlineSnapshot(`10240`);
	});
});