@bastani/atomic 0.9.16-alpha.2 → 0.9.16-alpha.4

package/dist/builtin/web-access/github-config.ts

@@ -1,156 +0,0 @@
-import { existsSync, readFileSync } from "node:fs";
-import { findReadableConfigPath } from "./config-paths.ts";
-
-const CONFIG_PATH = findReadableConfigPath();
-
-export const BINARY_EXTENSIONS = new Set([
-	".png", ".jpg", ".jpeg", ".gif", ".bmp", ".ico", ".webp", ".svg", ".tiff", ".tif",
-	".mp3", ".mp4", ".avi", ".mov", ".mkv", ".flv", ".wmv", ".wav", ".ogg", ".webm", ".flac", ".aac",
-	".zip", ".tar", ".gz", ".bz2", ".xz", ".7z", ".rar", ".zst",
-	".exe", ".dll", ".so", ".dylib", ".bin", ".o", ".a", ".lib",
-	".woff", ".woff2", ".ttf", ".otf", ".eot",
-	".pdf", ".doc", ".docx", ".xls", ".xlsx", ".ppt", ".pptx",
-	".sqlite", ".db", ".sqlite3",
-	".pyc", ".pyo", ".class", ".jar", ".war",
-	".iso", ".img", ".dmg",
-]);
-
-export const NOISE_DIRS = new Set([
-	"node_modules", "vendor", ".next", "dist", "build", "__pycache__",
-	".venv", "venv", ".tox", ".mypy_cache", ".pytest_cache",
-	"target", ".gradle", ".idea", ".vscode",
-]);
-
-export const MAX_INLINE_FILE_CHARS = 100_000;
-export const MAX_TREE_ENTRIES = 200;
-
-export interface GitHubUrlInfo {
-	owner: string;
-	repo: string;
-	ref?: string;
-	refIsFullSha: boolean;
-	path?: string;
-	type: "root" | "blob" | "tree";
-}
-
-export interface GitHubCloneConfig {
-	enabled: boolean;
-	maxRepoSizeMB: number;
-	cloneTimeoutSeconds: number;
-	clonePath: string;
-}
-
-let cachedConfig: GitHubCloneConfig | null = null;
-
-function normalizeEnabled(value: unknown, fallback: boolean): boolean {
-	return typeof value === "boolean" ? value : fallback;
-}
-
-function normalizePositiveNumber(value: unknown, fallback: number): number {
-	if (typeof value !== "number" || !Number.isFinite(value)) return fallback;
-	return value > 0 ? value : fallback;
-}
-
-function normalizeClonePath(value: unknown, fallback: string): string {
-	if (typeof value !== "string") return fallback;
-	const normalized = value.trim();
-	return normalized.length > 0 ? normalized : fallback;
-}
-
-export function loadGitHubConfig(): GitHubCloneConfig {
-	if (cachedConfig) return cachedConfig;
-
-	const defaults: GitHubCloneConfig = {
-		enabled: true,
-		maxRepoSizeMB: 350,
-		cloneTimeoutSeconds: 30,
-		clonePath: "/tmp/atomic-github-repos",
-	};
-
-	if (!existsSync(CONFIG_PATH)) {
-		cachedConfig = defaults;
-		return cachedConfig;
-	}
-
-	const rawText = readFileSync(CONFIG_PATH, "utf-8");
-	let raw: { githubClone?: { enabled?: unknown; maxRepoSizeMB?: unknown; cloneTimeoutSeconds?: unknown; clonePath?: unknown } };
-	try {
-		raw = JSON.parse(rawText) as { githubClone?: { enabled?: unknown; maxRepoSizeMB?: unknown; cloneTimeoutSeconds?: unknown; clonePath?: unknown } };
-	} catch (err) {
-		const message = err instanceof Error ? err.message : String(err);
-		throw new Error(`Failed to parse ${CONFIG_PATH}: ${message}`);
-	}
-
-	const gc = raw.githubClone ?? {};
-	cachedConfig = {
-		enabled: normalizeEnabled(gc.enabled, defaults.enabled),
-		maxRepoSizeMB: normalizePositiveNumber(gc.maxRepoSizeMB, defaults.maxRepoSizeMB),
-		cloneTimeoutSeconds: normalizePositiveNumber(gc.cloneTimeoutSeconds, defaults.cloneTimeoutSeconds),
-		clonePath: normalizeClonePath(gc.clonePath, defaults.clonePath),
-	};
-	return cachedConfig;
-}
-
-const NON_CODE_SEGMENTS = new Set([
-	"issues", "pull", "pulls", "discussions", "releases", "wiki",
-	"actions", "settings", "security", "projects", "graphs",
-	"compare", "commits", "tags", "branches", "stargazers",
-	"watchers", "network", "forks", "milestone", "labels",
-	"packages", "codespaces", "contribute", "community",
-	"sponsors", "invitations", "notifications", "insights",
-]);
-
-export function parseGitHubUrl(url: string): GitHubUrlInfo | null {
-	let parsed: URL;
-	try {
-		parsed = new URL(url);
-	} catch {
-		return null;
-	}
-
-	const host = parsed.hostname.toLowerCase();
-	if (host !== "github.com" && host !== "www.github.com") return null;
-
-	const segments = parsed.pathname
-		.split("/")
-		.filter(Boolean)
-		.map((segment) => {
-			try {
-				return decodeURIComponent(segment);
-			} catch {
-				return segment;
-			}
-		});
-	if (segments.length < 2) return null;
-
-	const owner = segments[0];
-	const repo = segments[1].replace(/\.git$/, "");
-
-	if (NON_CODE_SEGMENTS.has(segments[2]?.toLowerCase())) return null;
-
-	if (segments.length === 2) {
-		return { owner, repo, refIsFullSha: false, type: "root" };
-	}
-
-	const action = segments[2];
-	if (action !== "blob" && action !== "tree") return null;
-	if (segments.length < 4) return null;
-
-	const ref = segments[3];
-	const refIsFullSha = /^[0-9a-f]{40}$/.test(ref);
-	const pathParts = segments.slice(4);
-	const path = pathParts.length > 0 ? pathParts.join("/") : "";
-
-	return {
-		owner,
-		repo,
-		ref,
-		refIsFullSha,
-		path,
-		type: action as "blob" | "tree",
-	};
-}
-
-export function resetGitHubConfig(): void {
-	cachedConfig = null;
-}

