/**
 * Triples query CLI commands.
 *
 * Query the knowledge graph for subject-predicate-object triples.
 */

import type { Command } from "commander";
import type { Term } from "@trustgraph/client";
import { Effect } from "effect";
import { cliCommandError, withSocket, writeJson } from "./util.js";

export function registerTriplesCommands(program: Command): void {
  program
    .command("triples")
    .description("Query knowledge graph triples")
    .option("-s, --subject <iri>", "Subject IRI")
    .option("-p, --predicate <iri>", "Predicate IRI")
    .option("-o, --object <iri>", "Object IRI or literal")
    .option("-l, --limit <n>", "Max results", "20")
    .option("--collection <name>", "Collection name")
    .action((cmdOpts, cmd) =>
      Effect.runPromise(withSocket(cmd, (socket, opts) =>
        Effect.gen(function* () {
        const flow = socket.flow(opts.flow);
        const subject = cmdOpts.subject as string | undefined;
        const predicate = cmdOpts.predicate as string | undefined;
        const object = cmdOpts.object as string | undefined;
        const s: Term | undefined = subject !== undefined && subject.length > 0
          ? { t: "i", i: subject }
          : undefined;
        const p: Term | undefined = predicate !== undefined && predicate.length > 0
          ? { t: "i", i: predicate }
          : undefined;
        const o: Term | undefined = object !== undefined && object.length > 0
          ? { t: "i", i: object }
          : undefined;

          const triples = yield* Effect.tryPromise({
            try: () =>
              flow.triplesQuery(
                s,
                p,
                o,
                parseInt(cmdOpts.limit as string, 10),
                cmdOpts.collection as string | undefined,
              ),
            catch: (error) => cliCommandError("triples", error),
          });
          yield* writeJson(triples);
        }),
      )),
    );
}