import colorWheelIcon from "../../assets/images/color-wheel.png";
import { getFormProps, getInputProps, useForm } from "@conform-to/react";
import { conformZodMessage, parseWithZod } from "@conform-to/zod";
import {
  CheckIcon,
  ExclamationTriangleIcon,
  FolderIcon,
  GlobeAltIcon,
  TrashIcon,
} from "@heroicons/react/20/solid";
import { Form, type MetaFunction, useActionData, useNavigation, useSubmit } from "@remix-run/react";
import { type ActionFunction, json, type LoaderFunctionArgs } from "@remix-run/server-runtime";
import { useEffect, useRef, useState } from "react";
import { redirect, typedjson, useTypedLoaderData } from "remix-typedjson";
import { z } from "zod";
import { InlineCode } from "~/components/code/InlineCode";
import {
  MainHorizontallyCenteredContainer,
  PageBody,
  PageContainer,
} from "~/components/layout/AppLayout";
import {
  Avatar,
  AvatarData,
  avatarIcons,
  AvatarType,
  defaultAvatar,
  parseAvatar,
  defaultAvatarHex,
  defaultAvatarColors,
  type Avatar as AvatarT,
} from "~/components/primitives/Avatar";
import { Button } from "~/components/primitives/Buttons";
import { Fieldset } from "~/components/primitives/Fieldset";
import { FormButtons } from "~/components/primitives/FormButtons";
import { FormError } from "~/components/primitives/FormError";
import { Header2 } from "~/components/primitives/Headers";
import { Hint } from "~/components/primitives/Hint";
import { Input } from "~/components/primitives/Input";
import { InputGroup } from "~/components/primitives/InputGroup";
import { Label } from "~/components/primitives/Label";
import { NavBar, PageTitle } from "~/components/primitives/PageHeader";
import { Popover, PopoverContent, PopoverTrigger } from "~/components/primitives/Popover";
import { SpinnerWhite } from "~/components/primitives/Spinner";
import { prisma } from "~/db.server";
import { useFaviconUrl } from "~/hooks/useFaviconUrl";
import { redirectWithErrorMessage, redirectWithSuccessMessage } from "~/models/message.server";
import { clearCurrentProject } from "~/services/dashboardPreferences.server";
import { DeleteOrganizationService } from "~/services/deleteOrganization.server";
import { logger } from "~/services/logger.server";
import { requireUser, requireUserId } from "~/services/session.server";
import { cn } from "~/utils/cn";
import { extractDomain, faviconUrl as buildFaviconUrl } from "~/utils/favicon";
import { OrganizationParamsSchema, organizationSettingsPath, rootPath } from "~/utils/pathBuilder";

export const meta: MetaFunction = () => {
  return [
    {
      title: `Organization settings | Trigger.dev`,
    },
  ];
};

export const loader = async ({ request, params }: LoaderFunctionArgs) => {
  const userId = await requireUserId(request);
  const { organizationSlug } = OrganizationParamsSchema.parse(params);

  const organization = await prisma.organization.findFirst({
    where: { slug: organizationSlug, members: { some: { userId } }, deletedAt: null },
    orderBy: { createdAt: "desc" },
    select: {
      id: true,
      slug: true,
      title: true,
      avatar: true,
      onboardingData: true,
    },
  });

  if (!organization) {
    throw new Response("Not found", { status: 404 });
  }

  const onboardingData = toRecord(organization.onboardingData);

  const parsedAvatar = parseAvatar(organization.avatar, defaultAvatar);
  const lastIconHex =
    parsedAvatar.type === "image" && parsedAvatar.lastIconHex
      ? parsedAvatar.lastIconHex
      : defaultAvatarHex;

  return typedjson({
    organization: {
      ...organization,
      avatar: parsedAvatar,
      companyUrl: typeof onboardingData.companyUrl === "string" ? onboardingData.companyUrl : "",
      lastIconHex,
    },
  });
};