package/dist/builtin/web-access/github-extract.ts

@@ -1,487 +0,0 @@
-import { existsSync, readFileSync, rmSync, statSync, readdirSync, openSync, readSync, closeSync, realpathSync } from "node:fs";
-import { execFile } from "node:child_process";
-import { extname, join, resolve as resolvePath, sep as pathSep } from "node:path";
-import { createChildProcessEnvironment } from "@bastani/atomic";
-import { activityMonitor } from "./activity.js";
-import type { ExtractedContent } from "./extract.js";
-import { flattenTruncatedString } from "./flat-string.js";
-import { checkGhAvailable, checkRepoSize, fetchViaApi, showGhHint } from "./github-api.js";
-import { BINARY_EXTENSIONS, MAX_INLINE_FILE_CHARS, MAX_TREE_ENTRIES, NOISE_DIRS, loadGitHubConfig, parseGitHubUrl, resetGitHubConfig, type GitHubCloneConfig, type GitHubUrlInfo } from "./github-config.js";
-export { parseGitHubUrl, type GitHubUrlInfo } from "./github-config.js";
-
-interface CachedClone {
-	localPath: string;
-	clonePromise: Promise<string | null>;
-}
-
-const cloneCache = new Map<string, CachedClone>();
-
-function cacheKey(owner: string, repo: string, ref?: string): string {
-	return ref ? `${owner}/${repo}@${ref}` : `${owner}/${repo}`;
-}
-
-function cloneDir(config: GitHubCloneConfig, owner: string, repo: string, ref?: string): string {
-	const dirName = ref ? `${repo}@${ref}` : repo;
-	return join(config.clonePath, owner, dirName);
-}
-
-function execClone(args: string[], localPath: string, timeoutMs: number, signal?: AbortSignal): Promise<string | null> {
-	return new Promise((resolve) => {
-		const child = execFile(args[0], args.slice(1), { timeout: timeoutMs, env: createChildProcessEnvironment() }, (err) => {
-			if (err) {
-				try {
-					rmSync(localPath, { recursive: true, force: true });
-				} catch {
-				}
-				resolve(null);
-				return;
-			}
-			resolve(localPath);
-		});
-
-		if (signal) {
-			const onAbort = () => child.kill();
-			signal.addEventListener("abort", onAbort, { once: true });
-			child.on("exit", () => signal.removeEventListener("abort", onAbort));
-		}
-	});
-}
-
-async function cloneRepo(
-	owner: string,
-	repo: string,
-	ref: string | undefined,
-	config: GitHubCloneConfig,
-	signal?: AbortSignal,
-): Promise<string | null> {
-	const localPath = cloneDir(config, owner, repo, ref);
-
-	try {
-		rmSync(localPath, { recursive: true, force: true });
-	} catch {
-	}
-
-	const timeoutMs = config.cloneTimeoutSeconds * 1000;
-	const hasGh = await checkGhAvailable();
-
-	if (hasGh) {
-		const args = ["gh", "repo", "clone", `${owner}/${repo}`, localPath, "--", "--depth", "1", "--single-branch"];
-		if (ref) args.push("--branch", ref);
-		return execClone(args, localPath, timeoutMs, signal);
-	}
-
-	showGhHint();
-
-	const gitUrl = `https://github.com/${owner}/${repo}.git`;
-	const args = ["git", "clone", "--depth", "1", "--single-branch"];
-	if (ref) args.push("--branch", ref);
-	args.push(gitUrl, localPath);
-	return execClone(args, localPath, timeoutMs, signal);
-}
-
-function isBinaryFile(filePath: string): boolean {
-	const ext = extname(filePath).toLowerCase();
-	if (BINARY_EXTENSIONS.has(ext)) return true;
-
-	let fd: number;
-	try {
-		fd = openSync(filePath, "r");
-	} catch {
-		return false;
-	}
-	try {
-		const buf = Buffer.alloc(512);
-		const bytesRead = readSync(fd, buf, 0, 512, 0);
-		for (let i = 0; i < bytesRead; i++) {
-			if (buf[i] === 0) return true;
-		}
-	} catch {
-		return false;
-	} finally {
-		closeSync(fd);
-	}
-
-	return false;
-}
-
-function formatFileSize(bytes: number): string {
-	if (bytes < 1024) return `${bytes} B`;
-	if (bytes < 1024 * 1024) return `${(bytes / 1024).toFixed(1)} KB`;
-	return `${(bytes / (1024 * 1024)).toFixed(1)} MB`;
-}
-
-function resolveWithinRepo(rootPath: string, relativePath: string): string | null {
-	const normalizedRoot = resolvePath(rootPath);
-	const candidate = resolvePath(normalizedRoot, relativePath);
-	if (candidate !== normalizedRoot) {
-		const rootPrefix = normalizedRoot.endsWith(pathSep) ? normalizedRoot : normalizedRoot + pathSep;
-		if (!candidate.startsWith(rootPrefix)) return null;
-	}
-
-	if (!existsSync(candidate)) return candidate;
-
-	try {
-		const realRoot = realpathSync(normalizedRoot);
-		const realCandidate = realpathSync(candidate);
-		if (realCandidate === realRoot) return candidate;
-		const realRootPrefix = realRoot.endsWith(pathSep) ? realRoot : realRoot + pathSep;
-		return realCandidate.startsWith(realRootPrefix) ? candidate : null;
-	} catch {
-		return null;
-	}
-}
-
-function readTextFile(path: string): string | null {
-	try {
-		return readFileSync(path, "utf-8");
-	} catch {
-		return null;
-	}
-}
-
-function buildTree(rootPath: string): string {
-	const entries: string[] = [];
-
-	function walk(dir: string, relPath: string): void {
-		if (entries.length >= MAX_TREE_ENTRIES) return;
-
-		let items: string[];
-		try {
-			items = readdirSync(dir).sort();
-		} catch {
-			return;
-		}
-
-		for (const item of items) {
-			if (entries.length >= MAX_TREE_ENTRIES) return;
-			if (item === ".git") continue;
-
-			const rel = relPath ? `${relPath}/${item}` : item;
-			const safePath = resolveWithinRepo(rootPath, rel);
-			if (!safePath) {
-				entries.push(`${rel}  [outside repo skipped]`);
-				continue;
-			}
-
-			let stat;
-			try {
-				stat = statSync(safePath);
-			} catch {
-				continue;
-			}
-
-			if (stat.isDirectory()) {
-				if (NOISE_DIRS.has(item)) {
-					entries.push(`${rel}/  [skipped]`);
-					continue;
-				}
-				entries.push(`${rel}/`);
-				walk(safePath, rel);
-			} else {
-				entries.push(rel);
-			}
-		}
-	}
-
-	walk(rootPath, "");
-
-	if (entries.length >= MAX_TREE_ENTRIES) {
-		entries.push(`... (truncated at ${MAX_TREE_ENTRIES} entries)`);
-	}
-
-	return entries.join("\n");
-}
-
-function buildDirListing(rootPath: string, subPath: string): string {
-	const targetPath = resolveWithinRepo(rootPath, subPath);
-	if (!targetPath) return "(path escapes repository root)";
-	const lines: string[] = [];
-
-	let items: string[];
-	try {
-		items = readdirSync(targetPath).sort();
-	} catch {
-		return "(directory not readable)";
-	}
-
-	for (const item of items) {
-		if (item === ".git") continue;
-		const rel = subPath ? `${subPath}/${item}` : item;
-		const safePath = resolveWithinRepo(rootPath, rel);
-		if (!safePath) {
-			lines.push(`  ${item}  (outside repo)`);
-			continue;
-		}
-		try {
-			const stat = statSync(safePath);
-			if (stat.isDirectory()) {
-				lines.push(`  ${item}/`);
-			} else {
-				lines.push(`  ${item}  (${formatFileSize(stat.size)})`);
-			}
-		} catch {
-			lines.push(`  ${item}  (unreadable)`);
-		}
-	}
-
-	return lines.join("\n");
-}
-
-export function readReadme(localPath: string): string | null {
-	const candidates = ["README.md", "readme.md", "README", "README.txt", "README.rst"];
-	for (const name of candidates) {
-		const readmePath = join(localPath, name);
-		if (existsSync(readmePath)) {
-			try {
-				const content = readFileSync(readmePath, "utf-8");
-				return content.length > 8192 ? flattenTruncatedString(content.slice(0, 8192) + "\n\n[README truncated at 8K chars]") : content;
-			} catch {
-				continue;
-			}
-		}
-	}
-	return null;
-}
-
-function generateContent(localPath: string, info: GitHubUrlInfo): string {
-	const lines: string[] = [];
-	lines.push(`Repository cloned to: ${localPath}`);
-	lines.push("");
-
-	if (info.type === "root") {
-		lines.push("## Structure");
-		lines.push(buildTree(localPath));
-		lines.push("");
-
-		const readme = readReadme(localPath);
-		if (readme) {
-			lines.push("## README.md");
-			lines.push(readme);
-			lines.push("");
-		}
-
-		lines.push("Use `read` and `bash` tools at the path above to explore further.");
-		return lines.join("\n");
-	}
-
-	if (info.type === "tree") {
-		const dirPath = info.path || "";
-		const fullDirPath = resolveWithinRepo(localPath, dirPath);
-
-		if (!fullDirPath || !existsSync(fullDirPath)) {
-			lines.push(`Path \`${dirPath}\` not found in clone. Showing repository root instead.`);
-			lines.push("");
-			lines.push("## Structure");
-			lines.push(buildTree(localPath));
-		} else {
-			lines.push(`## ${dirPath || "/"}`);
-			lines.push(buildDirListing(localPath, dirPath));
-		}
-
-		lines.push("");
-		lines.push("Use `read` and `bash` tools at the path above to explore further.");
-		return lines.join("\n");
-	}
-
-	if (info.type === "blob") {
-		const filePath = info.path || "";
-		const fullFilePath = resolveWithinRepo(localPath, filePath);
-
-		if (!fullFilePath || !existsSync(fullFilePath)) {
-			lines.push(`Path \`${filePath}\` not found in clone. Showing repository root instead.`);
-			lines.push("");
-			lines.push("## Structure");
-			lines.push(buildTree(localPath));
-			lines.push("");
-			lines.push("Use `read` and `bash` tools at the path above to explore further.");
-			return lines.join("\n");
-		}
-
-		let stat: ReturnType<typeof statSync>;
-		try {
-			stat = statSync(fullFilePath);
-		} catch (err) {
-			const message = err instanceof Error ? err.message : String(err);
-			lines.push(`Could not inspect \`${filePath}\`: ${message}`);
-			lines.push("");
-			lines.push("Use `read` and `bash` tools at the path above to explore further.");
-			return lines.join("\n");
-		}
-
-		if (stat.isDirectory()) {
-			lines.push(`## ${filePath || "/"}`);
-			lines.push(buildDirListing(localPath, filePath));
-			lines.push("");
-			lines.push("Use `read` and `bash` tools at the path above to explore further.");
-			return lines.join("\n");
-		}
-
-		if (isBinaryFile(fullFilePath)) {
-			const ext = extname(filePath).replace(".", "");
-			lines.push(`## ${filePath}`);
-			lines.push(`Binary file (${ext}, ${formatFileSize(stat.size)}). Use \`read\` or \`bash\` tools at the path above to inspect.`);
-			return lines.join("\n");
-		}
-
-		const content = readTextFile(fullFilePath);
-		if (content === null) {
-			lines.push(`Could not read \`${filePath}\` as UTF-8 text.`);
-			lines.push("");
-			lines.push("Use `read` and `bash` tools at the path above to explore further.");
-			return lines.join("\n");
-		}
-		lines.push(`## ${filePath}`);
-
-		if (content.length > MAX_INLINE_FILE_CHARS) {
-			lines.push(content.slice(0, MAX_INLINE_FILE_CHARS));
-			lines.push("");
-			lines.push(`[File truncated at 100K chars. Full file: ${fullFilePath}]`);
-		} else {
-			lines.push(content);
-		}
-
-		lines.push("");
-		lines.push("Use `read` and `bash` tools at the path above to explore further.");
-		return lines.join("\n");
-	}
-
-	return lines.join("\n");
-}
-
-async function awaitCachedClone(
-	cached: CachedClone,
-	url: string,
-	owner: string,
-	repo: string,
-	info: GitHubUrlInfo,
-	signal?: AbortSignal,
-): Promise<ExtractedContent | null> {
-	if (signal?.aborted) return null;
-	const result = await cached.clonePromise;
-	if (signal?.aborted) return null;
-	if (result) {
-		const content = generateContent(result, info);
-		const title = info.path ? `${owner}/${repo} - ${info.path}` : `${owner}/${repo}`;
-		return { url, title, content, error: null };
-	}
-	return fetchViaApi(url, owner, repo, info);
-}
-
-export async function extractGitHub(
-	url: string,
-	signal?: AbortSignal,
-	forceClone?: boolean,
-): Promise<ExtractedContent | null> {
-	const info = parseGitHubUrl(url);
-	if (!info) return null;
-
-	if (signal?.aborted) return null;
-
-	const config = loadGitHubConfig();
-	if (!config.enabled) return null;
-
-	const { owner, repo } = info;
-	const key = cacheKey(owner, repo, info.ref);
-
-	const cached = cloneCache.get(key);
-	if (cached) return awaitCachedClone(cached, url, owner, repo, info, signal);
-
-	if (info.refIsFullSha) {
-		if (signal?.aborted) return null;
-		const sizeNote = `Note: Commit SHA URLs use the GitHub API instead of cloning.`;
-		return fetchViaApi(url, owner, repo, info, sizeNote);
-	}
-
-	const activityId = activityMonitor.logStart({ type: "fetch", url: `github.com/${owner}/${repo}` });
-
-	if (!forceClone) {
-		const sizeKB = await checkRepoSize(owner, repo);
-		if (signal?.aborted) {
-			activityMonitor.logComplete(activityId, 0);
-			return null;
-		}
-		if (sizeKB !== null) {
-			const sizeMB = sizeKB / 1024;
-			if (sizeMB > config.maxRepoSizeMB) {
-				if (signal?.aborted) {
-					activityMonitor.logComplete(activityId, 0);
-					return null;
-				}
-				const sizeNote =
-					`Note: Repository is ${Math.round(sizeMB)}MB (threshold: ${config.maxRepoSizeMB}MB). ` +
-					`Showing API-fetched content instead of full clone. Ask the user if they'd like to clone the full repo -- ` +
-					`if yes, call fetch_content again with the same URL and add forceClone: true to the params.`;
-				const apiView = await fetchViaApi(url, owner, repo, info, sizeNote);
-				if (apiView) {
-					activityMonitor.logComplete(activityId, 200);
-					return apiView;
-				}
-				activityMonitor.logError(activityId, "api fallback unavailable for oversized repository");
-				return null;
-			}
-		}
-	}
-
-	if (signal?.aborted) {
-		activityMonitor.logComplete(activityId, 0);
-		return null;
-	}
-
-	// Re-check: another concurrent caller may have started a clone while we awaited the size check
-	const cachedAfterSizeCheck = cloneCache.get(key);
-	if (cachedAfterSizeCheck) {
-		const cachedResult = await awaitCachedClone(cachedAfterSizeCheck, url, owner, repo, info, signal);
-		if (signal?.aborted) {
-			activityMonitor.logComplete(activityId, 0);
-		} else if (cachedResult) {
-			activityMonitor.logComplete(activityId, 200);
-		} else {
-			activityMonitor.logError(activityId, "clone failed");
-		}
-		return cachedResult;
-	}
-
-	const clonePromise = cloneRepo(owner, repo, info.ref, config, signal);
-	const localPath = cloneDir(config, owner, repo, info.ref);
-	cloneCache.set(key, { localPath, clonePromise });
-
-	const result = await clonePromise;
-	if (signal?.aborted) {
-		if (!result) cloneCache.delete(key);
-		activityMonitor.logComplete(activityId, 0);
-		return null;
-	}
-
-	if (!result) {
-		cloneCache.delete(key);
-		if (signal?.aborted) {
-			activityMonitor.logComplete(activityId, 0);
-			return null;
-		}
-
-		const apiFallback = await fetchViaApi(url, owner, repo, info);
-		if (apiFallback) {
-			activityMonitor.logComplete(activityId, 200);
-			return apiFallback;
-		}
-
-		activityMonitor.logError(activityId, "clone and API fallback failed");
-		return null;
-	}
-
-	activityMonitor.logComplete(activityId, 200);
-	const content = generateContent(result, info);
-	const title = info.path ? `${owner}/${repo} - ${info.path}` : `${owner}/${repo}`;
-	return { url, title, content, error: null };
-}
-
-export function clearCloneCache(): void {
-	for (const entry of cloneCache.values()) {
-		try {
-			rmSync(entry.localPath, { recursive: true, force: true });
-		} catch {
-		}
-	}
-	cloneCache.clear();
-	resetGitHubConfig();
-}