import { Events } from "@openstatus/analytics";
import {
  PreviewImportInput,
  RunImportInput,
  previewImport,
  runImport,
} from "@openstatus/services/import";

import { toServiceCtx, toTRPCError } from "../service-adapter";
import { createTRPCRouter, protectedProcedure } from "../trpc";

export const importRouter = createTRPCRouter({
  preview: protectedProcedure
    .meta({ track: Events.PreviewImport, trackProps: ["provider"] })
    .input(PreviewImportInput)
    .mutation(async ({ ctx, input }) => {
      try {
        return await previewImport({ ctx: toServiceCtx(ctx), input });
      } catch (err) {
        toTRPCError(err);
      }
    }),

  run: protectedProcedure
    .meta({
      track: Events.RunImport,
      trackProps: ["provider"],
      // A summary is returned for failed runs too; only count imports that
      // actually wrote something.
      trackResult: ({ data }) => {
        const status = (data as { status?: string } | undefined)?.status;
        return status === "completed" || status === "partial"
          ? { status }
          : null;
      },
    })
    .input(RunImportInput)
    .mutation(async ({ ctx, input }) => {
      try {
        return await runImport({ ctx: toServiceCtx(ctx), input });
      } catch (err) {
        toTRPCError(err);
      }
    }),
});