import { ChevronRightIcon, Cog8ToothIcon } from "@heroicons/react/20/solid";
import { DEFAULT_DEV_BRANCH } from "@trigger.dev/core/v3/utils/gitBranch";
import { isBranchableEnvironment } from "~/utils/branchableEnvironment";
import { DropdownIcon } from "~/assets/icons/DropdownIcon";
import { useNavigation } from "@remix-run/react";
import { useEffect, useRef, useState } from "react";
import { BranchEnvironmentIconSmall } from "~/assets/icons/EnvironmentIcons";
import { useEnvironment } from "~/hooks/useEnvironment";
import { useEnvironmentSwitcher } from "~/hooks/useEnvironmentSwitcher";
import { useFeatures } from "~/hooks/useFeatures";
import { useOrganization, type MatchedOrganization } from "~/hooks/useOrganizations";
import { useProject } from "~/hooks/useProject";
import { cn } from "~/utils/cn";
import { branchesPath, branchesDevPath, docsPath, v3BillingPath } from "~/utils/pathBuilder";
import {
  EnvironmentCombo,
  EnvironmentIcon,
  EnvironmentLabel,
  environmentFullTitle,
  environmentTextClassName,
} from "../environments/EnvironmentLabel";
import { ButtonContent } from "../primitives/Buttons";
import { Header2 } from "../primitives/Headers";
import { Paragraph } from "../primitives/Paragraph";
import {
  Popover,
  PopoverContent,
  PopoverMenuItem,
  PopoverSectionHeader,
  PopoverTrigger,
} from "../primitives/Popover";
import { TextLink } from "../primitives/TextLink";
import { SimpleTooltip } from "../primitives/Tooltip";
import { V4Badge } from "../V4Badge";
import { type SideMenuEnvironment, type SideMenuProject } from "./SideMenu";
import { Badge } from "../primitives/Badge";

