p-stream/src/pages/parts/settings/SetupPart.tsx

/* eslint-disable no-console */
import classNames from "classnames";
import { FetchError } from "ofetch";
import { ReactNode } from "react";
import { useTranslation } from "react-i18next";
import { useNavigate } from "react-router-dom";
import { useAsync } from "react-use";

import { isExtensionActive } from "@/backend/extension/messaging";
import { proxiedFetch, singularProxiedFetch } from "@/backend/helpers/fetch";
import { Button } from "@/components/buttons/Button";
import { Icon, Icons } from "@/components/Icon";
import { Loading } from "@/components/layout/Loading";
import { SettingsCard } from "@/components/layout/SettingsCard";
import {
  StatusCircle,
  StatusCircleProps,
} from "@/components/player/internals/StatusCircle";
import { Heading3 } from "@/components/utils/Text";
import { useIsDesktopApp } from "@/hooks/useIsDesktopApp";
import { conf } from "@/setup/config";
import { useAuthStore } from "@/stores/auth";
import { usePreferencesStore } from "@/stores/preferences";

const testUrl = "https://postman-echo.com/get";

const sleep = (ms: number): Promise<void> => {
  return new Promise((resolve) => {
    setTimeout(() => {
      resolve();
    }, ms);
  });
};

export type Status =
  | "success"
  | "unset"
  | "error"
  | "api_down"
  | "invalid_token";

type SetupData = {
  extension: Status;
  proxy: Status;
  defaultProxy: Status;
  febboxKeyTest?: Status;
  debridTokenTest?: Status;
};

function testProxy(url: string) {
  return new Promise<void>((resolve, reject) => {
    setTimeout(() => reject(new Error("Timed out!")), 3000);
    singularProxiedFetch(url, testUrl, {})
      .then((res) => {
        if (res.url !== testUrl) return reject(new Error("Not a proxy"));
        resolve();
      })
      .catch(reject);
  });
}

export async function fetchFebboxQuota(febboxKey: string | null): Promise<any> {
  if (!febboxKey) {
    return null;
  }

  console.log("SetupPart.tsx: Fetching Febbox quota");
  try {
    const response = await fetch("https://fed-api.pstream.mov/quota", {
      headers: {
        "ui-token": febboxKey,
      },
    });

    if (!response.ok) {
      console.error("Febbox quota API failed with status:", response.status);
      return null;
    }

    const data = await response.json();
    console.log("SetupPart.tsx: Febbox quota fetched successfully");
    return data;
  } catch (error) {
    console.error("SetupPart.tsx: Error fetching Febbox quota:", error);
    return null;
  }
}

export async function testFebboxKey(febboxKey: string | null): Promise<Status> {
  const febboxApiTestUrl = `https://fed-api.pstream.mov/movie/tt0325980`;

  if (!febboxKey) {
    return "unset";
  }

  let attempts = 0;
  const maxAttempts = 2;

  while (attempts < maxAttempts) {
    console.log(
      `Attempt ${attempts + 1} of ${maxAttempts} to check Febbox token`,
    );
    try {
      const response = await fetch(febboxApiTestUrl, {
        headers: {
          "ui-token": febboxKey,
        },
      });

      if (!response.ok) {
        console.error("Febbox API test failed with status:", response.status);
        if (response.status === 503 || response.status === 502) {
          return "api_down";
        }
        attempts += 1;
        if (attempts === maxAttempts) {
          console.log("Max attempts reached, returning error");
          return "invalid_token";
        }
        console.log("Retrying after failed response...");
        await sleep(3000);
        continue;
      }

      const data = (await response.json()) as any;
      if (!data || !data.streams) {
        console.error("Invalid response format from Febbox API:", data);
        attempts += 1;
        if (attempts === maxAttempts) {
          console.log("Max attempts reached, returning error");
          return "invalid_token";
        }
        console.log("Retrying after invalid response format...");
        await sleep(3000);
        continue;
      }

      const isVIPLink = Object.values(data.streams).some((stream: any) => {
        if (typeof stream === "object" && stream.download) {
          return stream.download.includes("/vip/");
        }
        return false;
      });

      if (isVIPLink) {
        console.log("VIP link found, returning success");
        return "success";
      }

      console.log("No VIP link found in attempt", attempts + 1);
      attempts += 1;
      if (attempts === maxAttempts) {
        console.log("Max attempts reached, returning error");
        return "invalid_token";
      }
      console.log("Retrying after no VIP link found...");
      await sleep(3000);
    } catch (error: any) {
      console.error("Error testing Febbox token:", error);
      attempts += 1;
      if (attempts === maxAttempts) {
        console.log("Max attempts reached, returning error");
        return "api_down";
      }
      console.log("Retrying after error...");
      await sleep(3000);
    }
  }

  console.log("All attempts exhausted, returning error");
  return "api_down";
}

