@harryisfish/gitt 1.5.0 → 1.6.0

package/dist/commands/clean.js

@@ -6,7 +6,6 @@ const listr2_1 = require("listr2");
 const prompts_1 = require("@inquirer/prompts");
 const minimatch_1 = require("minimatch");
 const errors_1 = require("../errors");
-const errors_2 = require("../errors");
 const git_1 = require("../utils/git");
 const config_1 = require("../utils/config");
 const git = (0, simple_git_1.simpleGit)();
@@ -33,16 +32,18 @@ async function cleanDeletedBranches(options = {}) {
             {
                 title: 'Switch to main branch',
                 task: async (ctx) => {
-                    // Check if we are on a branch that will be deleted? 
-                    // For now, just try to switch to main to be safe.
-                    // But if main is checked out in another worktree, this might fail?
-                    // Let's just try.
+                    const currentBranch = (await git.branchLocal()).current;
+                    const willDeleteCurrent = state.deletedBranches.some(b => b.name === currentBranch);
                     try {
                         await git.checkout(ctx.mainBranch);
                     }
                     catch (e) {
-                        // If we can't checkout main (e.g. dirty state), warn but continue?
-                        // Ideally we should be on main to delete other branches safely.
+                        // If current branch will be deleted, we must switch - fail the operation
+                        if (willDeleteCurrent) {
+                            throw new errors_1.GitError(`Cannot switch to ${ctx.mainBranch}. Current branch "${currentBranch}" will be deleted but checkout failed.`);
+                        }
+                        // Otherwise, warn but continue (we can delete other branches)
+                        console.warn(`Warning: Could not switch to ${ctx.mainBranch}, but continuing as current branch is not being deleted.`);
                     }
                 }
             },
@@ -110,7 +111,7 @@ async function cleanDeletedBranches(options = {}) {
         await discoveryTasks.run(state);
         // Phase 2: Interaction / Filtering
         if (state.deletedBranches.length === 0) {
-            (0, errors_2.printSuccess)('No branches need to be cleaned up');
+            (0, errors_1.printSuccess)('No branches need to be cleaned up');
             return;
         }
         if (options.interactive) {
@@ -128,7 +129,7 @@ async function cleanDeletedBranches(options = {}) {
             }
             catch (e) {
                 // User cancelled
-                throw new Error('Operation cancelled');
+                throw new errors_1.UserCancelError('Operation cancelled');
             }
         }
         else {
@@ -141,7 +142,7 @@ async function cleanDeletedBranches(options = {}) {
             state.deletedBranches = state.deletedBranches.filter(b => b.isMerged);
         }
         if (state.deletedBranches.length === 0) {
-            (0, errors_2.printSuccess)('No branches selected for deletion');
+            (0, errors_1.printSuccess)('No branches selected for deletion');
             return;
         }
         if (options.dryRun) {
@@ -153,7 +154,7 @@ async function cleanDeletedBranches(options = {}) {
         const deleteTasks = new listr2_1.Listr([
             {
                 title: 'Delete branches',
-                task: (ctx) => {
+                task: (_ctx) => {
                     return new listr2_1.Listr(state.deletedBranches.map(branch => ({
                         title: `Delete ${branch.name}`,
                         task: async () => {
@@ -165,7 +166,7 @@ async function cleanDeletedBranches(options = {}) {
             }
         ]);
         await deleteTasks.run(state);
-        (0, errors_2.printSuccess)('Branch cleanup completed');
+        (0, errors_1.printSuccess)('Branch cleanup completed');
     }
     catch (error) {
         throw new errors_1.GitError(error instanceof Error ? error.message : 'Unknown error occurred while cleaning branches');

package/dist/errors.js

@@ -1,9 +1,13 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.UserCancelError = exports.GitError = void 0;
 exports.handleError = handleError;
 exports.printSuccess = printSuccess;
 exports.printError = printError;
+const chalk_1 = __importDefault(require("chalk"));
 // 自定义错误类型
 class GitError extends Error {
     constructor(message) {
@@ -19,10 +23,6 @@ class UserCancelError extends Error {
     }
 }
 exports.UserCancelError = UserCancelError;
-// 错误消息颜色
-const ERROR_COLOR = '\x1b[31m';
-const SUCCESS_COLOR = '\x1b[32m';
-const RESET_COLOR = '\x1b[0m';
 // 统一错误处理函数
 function handleError(error) {
     if (error instanceof UserCancelError) {
@@ -30,21 +30,21 @@ function handleError(error) {
         process.exit(0);
     }
     if (error instanceof GitError) {
-        console.error(`${ERROR_COLOR}Error: ${RESET_COLOR}${error.message}`);
+        console.error(chalk_1.default.red('Error:'), error.message);
         process.exit(1);
     }
     if (error instanceof Error) {
-        console.error(`${ERROR_COLOR}Program error: ${RESET_COLOR}${error.message}`);
+        console.error(chalk_1.default.red('Program error:'), error.message);
         process.exit(1);
     }
-    console.error(`${ERROR_COLOR}Unknown error occurred${RESET_COLOR}`);
+    console.error(chalk_1.default.red('Unknown error occurred'));
     process.exit(1);
 }
 // 成功消息处理
 function printSuccess(message) {
-    console.log(`${SUCCESS_COLOR}✓ ${message}${RESET_COLOR}`);
+    console.log(chalk_1.default.green('✓'), message);
 }
 // 错误消息处理
 function printError(message) {
-    console.error(`${ERROR_COLOR}${message}${RESET_COLOR}`);
+    console.error(chalk_1.default.red(message));
 }

package/dist/index.js

@@ -34,10 +34,12 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
+const commander_1 = require("commander");
 const simple_git_1 = require("simple-git");
 const errors_1 = require("./errors");
 const clean_1 = require("./commands/clean");
 const git = (0, simple_git_1.simpleGit)();
+const packageJson = require('../package.json');
 // 处理 Ctrl+C 和其他终止信号
 process.on('SIGINT', () => {
     throw new errors_1.UserCancelError('\nOperation cancelled');
@@ -45,16 +47,6 @@ process.on('SIGINT', () => {
 process.on('SIGTERM', () => {
     throw new errors_1.UserCancelError('\nProgram terminated');
 });
-// 初始化 Git 仓库
-async function initGitRepo() {
-    try {
-        await git.init();
-        (0, errors_1.printSuccess)('Git repository initialized successfully');
-    }
-    catch (error) {
-        throw new errors_1.GitError('Failed to initialize Git repository');
-    }
-}
 // 检查当前目录是否是 Git 仓库
 async function checkGitRepo() {
     const isRepo = await git.checkIsRepo();
@@ -74,104 +66,70 @@ async function checkGitRepo() {
         throw new errors_1.GitError('Cannot access remote repository, please check network connection or repository permissions');
     }
 }
-function printHelp() {
-    console.log(`
-Usage: gitt [command] [options]
-
-Commands:
-  (default)              Clean up local branches that have been deleted on remote
-  set-main <branch>      Set the main branch for the current project
-  ignore <pattern>       Add a branch pattern to the ignore list (e.g., "release/*")
-
-Options:
-  -i, --interactive      Interactive mode: Select branches to delete
-  -d, --dry-run          Dry run: Show what would be deleted without deleting
-  --stale [days]         Find stale branches (default: 90 days, use with a number for custom)
-  -v, --version          Show version number
-  -h, --help             Show this help message
-
-Examples:
-  gitt                   # Auto-clean deleted branches
-  gitt -i                # Select branches to delete interactively
-  gitt -d                # Preview deletion
-  gitt --stale           # Find branches inactive for 90+ days
-  gitt --stale 30        # Find branches inactive for 30+ days
-  gitt ignore "temp/*"   # Ignore branches matching "temp/*"
-  gitt set-main master   # Set main branch to 'master'
-`);
-}
 async function main() {
     try {
-        const args = process.argv.slice(2);
-        const command = args[0];
-        // Check for version command
-        if (args.includes('-v') || args.includes('--version')) {
-            const packageJson = require('../package.json');
-            console.log(`v${packageJson.version}`);
-            process.exit(0);
-        }
-        // Check for help command
-        if (args.includes('-h') || args.includes('--help')) {
-            printHelp();
-            process.exit(0);
-        }
-        // Check for updates
+        // Check for updates before parsing commands
         try {
             // Use eval to prevent TypeScript from transpiling dynamic import to require()
             const { default: updateNotifier } = await eval('import("update-notifier")');
-            const pkg = require('../package.json');
-            updateNotifier({ pkg }).notify();
+            updateNotifier({ pkg: packageJson }).notify();
         }
         catch (e) {
             // Ignore update check errors
         }
-        // 检查 Git 仓库
-        await checkGitRepo();
-        if (command === 'set-main') {
-            const branch = args[1];
-            if (!branch) {
-                throw new Error('Please specify a branch name');
-            }
-            await Promise.resolve().then(() => __importStar(require('./commands/config'))).then(m => m.configMainBranch(branch));
-        }
-        else if (command === 'ignore') {
-            const pattern = args[1];
-            if (!pattern) {
-                throw new Error('Please specify a branch pattern to ignore');
-            }
-            await Promise.resolve().then(() => __importStar(require('./commands/config'))).then(m => m.configIgnoreBranch(pattern));
-        }
-        else {
-            // Parse options
-            const isInteractive = args.includes('-i') || args.includes('--interactive');
-            const isDryRun = args.includes('-d') || args.includes('--dry-run');
-            // Parse stale options
-            const staleIndex = args.indexOf('--stale');
-            const isStale = staleIndex !== -1;
-            let staleDays = 90; // Default 3 months
-            if (isStale) {
-                // Check if next arg is a number (days)
-                // Note: This is a simple check, ideally use a proper arg parser like commander or yargs
-                const nextArg = args[staleIndex + 1];
-                if (nextArg && !nextArg.startsWith('-') && !isNaN(Number(nextArg))) {
-                    staleDays = Number(nextArg);
-                }
-            }
-            // 默认执行清理操作
+        const program = new commander_1.Command();
+        program
+            .name('gitt')
+            .description('A CLI tool for Git branch management')
+            .version(packageJson.version, '-v, --version', 'Show version number');
+        // Default command: clean deleted branches
+        program
+            .option('-i, --interactive', 'Interactive mode: Select branches to delete')
+            .option('-d, --dry-run', 'Dry run: Show what would be deleted without deleting')
+            .option('--stale [days]', 'Find stale branches (default: 90 days)', '90')
+            .action(async (options) => {
+            await checkGitRepo();
+            const staleDays = options.stale === true ? 90 : parseInt(options.stale, 10);
+            const isStale = options.stale !== undefined;
             await (0, clean_1.cleanDeletedBranches)({
-                interactive: isInteractive,
-                dryRun: isDryRun,
+                interactive: options.interactive || false,
+                dryRun: options.dryRun || false,
                 stale: isStale,
                 staleDays: staleDays
             });
-        }
-        // 退出程序
-        process.exit(0);
+        });
+        // set-main command
+        program
+            .command('set-main <branch>')
+            .description('Set the main branch for the current project')
+            .action(async (branch) => {
+            await checkGitRepo();
+            await Promise.resolve().then(() => __importStar(require('./commands/config'))).then(m => m.configMainBranch(branch));
+        });
+        // ignore command
+        program
+            .command('ignore <pattern>')
+            .description('Add a branch pattern to the ignore list (e.g., "release/*")')
+            .action(async (pattern) => {
+            await checkGitRepo();
+            await Promise.resolve().then(() => __importStar(require('./commands/config'))).then(m => m.configIgnoreBranch(pattern));
+        });
+        // Add examples to help
+        program.addHelpText('after', `
+Examples:
+  $ gitt                   # Auto-clean deleted branches
+  $ gitt -i                # Select branches to delete interactively
+  $ gitt -d                # Preview deletion
+  $ gitt --stale           # Find branches inactive for 90+ days
+  $ gitt --stale 30        # Find branches inactive for 30+ days
+  $ gitt ignore "temp/*"   # Ignore branches matching "temp/*"
+  $ gitt set-main master   # Set main branch to 'master'
+`);
+        await program.parseAsync(process.argv);
     }
     catch (error) {
         (0, errors_1.handleError)(error);
-        process.exit(1);
     }
 }
 // 启动程序
-main().catch(errors_1.handleError);
+main();

package/dist/utils/config.js

@@ -39,7 +39,50 @@ exports.writeConfigFile = writeConfigFile;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 const simple_git_1 = require("simple-git");
+const minimatch_1 = require("minimatch");
 const CONFIG_FILE_NAME = '.gitt';
+/**
+ * Validate configuration values.
+ */
+function validateConfig(config) {
+    const validated = {};
+    // Validate mainBranch
+    if (config.mainBranch !== undefined) {
+        if (typeof config.mainBranch !== 'string' || config.mainBranch.trim() === '') {
+            throw new Error('Invalid config: mainBranch must be a non-empty string');
+        }
+        validated.mainBranch = config.mainBranch.trim();
+    }
+    // Validate ignoreBranches
+    if (config.ignoreBranches !== undefined) {
+        if (!Array.isArray(config.ignoreBranches)) {
+            throw new Error('Invalid config: ignoreBranches must be an array');
+        }
+        const validPatterns = [];
+        for (const pattern of config.ignoreBranches) {
+            if (typeof pattern !== 'string' || pattern.trim() === '') {
+                throw new Error('Invalid config: ignoreBranches must contain non-empty strings');
+            }
+            // Test if it's a valid glob pattern by trying to use it
+            try {
+                (0, minimatch_1.minimatch)('test', pattern);
+                validPatterns.push(pattern);
+            }
+            catch (e) {
+                throw new Error(`Invalid config: "${pattern}" is not a valid glob pattern`);
+            }
+        }
+        validated.ignoreBranches = validPatterns;
+    }
+    // Validate staleDays
+    if (config.staleDays !== undefined) {
+        if (typeof config.staleDays !== 'number' || config.staleDays < 1 || config.staleDays > 365 || !Number.isInteger(config.staleDays)) {
+            throw new Error('Invalid config: staleDays must be an integer between 1 and 365');
+        }
+        validated.staleDays = config.staleDays;
+    }
+    return validated;
+}
 /**
  * Get the project root directory (where .git is located).
  */
@@ -64,9 +107,15 @@ async function readConfigFile() {
             return {};
         }
         const content = fs.readFileSync(configPath, 'utf-8');
-        return JSON.parse(content);
+        const parsedConfig = JSON.parse(content);
+        return validateConfig(parsedConfig);
     }
     catch (error) {
+        // If validation fails, throw the error instead of returning empty config
+        if (error instanceof Error && error.message.startsWith('Invalid config:')) {
+            throw error;
+        }
+        // For other errors (file read, JSON parse), return empty config
         return {};
     }
 }
@@ -74,10 +123,12 @@ async function readConfigFile() {
  * Write to the .gitt configuration file.
  */
 async function writeConfigFile(config) {
+    // Validate new config before writing
+    const validatedConfig = validateConfig(config);
     const root = await getProjectRoot();
     const configPath = path.join(root, CONFIG_FILE_NAME);
     // Read existing config to merge
     const currentConfig = await readConfigFile();
-    const newConfig = { ...currentConfig, ...config };
+    const newConfig = { ...currentConfig, ...validatedConfig };
     fs.writeFileSync(configPath, JSON.stringify(newConfig, null, 2));
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@harryisfish/gitt",
-  "version": "1.5.0",
+  "version": "1.6.0",
   "description": "A command-line tool to help you manage Git repositories and remote repositories, such as keeping in sync, pushing, pulling, etc.",
   "main": "dist/index.js",
   "bin": {
@@ -10,6 +10,9 @@
     "build": "tsc",
     "start": "tsx src/index.ts",
     "dev": "tsx watch src/index.ts",
+    "test": "vitest",
+    "test:ui": "vitest --ui",
+    "test:coverage": "vitest --coverage",
     "prepare": "pnpm run build",
     "dev:link": "pnpm build && pnpm link --global",
     "dev:unlink": "pnpm unlink --global"
@@ -32,14 +35,18 @@
   "homepage": "https://github.com/harryisfish/gitt#readme",
   "devDependencies": {
     "@types/node": "^20.11.24",
+    "@vitest/ui": "^4.0.14",
     "ts-node": "^10.9.2",
     "tsx": "^4.7.1",
-    "typescript": "^5.3.3"
+    "typescript": "^5.3.3",
+    "vitest": "^4.0.14"
   },
   "dependencies": {
     "@inquirer/prompts": "^3.3.0",
     "@types/minimatch": "^6.0.0",
     "@types/update-notifier": "^6.0.8",
+    "chalk": "4",
+    "commander": "^14.0.2",
     "listr2": "^8.0.0",
     "minimatch": "^10.1.1",
     "simple-git": "^3.22.0",