import { GridCellKind, isTextEditableGridCell } from "../src/internal/data-grid/data-grid-types.js";
import { CompactSelection, isEditableGridCell } from "../src/index.js";
import { sprites } from "../src/internal/data-grid/sprites.js";
import { expect, describe, test } from "vitest";

describe("data-grid-types", () => {
    test("Smoke test type checks", () => {
        expect(
            isEditableGridCell({
                kind: GridCellKind.Loading,
                allowOverlay: false,
            })
        ).toBe(false);

        expect(
            isEditableGridCell({
                kind: GridCellKind.Text,
                data: "Test",
                displayData: "Test",
                allowOverlay: false,
            })
        ).toBe(true);

        expect(
            isTextEditableGridCell({
                kind: GridCellKind.Loading,
                allowOverlay: false,
            })
        ).toBe(false);

        expect(
            isTextEditableGridCell({
                kind: GridCellKind.Text,
                data: "Test",
                displayData: "Test",
                allowOverlay: false,
            })
        ).toBe(true);
    });

    test("Smoke test compact selection", () => {
        const sel = CompactSelection.fromSingleSelection([3, 5]);

        expect(sel.length).toBe(2);

        expect(sel.first()).toBe(3);

        expect(sel.last()).toBe(4);

        expect(sel.offset(2).first()).toBe(5);

        expect(sel.hasAll([1, 4])).toBe(false);
        expect(sel.hasAll([3, 4])).toBe(true);

        expect(sel.remove(4).length).toBe(1);

        expect([...sel]).toEqual([3, 4]);
        expect([...sel.add([4, 9]).remove(5)]).toEqual([3, 4, 6, 7, 8]);

        expect(CompactSelection.empty().first()).toBeUndefined();
        expect(CompactSelection.empty().last()).toBeUndefined();
        expect(CompactSelection.empty().add(5).length).toBe(1);
    });

    test("Compact selection equals", () => {
        const sel = CompactSelection.fromSingleSelection([3, 5]);
        const other = CompactSelection.fromSingleSelection([3, 5]);

        expect(sel.equals(other)).toBe(true);
        expect(sel.equals(sel)).toBe(true);
        expect(sel.equals(CompactSelection.fromSingleSelection([3, 6]))).toBe(false);
        expect(sel.equals(CompactSelection.fromSingleSelection([4, 6]))).toBe(false);

        expect(sel.some(x => x > 3)).toBe(true);
    });

    test("Compact selection toArray", () => {
        const sel = CompactSelection.fromSingleSelection([3, 5]);
        expect(sel.toArray()).toEqual([3, 4]);
    });

    test("Smoke test compact selection remove", () => {
        const sel = CompactSelection.fromSingleSelection([3, 8]);

        expect([...sel]).toEqual([3, 4, 5, 6, 7]);

        // Remove entire selection

        expect([...sel.remove([3, 8])]).toEqual([]);
        expect(sel.remove([3, 8]).length).toBe(0);

        expect([...sel.remove([2, 9])]).toEqual([]);
        expect(sel.remove([2, 9]).length).toBe(0);

        // Remove ends of selection

        expect([...sel.remove([2, 6])]).toEqual([6, 7]);
        expect(sel.remove([2, 6]).length).toBe(2);

        expect([...sel.remove([5, 9])]).toEqual([3, 4]);
        expect(sel.remove([5, 9]).length).toBe(2);

        expect([...sel.remove([2, 3])]).toEqual([3, 4, 5, 6, 7]);
        expect(sel.remove([2, 3]).length).toBe(5);

        expect([...sel.remove([8, 9])]).toEqual([3, 4, 5, 6, 7]);
        expect(sel.remove([8, 9]).length).toBe(5);

        expect([...sel.remove(3)]).toEqual([4, 5, 6, 7]);
        expect(sel.remove(3).length).toBe(4);

        expect([...sel.remove(7)]).toEqual([3, 4, 5, 6]);
        expect(sel.remove(7).length).toBe(4);

        // Remove end and start of 2 different slices
        const altSel = CompactSelection.fromSingleSelection([0, 5]).add([8, 12]);
        expect([...altSel]).toEqual([0, 1, 2, 3, 4, 8, 9, 10, 11]);
        expect([...altSel.remove([3, 10])]).toEqual([0, 1, 2, 10, 11]);

        // Remove middle of selection

        expect([...sel.remove([4, 7])]).toEqual([3, 7]);
        expect(sel.remove([4, 7]).length).toBe(2);

        expect([...sel.remove(5)]).toEqual([3, 4, 6, 7]);
        expect(sel.remove(5).length).toBe(4);

        // Remove nothing from selection

        expect([...sel.remove([1, 2])]).toEqual([3, 4, 5, 6, 7]);
        expect(sel.remove([1, 2]).length).toBe(5);

        expect([...sel.remove([8, 9])]).toEqual([3, 4, 5, 6, 7]);
        expect(sel.remove([8, 9]).length).toBe(5);
    });

    test("Load sprites", () => {
        const keys = Object.keys(sprites) as (keyof typeof sprites)[];
        for (const k of keys) {
            const cb = sprites[k];
            cb({
                bgColor: "red",
                fgColor: "blue",
            });
        }
    });
});