ai-cmg 0.0.1 → 0.0.2

package/dist/index.js

@@ -1,22 +1,214 @@
 #!/usr/bin/env node
 import { execSync, spawnSync } from 'child_process';
+import { readFileSync } from 'fs';
+import { dirname, join } from 'path';
+import { fileURLToPath } from 'url';
 import prompts from 'prompts';
-import pc from 'picocolors';
 import clipboardy from 'clipboardy';
 import Conf from 'conf';
 const WORKER_URL = 'https://commit.d-code.workers.dev';
 const config = new Conf({ projectName: 'ai-cmg' });
+const MEDIA_EXTENSIONS = new Set([
+    '.png', '.jpg', '.jpeg', '.gif', '.webp', '.svg', '.ico', '.bmp', '.tiff',
+    '.mp4', '.mov', '.webm', '.mp3', '.wav', '.flac', '.ogg',
+    '.pdf', '.zip', '.tar', '.gz', '.7z', '.rar', '.woff', '.woff2', '.ttf', '.eot',
+]);
+const LOCK_FILES = new Set([
+    'package-lock.json',
+    'npm-shrinkwrap.json',
+    'yarn.lock',
+    'pnpm-lock.yaml',
+    'bun.lockb',
+]);
+const GENERATED_PREFIXES = ['dist/', 'build/', '.next/', 'coverage/', 'node_modules/'];
+const MAX_BLOCK_LINES = 400;
+const MAX_BLOCK_CHARS = 20000;
+function getPackageVersion() {
+    try {
+        const currentFile = fileURLToPath(import.meta.url);
+        const currentDir = dirname(currentFile);
+        const packageJsonPath = join(currentDir, '..', 'package.json');
+        const raw = readFileSync(packageJsonPath, 'utf8');
+        const parsed = JSON.parse(raw);
+        return parsed.version ?? null;
+    }
+    catch {
+        return null;
+    }
+}
+function getNameStatus() {
+    const map = new Map();
+    try {
+        const output = execSync('git diff --cached --name-status').toString();
+        output.split('\n').forEach((line) => {
+            if (!line.trim())
+                return;
+            const [status, ...rest] = line.split('\t');
+            const filePath = rest[rest.length - 1];
+            if (filePath)
+                map.set(filePath, status);
+        });
+    }
+    catch {
+        // ignore status errors; diff will still be used
+    }
+    return map;
+}
+function getChangedFiles() {
+    try {
+        const output = execSync('git status --porcelain -z').toString();
+        if (!output)
+            return [];
+        const parts = output.split('\0').filter(Boolean);
+        const entries = [];
+        for (let i = 0; i < parts.length; i += 1) {
+            const entry = parts[i];
+            const status = entry.slice(0, 2);
+            let path = entry.slice(3);
+            if (status.startsWith('R') || status.startsWith('C')) {
+                const newPath = parts[i + 1];
+                if (newPath) {
+                    path = newPath;
+                    i += 1;
+                }
+            }
+            if (path)
+                entries.push({ path, status: status.trim() });
+        }
+        const unique = new Map();
+        entries.forEach((entry) => unique.set(entry.path, entry));
+        return Array.from(unique.values());
+    }
+    catch {
+        return [];
+    }
+}
+async function promptStageFiles() {
+    const candidates = getChangedFiles();
+    if (candidates.length === 0) {
+        console.log('No local changes to stage.');
+        return false;
+    }
+    console.log('No staged changes detected.');
+    console.log('Choose how to stage changes:');
+    const options = [
+        { label: 'Stage all files (git add .)', value: 'all' },
+        { label: 'Stage selected files', value: 'select' },
+        { label: 'Cancel', value: 'cancel' }
+    ];
+    options.forEach((option, index) => {
+        console.log(`${index + 1}. ${option.label}`);
+    });
+    const selection = await prompts({
+        type: 'number',
+        name: 'choice',
+        message: 'Enter a number (1-3):',
+        min: 1,
+        max: options.length
+    });
+    const choice = selection.choice;
+    if (!choice || choice < 1 || choice > options.length)
+        return false;
+    const action = options[choice - 1].value;
+    if (action === 'cancel')
+        return false;
+    if (action === 'all') {
+        spawnSync('git', ['add', '.'], { stdio: 'inherit' });
+        return true;
+    }
+    const fileChoices = candidates.map((entry) => ({
+        title: `${entry.status.padEnd(2, ' ')} ${entry.path}`,
+        value: entry.path
+    }));
+    const picked = await prompts({
+        type: 'multiselect',
+        name: 'files',
+        message: 'Select files to stage (space to toggle, enter to submit):',
+        choices: fileChoices,
+        min: 1
+    });
+    const files = picked.files;
+    if (!files || files.length === 0)
+        return false;
+    spawnSync('git', ['add', '--', ...files], { stdio: 'inherit' });
+    return true;
+}
+function getExtension(filePath) {
+    const lower = filePath.toLowerCase();
+    const idx = lower.lastIndexOf('.');
+    return idx >= 0 ? lower.slice(idx) : '';
+}
+function shouldOmitFile(path, block) {
+    const lowerPath = path.toLowerCase();
+    if (LOCK_FILES.has(lowerPath))
+        return { omit: true, reason: 'lockfile' };
+    if (MEDIA_EXTENSIONS.has(getExtension(lowerPath)))
+        return { omit: true, reason: 'media/binary asset' };
+    if (GENERATED_PREFIXES.some((prefix) => lowerPath.startsWith(prefix))) {
+        return { omit: true, reason: 'generated artifact' };
+    }
+    if (block.includes('GIT binary patch') || block.includes('Binary files')) {
+        return { omit: true, reason: 'binary diff' };
+    }
+    const lines = block.split('\n').length;
+    if (lines > MAX_BLOCK_LINES || block.length > MAX_BLOCK_CHARS) {
+        return { omit: true, reason: 'large diff' };
+    }
+    return { omit: false };
+}
+function summarizeDiff(diff) {
+    const statusMap = getNameStatus();
+    const blocks = diff.split(/^diff --git /m);
+    const keptBlocks = [];
+    const omitted = [];
+    for (const block of blocks) {
+        if (!block.trim())
+            continue;
+        const headerLine = block.split('\n')[0] ?? '';
+        const match = headerLine.match(/^a\/(.+?) b\/(.+?)$/);
+        if (!match) {
+            keptBlocks.push(`diff --git ${block}`);
+            continue;
+        }
+        const aPath = match[1];
+        const bPath = match[2];
+        const filePath = bPath === '/dev/null' ? aPath : bPath;
+        const status = statusMap.get(filePath);
+        const originalBlock = `diff --git ${block}`;
+        const { omit, reason } = shouldOmitFile(filePath, originalBlock);
+        if (omit && reason) {
+            omitted.push({ path: filePath, reason, status });
+            keptBlocks.push(`diff --git a/${aPath} b/${bPath}\n# content omitted (${reason})\n`);
+            continue;
+        }
+        keptBlocks.push(originalBlock);
+    }
+    const summaryLines = omitted.map((item) => {
+        const statusLabel = item.status ? `${item.status} ` : '';
+        return `- ${statusLabel}${item.path} (${item.reason})`;
+    });
+    const summaryHeader = summaryLines.length
+        ? `# Omitted file contents (summarized)\n${summaryLines.join('\n')}\n\n`
+        : '';
+    return `${summaryHeader}${keptBlocks.join('\n')}`.trim();
+}
 async function main() {
     // 1. 사용자 힌트(Arguments) 가져오기
-    const args = process.argv.slice(2).join(' ');
+    const rawArgs = process.argv.slice(2);
+    if (rawArgs.includes('--version') || rawArgs.includes('-v')) {
+        const version = getPackageVersion();
+        console.log(version ?? 'unknown');
+        return;
+    }
+    const args = rawArgs.filter((arg) => arg !== '--version' && arg !== '-v').join(' ');
     if (args) {
-        console.log(pc.cyan(`💡 사용자 힌트 감지: "${args}"`));
+        console.log(`Hint detected: "${args}"`);
     }
     // 2. 인증 토큰 확인 (없으면 물어봄)
     let authToken = config.get('authToken');
     if (!authToken) {
-        console.log(pc.bgMagenta(pc.white(' 🔐 최초 인증 설정 ')));
-        console.log(pc.gray('팀에서 공유된 Access Token을 입력해주세요. (내 컴퓨터에 안전하게 저장됩니다)'));
+        console.log('Initial authentication setup');
+        console.log('Enter the shared access token. It will be stored locally.');
         const response = await prompts({
             type: 'password',
             name: 'token',
@@ -27,11 +219,11 @@ async function main() {
             return;
         authToken = response.token;
         config.set('authToken', authToken);
-        console.log(pc.green('✅ 인증 정보가 저장되었습니다.\n'));
+        console.log('Token saved.\n');
     }
     if (!authToken)
         return;
-    console.log(pc.cyan('ℹ️  Staged 변경 사항 분석 중...'));
+    console.log('Analyzing staged changes...');
     try {
         // 3. Git Diff 가져오기
         let diff;
@@ -39,13 +231,20 @@ async function main() {
             diff = execSync('git diff --cached').toString();
         }
         catch (e) {
-            console.log(pc.red('❌ Git 저장소가 아니거나 git 명령어를 찾을 수 없습니다.'));
+            console.log('Error: not a git repository or git is unavailable.');
             return;
         }
         if (!diff.trim()) {
-            console.log(pc.yellow('⚠️  커밋할 변경 사항이 없습니다. "git add"를 먼저 실행해주세요.'));
-            return;
+            const staged = await promptStageFiles();
+            if (!staged)
+                return;
+            diff = execSync('git diff --cached').toString();
+            if (!diff.trim()) {
+                console.log('No staged changes. Nothing to commit.');
+                return;
+            }
         }
+        const filteredDiff = summarizeDiff(diff);
         // 4. API 요청 (Diff + Hint + Auth)
         const response = await fetch(WORKER_URL, {
             method: 'POST',
@@ -53,12 +252,12 @@ async function main() {
                 'Content-Type': 'application/json',
                 'X-AUTH-TOKEN': authToken
             },
-            body: JSON.stringify({ diff, hint: args }) // 힌트도 같이 전송
+            body: JSON.stringify({ diff: filteredDiff, hint: args }) // 힌트도 같이 전송
         });
         // 인증 실패 처리 (401)
         if (response.status === 401) {
-            console.log(pc.red('\n⛔️ 인증 실패! 토큰이 만료되었거나 틀렸습니다.'));
-            console.log(pc.gray('기존 설정 파일을 삭제합니다. 다시 실행하여 토큰을 입력해주세요.'));
+            console.log('\nAuthentication failed. The token is invalid or expired.');
+            console.log('Stored token will be cleared. Run again to enter a new token.');
             config.delete('authToken');
             return;
         }
@@ -67,44 +266,51 @@ async function main() {
         }
         const { message } = await response.json();
         // 5. 결과 출력 및 액션 선택
-        console.log('\n' + pc.bgGreen(pc.black(' 🤖 AI 생성 커밋 메시지 ')) + '\n');
-        console.log(pc.bold(message));
-        console.log('\n' + pc.dim('-----------------------------------') + '\n');
+        console.log('\nCommit message:\n');
+        console.log(message);
+        console.log('\n-----------------------------------\n');
+        const actions = [
+            { label: 'Commit with this message', value: 'commit' },
+            { label: 'Edit in editor and commit', value: 'edit' },
+            { label: 'Copy to clipboard', value: 'copy' },
+            { label: 'Cancel', value: 'cancel' }
+        ];
+        console.log('Select an action:');
+        actions.forEach((action, index) => {
+            console.log(`${index + 1}. ${action.label}`);
+        });
         const result = await prompts({
-            type: 'select',
-            name: 'action',
-            message: '어떻게 하시겠습니까?',
-            choices: [
-                { title: '🚀 이대로 커밋하기', value: 'commit' },
-                { title: '✏️  에디터에서 수정 후 커밋', value: 'edit' },
-                { title: '📋 클립보드에 복사', value: 'copy' },
-                { title: '❌ 취소', value: 'cancel' }
-            ]
+            type: 'number',
+            name: 'choice',
+            message: 'Enter a number (1-4):',
+            min: 1,
+            max: actions.length
         });
-        const action = result.action;
-        if (!action)
+        const choice = result.choice;
+        if (!choice || choice < 1 || choice > actions.length)
             return;
+        const action = actions[choice - 1].value;
         switch (action) {
             case 'commit':
                 spawnSync('git', ['commit', '-m', message], { stdio: 'inherit' });
-                console.log(pc.green('✅ 커밋 완료!'));
+                console.log('Commit complete.');
                 break;
             case 'edit':
                 spawnSync('git', ['commit', '-e', '-m', message], { stdio: 'inherit' });
-                console.log(pc.green('✅ 커밋 완료!'));
+                console.log('Commit complete.');
                 break;
             case 'copy':
                 clipboardy.writeSync(message);
-                console.log(pc.green('✅ 복사 완료!'));
+                console.log('Copied to clipboard.');
                 break;
             case 'cancel':
-                console.log(pc.gray('취소됨'));
+                console.log('Cancelled.');
                 break;
         }
     }
     catch (error) {
         const message = error instanceof Error ? error.message : String(error);
-        console.error(pc.red('오류 발생:'), message);
+        console.error('Error:', message);
     }
 }
 main();

package/package.json

@@ -1,10 +1,12 @@
 {
   "name": "ai-cmg",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "AI Commit Message Generator",
   "type": "module",
   "bin": {
-    "ai-commit": "dist/index.js"
+    "ai-cmg": "dist/index.js",
+    "ai-commit": "dist/index.js",
+    "d-commit": "dist/index.js"
   },
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -20,7 +22,6 @@
   "dependencies": {
     "clipboardy": "^5.0.2",
     "conf": "^15.0.2",
-    "picocolors": "^1.1.1",
     "prompts": "^2.4.2"
   },
   "devDependencies": {