export function createSchema(
  constraints: {
    getSlugMatch?: (slug: string) => { isMatch: boolean; organizationSlug: string };
  } = {}
) {
  return z.discriminatedUnion("action", [
    z.object({
      action: z.literal("avatar"),
      type: AvatarType,
      name: z.string().optional(),
      hex: z.string().optional(),
      url: z.string().optional(),
    }),
    z.object({
      action: z.literal("rename"),
      organizationName: z
        .string()
        .min(3, "Organization name must have at least 3 characters")
        .max(50),
    }),
    z.object({
      action: z.literal("delete"),
      organizationSlug: z.string().superRefine((slug, ctx) => {
        if (constraints.getSlugMatch === undefined) {
          ctx.addIssue({
            code: z.ZodIssueCode.custom,
            message: conformZodMessage.VALIDATION_UNDEFINED,
          });
        } else {
          const { isMatch, organizationSlug } = constraints.getSlugMatch(slug);
          if (isMatch) {
            return;
          }

          ctx.addIssue({
            code: z.ZodIssueCode.custom,
            message: `The slug must match ${organizationSlug}`,
          });
        }
      }),
    }),
  ]);
}

export const action: ActionFunction = async ({ request, params }) => {
  const user = await requireUser(request);
  const { organizationSlug } = params;
  if (!organizationSlug) {
    return json({ errors: { body: "organizationSlug is required" } }, { status: 400 });
  }

  const formData = await request.formData();
  const schema = createSchema({
    getSlugMatch: (slug) => {
      return { isMatch: slug === organizationSlug, organizationSlug };
    },
  });
  const submission = parseWithZod(formData, { schema });

  if (submission.status !== "success") {
    return json(submission.reply());
  }

  try {
    switch (submission.value.action) {
      case "rename": {
        await prisma.organization.update({
          where: {
            slug: organizationSlug,
            members: {
              some: {
                userId: user.id,
              },
            },
          },
          data: {
            title: submission.value.organizationName,
          },
        });

        return redirectWithSuccessMessage(
          organizationSettingsPath({ slug: organizationSlug }),
          request,
          `Organization renamed to ${submission.value.organizationName}`
        );
      }
      case "delete": {
        const deleteOrganizationService = new DeleteOrganizationService();
        try {
          await deleteOrganizationService.call({ organizationSlug, userId: user.id, request });

          //we need to clear the project from the session
          await clearCurrentProject({
            user,
          });
          return redirect(rootPath());
        } catch (error: unknown) {
          const errorMessage = error instanceof Error ? error.message : JSON.stringify(error);
          logger.error("Organization could not be deleted", {
            error: errorMessage,
          });
          return redirectWithErrorMessage(
            organizationSettingsPath({ slug: organizationSlug }),
            request,
            errorMessage
          );
        }
      }
      case "avatar": {
        const orgWhere = {
          slug: organizationSlug,
          members: { some: { userId: user.id } },
        };

        if (submission.value.type === "image") {
          const url = submission.value.url ?? "";
          const domain = url ? extractDomain(url) : null;

          const existing = await prisma.organization.findFirst({
            where: orgWhere,
            select: { avatar: true, onboardingData: true },
          });

          const existingData = toRecord(existing?.onboardingData);
          const existingAvatar = parseAvatar(existing?.avatar ?? null, defaultAvatar);
          const lastIconHex = extractLastIconHex(existingAvatar);

          await prisma.organization.update({
            where: orgWhere,
            data: {
              avatar: {
                type: "image",
                url: domain ? buildFaviconUrl(domain) : "",
                ...(lastIconHex ? { lastIconHex } : {}),
              },
              onboardingData: { ...existingData, companyUrl: url },
            },
          });

          return redirectWithSuccessMessage(
            organizationSettingsPath({ slug: organizationSlug }),
            request,
            `Updated logo`
          );
        }

        const avatar = AvatarData.safeParse(submission.value);

        if (!avatar.success) {
          return redirectWithErrorMessage(
            organizationSettingsPath({ slug: organizationSlug }),
            request,
            avatar.error.message
          );
        }

        await prisma.organization.update({
          where: orgWhere,
          data: {
            avatar: avatar.data,
          },
        });

        return redirectWithSuccessMessage(
          organizationSettingsPath({ slug: organizationSlug }),
          request,
          `Updated logo`
        );
      }
    }
  } catch (error: unknown) {
    const message = error instanceof Error ? error.message : "An unexpected error occurred";
    return json({ errors: { body: message } }, { status: 400 });
  }
};

