import type { CidString } from "@atp/lex";
import { InvalidRequestError, Server } from "@atp/xrpc-server";

import { AppContext } from "../../../../context.ts";
import * as com from "../../../../lex/com.ts";
import { $OutputBody } from "../../../../lex/com/atproto/admin/getSubjectStatus.ts";

export default function (server: Server, ctx: AppContext) {
  server.add(com.atproto.admin.getSubjectStatus, {
    auth: ctx.authVerifier.roleOrModService,
    handler: async ({ params }) => {
      const { did, uri, blob } = params;

      let body: $OutputBody | null = null;
      if (blob) {
        if (!did) {
          throw new InvalidRequestError(
            "Must provide a did to request blob state",
          );
        }
        const res = await ctx.dataplane.moderation.getBlobTakedown(
          did,
          blob,
        );
        body = {
          subject: {
            $type: "com.atproto.admin.defs#repoBlobRef",
            did: did,
            cid: blob,
          },
          takedown: {
            applied: res.takenDown,
            ref: res.takedownRef ? "TAKEDOWN" : undefined,
          },
        };
      } else if (uri) {
        const res = await ctx.hydrator.getRecord(uri, true);
        if (res) {
          body = {
            subject: {
              $type: "com.atproto.repo.strongRef",
              uri,
              cid: res.cid as CidString,
            },
            takedown: {
              applied: !!res.takedownRef,
              ref: res.takedownRef || undefined,
            },
          };
        }
      } else if (did) {
        const res = (
          await ctx.hydrator.actor.getActors([did], {
            includeTakedowns: true,
          })
        ).get(did);
        if (res) {
          body = {
            subject: {
              $type: "com.atproto.admin.defs#repoRef",
              did: did,
            },
            takedown: {
              applied: !!res.takedownRef,
              ref: res.takedownRef || undefined,
            },
          };
        }
      } else {
        throw new InvalidRequestError("No provided subject");
      }
      if (body === null) {
        throw new InvalidRequestError("Subject not found", "NotFound");
      }
      return {
        encoding: "application/json",
        body,
      };
    },
  });
}