gui/modules/desktop/electron/libs/package.ts

import path from "path";
import { mkdirp } from "mkdirp";
import fs from "fs";
import log from "./logger";
import { getTeaPath } from "./tea-dir";
import { GUIPackage, Packages } from "../../src/libs/types";
import { isDev } from "./auto-updater";

const pkgsFilePath = path.join(getTeaPath(), "tea.xyz/gui/pkgs.json");
const pkgsFolder = path.join(getTeaPath(), "tea.xyz/gui");
const dev = isDev();

export async function writePackageCache(pkgs: Packages) {
  try {
    if (!pkgs || !Object.keys(pkgs.packages).length) {
      return;
    }

    log.info(`writing data for ${Object.keys(pkgs.packages).length} packages to ${pkgsFilePath}`);
    await mkdirp(pkgsFolder);
    fs.writeFileSync(pkgsFilePath, JSON.stringify(pkgs), {
      encoding: "utf-8"
    });
  } catch (error) {
    log.error(error);
  }
}

export async function loadPackageCache(): Promise<Packages> {
  try {
    log.info(`loading package cache from ${pkgsFilePath}`);
    const pkgData = fs.readFileSync(pkgsFilePath);
    const pkgs = JSON.parse(pkgData.toString()) as Packages;
    if (pkgs?.packages) {
      // Remove any temporary properties that may have been added to the package (like installation progress)
      for (const [key, value] of Object.entries(pkgs.packages)) {
        const { install_progress_percentage, isUninstalling, synced, displayState, ...rest } =
          value;
        pkgs.packages[key] = rest as GUIPackage;
        // possible user deletes cache files
        delete pkgs.packages[key].cached_image_url;
        if (rest.image_added_at) {
          const prefix = `https://gui.tea.xyz/${dev ? "dev" : "prod"}/${rest.full_name}`;
          pkgs.packages[key].image_128_url = `${prefix}/128x128.webp`;
          pkgs.packages[key].image_512_url = `${prefix}/512x512.webp`;
        }
      }
    }
    return pkgs;
  } catch (err) {
    if (err.code !== "ENOENT") {
      log.error(err);
    }
    return { version: "1", packages: {} };
  }
}

export const nameToSlug = (name: string) => {
  // github.com/Pypa/twine -> github_com_pypa_twine
  const [nameOnly] = name.split("@");
  const slug = nameOnly.replace(/[^\w\s]/gi, "_").toLocaleLowerCase();
  return slug;
};

export function isInstalled(fullName: string) {
  const folderPath = path.join(getTeaPath(), fullName);
  return fs.existsSync(folderPath);
}