export default function Page() {
  const { organization } = useTypedLoaderData<typeof loader>();
  const lastSubmission = useActionData();
  const navigation = useNavigation();

  const [renameForm, { organizationName }] = useForm({
    id: "rename-organization",
    // TODO: type this
    lastResult: lastSubmission as any,
    shouldRevalidate: "onSubmit",
    onValidate({ formData }) {
      return parseWithZod(formData, {
        schema: createSchema(),
      });
    },
  });

  const [deleteForm, { organizationSlug }] = useForm({
    id: "delete-organization",
    // TODO: type this
    lastResult: lastSubmission as any,
    shouldValidate: "onInput",
    shouldRevalidate: "onSubmit",
    onValidate({ formData }) {
      return parseWithZod(formData, {
        schema: createSchema({
          getSlugMatch: (slug) => ({
            isMatch: slug === organization.slug,
            organizationSlug: organization.slug,
          }),
        }),
      });
    },
  });

  const isRenameLoading =
    navigation.formData?.get("action") === "rename" &&
    (navigation.state === "submitting" || navigation.state === "loading");

  const isDeleteLoading =
    navigation.formData?.get("action") === "delete" &&
    (navigation.state === "submitting" || navigation.state === "loading");

  return (
    <PageContainer>
      <NavBar>
        <PageTitle title={`${organization.title} organization settings`} />
      </NavBar>

      <PageBody>
        <MainHorizontallyCenteredContainer>
          <div className="mb-3 border-b border-grid-dimmed pb-3">
            <Header2>Settings</Header2>
          </div>
          <div className="flex flex-col gap-6">
            <div>
              <LogoForm organization={organization} />
            </div>

            <div>
              <Form method="post" {...getFormProps(renameForm)}>
                <input type="hidden" name="action" value="rename" />
                <Fieldset className="gap-y-0">
                  <InputGroup fullWidth>
                    <Label htmlFor={organizationName.id}>Organization name</Label>
                    <Input
                      {...getInputProps(organizationName, { type: "text" })}
                      defaultValue={organization.title}
                      placeholder="Your organization name"
                      icon={FolderIcon}
                      autoFocus
                    />
                    <FormError id={organizationName.errorId}>{organizationName.errors}</FormError>
                  </InputGroup>
                  <FormButtons
                    confirmButton={
                      <Button
                        type="submit"
                        variant={"secondary/small"}
                        disabled={isRenameLoading}
                        LeadingIcon={isRenameLoading ? SpinnerWhite : undefined}
                      >
                        Rename organization
                      </Button>
                    }
                    className="border-t-0"
                  />
                </Fieldset>
              </Form>
            </div>

            <div>
              <Header2 spacing>Danger zone</Header2>
              <Form
                method="post"
                {...getFormProps(deleteForm)}
                className="w-full rounded-sm border border-rose-500/40"
              >
                <input type="hidden" name="action" value="delete" />
                <Fieldset className="p-4">
                  <InputGroup>
                    <Label htmlFor={organizationSlug.id}>Delete organization</Label>
                    <Input
                      {...getInputProps(organizationSlug, { type: "text" })}
                      placeholder="Your organization slug"
                      icon={ExclamationTriangleIcon}
                      fullWidth
                    />
                    <FormError id={organizationSlug.errorId}>{organizationSlug.errors}</FormError>
                    <FormError>{deleteForm.errors}</FormError>
                    <Hint>
                      This change is irreversible, so please be certain. Type in the Organization
                      slug <InlineCode variant="extra-small">{organization.slug}</InlineCode> and
                      then press Delete.
                    </Hint>
                  </InputGroup>
                  <FormButtons
                    confirmButton={
                      <Button
                        type="submit"
                        variant={"danger/small"}
                        LeadingIcon={isDeleteLoading ? SpinnerWhite : TrashIcon}
                        leadingIconClassName="text-white"
                        disabled={isDeleteLoading}
                      >
                        Delete organization
                      </Button>
                    }
                  />
                </Fieldset>
              </Form>
            </div>
          </div>
        </MainHorizontallyCenteredContainer>
      </PageBody>
    </PageContainer>
  );
}