export function EnvironmentSelector({
  organization,
  project,
  environment,
  className,
  isCollapsed = false,
}: {
  organization: MatchedOrganization;
  project: SideMenuProject;
  environment: SideMenuEnvironment;
  className?: string;
  isCollapsed?: boolean;
}) {
  const { isManagedCloud } = useFeatures();
  const [isMenuOpen, setIsMenuOpen] = useState(false);
  const navigation = useNavigation();
  const { urlForEnvironment } = useEnvironmentSwitcher();

  useEffect(() => {
    setIsMenuOpen(false);
  }, [navigation.location?.pathname]);

  const hasStaging = project.environments.some((env) => env.type === "STAGING");
  const devBranchesEnabled = Boolean(organization.featureFlags?.devBranchesEnabled);

  return (
    <Popover onOpenChange={(open) => setIsMenuOpen(open)} open={isMenuOpen}>
      <SimpleTooltip
        button={
          <PopoverTrigger
            className={cn(
              "group flex h-8 items-center rounded pl-[0.4375rem] transition-colors hover:bg-charcoal-750",
              isCollapsed ? "justify-center pr-0.5" : "justify-between pr-1",
              className
            )}
          >
            <span className="flex min-w-0 flex-1 items-center gap-1.5 overflow-hidden">
              <EnvironmentIcon environment={environment} className="size-5 shrink-0" />
              <span
                className={cn(
                  "flex min-w-0 items-center overflow-hidden transition-all duration-200",
                  isCollapsed ? "max-w-0 opacity-0" : "max-w-[200px] opacity-100"
                )}
              >
                <EnvironmentLabel
                  environment={environment}
                  className="text-[0.90625rem] font-medium tracking-[-0.01em]"
                  disableTooltip
                />
              </span>
            </span>
            <span
              className={cn(
                "overflow-hidden transition-all duration-200",
                isCollapsed ? "max-w-0 opacity-0" : "max-w-[16px] opacity-100"
              )}
            >
              <DropdownIcon className="size-4 min-w-4 text-text-dimmed transition group-hover:text-text-bright" />
            </span>
          </PopoverTrigger>
        }
        content={environmentFullTitle(environment)}
        side="right"
        sideOffset={8}
        hidden={!isCollapsed}
        buttonClassName="!h-8"
        asChild
        disableHoverableContent
      />
      <PopoverContent
        className="min-w-[14rem] overflow-y-auto p-0 scrollbar-thin scrollbar-track-transparent scrollbar-thumb-charcoal-600"
        side={isCollapsed ? "right" : "bottom"}
        sideOffset={isCollapsed ? 8 : 4}
        align="start"
        style={{ maxHeight: `calc(var(--radix-popover-content-available-height) - 10vh)` }}
      >
        <div className="flex flex-col gap-1 p-1">
          {project.environments
            .filter((env) => env.parentEnvironmentId === null)
            .map((env) => {
              // DEVELOPMENT is only branchable in the UI when the org has the
              // multi-branch dev flag on. Without it, dev renders as a plain
              // selector button (the original behavior). PREVIEW is unaffected.
              const renderAsBranchable =
                isBranchableEnvironment(env) && (env.type !== "DEVELOPMENT" || devBranchesEnabled);

              if (renderAsBranchable) {
                const branchEnvironments = project.environments.filter(
                  (e) => e.parentEnvironmentId === env.id
                );
                const allBranchEnvironments =
                  env.type === "DEVELOPMENT" ? [env, ...branchEnvironments] : branchEnvironments;
                return (
                  <Branches
                    key={env.id}
                    parentEnvironment={env}
                    branchEnvironments={allBranchEnvironments}
                    currentEnvironment={environment}
                  />
                );
              }

              return (
                <PopoverMenuItem
                  key={env.id}
                  to={urlForEnvironment(env)}
                  title={<EnvironmentCombo environment={env} className="mx-auto grow text-2sm" />}
                  isSelected={env.id === environment.id}
                />
              );
            })}
        </div>
        {!hasStaging && isManagedCloud && (
          <>
            <PopoverSectionHeader title="Additional environments" />
            <div className="p-1">
              <PopoverMenuItem
                key="staging"
                to={v3BillingPath(
                  organization,
                  "Upgrade to unlock a Staging environment for your projects."
                )}
                title={
                  <div className="flex w-full items-center justify-between">
                    <EnvironmentCombo environment={{ type: "STAGING" }} className="text-2sm" />
                    <span className="text-indigo-500">Upgrade</span>
                  </div>
                }
                isSelected={false}
              />
              <PopoverMenuItem
                key="preview"
                to={v3BillingPath(
                  organization,
                  "Upgrade to unlock Preview environments for your projects."
                )}
                title={
                  <div className="flex w-full items-center justify-between">
                    <EnvironmentCombo environment={{ type: "PREVIEW" }} className="text-2sm" />
                    <span className="text-indigo-500">Upgrade</span>
                  </div>
                }
                isSelected={false}
              />
            </div>
          </>
        )}
      </PopoverContent>
    </Popover>
  );
}

