rea/src/application/javascript/JavaScriptSemanticGraphObjectProjection.ts

import { createJavaScriptSemanticGraphUnknown } from "../../domain/javascript/javascriptSemanticGraph.js";
import type { JavaScriptSemanticGraphNode } from "../../domain/javascript/javascriptSemanticGraph.js";
import type { JavaScriptSemanticObjectOperation } from "../../domain/javascript/javascriptSemanticIr.js";
import {
  retainSemanticGraphNode,
  addSemanticGraphRelation,
  addSemanticGraphUnknown,
} from "./JavaScriptSemanticGraphConstruction.js";
import { unknownSemanticEvidence } from "./JavaScriptSemanticGraphEvidence.js";
import type { SemanticFlowProjectionContext } from "./JavaScriptSemanticGraphFlowProjection.js";
import { semanticPropertySlot } from "./JavaScriptSemanticGraphValueProjection.js";

/** Project static object reads, writes, spreads, and destructuring. */
export const projectSemanticObjects = (
  context: SemanticFlowProjectionContext,
): void => {
  for (const operation of context.ir.objectOperations) {
    const occurrence = addObjectOccurrence(context, operation);
    const slot =
      operation.objectBindingId === null || operation.propertyName === null
        ? null
        : semanticPropertySlot(
            context,
            operation.objectBindingId,
            operation.propertyName,
          );
    if (operation.kind === "read")
      addSemanticGraphRelation(context.state, {
        source: slot,
        target: occurrence,
        relation: "reads-property",
        resolution:
          operation.resolution === "complete" ? "resolved" : "candidate",
      });
    else if (operation.kind === "write")
      addSemanticGraphRelation(context.state, {
        source: occurrence,
        target: slot,
        relation: "writes-property",
        resolution:
          operation.resolution === "complete" ? "resolved" : "candidate",
      });
    else if (operation.kind === "destructure")
      addSemanticGraphRelation(context.state, {
        source: slot,
        target:
          operation.targetBindingId === null
            ? undefined
            : context.bindingNodes.get(operation.targetBindingId),
        relation: "destructures",
        resolution:
          operation.resolution === "complete" ? "resolved" : "candidate",
      });
    else
      addSemanticGraphRelation(context.state, {
        source:
          operation.objectBindingId === null
            ? undefined
            : context.bindingNodes.get(operation.objectBindingId),
        target: occurrence,
        relation: "spreads",
        resolution:
          operation.resolution === "complete" ? "resolved" : "candidate",
      });
    if (operation.resolution !== "complete")
      addObjectUnknown(context, operation, slot ?? occurrence);
  }
};

const addObjectOccurrence = (
  context: SemanticFlowProjectionContext,
  operation: JavaScriptSemanticObjectOperation,
): JavaScriptSemanticGraphNode | null =>
  retainSemanticGraphNode(context.state, context.file, {
    kind: "expression",
    roleKey: operation.objectOperationId,
    location: operation.location,
    label:
      operation.propertyName === null
        ? operation.kind
        : `${operation.kind}:${operation.propertyName}`,
    functionNodeId:
      operation.ownerCallableId === null
        ? null
        : (context.callableNodes.get(operation.ownerCallableId)?.node_id ??
          null),
    properties: {
      operation_kind: operation.kind,
      property_name: operation.propertyName,
    },
  });

const addObjectUnknown = (
  context: SemanticFlowProjectionContext,
  operation: JavaScriptSemanticObjectOperation,
  node: JavaScriptSemanticGraphNode | null,
): void => {
  const relation =
    operation.kind === "read"
      ? "reads-property"
      : operation.kind === "write"
        ? "writes-property"
        : operation.kind === "spread"
          ? "spreads"
          : "destructures";
  addSemanticGraphUnknown(
    context.state,
    createJavaScriptSemanticGraphUnknown({
      node_id: node?.node_id ?? null,
      family: "object-flow",
      relation_kinds: [relation],
      reason: "ambiguous-target",
      detail: `Static ${operation.kind} object identity is partial.`,
      candidate_node_ids: [],
      evidence: unknownSemanticEvidence(context.file, operation.location),
    }),
  );
};