@oh-my-pi/pi-coding-agent 3.25.0 → 3.30.0

package/src/core/tools/web-fetch-handlers/gitlab.ts

@@ -0,0 +1,444 @@
+import {
+	finalizeOutput,
+	formatCount,
+	htmlToBasicMarkdown,
+	loadPage,
+	type RenderResult,
+	type SpecialHandler,
+} from "./types";
+
+interface GitLabUrl {
+	namespace: string;
+	project: string;
+	type: "repo" | "blob" | "tree" | "issue" | "merge_request";
+	ref?: string;
+	path?: string;
+	id?: number;
+}
+
+/**
+ * Parse GitLab URL into structured data
+ */
+function parseGitLabUrl(url: string): GitLabUrl | null {
+	try {
+		const parsed = new URL(url);
+		if (parsed.hostname !== "gitlab.com") return null;
+
+		const segments = parsed.pathname.split("/").filter(Boolean);
+		if (segments.length < 2) return null;
+
+		const [namespace, project, ...rest] = segments;
+
+		// Repo root
+		if (rest.length === 0) {
+			return { namespace, project, type: "repo" };
+		}
+
+		// Skip - prefix
+		if (rest[0] !== "-") return null;
+
+		const [, type, ...remaining] = rest;
+
+		// File: gitlab.com/{ns}/{proj}/-/blob/{ref}/{path}
+		if (type === "blob" && remaining.length >= 2) {
+			const [ref, ...pathParts] = remaining;
+			return {
+				namespace,
+				project,
+				type: "blob",
+				ref,
+				path: pathParts.join("/"),
+			};
+		}
+
+		// Directory: gitlab.com/{ns}/{proj}/-/tree/{ref}/{path}
+		if (type === "tree" && remaining.length >= 1) {
+			const [ref, ...pathParts] = remaining;
+			return {
+				namespace,
+				project,
+				type: "tree",
+				ref,
+				path: pathParts.length > 0 ? pathParts.join("/") : undefined,
+			};
+		}
+
+		// Issue: gitlab.com/{ns}/{proj}/-/issues/{id}
+		if (type === "issues" && remaining.length === 1) {
+			const id = parseInt(remaining[0], 10);
+			if (Number.isNaN(id)) return null;
+			return { namespace, project, type: "issue", id };
+		}
+
+		// MR: gitlab.com/{ns}/{proj}/-/merge_requests/{id}
+		if (type === "merge_requests" && remaining.length === 1) {
+			const id = parseInt(remaining[0], 10);
+			if (Number.isNaN(id)) return null;
+			return { namespace, project, type: "merge_request", id };
+		}
+
+		return null;
+	} catch {
+		return null;
+	}
+}
+
+/**
+ * Get project ID from namespace/project path
+ */
+async function getProjectId(gl: GitLabUrl, timeout: number): Promise<number | null> {
+	const encodedPath = encodeURIComponent(`${gl.namespace}/${gl.project}`);
+	const apiUrl = `https://gitlab.com/api/v4/projects/${encodedPath}`;
+
+	const result = await loadPage(apiUrl, { timeout });
+	if (!result.ok) return null;
+
+	try {
+		const data = JSON.parse(result.content) as { id: number };
+		return data.id;
+	} catch {
+		return null;
+	}
+}
+
+/**
+ * Render GitLab repository
+ */
+async function renderGitLabRepo(gl: GitLabUrl, timeout: number): Promise<{ content: string; ok: boolean }> {
+	const encodedPath = encodeURIComponent(`${gl.namespace}/${gl.project}`);
+	const apiUrl = `https://gitlab.com/api/v4/projects/${encodedPath}`;
+
+	const result = await loadPage(apiUrl, { timeout });
+	if (!result.ok) return { content: "", ok: false };
+
+	try {
+		const repo = JSON.parse(result.content) as {
+			name: string;
+			description?: string;
+			star_count: number;
+			forks_count: number;
+			open_issues_count: number;
+			default_branch: string;
+			visibility: string;
+			created_at: string;
+			last_activity_at: string;
+			topics?: string[];
+			readme_url?: string;
+		};
+
+		let md = `# ${repo.name}\n\n`;
+		if (repo.description) md += `${repo.description}\n\n`;
+		md += `**Stars:** ${formatCount(repo.star_count)} · **Forks:** ${formatCount(repo.forks_count)} · **Issues:** ${formatCount(repo.open_issues_count)}\n`;
+		md += `**Visibility:** ${repo.visibility} · **Default Branch:** ${repo.default_branch}\n`;
+		if (repo.topics && repo.topics.length > 0) {
+			md += `**Topics:** ${repo.topics.join(", ")}\n`;
+		}
+		md += `**Created:** ${new Date(repo.created_at).toISOString().split("T")[0]} · **Last Activity:** ${new Date(repo.last_activity_at).toISOString().split("T")[0]}\n\n`;
+
+		// Try to fetch README
+		if (repo.readme_url) {
+			const readmeResult = await loadPage(repo.readme_url, { timeout });
+			if (readmeResult.ok && readmeResult.content.trim().length > 0) {
+				md += `---\n\n## README\n\n${readmeResult.content}\n`;
+			}
+		}
+
+		return { content: md, ok: true };
+	} catch {
+		return { content: "", ok: false };
+	}
+}
+
+/**
+ * Render GitLab file
+ */
+async function renderGitLabFile(
+	gl: GitLabUrl,
+	projectId: number,
+	timeout: number,
+): Promise<{ content: string; ok: boolean }> {
+	const encodedPath = encodeURIComponent(gl.path!);
+	const apiUrl = `https://gitlab.com/api/v4/projects/${projectId}/repository/files/${encodedPath}/raw?ref=${gl.ref}`;
+
+	const result = await loadPage(apiUrl, { timeout });
+	if (!result.ok) return { content: "", ok: false };
+
+	return { content: result.content, ok: true };
+}
+
+/**
+ * Render GitLab directory tree
+ */
+async function renderGitLabTree(
+	gl: GitLabUrl,
+	projectId: number,
+	timeout: number,
+): Promise<{ content: string; ok: boolean }> {
+	const apiUrl = `https://gitlab.com/api/v4/projects/${projectId}/repository/tree?ref=${gl.ref}&path=${gl.path || ""}&per_page=100`;
+
+	const result = await loadPage(apiUrl, { timeout });
+	if (!result.ok) return { content: "", ok: false };
+
+	try {
+		const tree = JSON.parse(result.content) as Array<{
+			name: string;
+			type: "tree" | "blob";
+			path: string;
+			mode: string;
+		}>;
+
+		let md = `# Directory: ${gl.path || "/"}\n\n`;
+		md += `**Ref:** ${gl.ref}\n\n`;
+
+		// Separate directories and files
+		const dirs = tree.filter((item) => item.type === "tree");
+		const files = tree.filter((item) => item.type === "blob");
+
+		if (dirs.length > 0) {
+			md += `## Directories (${dirs.length})\n\n`;
+			for (const dir of dirs) {
+				md += `- 📁 ${dir.name}/\n`;
+			}
+			md += `\n`;
+		}
+
+		if (files.length > 0) {
+			md += `## Files (${files.length})\n\n`;
+			for (const file of files) {
+				md += `- 📄 ${file.name}\n`;
+			}
+		}
+
+		return { content: md, ok: true };
+	} catch {
+		return { content: "", ok: false };
+	}
+}
+
+/**
+ * Render GitLab issue
+ */
+async function renderGitLabIssue(
+	gl: GitLabUrl,
+	projectId: number,
+	timeout: number,
+): Promise<{ content: string; ok: boolean }> {
+	const apiUrl = `https://gitlab.com/api/v4/projects/${projectId}/issues/${gl.id}`;
+
+	const result = await loadPage(apiUrl, { timeout });
+	if (!result.ok) return { content: "", ok: false };
+
+	try {
+		const issue = JSON.parse(result.content) as {
+			title: string;
+			description?: string;
+			state: string;
+			author: { name: string; username: string };
+			created_at: string;
+			updated_at: string;
+			labels: string[];
+			upvotes: number;
+			downvotes: number;
+			user_notes_count: number;
+			assignees?: Array<{ name: string }>;
+		};
+
+		let md = `# Issue #${gl.id}: ${issue.title}\n\n`;
+		md += `**State:** ${issue.state.toUpperCase()} · **Author:** ${issue.author.name} (@${issue.author.username})\n`;
+		md += `**Created:** ${new Date(issue.created_at).toISOString().split("T")[0]} · **Updated:** ${new Date(issue.updated_at).toISOString().split("T")[0]}\n`;
+		md += `**Upvotes:** ${issue.upvotes} · **Downvotes:** ${issue.downvotes} · **Comments:** ${issue.user_notes_count}\n`;
+
+		if (issue.labels.length > 0) {
+			md += `**Labels:** ${issue.labels.join(", ")}\n`;
+		}
+
+		if (issue.assignees && issue.assignees.length > 0) {
+			md += `**Assignees:** ${issue.assignees.map((a) => a.name).join(", ")}\n`;
+		}
+
+		md += `\n---\n\n## Description\n\n`;
+		md += issue.description ? htmlToBasicMarkdown(issue.description) : "*No description*";
+
+		return { content: md, ok: true };
+	} catch {
+		return { content: "", ok: false };
+	}
+}
+
+/**
+ * Render GitLab merge request
+ */
+async function renderGitLabMR(
+	gl: GitLabUrl,
+	projectId: number,
+	timeout: number,
+): Promise<{ content: string; ok: boolean }> {
+	const apiUrl = `https://gitlab.com/api/v4/projects/${projectId}/merge_requests/${gl.id}`;
+
+	const result = await loadPage(apiUrl, { timeout });
+	if (!result.ok) return { content: "", ok: false };
+
+	try {
+		const mr = JSON.parse(result.content) as {
+			title: string;
+			description?: string;
+			state: string;
+			author: { name: string; username: string };
+			created_at: string;
+			updated_at: string;
+			source_branch: string;
+			target_branch: string;
+			labels: string[];
+			upvotes: number;
+			downvotes: number;
+			user_notes_count: number;
+			assignees?: Array<{ name: string }>;
+			draft: boolean;
+			merge_status: string;
+		};
+
+		let md = `# MR !${gl.id}: ${mr.title}\n\n`;
+		if (mr.draft) md += `**[DRAFT]** `;
+		md += `**State:** ${mr.state.toUpperCase()} · **Author:** ${mr.author.name} (@${mr.author.username})\n`;
+		md += `**Branch:** ${mr.source_branch} → ${mr.target_branch}\n`;
+		md += `**Created:** ${new Date(mr.created_at).toISOString().split("T")[0]} · **Updated:** ${new Date(mr.updated_at).toISOString().split("T")[0]}\n`;
+		md += `**Merge Status:** ${mr.merge_status} · **Upvotes:** ${mr.upvotes} · **Downvotes:** ${mr.downvotes} · **Comments:** ${mr.user_notes_count}\n`;
+
+		if (mr.labels.length > 0) {
+			md += `**Labels:** ${mr.labels.join(", ")}\n`;
+		}
+
+		if (mr.assignees && mr.assignees.length > 0) {
+			md += `**Assignees:** ${mr.assignees.map((a) => a.name).join(", ")}\n`;
+		}
+
+		md += `\n---\n\n## Description\n\n`;
+		md += mr.description ? htmlToBasicMarkdown(mr.description) : "*No description*";
+
+		return { content: md, ok: true };
+	} catch {
+		return { content: "", ok: false };
+	}
+}
+
+/**
+ * Handle GitLab URLs specially
+ */
+export const handleGitLab: SpecialHandler = async (url: string, timeout: number): Promise<RenderResult | null> => {
+	const gl = parseGitLabUrl(url);
+	if (!gl) return null;
+
+	const fetchedAt = new Date().toISOString();
+	const notes: string[] = [];
+
+	switch (gl.type) {
+		case "blob": {
+			const projectId = await getProjectId(gl, timeout);
+			if (!projectId) break;
+
+			notes.push(`Fetched raw file via GitLab API`);
+			const result = await renderGitLabFile(gl, projectId, timeout);
+			if (result.ok) {
+				const output = finalizeOutput(result.content);
+				return {
+					url,
+					finalUrl: url,
+					contentType: "text/plain",
+					method: "gitlab-raw",
+					content: output.content,
+					fetchedAt,
+					truncated: output.truncated,
+					notes,
+				};
+			}
+			break;
+		}
+
+		case "tree": {
+			const projectId = await getProjectId(gl, timeout);
+			if (!projectId) break;
+
+			notes.push(`Fetched directory tree via GitLab API`);
+			const result = await renderGitLabTree(gl, projectId, timeout);
+			if (result.ok) {
+				const output = finalizeOutput(result.content);
+				return {
+					url,
+					finalUrl: url,
+					contentType: "text/markdown",
+					method: "gitlab-tree",
+					content: output.content,
+					fetchedAt,
+					truncated: output.truncated,
+					notes,
+				};
+			}
+			break;
+		}
+
+		case "issue": {
+			const projectId = await getProjectId(gl, timeout);
+			if (!projectId) break;
+
+			notes.push(`Fetched issue via GitLab API`);
+			const result = await renderGitLabIssue(gl, projectId, timeout);
+			if (result.ok) {
+				const output = finalizeOutput(result.content);
+				return {
+					url,
+					finalUrl: url,
+					contentType: "text/markdown",
+					method: "gitlab-issue",
+					content: output.content,
+					fetchedAt,
+					truncated: output.truncated,
+					notes,
+				};
+			}
+			break;
+		}
+
+		case "merge_request": {
+			const projectId = await getProjectId(gl, timeout);
+			if (!projectId) break;
+
+			notes.push(`Fetched merge request via GitLab API`);
+			const result = await renderGitLabMR(gl, projectId, timeout);
+			if (result.ok) {
+				const output = finalizeOutput(result.content);
+				return {
+					url,
+					finalUrl: url,
+					contentType: "text/markdown",
+					method: "gitlab-mr",
+					content: output.content,
+					fetchedAt,
+					truncated: output.truncated,
+					notes,
+				};
+			}
+			break;
+		}
+
+		case "repo": {
+			notes.push(`Fetched repository via GitLab API`);
+			const result = await renderGitLabRepo(gl, timeout);
+			if (result.ok) {
+				const output = finalizeOutput(result.content);
+				return {
+					url,
+					finalUrl: url,
+					contentType: "text/markdown",
+					method: "gitlab-repo",
+					content: output.content,
+					fetchedAt,
+					truncated: output.truncated,
+					notes,
+				};
+			}
+			break;
+		}
+	}
+
+	return null;
+};