function LogoForm({
  organization,
}: {
  organization: { avatar: AvatarT; title: string; companyUrl: string; lastIconHex: string };
}) {
  const navigation = useNavigation();

  const avatar = navigation.formData
    ? (avatarFromFormData(navigation.formData) ?? organization.avatar)
    : organization.avatar;

  const hex =
    "hex" in avatar
      ? avatar.hex
      : avatar.type === "image" && avatar.lastIconHex
        ? avatar.lastIconHex
        : organization.lastIconHex;
  const mode: "logo" | "icon" = avatar.type === "image" ? "logo" : "icon";

  const [companyUrl, setCompanyUrl] = useState(organization.companyUrl);
  const faviconPreview = useFaviconUrl(companyUrl);
  const [faviconError, setFaviconError] = useState(false);
  const logoFormRef = useRef<HTMLFormElement>(null);
  const submit = useSubmit();
  const prevFaviconRef = useRef(faviconPreview);

  useEffect(() => {
    if (faviconPreview === prevFaviconRef.current) return;
    prevFaviconRef.current = faviconPreview;
    if (mode === "logo" && logoFormRef.current) {
      submit(logoFormRef.current);
    }
  }, [faviconPreview, mode, submit]);

  const showFavicon = faviconPreview && !faviconError;

  return (
    <Fieldset>
      <InputGroup fullWidth>
        <Label>Logo</Label>
        <div className="flex flex-col gap-3">
          {/* Row 1: Logo from URL */}
          <Form ref={logoFormRef} method="post" className="flex items-center gap-3">
            <input type="hidden" name="action" value="avatar" />
            <input type="hidden" name="type" value="image" />
            <input type="hidden" name="url" value={companyUrl} />
            <button type="submit" className="flex shrink-0 items-center gap-3">
              <RadioDot active={mode === "logo"} />
            </button>
            <div className="flex flex-1 items-center gap-1.5">
              <button
                type="submit"
                className={cn(
                  iconTileClass,
                  mode === "logo"
                    ? "border-indigo-500"
                    : "border-grid-dimmed hover:border-border-bright"
                )}
              >
                {showFavicon ? (
                  <img
                    src={faviconPreview}
                    alt=""
                    width={28}
                    height={28}
                    className="rounded-sm"
                    onError={() => setFaviconError(true)}
                    onLoad={() => setFaviconError(false)}
                  />
                ) : (
                  <GlobeAltIcon className="size-6 text-text-dimmed" />
                )}
              </button>
              <Input
                type="text"
                value={companyUrl}
                onChange={(e) => {
                  setCompanyUrl(e.target.value);
                  setFaviconError(false);
                }}
                onFocus={() => {
                  if (mode !== "logo" && logoFormRef.current) {
                    submit(logoFormRef.current);
                  }
                }}
                placeholder="Enter your company URL to generate a logo"
                variant="medium"
                containerClassName="flex-1"
              />
            </div>
          </Form>

          {/* Row 2: Icon picker */}
          <div className="flex items-center gap-3">
            <Form method="post" className="shrink-0">
              <input type="hidden" name="action" value="avatar" />
              <input type="hidden" name="type" value="letters" />
              <input type="hidden" name="hex" value={hex} />
              <button type="submit">
                <RadioDot active={mode === "icon"} />
              </button>
            </Form>
            <div className="flex flex-wrap items-center gap-1.5">
              {/* Letters */}
              <Form method="post">
                <input type="hidden" name="action" value="avatar" />
                <input type="hidden" name="type" value="letters" />
                <input type="hidden" name="hex" value={hex} />
                <button
                  type="submit"
                  className={cn(
                    iconTileClass,
                    avatar.type !== "letters" && "border-grid-dimmed hover:border-border-bright"
                  )}
                  style={{
                    borderColor: avatar.type === "letters" ? hex : undefined,
                  }}
                >
                  <Avatar
                    avatar={{ type: "letters", hex }}
                    size={2.5}
                    includePadding
                    orgName={organization.title}
                  />
                </button>
              </Form>
              {/* Icons */}
              {Object.entries(avatarIcons).map(([name]) => (
                <Form key={name} method="post">
                  <input type="hidden" name="action" value="avatar" />
                  <input type="hidden" name="type" value="icon" />
                  <input type="hidden" name="name" value={name} />
                  <input type="hidden" name="hex" value={hex} />
                  <button
                    type="submit"
                    className={cn(
                      iconTileClass,
                      !(avatar.type === "icon" && avatar.name === name) &&
                        "border-grid-dimmed hover:border-border-bright"
                    )}
                    style={{
                      borderColor: avatar.type === "icon" && avatar.name === name ? hex : undefined,
                    }}
                  >
                    <Avatar
                      avatar={{ type: "icon", name, hex }}
                      size={2.5}
                      includePadding
                      orgName={organization.title}
                    />
                  </button>
                </Form>
              ))}
              {/* Color picker */}
              <HexPopover avatar={avatar} hex={hex} />
            </div>
          </div>
        </div>
      </InputGroup>
    </Fieldset>
  );
}