export async function testdebridToken(
  debridToken: string | null,
): Promise<Status> {
  if (!debridToken) {
    return "unset";
  }

  const maxAttempts = 2;
  let attempts = 0;

  while (attempts < maxAttempts) {
    try {
      console.log(`RD API attempt ${attempts + 1}`);
      const data = await proxiedFetch(
        "https://api.real-debrid.com/rest/1.0/user",
        {
          method: "GET",
          headers: {
            Authorization: `Bearer ${debridToken}`,
            "Content-Type": "application/json",
          },
        },
      );

      // If we have data and it indicates premium status, return success immediately
      if (data && typeof data === "object" && data.type === "premium") {
        console.log("RD premium status confirmed");
        return "success";
      }

      console.log("RD response did not indicate premium status");
      attempts += 1;
      if (attempts === maxAttempts || data?.error_code === 8) {
        return "invalid_token";
      }
      await sleep(3000);
    } catch (error) {
      console.error("RD API error:", error);

      // Check if it's a FetchError with error_code 8 (bad_token)
      if (error instanceof FetchError) {
        try {
          const errorData = error.data;
          if (errorData?.error_code === 8) {
            console.log("RD token is invalid (error_code 8)");
            return "invalid_token";
          }
        } catch (parseError) {
          console.error("Failed to parse RD error response:", parseError);
        }

        // For other HTTP errors (like 500, 502, etc.), treat as API down
        if (error.statusCode && error.statusCode >= 500) {
          console.log(`RD API down (status ${error.statusCode})`);
          return "api_down";
        }
      }

      attempts += 1;
      if (attempts === maxAttempts) {
        return "api_down";
      }
      await sleep(3000);
    }
  }

  return "api_down";
}

export async function testTorboxToken(
  torboxToken: string | null,
): Promise<Status> {
  if (!torboxToken) {
    return "unset";
  }

  // TODO: Implement Torbox token test
  return "success";
}

function useIsSetup() {
  const proxyUrls = useAuthStore((s) => s.proxySet);
  const febboxKey = usePreferencesStore((s) => s.febboxKey);
  const debridToken = usePreferencesStore((s) => s.debridToken);
  const debridService = usePreferencesStore((s) => s.debridService);
  const { loading, value } = useAsync(async (): Promise<SetupData> => {
    const extensionStatus: Status = (await isExtensionActive())
      ? "success"
      : "unset";
    let proxyStatus: Status = "unset";
    if (proxyUrls && proxyUrls.length > 0) {
      try {
        await testProxy(proxyUrls[0]);
        proxyStatus = "success";
      } catch {
        proxyStatus = "error";
      }
    }

    const febboxKeyStatus: Status = await testFebboxKey(febboxKey);
    const debridTokenStatus: Status =
      debridService === "torbox"
        ? await testTorboxToken(debridToken)
        : await testdebridToken(debridToken);

    return {
      extension: extensionStatus,
      proxy: proxyStatus,
      defaultProxy: "success",
      ...(conf().ALLOW_FEBBOX_KEY && {
        febboxKeyTest: febboxKeyStatus,
      }),
      debridTokenTest: debridTokenStatus,
    };
  }, [proxyUrls, febboxKey, debridToken, debridService]);

  let globalState: Status = "unset";
  if (
    value?.extension === "success" ||
    value?.proxy === "success" ||
    value?.febboxKeyTest === "success" ||
    value?.debridTokenTest === "success"
  )
    globalState = "success";
  if (
    value?.proxy === "error" ||
    value?.extension === "error" ||
    value?.febboxKeyTest === "error" ||
    value?.debridTokenTest === "error"
  )
    globalState = "error";

  return {
    setupStates: value,
    globalState,
    loading,
  };
}