package/src/core/tools/web-fetch-handlers/go-pkg.ts

@@ -0,0 +1,271 @@
+import { parse as parseHtml } from "node-html-parser";
+import type { RenderResult, SpecialHandler } from "./types";
+import { finalizeOutput, htmlToBasicMarkdown, loadPage } from "./types";
+
+interface GoModuleInfo {
+	Version: string;
+	Time: string;
+}
+
+/**
+ * Handle pkg.go.dev URLs via proxy API and page parsing
+ */
+export const handleGoPkg: SpecialHandler = async (url: string, timeout: number): Promise<RenderResult | null> => {
+	try {
+		const parsed = new URL(url);
+		if (parsed.hostname !== "pkg.go.dev") return null;
+
+		// Extract module path and version from URL
+		// Patterns: /module, /module@version, /module/subpackage
+		const pathname = parsed.pathname.slice(1); // remove leading /
+		if (!pathname) return null;
+
+		let modulePath: string;
+		let version = "latest";
+		let _subpackage = "";
+
+		// Parse @version if present
+		const atIndex = pathname.indexOf("@");
+		if (atIndex !== -1) {
+			const beforeAt = pathname.slice(0, atIndex);
+			const afterAt = pathname.slice(atIndex + 1);
+
+			// Check if there's a subpackage after version
+			const slashIndex = afterAt.indexOf("/");
+			if (slashIndex !== -1) {
+				version = afterAt.slice(0, slashIndex);
+				const remainder = afterAt.slice(slashIndex + 1);
+				modulePath = beforeAt;
+				_subpackage = remainder;
+			} else {
+				version = afterAt;
+				modulePath = beforeAt;
+			}
+		} else {
+			// No version specified, check for subpackage
+			// Need to determine where module ends and subpackage begins
+			// For now, treat the whole path as module path (we'll refine from proxy response)
+			modulePath = pathname;
+		}
+
+		const notes: string[] = [];
+		const sections: string[] = [];
+
+		// Fetch module info from proxy
+		let moduleInfo: GoModuleInfo | null = null;
+		let actualModulePath = modulePath;
+
+		if (version === "latest") {
+			try {
+				const proxyUrl = `https://proxy.golang.org/${encodeURIComponent(modulePath)}/@latest`;
+				const proxyResult = await loadPage(proxyUrl, { timeout });
+
+				if (proxyResult.ok) {
+					moduleInfo = JSON.parse(proxyResult.content) as GoModuleInfo;
+					version = moduleInfo.Version;
+				}
+			} catch {
+				// If @latest fails, might be a subpackage - will extract from page
+			}
+		} else {
+			try {
+				const proxyUrl = `https://proxy.golang.org/${encodeURIComponent(modulePath)}/@v/${encodeURIComponent(version)}.info`;
+				const proxyResult = await loadPage(proxyUrl, { timeout });
+
+				if (proxyResult.ok) {
+					moduleInfo = JSON.parse(proxyResult.content) as GoModuleInfo;
+				}
+			} catch {
+				// Proxy lookup failed, will rely on page data
+			}
+		}
+
+		// Fetch the pkg.go.dev page
+		const pageResult = await loadPage(url, { timeout });
+		if (!pageResult.ok) {
+			return {
+				url,
+				finalUrl: pageResult.finalUrl,
+				contentType: "text/plain",
+				method: "go-pkg",
+				content: `Failed to fetch pkg.go.dev page (status: ${pageResult.status ?? "unknown"})`,
+				fetchedAt: new Date().toISOString(),
+				truncated: false,
+				notes: ["error"],
+			};
+		}
+
+		const doc = parseHtml(pageResult.content);
+
+		// Extract module/package information
+		const breadcrumb = doc.querySelector(".go-Breadcrumb");
+		const _headerDiv = doc.querySelector(".go-Main-header");
+
+		// Extract actual module path from breadcrumb or header
+		if (breadcrumb) {
+			const moduleLink = breadcrumb.querySelector("a[href^='/']");
+			if (moduleLink) {
+				const href = moduleLink.getAttribute("href");
+				if (href) {
+					actualModulePath = href.slice(1).split("@")[0];
+				}
+			}
+		}
+
+		// Extract version if not from proxy
+		if (!moduleInfo) {
+			const versionBadge = doc.querySelector(".go-Chip");
+			if (versionBadge) {
+				const versionText = versionBadge.textContent?.trim();
+				if (versionText?.startsWith("v")) {
+					version = versionText;
+				}
+			}
+		}
+
+		// Extract license
+		const licenseLink = doc.querySelector("a[data-test-id='UnitHeader-license']");
+		const license = licenseLink?.textContent?.trim() || "Unknown";
+
+		// Extract import path
+		const importPathInput = doc.querySelector("input[data-test-id='UnitHeader-importPath']");
+		const importPath = importPathInput?.getAttribute("value") || actualModulePath;
+
+		// Build header
+		sections.push(`# ${importPath}`);
+		sections.push("");
+		sections.push(`**Module:** ${actualModulePath}`);
+		sections.push(`**Version:** ${version}`);
+		sections.push(`**License:** ${license}`);
+		sections.push("");
+
+		// Extract package synopsis
+		const synopsis = doc.querySelector(".go-Main-headerContent p");
+		if (synopsis) {
+			const synopsisText = synopsis.textContent?.trim();
+			if (synopsisText) {
+				sections.push(`## Synopsis`);
+				sections.push("");
+				sections.push(synopsisText);
+				sections.push("");
+			}
+		}
+
+		// Extract documentation overview
+		const docSection = doc.querySelector("#section-documentation");
+		if (docSection) {
+			sections.push("## Documentation");
+			sections.push("");
+
+			// Get overview paragraph
+			const overview = docSection.querySelector(".go-Message");
+			if (overview) {
+				const overviewMd = htmlToBasicMarkdown(overview.innerHTML);
+				sections.push(overviewMd);
+				sections.push("");
+			}
+
+			// Get package-level documentation
+			const docContent = docSection.querySelector(".Documentation-content");
+			if (docContent) {
+				// Extract first few paragraphs
+				const paragraphs = docContent.querySelectorAll("p");
+				const docParts: string[] = [];
+				for (let i = 0; i < Math.min(3, paragraphs.length); i++) {
+					const p = paragraphs[i];
+					const text = htmlToBasicMarkdown(p.innerHTML).trim();
+					if (text) {
+						docParts.push(text);
+					}
+				}
+
+				if (docParts.length > 0) {
+					sections.push(docParts.join("\n\n"));
+					sections.push("");
+				}
+			}
+		}
+
+		// Extract index of exported identifiers
+		const indexSection = doc.querySelector("#section-index");
+		if (indexSection) {
+			const indexList = indexSection.querySelector(".Documentation-indexList");
+			if (indexList) {
+				sections.push("## Index");
+				sections.push("");
+
+				const items = indexList.querySelectorAll("li");
+				const exported: string[] = [];
+
+				for (const item of items) {
+					const link = item.querySelector("a");
+					if (link) {
+						const name = link.textContent?.trim();
+						if (name) {
+							exported.push(`- ${name}`);
+						}
+					}
+				}
+
+				if (exported.length > 0) {
+					// Limit to first 50 exports
+					sections.push(exported.slice(0, 50).join("\n"));
+					if (exported.length > 50) {
+						notes.push(`showing 50 of ${exported.length} exports`);
+						sections.push(`\n... and ${exported.length - 50} more`);
+					}
+					sections.push("");
+				}
+			}
+		}
+
+		// Extract dependencies/imports
+		const importsSection = doc.querySelector("#section-imports");
+		if (importsSection) {
+			const importsList = importsSection.querySelector(".go-Message");
+			if (importsList) {
+				sections.push("## Imports");
+				sections.push("");
+
+				const links = importsList.querySelectorAll("a");
+				const imports: string[] = [];
+
+				for (const link of links) {
+					const imp = link.textContent?.trim();
+					if (imp) {
+						imports.push(`- ${imp}`);
+					}
+				}
+
+				if (imports.length > 0) {
+					sections.push(imports.slice(0, 20).join("\n"));
+					if (imports.length > 20) {
+						notes.push(`showing 20 of ${imports.length} imports`);
+						sections.push(`\n... and ${imports.length - 20} more`);
+					}
+					sections.push("");
+				}
+			}
+		}
+
+		if (moduleInfo) {
+			notes.push(`published ${moduleInfo.Time}`);
+		}
+
+		const content = sections.join("\n");
+		const { content: finalContent, truncated } = finalizeOutput(content);
+
+		return {
+			url,
+			finalUrl: pageResult.finalUrl,
+			contentType: "text/markdown",
+			method: "go-pkg",
+			content: finalContent,
+			fetchedAt: new Date().toISOString(),
+			truncated,
+			notes,
+		};
+	} catch {
+		return null;
+	}
+};