function HexPopover({ avatar, hex }: { avatar: Avatar; hex: string }) {
  return (
    <Popover>
      <PopoverTrigger
        className={cn(iconTileClass, "border-grid-dimmed hover:border-border-bright")}
      >
        <img src={colorWheelIcon} className="m-0 block size-[30px] p-0" />
      </PopoverTrigger>
      <PopoverContent
        className="overflow-y-auto p-1 scrollbar-thin scrollbar-track-transparent scrollbar-thumb-surface-control"
        align="start"
        style={{ maxHeight: `calc(var(--radix-popover-content-available-height) - 10vh)` }}
      >
        <Form method="post" className="flex w-fit min-w-40 flex-col gap-1">
          <input type="hidden" name="action" value="avatar" />
          <input
            type="hidden"
            name="type"
            value={avatar.type === "image" ? "letters" : avatar.type}
          />
          {avatar.type === "icon" && <input type="hidden" name="name" value={avatar.name} />}
          {defaultAvatarColors.map((color) => (
            <Button
              key={color.hex}
              name="hex"
              value={color.hex}
              type="submit"
              variant="small-menu-item"
              LeadingIcon={
                <div
                  className="size-4 rounded-full"
                  style={{
                    backgroundColor: color.hex,
                  }}
                />
              }
              TrailingIcon={hex === color.hex && <CheckIcon className="size-4 text-text-dimmed" />}
              trailingIconClassName="ml-4"
              fullWidth
              textAlignLeft
              className={cn(
                "group-hover:bg-background-raised",
                hex === color.hex
                  ? "bg-background-hover group-hover:bg-surface-control/50"
                  : undefined
              )}
            >
              {color.name}
            </Button>
          ))}
        </Form>
      </PopoverContent>
    </Popover>
  );
}

function RadioDot({ active }: { active: boolean }) {
  return (
    <div
      className={cn(
        "flex shrink-0 items-center justify-center rounded-full border-2 p-0.5 transition",
        active ? "border-indigo-500" : "border-grid-bright hover:border-border-bright"
      )}
    >
      <div
        className="size-2 rounded-full"
        style={{ backgroundColor: active ? "#6366f1" : "transparent" }}
      />
    </div>
  );
}

const iconTileClass =
  "box-content grid size-10 shrink-0 place-items-center rounded-sm border-2 bg-charcoal-775";

function toRecord(json: unknown): Record<string, unknown> {
  return json && typeof json === "object" ? (json as Record<string, unknown>) : {};
}

function extractLastIconHex(avatar: AvatarT): string | undefined {
  if ("hex" in avatar) return avatar.hex;
  if (avatar.type === "image") return avatar.lastIconHex;
  return undefined;
}

function avatarFromFormData(formData: FormData): AvatarT | undefined {
  const action = formData.get("action");
  if (action !== "avatar") return undefined;

  const type = formData.get("type");
  const hex = formData.get("hex") as string;

  switch (type) {
    case "letters":
      return { type: "letters", hex };
    case "icon":
      return { type: "icon", name: formData.get("name") as string, hex };
    case "image": {
      const url = formData.get("url") as string;
      const domain = url ? extractDomain(url) : null;
      return { type: "image", url: domain ? buildFaviconUrl(domain) : "" };
    }
    default:
      return undefined;
  }
}