function Branches({
  parentEnvironment,
  branchEnvironments,
  currentEnvironment,
}: {
  parentEnvironment: SideMenuEnvironment;
  branchEnvironments: SideMenuEnvironment[];
  currentEnvironment: SideMenuEnvironment;
}) {
  const organization = useOrganization();
  const project = useProject();
  const environment = useEnvironment();
  const { urlForEnvironment } = useEnvironmentSwitcher();
  const navigation = useNavigation();
  const [isMenuOpen, setMenuOpen] = useState(false);
  const timeoutRef = useRef<NodeJS.Timeout | null>(null);

  // Clear timeout on unmount
  useEffect(() => {
    return () => {
      if (timeoutRef.current) {
        clearTimeout(timeoutRef.current);
      }
    };
  }, []);

  useEffect(() => {
    setMenuOpen(false);
  }, [navigation.location?.pathname]);

  const handleMouseEnter = () => {
    if (timeoutRef.current) {
      clearTimeout(timeoutRef.current);
    }
    setMenuOpen(true);
  };

  const handleMouseLeave = () => {
    // Small delay before closing to allow moving to the content
    timeoutRef.current = setTimeout(() => {
      setMenuOpen(false);
    }, 150);
  };

  const activeBranches = branchEnvironments.filter((env) => env.archivedAt === null);
  const state =
    branchEnvironments.length === 0
      ? "no-branches"
      : activeBranches.length === 0
        ? "no-active-branches"
        : "has-branches";

  // Only surface the active environment's archived-branch item in the submenu it
  // actually belongs to. Both Development and Preview render this component, so
  // without the parent check an archived dev branch would leak into the Preview
  // submenu (and vice-versa).
  const currentBranchIsArchived =
    environment.archivedAt !== null && environment.parentEnvironmentId === parentEnvironment.id;

  const envTextClassName = environmentTextClassName(parentEnvironment);

  return (
    <Popover onOpenChange={(open) => setMenuOpen(open)} open={isMenuOpen}>
      <div onMouseEnter={handleMouseEnter} onMouseLeave={handleMouseLeave} className="flex">
        <PopoverTrigger className="w-full justify-between overflow-hidden focus-custom">
          <ButtonContent
            variant="small-menu-item"
            className="hover:bg-charcoal-750"
            TrailingIcon={ChevronRightIcon}
            trailingIconClassName="text-text-dimmed"
            textAlignLeft
            fullWidth
          >
            <EnvironmentCombo environment={parentEnvironment} className="mx-auto grow text-2sm" />
          </ButtonContent>
        </PopoverTrigger>
        <PopoverContent
          className="min-w-[16rem] overflow-y-auto p-0 scrollbar-thin scrollbar-track-transparent scrollbar-thumb-charcoal-600"
          align="start"
          style={{ maxHeight: `calc(var(--radix-popover-content-available-height) - 10vh)` }}
          side="right"
          alignOffset={0}
          sideOffset={-4}
          onMouseEnter={handleMouseEnter}
          onMouseLeave={handleMouseLeave}
        >
          <div className="flex flex-col gap-1 p-1">
            {currentBranchIsArchived && (
              <PopoverMenuItem
                key={environment.id}
                to={urlForEnvironment(environment)}
                title={
                  <>
                    <span className={cn("block w-full", envTextClassName)}>
                      {environment.branchName}
                    </span>
                    <Badge variant="extra-small">Archived</Badge>
                  </>
                }
                icon={
                  <BranchEnvironmentIconSmall className={cn("size-4 shrink-0", envTextClassName)} />
                }
                isSelected={environment.id === currentEnvironment.id}
              />
            )}
            {state === "has-branches" ? (
              <>
                {branchEnvironments
                  .filter((env) => env.archivedAt === null)
                  .map((env) => (
                    <PopoverMenuItem
                      key={env.id}
                      to={urlForEnvironment(env)}
                      title={
                        <span className={cn("block w-full", envTextClassName)}>
                          {env.branchName ?? DEFAULT_DEV_BRANCH}
                        </span>
                      }
                      icon={
                        <BranchEnvironmentIconSmall
                          className={cn("size-4 shrink-0", envTextClassName)}
                        />
                      }
                      isSelected={env.id === currentEnvironment.id}
                    />
                  ))}
              </>
            ) : state === "no-branches" ? (
              <div className="flex max-w-sm flex-col gap-1 p-2">
                <div className="flex items-center gap-1">
                  <BranchEnvironmentIconSmall className={cn("size-4", envTextClassName)} />
                  <Header2>Create your first branch</Header2>
                </div>
                <Paragraph spacing variant="small">
                  Branches are a way to test new features in isolation before merging them into the
                  main environment.
                </Paragraph>
                <Paragraph variant="small">
                  Branches are only available when using <V4Badge inline /> or above. Read our{" "}
                  <TextLink to={docsPath("upgrade-to-v4")}>v4 upgrade guide</TextLink> to learn
                  more.
                </Paragraph>
              </div>
            ) : (
              <div className="flex max-w-sm flex-col gap-1 p-2">
                <Paragraph variant="extra-small">All branches are archived.</Paragraph>
              </div>
            )}
          </div>
          <div className="border-t border-charcoal-700 p-1">
            {parentEnvironment.type === "DEVELOPMENT" ? (
              <PopoverMenuItem
                to={branchesDevPath(organization, project, environment)}
                title="Manage dev branches"
                icon={<Cog8ToothIcon className="size-4 text-text-dimmed" />}
                leadingIconClassName="text-text-dimmed"
              />
            ) : (
              <PopoverMenuItem
                to={branchesPath(organization, project, environment)}
                title="Manage preview branches"
                icon={<Cog8ToothIcon className="size-4 text-text-dimmed" />}
                leadingIconClassName="text-text-dimmed"
              />
            )}
          </div>
        </PopoverContent>
      </div>
    </Popover>
  );
}
