atlasia-ghost 0.1.0 → 0.3.0

package/README.md

@@ -39,10 +39,26 @@ ghost
 | Option | Description |
 | :--- | :--- |
 | `--model <name>` | Utiliser un modèle spécifique (ex: `llama-3.1-8b-instant`) |
+| `--provider <name>` | Choisir le fournisseur (`groq` [défaut], `openai`) |
 | `--no-security` | Désactiver l'audit de sécurité (scan de secrets) |
 | `--dry-run` | Générer le message sans effectuer le commit |
 | `--help`, `-h` | Afficher l'aide |
 
+## 📝 Personnalisation
+
+### Sélection de fichiers
+Ghost propose désormais une sélection interactive si plusieurs fichiers sont modifiés. Vous pouvez spécifier les fichiers à analyser (ex: `1,3,5`) ou tout analyser (`all`).
+
+### Configuration locale (`.ghostrc`)
+Vous pouvez personnaliser le comportement de Ghost par projet en créant un fichier `.ghostrc` à la racine :
+
+```json
+{
+  "prompt": "Ton prompt système personnalisé pour l'IA ici",
+  "provider": "openai"
+}
+```
+
 ## 🛡️ Sécurité
 
 Ghost effectue un double audit :

package/ghost.js

@@ -17,6 +17,7 @@ const os = require('os');
 // ⚙️ CONFIGURATION & CONSTANTES
 // ==============================================================================
 const CONFIG_FILE = path.join(os.homedir(), '.ghost');
+const HISTORY_FILE = path.join(os.homedir(), '.ghost_history');
 const SAFE_EXTENSIONS = new Set(['.md', '.txt', '.csv', '.html', '.css', '.scss', '.lock', '.xml', '.json']);
 const SAFE_FILES = new Set(['mvnw', 'gradlew', 'package-lock.json', 'yarn.lock', 'pom.xml']);
 
@@ -62,21 +63,29 @@ class ConfigManager {
         console.log(`${Colors.DIM}Configuration sauvegardée dans ${CONFIG_FILE}${Colors.ENDC}`);
     }
 
