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

import fs from "fs";
import { getGuiPath } from "./tea-dir";
import log from "./logger";
import { cliBinPath, asyncExec } from "./cli";
import { createInitialSessionFile } from "./auth";
import { SemVer, isValidSemVer } from "@tea/libtea";

const MINIMUM_TEA_VERSION = "0.31.2";

const destinationDirectory = getGuiPath();

// TODO: move device_id generation here

// Get the binary path from the current app directory
const binaryUrl = "https://tea.xyz/$(uname)/$(uname -m)";

let initializePromise: Promise<string> | null = null;

export async function initializeTeaCli(): Promise<string> {
  if (initializePromise) {
    return initializePromise;
  }

  log.info("Initializing tea cli");
  initializePromise = initializeTeaCliInternal();

  initializePromise.catch((error) => {
    log.info("Error initializing tea cli, resetting promise:", error);
    initializePromise = null;
  });

  return initializePromise;
}

async function initializeTeaCliInternal(): Promise<string> {
  let binCheck = "";
  let needsUpdate = false;

  // Create the destination directory if it doesn't exist
  if (!fs.existsSync(destinationDirectory)) {
    fs.mkdirSync(destinationDirectory, { recursive: true });
  }

  // replace this with max's pr
  const curlCommand = `curl --insecure -L -o "${cliBinPath}" "${binaryUrl}"`;

  const exists = fs.existsSync(cliBinPath);
  if (exists) {
    log.info("binary tea already exists at", cliBinPath);
    try {
      binCheck = await asyncExec(`cd ${destinationDirectory} && ./tea --version`);
      const teaVersion = binCheck.toString().split(" ")[1].trim();
      if (new SemVer(teaVersion).compare(new SemVer(MINIMUM_TEA_VERSION)) < 0) {
        log.info("binary tea version is too old, updating");
        needsUpdate = true;
      }
    } catch (error) {
      // probably binary is not executable or no permission
      log.error("Error checking tea binary version:", error);
      needsUpdate = true;
      await asyncExec(`cd ${destinationDirectory} && rm tea`);
    }
  }

  if (!exists || needsUpdate) {
    await asyncExec(curlCommand);
    log.info("Binary downloaded and saved to", cliBinPath);
    await asyncExec("chmod u+x " + cliBinPath);
    log.info("Binary is now ready for use at", cliBinPath);
    binCheck = await asyncExec(`cd ${destinationDirectory} && ./tea --version`);
  }

  const version = binCheck.toString().split(" ")[1];
  log.info("binary tea version:", version);
  return isValidSemVer(version.trim()) ? version : "";
}

export default async function initialize(): Promise<string> {
  const [version] = await Promise.all([initializeTeaCli(), createInitialSessionFile()]);
  return version;
}