@mars167/git-ai 2.3.0

package/dist/src/core/workspace.js

@@ -0,0 +1,116 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ensureProjectRepo = ensureProjectRepo;
+exports.ensureRepoIndexReady = ensureRepoIndexReady;
+exports.queryManifestWorkspace = queryManifestWorkspace;
+const fs_extra_1 = __importDefault(require("fs-extra"));
+const path_1 = __importDefault(require("path"));
+const simple_git_1 = __importDefault(require("simple-git"));
+const git_1 = require("./git");
+const archive_1 = require("./archive");
+const lancedb_1 = require("./lancedb");
+const indexer_1 = require("./indexer");
+const manifest_1 = require("./manifest");
+const indexCheck_1 = require("./indexCheck");
+function sanitizeSegment(s) {
+    return s.replace(/[^a-zA-Z0-9._-]+/g, '_').slice(0, 120);
+}
+async function isGitRepo(dir) {
+    return fs_extra_1.default.pathExists(path_1.default.join(dir, '.git'));
+}
+async function ensureCheckedOut(repoDir, revision) {
+    if (!revision)
+        return;
+    const git = (0, simple_git_1.default)(repoDir);
+    await git.fetch(['--all', '--tags']).catch(() => undefined);
+    try {
+        await git.checkout(revision);
+        return;
+    }
+    catch {
+        await git.checkout(['-f', `origin/${revision}`]);
+    }
+}
+async function ensureProjectRepo(params) {
+    const workspaceRoot = (0, git_1.inferWorkspaceRoot)(params.manifestRepoRoot);
+    if (workspaceRoot) {
+        const direct = path_1.default.resolve(workspaceRoot, params.project.path);
+        if (await isGitRepo(direct)) {
+            await ensureCheckedOut(direct, params.revision);
+            return { repoDir: direct, from: 'workspace' };
+        }
+    }
+    const cacheBase = path_1.default.join(params.manifestRepoRoot, '.git-ai', 'workspace-cache');
+    const cacheDir = path_1.default.join(cacheBase, sanitizeSegment(params.project.path || params.project.name));
+    if (await isGitRepo(cacheDir)) {
+        await ensureCheckedOut(cacheDir, params.revision);
+        return { repoDir: cacheDir, from: 'cache' };
+    }
+    if (!params.fetch)
+        throw new Error(`No remote fetch configured for project: ${params.project.name}`);
+    await fs_extra_1.default.ensureDir(cacheBase);
+    const url = (0, manifest_1.buildCloneUrl)(params.fetch, params.project.name);
+    await (0, simple_git_1.default)().clone(url, cacheDir);
+    await ensureCheckedOut(cacheDir, params.revision);
+    return { repoDir: cacheDir, from: 'cloned' };
+}
+async function ensureRepoIndexReady(repoRoot, dim = 256) {
+    const dbDir = (0, lancedb_1.defaultDbDir)(repoRoot);
+    const lancedbDir = path_1.default.join(dbDir);
+    if (await fs_extra_1.default.pathExists(lancedbDir))
+        return { source: 'present' };
+    const archive = path_1.default.join(repoRoot, '.git-ai', 'lancedb.tar.gz');
+    if (await fs_extra_1.default.pathExists(archive)) {
+        await (0, archive_1.unpackLanceDb)(repoRoot);
+        return { source: 'unpacked' };
+    }
+    const indexer = new indexer_1.IndexerV2({ repoRoot, scanRoot: repoRoot, dim, overwrite: true });
+    await indexer.run();
+    return { source: 'built' };
+}
+async function queryManifestWorkspace(params) {
+    const workspaceRoot = (0, git_1.inferWorkspaceRoot)(params.manifestRepoRoot);
+    if (!workspaceRoot)
+        throw new Error('Not a manifests repo (.repo/manifests)');
+    const manifest = await (0, manifest_1.loadRepoManifestFromManifestsRepo)(params.manifestRepoRoot);
+    const q = params.keyword.replace(/'/g, "''");
+    const rows = [];
+    for (const project of manifest.projects) {
+        const fetch = (0, manifest_1.resolveProjectFetch)(project, manifest);
+        const revision = (0, manifest_1.resolveProjectRevision)(project, manifest);
+        const ensured = await ensureProjectRepo({ manifestRepoRoot: params.manifestRepoRoot, project, fetch, revision });
+        await ensureRepoIndexReady(ensured.repoDir, 256);
+        const status = await (0, indexCheck_1.checkIndex)(ensured.repoDir);
+        if (!status.ok) {
+            const indexer = new indexer_1.IndexerV2({ repoRoot: ensured.repoDir, scanRoot: ensured.repoDir, dim: 256, overwrite: true });
+            await indexer.run();
+        }
+        const dbDir = (0, lancedb_1.defaultDbDir)(ensured.repoDir);
+        const status2 = await (0, indexCheck_1.checkIndex)(ensured.repoDir);
+        const langs = (0, indexCheck_1.resolveLangs)(status2.found.meta ?? null, 'all');
+        const { byLang } = await (0, lancedb_1.openTablesByLang)({ dbDir, dim: 256, mode: 'open_only', languages: langs });
+        for (const lang of langs) {
+            const t = byLang[lang];
+            if (!t)
+                continue;
+            const projectRows = await t.refs.query().where(`symbol ILIKE '%${q}%'`).limit(params.limit).toArray();
+            for (const r of projectRows) {
+                rows.push({
+                    project: { name: project.name, path: project.path, repoRoot: ensured.repoDir, from: ensured.from },
+                    lang,
+                    ...r,
+                });
+                if (rows.length >= params.limit)
+                    break;
+            }
+            if (rows.length >= params.limit)
+                break;
+        }
+        if (rows.length >= params.limit)
+            break;
+    }
+    return { repoRoot: params.manifestRepoRoot, workspaceRoot, count: rows.length, rows };
+}