Spaces:
Running
Running
| import { access, mkdir } from "node:fs/promises"; | |
| import path from "node:path"; | |
| import { simpleGit } from "simple-git"; | |
| import { brainapiHome, sourcePath } from "./paths.js"; | |
| import { updateState } from "./state.js"; | |
| export interface CloneOptions { | |
| repoUrl: string; | |
| branch: string; | |
| depth?: number; | |
| onProgress?: (line: string) => void; | |
| } | |
| export async function isClonedRepo(): Promise<boolean> { | |
| try { | |
| await access(path.join(sourcePath(), ".git")); | |
| return true; | |
| } catch { | |
| return false; | |
| } | |
| } | |
| export async function cloneRepo(opts: CloneOptions): Promise<void> { | |
| await mkdir(brainapiHome(), { recursive: true }); | |
| const target = sourcePath(); | |
| const git = simpleGit({ | |
| progress: ({ method, stage, progress }) => { | |
| opts.onProgress?.(`${method} ${stage} ${progress}%`); | |
| }, | |
| }); | |
| await git.clone(opts.repoUrl, target, [ | |
| "--branch", | |
| opts.branch, | |
| ...(opts.depth ? ["--depth", String(opts.depth)] : []), | |
| ]); | |
| await updateState({ cloned: true, repoUrl: opts.repoUrl, branch: opts.branch }); | |
| } | |
| export async function pullRepo(branch: string): Promise<void> { | |
| const git = simpleGit(sourcePath()); | |
| const remoteRef = `origin/${branch}`; | |
| await git.fetch("origin", branch); | |
| await git.clean("f", ["-d", "-e", ".env", "-e", ".venv"]); | |
| await git.checkout(["-B", branch, "-f"]); | |
| await git.reset(["--hard", remoteRef]); | |
| } | |
| export async function ensureRepo(opts: CloneOptions): Promise<"cloned" | "pulled"> { | |
| if (await isClonedRepo()) { | |
| await pullRepo(opts.branch); | |
| return "pulled"; | |
| } | |
| await cloneRepo(opts); | |
| return "cloned"; | |
| } | |