@abacus-ai/cli 1.106.25007 → 2.0.0-canary.0

package/src/lib/utils.ts

@@ -0,0 +1,182 @@
+import { execSync } from "node:child_process";
+
+export function clamp(value: number, min: number, max: number): number {
+  return Math.max(min, Math.min(value, max));
+}
+
+export function deepClone<T>(obj: T): T {
+  if (!obj || typeof obj !== "object") {
+    return obj;
+  }
+  if (obj instanceof RegExp) {
+    return obj;
+  }
+  const result: any = Array.isArray(obj) ? [] : {};
+  Object.entries(obj).forEach(([key, value]) => {
+    result[key] = value && typeof value === "object" ? deepClone(value) : value;
+  });
+  return result;
+}
+
+export function createDeferredPromise<T = void>() {
+  let resolve: (value: T) => void;
+  let reject: (reason?: any) => void;
+
+  const promise = new Promise<T>((res, rej) => {
+    resolve = res;
+    reject = rej;
+  });
+
+  return {
+    promise,
+    resolve: resolve!,
+    reject: reject!,
+  };
+}
+
+type SegmentContent = string | { segment: string };
+
+export function getSegmentContent(segment: SegmentContent | undefined): string | null {
+  if (typeof segment === "string") {
+    return segment;
+  }
+
+  if (typeof segment === "object" && segment && "segment" in segment) {
+    return segment.segment;
+  }
+
+  return null;
+}
+
+export function isWindows(): boolean {
+  return process.platform === "win32";
+}
+
+/**
+ * Get git information for the current workspace
+ * Returns current branch name and main branch name
+ */
+export interface GitInfo {
+  branch?: string;
+  main?: string;
+  status?: string;
+  commits?: string;
+}
+
+export function getGitInfo(cwd: string): GitInfo {
+  const gitInfo: GitInfo = {};
+
+  try {
+    // Get current branch
+    const branch = execSync("git rev-parse --abbrev-ref HEAD", {
+      cwd,
+      encoding: "utf8",
+      stdio: ["pipe", "pipe", "pipe"],
+      timeout: 5000,
+    }).trim();
+    gitInfo.branch = branch;
+  } catch {
+    // Not a git repo or git not available
+  }
+
+  try {
+    // Get main branch name - try common names
+    // First check if there's a default branch configured
+    let mainBranch: string | undefined;
+
+    // Try to get the default branch from remote
+    try {
+      const remoteBranch = execSync("git symbolic-ref refs/remotes/origin/HEAD", {
+        cwd,
+        encoding: "utf8",
+        stdio: ["pipe", "pipe", "pipe"],
+        timeout: 5000,
+      }).trim();
+      // Extract branch name from refs/remotes/origin/HEAD -> refs/remotes/origin/main
+      mainBranch = remoteBranch.split("/").pop();
+    } catch {
+      // If that fails, check which of the common branch names exists
+      const commonNames = ["main", "master", "develop"];
+      for (const name of commonNames) {
+        try {
+          execSync(`git rev-parse --verify ${name}`, {
+            cwd,
+            encoding: "utf8",
+            stdio: ["pipe", "pipe", "pipe"],
+            timeout: 5000,
+          });
+          mainBranch = name;
+          break;
+        } catch {
+          // Branch doesn't exist, try next
+        }
+      }
+    }
+
+    gitInfo.main = mainBranch;
+  } catch {
+    // Ignore errors
+  }
+
+  return gitInfo;
+}
+
+export function getGitStatus(cwd: string): string | undefined {
+  try {
+    const status = execSync("git status --short", {
+      cwd,
+      encoding: "utf8",
+      stdio: ["pipe", "pipe", "pipe"],
+      timeout: 5000,
+    }).trim();
+
+    if (!status) {
+      return undefined;
+    }
+
+    const lines = status.split("\n");
+    const maxLines = 20;
+    const maxChars = 500;
+
+    let truncated = lines.slice(0, maxLines).join("\n");
+    if (lines.length > maxLines) {
+      truncated += `\n... (${lines.length - maxLines} more files)`;
+    }
+
+    if (truncated.length > maxChars) {
+      truncated = truncated.substring(0, maxChars) + "...";
+    }
+
+    return truncated;
+  } catch {
+    return undefined;
+  }
+}
+
+export function getGitCommits(cwd: string): string | undefined {
+  try {
+    const commits = execSync("git log -5 --oneline --no-decorate", {
+      cwd,
+      encoding: "utf8",
+      stdio: ["pipe", "pipe", "pipe"],
+      timeout: 5000,
+    }).trim();
+
+    if (!commits) {
+      return undefined;
+    }
+
+    const maxChars = 500;
+    if (commits.length > maxChars) {
+      return commits.substring(0, maxChars) + "...";
+    }
+
+    return commits;
+  } catch {
+    return undefined;
+  }
+}
+
+export async function sleep(ms: number): Promise<void> {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}