import { z } from "zod";

import { AnalysisInputError } from "../../domain/analysisErrorCore.js";
import type { AnalysisError } from "../../domain/analysisErrorBase.js";
import {
  applicationGraphEvidenceInputError,
  parseApplicationGraphEvidence,
} from "./JavaScriptApplicationEvidenceGraph.js";
import type { Evidence } from "../../domain/evidence.js";
import { queryJavaScriptSemanticGraphValidated } from "../../domain/javascript/javascriptSemanticQuery.js";
import {
  traceJavaScriptSemanticsInputSchema,
  type JavaScriptSemanticTraceResult,
} from "../../domain/javascript/javascriptSemanticTraceSchemas.js";
import { err, ok, type Result } from "../../domain/result.js";
import { createJavaScriptSemanticTraceEvidence } from "./JavaScriptApplicationWorkflowEvidence.js";

const OPERATION = "trace_javascript_semantics" as const;

/** Trace semantic relations from input already parsed by a trusted adapter. */
export const traceJavaScriptSemanticsEvidenceValidated = (
  input: z.output<typeof traceJavaScriptSemanticsInputSchema>,
): Result<Evidence, AnalysisError> => {
  const sourceResult = parseApplicationGraphEvidence(input.application, [
    "application",
  ]);
  if (!sourceResult.ok)
    return err(
      applicationGraphEvidenceInputError(OPERATION, sourceResult.error),
    );
  const source = sourceResult.value;
  if (source.semanticGraph === null)
    return err(
      new AnalysisInputError(OPERATION, undefined, [
        {
          path: ["application"],
          reason: "invalid_value",
          expected:
            "inline Evidence from analyze_javascript_application with a semantic graph",
        },
      ]),
    );
  const query = queryJavaScriptSemanticGraphValidated(
    source.semanticGraph,
    input.query,
  );
  const result: JavaScriptSemanticTraceResult = {
    ...query,
    source_evidence_id: source.evidence.evidence_id,
    evidence_links: [source.evidence.evidence_id],
  };
  const { allowed_relations, ...queryParameters } = input.query;
  return ok(
    createJavaScriptSemanticTraceEvidence(
      {
        application_evidence_id: source.evidence.evidence_id,
        query:
          allowed_relations === undefined
            ? queryParameters
            : { ...queryParameters, allowed_relations },
      },
      result,
    ),
  );
};