function SetupCheckList(props: {
  status: Status;
  grey?: boolean;
  highlight?: boolean;
  children?: ReactNode;
}) {
  const { t } = useTranslation();
  const statusMap: Record<Status, StatusCircleProps["type"]> = {
    error: "error",
    success: "success",
    unset: "noresult",
    api_down: "error",
    invalid_token: "error",
  };

  return (
    <div className="flex items-start text-type-dimmed my-4">
      <StatusCircle
        type={statusMap[props.status]}
        className={classNames({
          "!text-video-scraping-noresult !bg-video-scraping-noresult opacity-50":
            props.grey,
          "scale-90 mr-3": true,
        })}
      />
      <div>
        <p
          className={classNames({
            "!text-white": props.grey && props.highlight,
            "!text-type-dimmed opacity-75": props.grey && !props.highlight,
            "text-type-danger": props.status === "error",
            "text-white": props.status === "success",
          })}
        >
          {props.children}
        </p>
        {props.status === "error" ? (
          <p className="max-w-96">
            {t("settings.connections.setup.itemError")}
          </p>
        ) : null}
      </div>
    </div>
  );
}

export function SetupPart() {
  const { t } = useTranslation();
  const navigate = useNavigate();
  const { loading, setupStates, globalState } = useIsSetup();
  const isDesktopApp = useIsDesktopApp();
  if (loading || !setupStates) {
    return (
      <SettingsCard>
        <div className="flex py-6 items-center justify-center">
          <Loading />
        </div>
      </SettingsCard>
    );
  }

  const textLookupMap: Record<
    Status,
    { title: string; desc: string; button: string }
  > = {
    error: {
      title: "settings.connections.setup.errorStatus.title",
      desc: "settings.connections.setup.errorStatus.description",
      button: "settings.connections.setup.redoSetup",
    },
    success: {
      title: "settings.connections.setup.successStatus.title",
      desc: "settings.connections.setup.successStatus.description",
      button: "settings.connections.setup.redoSetup",
    },
    unset: {
      title: "settings.connections.setup.unsetStatus.title",
      desc: "settings.connections.setup.unsetStatus.description",
      button: "settings.connections.setup.doSetup",
    },
    api_down: {
      title: "settings.connections.setup.errorStatus.title",
      desc: "settings.connections.setup.errorStatus.description",
      button: "settings.connections.setup.redoSetup",
    },
    invalid_token: {
      title: "settings.connections.setup.errorStatus.title",
      desc: "settings.connections.setup.errorStatus.description",
      button: "settings.connections.setup.redoSetup",
    },
  };

  return (
    <SettingsCard>
      <div className="flex flex-col md:flex-row items-start gap-4">
        <div>
          <div
            className={classNames({
              "rounded-full h-12 w-12 flex bg-opacity-15 justify-center items-center": true,
              "text-type-success bg-type-success": globalState === "success",
              "text-type-danger bg-type-danger":
                globalState === "error" || globalState === "unset",
            })}
          >
            <Icon
              icon={globalState === "success" ? Icons.CHECKMARK : Icons.X}
              className="text-xl"
            />
          </div>
        </div>
        <div className="flex-1">
          <Heading3 className="!mb-3">
            {t(textLookupMap[globalState].title)}
          </Heading3>
          <p className="max-w-[20rem] font-medium mb-6">
            {t(textLookupMap[globalState].desc)}
          </p>
          {!isDesktopApp ? (
            <>
              <SetupCheckList status={setupStates.extension}>
                {t("settings.connections.setup.items.extension")}
              </SetupCheckList>
              <SetupCheckList status={setupStates.proxy}>
                {t("settings.connections.setup.items.proxy")}
              </SetupCheckList>
              <SetupCheckList
                grey
                highlight={globalState === "unset"}
                status={setupStates.defaultProxy}
              >
                {t("settings.connections.setup.items.default")}
              </SetupCheckList>
            </>
          ) : (
            <SetupCheckList status={setupStates.extension}>
              {t("settings.connections.setup.items.desktopapp")}
            </SetupCheckList>
          )}
          {conf().ALLOW_DEBRID_KEY && (
            <SetupCheckList status={setupStates.debridTokenTest || "unset"}>
              Debrid Service
            </SetupCheckList>
          )}
          {conf().ALLOW_FEBBOX_KEY && (
            <SetupCheckList status={setupStates.febboxKeyTest || "unset"}>
              Febbox UI token
            </SetupCheckList>
          )}
        </div>
        <div className="md:mt-5">
          <Button theme="purple" onClick={() => navigate("/onboarding")}>
            {t(textLookupMap[globalState].button)}
          </Button>
        </div>
      </div>
    </SettingsCard>
  );
}