import { assertEquals } from "@std/assert";
import {
  CallGraph,
  moduleToDbSchema,
  SecretGraph,
  validateEdge,
  validateNode,
} from "../mod.ts";
import type { TModule, TProperties } from "@alkdev/typebox";

Deno.test("CallGraph has required entries", () => {
  const entries = Object.keys(
    (CallGraph.Import("Config" as never).$defs as Record<string, unknown>),
  );
  const required = [
    "Config",
    "CallNode",
    "SubcallNode",
    "TriggeredEdge",
    "DependsOnEdge",
    "TriggeredEdgeConstraints",
    "DependsOnEdgeConstraints",
    "CallStatus",
    "Identity",
  ];
  for (const name of required) {
    assertEquals(entries.includes(name), true, `Missing entry: ${name}`);
  }
});

Deno.test("CallGraph.Config uses literal values", () => {
  const configImport = CallGraph.Import("Config" as never);
  const configSchema = configImport.$defs.Config as Record<string, unknown>;
  const properties = configSchema.properties as Record<string, unknown>;

  const typeProp = properties.type as Record<string, unknown>;
  assertEquals(typeProp.const, "directed");

  const multiProp = properties.multi as Record<string, unknown>;
  assertEquals(multiProp.const, false);

  const allowSelfLoopsProp = properties.allowSelfLoops as Record<string, unknown>;
  assertEquals(allowSelfLoopsProp.const, false);
});

Deno.test("validateNode with CallGraph CallNode returns true for valid data", () => {
  const result = validateNode(
    CallGraph as unknown as TModule<TProperties>,
    "CallNode",
    {
      requestId: "req-001",
      operationId: "op-001",
      status: "pending",
      input: {},
    },
  );
  assertEquals(result, true);
});

Deno.test("validateNode with CallGraph CallNode returns false for invalid data", () => {
  const result = validateNode(
    CallGraph as unknown as TModule<TProperties>,
    "CallNode",
    {
      requestId: "req-001",
    },
  );
  assertEquals(result, false);
});

Deno.test("moduleToDbSchema with CallGraph produces valid DbSchema", () => {
  const schema = moduleToDbSchema(
    CallGraph as unknown as TModule<TProperties>,
    "call-graph",
  );

  assertEquals(schema.graphType.name, "call-graph");
  assertEquals(schema.graphType.config.properties !== undefined, true);

  assertEquals(schema.nodeTypes.length, 2);
  assertEquals(schema.nodeTypes[0].name, "Call");
  assertEquals(schema.nodeTypes[1].name, "Subcall");

  assertEquals(schema.edgeTypes.length, 2);
  assertEquals(schema.edgeTypes[0].name, "Triggered");
  assertEquals(schema.edgeTypes[1].name, "DependsOn");
});

Deno.test("SecretGraph has required entries", () => {
  const entries = Object.keys(
    (SecretGraph.Import("Config" as never).$defs as Record<string, unknown>),
  );
  const required = [
    "Config",
    "SecretNode",
    "ClientNode",
    "HasSecretEdge",
    "HasSecretEdgeConstraints",
  ];
  for (const name of required) {
    assertEquals(entries.includes(name), true, `Missing entry: ${name}`);
  }
});

Deno.test("SecretGraph.Config uses literal values", () => {
  const configImport = SecretGraph.Import("Config" as never);
  const configSchema = configImport.$defs.Config as Record<string, unknown>;
  const properties = configSchema.properties as Record<string, unknown>;

  const typeProp = properties.type as Record<string, unknown>;
  assertEquals(typeProp.const, "undirected");

  const multiProp = properties.multi as Record<string, unknown>;
  assertEquals(multiProp.const, false);

  const allowSelfLoopsProp = properties.allowSelfLoops as Record<string, unknown>;
  assertEquals(allowSelfLoopsProp.const, false);
});

Deno.test("moduleToDbSchema with SecretGraph produces valid DbSchema", () => {
  const schema = moduleToDbSchema(
    SecretGraph as unknown as TModule<TProperties>,
    "secret-graph",
  );

  assertEquals(schema.graphType.name, "secret-graph");

  assertEquals(schema.nodeTypes.length, 2);
  assertEquals(schema.nodeTypes[0].name, "Secret");
  assertEquals(schema.nodeTypes[1].name, "Client");

  assertEquals(schema.edgeTypes.length, 1);
  assertEquals(schema.edgeTypes[0].name, "HasSecret");
});

Deno.test("validateNode with SecretGraph SecretNode validates encrypted data", () => {
  const result = validateNode(
    SecretGraph as unknown as TModule<TProperties>,
    "SecretNode",
    {
      key: "api_key",
      encryptedData: {
        keyVersion: 1,
        salt: "dGVzdA==",
        iv: "dGVzdA==dGVzdA==",
        data: "dGVzdA==",
      },
    },
  );
  assertEquals(result, true);
});

Deno.test("validateEdge with CallGraph TriggeredEdge returns true for valid data", () => {
  const result = validateEdge(
    CallGraph as unknown as TModule<TProperties>,
    "TriggeredEdge",
    {
      type: "triggered",
    },
  );
  assertEquals(result, true);
});

Deno.test("validateEdge with CallGraph TriggeredEdge returns false for wrong type", () => {
  const result = validateEdge(
    CallGraph as unknown as TModule<TProperties>,
    "TriggeredEdge",
    {
      type: "depends_on",
    },
  );
  assertEquals(result, false);
});