Spaces:

kirimdata369
/

task-manager-pro

Paused

task-manager-pro / rtk /filters /gitStatus.js

INTEGRATION: Added 9router RTK Token Saver and Smart 3-Tier Fallback

b180bd5 6 months ago

3.59 kB

// Port of git::format_status_output (rtk/src/cmds/git/git.rs:619-730)
// Output format:
//   * <branch>
//   + Staged: N files
//      path1
//      ... +K more
//   ~ Modified: N files
//   ? Untracked: N files
//   conflicts: N files
//   clean — nothing to commit
const { STATUS_MAX_FILES, STATUS_MAX_UNTRACKED } = require("../constants.js");

function gitStatus(input) {
  const lines = input.split("\n");
  if (lines.length === 0 || (lines.length === 1 && !lines[0].trim())) {
    return "Clean working tree";
  }

  let branch = "";
  const stagedFiles = [];
  const modifiedFiles = [];
  const untrackedFiles = [];
  let staged = 0;
  let modified = 0;
  let untracked = 0;
  let conflicts = 0;

  for (const raw of lines) {
    if (!raw.trim()) continue;

    // Long-form branch detection (LLM usually sends this, not porcelain)
    const longBranch = raw.match(/^On branch (\S+)/);
    if (longBranch) { branch = longBranch[1]; continue; }

    // Porcelain branch header: "## main...origin/main"
    if (raw.startsWith("##")) { branch = raw.replace(/^##\s*/, ""); continue; }

    // Porcelain status (2 chars + space + path)
    if (raw.length >= 3 && /^[ MADRCU?!][ MADRCU?!] /.test(raw)) {
      const x = raw[0];
      const y = raw[1];
      const file = raw.slice(3);

      if (raw.slice(0, 2) === "??") {
        untracked++;
        untrackedFiles.push(file);
        continue;
      }

      if ("MADRC".includes(x)) {
        staged++;
        stagedFiles.push(file);
      } else if (x === "U") {
        conflicts++;
      }

      if (y === "M" || y === "D") {
        modified++;
        modifiedFiles.push(file);
      }
      continue;
    }

    // Long form fallback ("modified:   path", "new file:   path", ...)
    const longMatch = raw.match(/^\s*(modified|new file|deleted|renamed|both modified):\s+(.+)$/);
    if (longMatch) {
      const kind = longMatch[1];
      const path = longMatch[2].trim();
      if (kind === "both modified") { conflicts++; }
      else if (kind === "modified" || kind === "deleted") { modified++; modifiedFiles.push(path); }
      else if (kind === "new file" || kind === "renamed") { staged++; stagedFiles.push(path); }
      continue;
    }

    // "Untracked files:" section — gather bare paths after this marker
    // Handled implicitly: plain paths without markers are skipped (safer).
  }

  let out = "";
  if (branch) out += `* ${branch}\n`;

  if (staged > 0) {
    out += `+ Staged: ${staged} files\n`;
    for (const f of stagedFiles.slice(0, STATUS_MAX_FILES)) out += `   ${f}\n`;
    if (stagedFiles.length > STATUS_MAX_FILES) {
      out += `   ... +${stagedFiles.length - STATUS_MAX_FILES} more\n`;
    }
  }

  if (modified > 0) {
    out += `~ Modified: ${modified} files\n`;
    for (const f of modifiedFiles.slice(0, STATUS_MAX_FILES)) out += `   ${f}\n`;
    if (modifiedFiles.length > STATUS_MAX_FILES) {
      out += `   ... +${modifiedFiles.length - STATUS_MAX_FILES} more\n`;
    }
  }

  if (untracked > 0) {
    out += `? Untracked: ${untracked} files\n`;
    for (const f of untrackedFiles.slice(0, STATUS_MAX_UNTRACKED)) out += `   ${f}\n`;
    if (untrackedFiles.length > STATUS_MAX_UNTRACKED) {
      out += `   ... +${untrackedFiles.length - STATUS_MAX_UNTRACKED} more\n`;
    }
  }

  if (conflicts > 0) {
    out += `conflicts: ${conflicts} files\n`;
  }

  if (staged === 0 && modified === 0 && untracked === 0 && conflicts === 0) {
    out += "clean — nothing to commit\n";
  }

  return out.replace(/\n+$/, "");
}

gitStatus.filterName = "git-status";

module.exports = { gitStatus };

	// Port of git::format_status_output (rtk/src/cmds/git/git.rs:619-730)
	// Output format:
	// * <branch>
	// + Staged: N files
	// path1
	// ... +K more
	// ~ Modified: N files
	// ? Untracked: N files
	// conflicts: N files
	// clean — nothing to commit
	const { STATUS_MAX_FILES, STATUS_MAX_UNTRACKED } = require("../constants.js");

	function gitStatus(input) {
	const lines = input.split("\n");
	if (lines.length === 0 \|\| (lines.length === 1 && !lines[0].trim())) {
	return "Clean working tree";
	}

	let branch = "";
	const stagedFiles = [];
	const modifiedFiles = [];
	const untrackedFiles = [];
	let staged = 0;
	let modified = 0;
	let untracked = 0;
	let conflicts = 0;

	for (const raw of lines) {
	if (!raw.trim()) continue;

	// Long-form branch detection (LLM usually sends this, not porcelain)
	const longBranch = raw.match(/^On branch (\S+)/);
	if (longBranch) { branch = longBranch[1]; continue; }

	// Porcelain branch header: "## main...origin/main"
	if (raw.startsWith("##")) { branch = raw.replace(/^##\s*/, ""); continue; }

	// Porcelain status (2 chars + space + path)
	if (raw.length >= 3 && /^[ MADRCU?!][ MADRCU?!] /.test(raw)) {
	const x = raw[0];
	const y = raw[1];
	const file = raw.slice(3);

	if (raw.slice(0, 2) === "??") {
	untracked++;
	untrackedFiles.push(file);
	continue;
	}

	if ("MADRC".includes(x)) {
	staged++;
	stagedFiles.push(file);
	} else if (x === "U") {
	conflicts++;
	}

	if (y === "M" \|\| y === "D") {
	modified++;
	modifiedFiles.push(file);
	}
	continue;
	}

	// Long form fallback ("modified: path", "new file: path", ...)
	const longMatch = raw.match(/^\s*(modified\|new file\|deleted\|renamed\|both modified):\s+(.+)$/);
	if (longMatch) {
	const kind = longMatch[1];
	const path = longMatch[2].trim();
	if (kind === "both modified") { conflicts++; }
	else if (kind === "modified" \|\| kind === "deleted") { modified++; modifiedFiles.push(path); }
	else if (kind === "new file" \|\| kind === "renamed") { staged++; stagedFiles.push(path); }
	continue;
	}

	// "Untracked files:" section — gather bare paths after this marker
	// Handled implicitly: plain paths without markers are skipped (safer).
	}

	let out = "";
	if (branch) out += `* ${branch}\n`;

	if (staged > 0) {
	out += `+ Staged: ${staged} files\n`;
	for (const f of stagedFiles.slice(0, STATUS_MAX_FILES)) out += ` ${f}\n`;
	if (stagedFiles.length > STATUS_MAX_FILES) {
	out += ` ... +${stagedFiles.length - STATUS_MAX_FILES} more\n`;
	}
	}

	if (modified > 0) {
	out += `~ Modified: ${modified} files\n`;
	for (const f of modifiedFiles.slice(0, STATUS_MAX_FILES)) out += ` ${f}\n`;
	if (modifiedFiles.length > STATUS_MAX_FILES) {
	out += ` ... +${modifiedFiles.length - STATUS_MAX_FILES} more\n`;
	}
	}

	if (untracked > 0) {
	out += `? Untracked: ${untracked} files\n`;
	for (const f of untrackedFiles.slice(0, STATUS_MAX_UNTRACKED)) out += ` ${f}\n`;
	if (untrackedFiles.length > STATUS_MAX_UNTRACKED) {
	out += ` ... +${untrackedFiles.length - STATUS_MAX_UNTRACKED} more\n`;
	}
	}

	if (conflicts > 0) {
	out += `conflicts: ${conflicts} files\n`;
	}

	if (staged === 0 && modified === 0 && untracked === 0 && conflicts === 0) {
	out += "clean — nothing to commit\n";
	}

	return out.replace(/\n+$/, "");
	}

	gitStatus.filterName = "git-status";

	module.exports = { gitStatus };