const defaultUser = "hatchet-dev";

const defaultRepo = "hatchet";

const localPaths: Record<string, string> = {
  ts: "sdks/typescript",
  py: "sdks/python",
  go: "",
};

export const extToLanguage: Record<string, string> = {
  ts: "typescript",
  py: "python",
  go: "go",
};

const defaultBranch = "main";

export type RepoProps = {
  user?: string;
  repo?: string;
  branch?: string;
  path?: string;
};

const getLocalUrl = (ext: string | undefined, { path }: RepoProps) => {
  if (!ext) return `http://localhost:4001/${path}`;
  return `http://localhost:4001/${localPaths[ext] || ''}/${path}`;
};

const isDev = process?.env?.NODE_ENV === "development";

const getRawUrl = ({ user, repo, branch, path }: RepoProps) => {
  const ext = path?.split(".").pop();
  if (isDev) {
    return getLocalUrl(ext, { path });
  }
  const extPath = ext && localPaths[ext] ? `${localPaths[ext]}/` : '';
  return `https://raw.githubusercontent.com/${user || defaultUser}/${repo || defaultRepo}/refs/heads/${branch || defaultBranch}/${extPath}${path}`;
};

const getUIUrl = ({ user, repo, branch, path }: RepoProps) => {
  const ext = path?.split(".").pop();
  if (isDev) {
    return getLocalUrl(ext, { path });
  }
  const extPath = ext && localPaths[ext] ? `${localPaths[ext]}/` : '';
  return `https://github.com/${user || defaultUser}/${repo || defaultRepo}/blob/${branch || defaultBranch}/${extPath}${path}`;
};

export type Src = {
  raw: string;
  props?: RepoProps;
  rawUrl?: string;
  githubUrl?: string;
  language?: string;
};
export const getSnippets = (
  props: RepoProps[]
): Promise<{ props: { ssg: { contents: Src[] } } }> => {
  return Promise.all(
    props.map(async (prop) => {
      const rawUrl = getRawUrl(prop);
      const githubUrl = getUIUrl(prop);
      const fileExt = prop.path?.split(".").pop() as keyof typeof extToLanguage;
      const language = extToLanguage[fileExt];

      try {
        const response = await fetch(rawUrl);
        const raw = await response.text();

        return {
          raw,
          props: prop,
          rawUrl,
          githubUrl,
          language,
        };
      } catch (error) {
        // Return object with empty raw content but preserve URLs on failure
        return {
          raw: "",
          props: prop,
          rawUrl,
          githubUrl,
          language,
        };
      }
    })
  ).then((results) => ({
    props: {
      ssg: {
        contents: results,
      },
      // revalidate every 60 seconds
      revalidate: 60,
    },
  }));
};