import { formatBillNumber } from "@/lib/format";

export type VoteSubjectKind =

  | "bill"

  | "nomination"

  | "treaty"

  | "ref"

  | "question"

  | "result";

export type VoteSubject = {

  primary: string;

  secondary: string;

  kind: VoteSubjectKind;

};

export type VoteSubjectInput = {

  billId?: number | null;

  billNumber?: string;

  billTitle?: string;

  billTextRef?: string;

  question?: string;

  result?: string;

};

export function resolveVoteSubject(v: VoteSubjectInput): VoteSubject {

  const ref = v.billTextRef ?? "";

  const question = v.question ?? "";

  const result = v.result ?? "";

  if (v.billNumber) {

    const title = v.billTitle ? ` -${v.billTitle}` : "";

    return {

      primary: `${formatBillNumber(v.billNumber)}${title}`,

      secondary: question || result,

      kind: "bill",

    };

  }

  if (ref.toUpperCase().startsWith("PN")) {

    return {

      primary: `Nomination -${question || ref}`,

      secondary: result,

      kind: "nomination",

    };

  }

  if (ref.toUpperCase().startsWith("TREATYDOC")) {

    return {

      primary: `Treaty document -${question || ref}`,

      secondary: result,

      kind: "treaty",

    };

  }

  if (ref) {

    return {

      primary: formatBillNumber(ref),

      secondary: question || result,

      kind: "ref",

    };

  }

  if (question) {

    return {

      primary: question,

      secondary: result,

      kind: "question",

    };

  }

  return {

    primary: result || "Roll-call vote",

    secondary: "",

    kind: "result",

  };

}