-    async getApiKey() {
-        let key = process.env.GROQ_API_KEY || this.config.groq_api_key;
+    async getApiKey(provider = 'groq') {
+        const keyMap = {
+            groq: { env: 'GROQ_API_KEY', config: 'groq_api_key', label: 'Groq', url: 'https://console.groq.com', prefix: 'gsk_' },
+            openai: { env: 'OPENAI_API_KEY', config: 'openai_api_key', label: 'OpenAI', url: 'https://platform.openai.com', prefix: 'sk-' },
+            anthropic: { env: 'ANTHROPIC_API_KEY', config: 'anthropic_api_key', label: 'Anthropic', url: 'https://console.anthropic.com', prefix: 'sk-ant' },
+            gemini: { env: 'GEMINI_API_KEY', config: 'gemini_api_key', label: 'Gemini', url: 'https://aistudio.google.com', prefix: '' }
+        };
+
+        const info = keyMap[provider] || keyMap.groq;
+        let key = process.env[info.env] || this.config[info.config];
 
         if (!key) {
-            console.log(`\n${Colors.WARNING}⚠️  Configuration manquante${Colors.ENDC}`);
-            console.log(`${Colors.DIM}Pour utiliser Ghost, vous avez besoin d'une clé API Groq (Gratuite).${Colors.ENDC}`);
-            console.log(`${Colors.BLUE}👉 Obtenir une clé : https://console.groq.com${Colors.ENDC}\n`);
+            console.log(`\n${Colors.WARNING}⚠️  Configuration manquante pour ${info.label}${Colors.ENDC}`);
+            console.log(`${Colors.DIM}Pour utiliser Ghost avec ${info.label}, vous avez besoin d'une clé API.${Colors.ENDC}`);
+            console.log(`${Colors.BLUE}👉 Obtenir une clé : ${info.url}${Colors.ENDC}\n`);
             
-            key = await promptUser(`${Colors.BOLD}Collez votre clé Groq (gsk_...) : ${Colors.ENDC}`);
+            key = await promptUser(`${Colors.BOLD}Collez votre clé ${info.label} : ${Colors.ENDC}`);
             
-            if (key && key.trim().startsWith('gsk_')) {
-                this.config.groq_api_key = key.trim();
+            if (key && key.trim()) {
+                this.config[info.config] = key.trim();
                 this.save();
             } else {
-                console.log(`${Colors.FAIL}❌ Clé invalide ou manquante. Abandon.${Colors.ENDC}`);
+                console.log(`${Colors.FAIL}❌ Clé manquante. Abandon.${Colors.ENDC}`);
                 process.exit(1);
             }
         }
@@ -96,14 +105,27 @@ class ConfigManager {
 // 🧠 MOTEUR IA (Client HTTPS Natif)
 // ==============================================================================
 class AIEngine {
-    constructor(apiKey, model) {
+    constructor(apiKey, model, provider = 'groq') {
         this.apiKey = apiKey;
-        this.hostname = "api.groq.com";
-        this.path = "/openai/v1/chat/completions";
         this.model = model || DEFAULT_MODEL;
+        this.provider = provider;
+        
+        // Configuration des providers
+        this.providers = {
+            groq: {
+                hostname: "api.groq.com",
+                path: "/openai/v1/chat/completions"
+            },
+            openai: {
+                hostname: "api.openai.com",
+                path: "/v1/chat/completions"
+            }
+        };
     }
 
     async call(systemPrompt, userPrompt, temperature = 0.3, jsonMode = false) {
+        const config = this.providers[this.provider] || this.providers.groq;
+        
         const payload = {
             model: this.model,
             messages: [
@@ -118,8 +140,8 @@ class AIEngine {
         }
 
         const options = {
-            hostname: this.hostname,
-            path: this.path,
+            hostname: config.hostname,
+            path: config.path,
             method: 'POST',
             headers: {
                 'Authorization': `Bearer ${this.apiKey}`,
@@ -136,9 +158,9 @@ class AIEngine {
                     if (res.statusCode >= 400) {
                         try {
                             const errBody = JSON.parse(data);
-                            reject(new Error(`API Error ${res.statusCode}: ${errBody.error?.message || data}`));
+                            reject(new Error(`API Error ${res.statusCode} (${this.provider}): ${errBody.error?.message || data}`));
                         } catch (e) {
-                            reject(new Error(`API Error ${res.statusCode}: ${data}`));
+                            reject(new Error(`API Error ${res.statusCode} (${this.provider}): ${data}`));
                         }
                     } else {
                         try {
@@ -165,6 +187,7 @@ function parseArgs() {
     const args = process.argv.slice(2);
     const flags = {
         model: null,
+        provider: null,
         noSecurity: false,
         dryRun: false,
         help: false
@@ -174,6 +197,9 @@ function parseArgs() {
         if (args[i] === '--model' && args[i + 1]) {
             flags.model = args[i + 1];
             i++;
+        } else if (args[i] === '--provider' && args[i + 1]) {
+            flags.provider = args[i + 1];
+            i++;
         } else if (args[i] === '--no-security') {
             flags.noSecurity = true;
         } else if (args[i] === '--dry-run') {
@@ -187,17 +213,22 @@ function parseArgs() {
 
 function showHelp() {
     console.log(`
-${Colors.BOLD}${Colors.CYAN}GHOST CLI v0.1.0${Colors.ENDC}
-Assistant Git Intelligent basé sur l'IA (Groq)
+${Colors.BOLD}${Colors.CYAN}GHOST CLI v0.2.0${Colors.ENDC}
+Assistant Git Intelligent basé sur l'IA (Groq/OpenAI)
 
 ${Colors.BOLD}USAGE:${Colors.ENDC}
   ghost [options]
 
 ${Colors.BOLD}OPTIONS:${Colors.ENDC}
-  --model <name>    Utiliser un modèle spécifique (ex: llama-3.1-8b-instant)
-  --no-security     Désactiver l'audit de sécurité
-  --dry-run         Générer le message sans effectuer le commit
-  --help, -h        Afficher cette aide
+  --model <name>     Utiliser un modèle spécifique (ex: llama-3.1-8b-instant)
+  --provider <name>  Choisir le fournisseur (groq [défaut], openai)
+  --no-security      Désactiver l'audit de sécurité
+  --dry-run          Générer le message sans effectuer le commit
+  --help, -h         Afficher cette aide
+
+${Colors.BOLD}CONFIGURATION LOCALE (.ghostrc):${Colors.ENDC}
+  Créez un fichier ${Colors.CYAN}.ghostrc${Colors.ENDC} JSON dans votre projet pour personnaliser le prompt :
+  { "prompt": "Ton prompt personnalisé ici" }
     `);
 }
 
@@ -338,9 +369,22 @@ async function main() {
     }
 
     console.clear();
-    console.log(`\n${Colors.BOLD}${Colors.CYAN} 👻 GHOST CLI ${Colors.ENDC}${Colors.DIM} v0.1.0${Colors.ENDC}`);
+    console.log(`\n${Colors.BOLD}${Colors.CYAN} 👻 GHOST CLI ${Colors.ENDC}${Colors.DIM} v0.2.0${Colors.ENDC}`);
     console.log(`${Colors.DIM} ─────────────────────────────────────${Colors.ENDC}\n`);
 
+    // 0. Chargement de la configuration locale .ghostrc
+    const localConfigPath = path.join(process.cwd(), '.ghostrc');
+    let customPrompt = null;
+    if (fs.existsSync(localConfigPath)) {
+        try {
+            const localConfig = JSON.parse(fs.readFileSync(localConfigPath, 'utf8'));
+            if (localConfig.prompt) customPrompt = localConfig.prompt;
+            console.log(`${Colors.DIM}📝 Configuration locale .ghostrc chargée${Colors.ENDC}`);
+        } catch (e) {
+            console.log(`${Colors.WARNING}⚠️  Erreur lecture .ghostrc: ${e.message}${Colors.ENDC}`);
+        }
+    }
+
     // 1. Vérification Git
     if (!checkGitRepo()) {
         console.log(`${Colors.FAIL}❌ Erreur : Ce dossier n'est pas un dépôt Git.${Colors.ENDC}`);
@@ -350,8 +394,9 @@ async function main() {
 
     const config = new ConfigManager();
     const apiKey = await config.getApiKey();
-    const model = flags.model || config.getModel(); // Priorité au flag --model
-    const ai = new AIEngine(apiKey, model);
+    const model = flags.model || config.getModel();
+    const provider = flags.provider || config.config.provider || 'groq';
+    const ai = new AIEngine(apiKey, model, provider);
 
     // 2. Récupération du Diff
     const { text: fullDiffText, map: diffMap, files: fileList } = getStagedDiff();
@@ -364,15 +409,36 @@ async function main() {
 
     // Affichage des fichiers détectés
     console.log(`${Colors.BOLD}📂 Fichiers détectés (${fileList.length}) :${Colors.ENDC}`);
-    fileList.forEach(f => console.log(`   ${Colors.DIM}• ${f}${Colors.ENDC}`));
-    console.log(""); // Saut de ligne
+    fileList.forEach((f, idx) => console.log(`   ${Colors.DIM}${idx + 1}. ${f}${Colors.ENDC}`));
+    console.log("");
+
+    let selectedFiles = fileList;
+    let finalDiffText = fullDiffText;
+    let finalDiffMap = diffMap;
+
+    if (fileList.length > 1) {
+        const selection = await promptUser(`${Colors.BOLD}Sélectionnez les fichiers (ex: 1,3,5 ou 'all' [par défaut]) : ${Colors.ENDC}`);
+        if (selection && selection.toLowerCase() !== 'all' && selection.trim() !== '') {
+            const indices = selection.split(',').map(s => parseInt(s.trim()) - 1).filter(idx => idx >= 0 && idx < fileList.length);
+            if (indices.length > 0) {
+                selectedFiles = indices.map(idx => fileList[idx]);
+                finalDiffMap = {};
+                finalDiffText = "";
+                selectedFiles.forEach(f => {
+                    finalDiffMap[f] = diffMap[f];
+                    finalDiffText += `\n--- ${f} ---\n${diffMap[f]}\n`;
+                });
+                console.log(`${Colors.GREEN}✅ ${selectedFiles.length} fichier(s) sélectionné(s).${Colors.ENDC}\n`);
+            }
+        }
+    }
 
     // 3. Audit de Sécurité
     if (!flags.noSecurity) {
         process.stdout.write(`${Colors.BLUE}🛡️  [1/2] Audit de Sécurité... ${Colors.ENDC}`);
         
         const potentialLeaks = {};
-        for (const [fname, content] of Object.entries(diffMap)) {
+        for (const [fname, content] of Object.entries(finalDiffMap)) {
             const suspects = scanForSecrets(content);
             if (suspects.length > 0) potentialLeaks[fname] = suspects;
         }
@@ -403,14 +469,14 @@ async function main() {
     }
 
     // 4. Génération
-    const tokensEstimates = Math.ceil(fullDiffText.length / 4);
+    const tokensEstimates = Math.ceil(finalDiffText.length / 4);
     console.log(`${Colors.BLUE}⚡ [2/2] Génération du message... ${Colors.DIM}(~${tokensEstimates} tokens)${Colors.ENDC}`);
-    console.log(`${Colors.DIM}   Modèle utilisé : ${model}${Colors.ENDC}`);
+    console.log(`${Colors.DIM}   Modèle utilisé : ${model} (${provider})${Colors.ENDC}`);
     
-    const sysPrompt = "Tu es un assistant Git expert. Génère UNIQUEMENT un message de commit suivant la convention 'Conventional Commits' (ex: feat: add login). Sois concis, descriptif et professionnel. N'utilise pas de markdown (pas de backticks), pas de guillemets autour du message.";
+    const sysPrompt = customPrompt || "Tu es un assistant Git expert. Génère UNIQUEMENT un message de commit suivant la convention 'Conventional Commits' (ex: feat: add login). Sois concis, descriptif et professionnel. N'utilise pas de markdown (pas de backticks), pas de guillemets autour du message.";
     
     try {
-        let commitMsg = await ai.call(sysPrompt, `Diff :\n${fullDiffText.substring(0, 12000)}`);
+        let commitMsg = await ai.call(sysPrompt, `Diff :\n${finalDiffText.substring(0, 12000)}`);
         commitMsg = commitMsg.trim().replace(/^['"`]|['"`]$/g, ''); // Nettoyage final
 
         console.log(`\n${Colors.CYAN}──────────────────────────────────────────────────${Colors.ENDC}`);

package/package.json

@@ -1,10 +1,15 @@
 {
   "name": "atlasia-ghost",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "description": "Utilitaire de verification, validation, et de proposition de commits basé sur l'IA",
   "main": "ghost.js",
+  "files": [
+    "ghost.js",
+    "LICENSE",
+    "README.md"
+  ],
   "bin": {
-    "ghost": "./ghost.js"
+    "ghost": "ghost.js"
   },
   "scripts": {
     "test": "node test.js",
@@ -21,7 +26,7 @@
     "automate",
     "ai",
     "cli",
-    "devsecops"
+    "npm"
   ],
   "author": "Adel Lamallam",
   "license": "MIT",

package/test.js

@@ -1,36 +0,0 @@
-const assert = require('assert');
-const { execSync } = require('child_process');
-const path = require('path');
-
-// Mock Colors if needed or import from ghost.js if it were exported
-const Colors = {
-    CYAN: '', BOLD: '', ENDC: '', DIM: '', GREEN: '', FAIL: ''
-};
-
-console.log('🧪 Running Ghost CLI Tests...\n');
-
-try {
-    // Test 1: Version check
-    const packageJson = require('./package.json');
-    assert.strictEqual(packageJson.version, '0.1.0', 'package.json version should be 0.1.0');
-    console.log('✅ Test 1: package.json version is correct');
-
-    // Test 2: Help flag
-    const helpOutput = execSync('node ghost.js --help', { encoding: 'utf8' });
-    assert.ok(helpOutput.includes('GHOST CLI v0.1.0'), 'Help output should contain version');
-    assert.ok(helpOutput.includes('--model'), 'Help output should contain --model flag');
-    assert.ok(helpOutput.includes('--no-security'), 'Help output should contain --no-security flag');
-    assert.ok(helpOutput.includes('--dry-run'), 'Help output should contain --dry-run flag');
-    console.log('✅ Test 2: Help flag output is correct');
-
-    // Test 3: Internal logic (Entropy)
-    // We can't easily import internal functions without exporting them
-    // But we can test if the file is valid JS
-    execSync('node --check ghost.js');
-    console.log('✅ Test 3: ghost.js syntax is valid');
-
-    console.log('\n🎉 All tests passed successfully!');
-} catch (error) {
-    console.error(`\n❌ Test failed: ${error.message}`);
-    process.exit(1);
-}