git-coco 0.5.0 → 0.6.0

package/dist/index.js

@@ -18,9 +18,9 @@ var prettyMilliseconds = require('pretty-ms');
 var path = require('path');
 var minimatch = require('minimatch');
 var fs = require('fs');
+var prompts$1 = require('@inquirer/prompts');
 var os = require('os');
 var ini = require('ini');
-var prompts$1 = require('@inquirer/prompts');
 var simpleGit = require('simple-git');
 
 function _interopNamespaceDefault(e) {
@@ -606,6 +606,82 @@ function removeUndefined(obj) {
     return Object.fromEntries(Object.entries(obj).filter(([, value]) => value !== undefined));
 }
 
+/**
+ * Default Config
+ *
+ * @type {Config}
+ */
+const DEFAULT_CONFIG = {
+    model: 'openai/gpt-4',
+    verbose: false,
+    tokenLimit: 1024,
+    summarizePrompt: SUMMARIZE_PROMPT.template,
+    temperature: 0.4,
+    mode: 'stdout',
+    ignoredFiles: ['package-lock.json'],
+    ignoredExtensions: ['.map', '.lock'],
+    defaultBranch: 'main',
+};
+/**
+ * Config keys
+ *
+ * @type {string[]}
+ */
+const CONFIG_KEYS = Object.keys({
+    ...DEFAULT_CONFIG,
+    huggingFaceHubApiKey: '',
+    openAIApiKey: '',
+    prompt: '',
+});
+
+async function updateFileSection(filePath, startComment, endComment, getNewContent, confirmUpdate = true) {
+    const lines = fs.existsSync(filePath) ? fs.readFileSync(filePath, 'utf-8').split(/\r?\n/) : [];
+    const newLines = [];
+    let foundSection = false;
+    for (let i = 0; i < lines.length; i++) {
+        if (lines[i].trim() === startComment) {
+            foundSection = true;
+            if (confirmUpdate) {
+                const confirmOverwrite = await prompts$1.confirm({
+                    message: `A section already exists in ${filePath}, do you want to override it?`,
+                    default: false,
+                });
+                if (!confirmOverwrite) {
+                    // keep all lines until the end comment
+                    while (i < lines.length && lines[i].trim() !== endComment) {
+                        newLines.push(lines[i]);
+                        i++;
+                    }
+                    newLines.push(endComment);
+                    continue;
+                }
+            }
+            newLines.push(startComment);
+            // Insert the new content
+            const newContent = await getNewContent();
+            newLines.push(newContent);
+            // Skip the existing content of the section
+            while (i < lines.length && lines[i].trim() !== endComment) {
+                i++;
+            }
+            newLines.push(endComment);
+            continue;
+        }
+        if (!foundSection || lines[i].trim() !== endComment) {
+            newLines.push(lines[i]);
+        }
+    }
+    // If section wasn't found, append it at the end
+    if (!foundSection) {
+        newLines.push('\n' + startComment);
+        const newContent = await getNewContent();
+        newLines.push(newContent);
+        newLines.push(endComment);
+    }
+    // Write the updated contents back to the file
+    fs.writeFileSync(filePath, newLines.join('\n'));
+}
+
 /**
  * Load environment variables
  *
@@ -613,27 +689,65 @@ function removeUndefined(obj) {
  * @returns {Config} Updated config
  **/
 function loadEnvConfig(config) {
-    const envConfig = {
-        model: process.env.COCO_MODEL || undefined,
-        openAIApiKey: process.env.OPENAI_API_KEY || undefined,
-        huggingFaceHubApiKey: process.env.HUGGINGFACE_HUB_API_KEY || undefined,
-        tokenLimit: process.env.COCO_TOKEN_LIMIT
-            ? parseInt(process.env.COCO_TOKEN_LIMIT)
-            : undefined,
-        prompt: process.env.COCO_PROMPT,
-        mode: process.env.COCO_MODE,
-        summarizePrompt: process.env.COCO_SUMMARIZE_PROMPT,
-        ignoredFiles: process.env.COCO_IGNORED_FILES
-            ? process.env.COCO_IGNORED_FILES.split(',')
-            : undefined,
-        ignoredExtensions: process.env.COCO_IGNORED_EXTENSIONS
-            ? process.env.COCO_IGNORED_EXTENSIONS.split(',')
-            : undefined,
-        defaultBranch: process.env.COCO_DEFAULT_BRANCH || undefined,
-    };
-    config = { ...config, ...removeUndefined(envConfig) };
-    return config;
+    const envConfig = {};
+    CONFIG_KEYS.forEach((key) => {
+        const envVarName = toEnvVarName(key);
+        const envValue = parseEnvValue(key, process.env[envVarName]);
+        if (envValue === undefined)
+            return;
+        // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+        // @ts-ignore
+        envConfig[key] = envValue;
+    });
+    return { ...config, ...removeUndefined(envConfig) };
+}
+function parseEnvValue(key, value) {
+    if (value === undefined) {
+        return undefined;
+    }
+    else if (key === 'tokenLimit' && typeof value === 'string') {
+        return parseInt(value);
+    }
+    else if ((key === 'ignoredFiles' || key === 'ignoredExtensions') &&
+        typeof value === 'string' &&
+        value.includes(',')) {
+        return value.split(',');
+    }
+    return value;
 }
+function toEnvVarName(key) {
+    switch (key) {
+        case 'openAIApiKey':
+            return 'OPENAI_API_KEY';
+        case 'huggingFaceHubApiKey':
+            return 'HUGGINGFACE_HUB_API_KEY';
+        default:
+            return 'COCO_' + key.replace(/([A-Z])/g, '_$1').toUpperCase();
+    }
+}
+function formatEnvValue(value) {
+    if (typeof value === 'number') {
+        return `${value}`;
+    }
+    else if (Array.isArray(value)) {
+        return `${value.join(',')}`;
+    }
+    else if (typeof value === 'string') {
+        // Escape newlines and tabs in strings
+        return `${value.replace(/\n/g, '\\n').replace(/\t/g, '\\t')}`;
+    }
+    return `${value}`;
+}
+const appendToEnvFile = async (filePath, config) => {
+    const startComment = '# -- Start coco config --';
+    const endComment = '# -- End coco config --';
+    const getNewContent = async () => {
+        return Object.entries(config)
+            .map(([key, value]) => `${toEnvVarName(key)}=${formatEnvValue(value)}`)
+            .join('\n');
+    };
+    await updateFileSection(filePath, startComment, endComment, getNewContent);
+};
 
 /**
  * Load git profile config (from ~/.gitconfig)
@@ -663,6 +777,38 @@ function loadGitConfig(config) {
     }
     return config;
 }
+/**
+ * Appends the provided configuration to a git config file.
+ *
+ * @param filePath - The path to the .gitconfig
+ * @param config - The configuration object to append.
+ */
+const appendToGitConfig = async (filePath, config) => {
+    if (!fs__namespace.existsSync(filePath)) {
+        throw new Error(`File ${filePath} does not exist.`);
+    }
+    const startComment = '# -- Start coco config --';
+    const endComment = '# -- End coco config --';
+    const header = '[coco]';
+    // Function to generate new content for the coco section
+    const getNewContent = async () => {
+        const contentLines = [header];
+        for (const key in config) {
+            // check if string has new lines, if so, wrap in quotes
+            if (typeof config[key] === 'string') {
+                const value = config[key];
+                if (value.includes('\n')) {
+                    contentLines.push(`\t${key} = ${JSON.stringify(value)}`);
+                    continue;
+                }
+            }
+            contentLines.push(`\t${key} = ${config[key]}`);
+        }
+        return contentLines.join('\n');
+    };
+    // Use the updateFileSection utility
+    await updateFileSection(filePath, startComment, endComment, getNewContent);
+};
 
 /**
  * Load .gitignore in project root
@@ -712,6 +858,12 @@ function loadProjectConfig(config) {
     }
     return config;
 }
+const appendToProjectConfig = (filePath, config) => {
+    fs__namespace.writeFileSync(filePath, JSON.stringify({
+        $schema: 'https://git-co.co/schema.json',
+        ...config,
+    }, null, 2));
+};
 
 /**
  * Load XDG config
@@ -729,22 +881,6 @@ function loadXDGConfig(config) {
     return config;
 }
 
-/**
- * Default Config
- *
- * @type {Config}
- */
-const DEFAULT_CONFIG = {
-    model: 'openai/gpt-4',
-    verbose: false,
-    tokenLimit: 1024,
-    summarizePrompt: SUMMARIZE_PROMPT.template,
-    temperature: 0.4,
-    mode: 'stdout',
-    ignoredFiles: ['package-lock.json'],
-    ignoredExtensions: ['.map', '.lock'],
-    defaultBranch: 'main',
-};
 /**
  * Load application config
  *
@@ -864,8 +1000,8 @@ const isInteractive = (argv) => {
 };
 const SEPERATOR = chalk.blue('----------------');
 
-function logResult(result) {
-    console.log(`\n${chalk.bgBlue(chalk.bold('Proposed Commit:'))}\n${SEPERATOR}\n${result}\n${SEPERATOR}\n`);
+function logResult(label, result) {
+    console.log(`\n${chalk.bgBlue(chalk.bold(`Proposed ${label}:`))}\n${SEPERATOR}\n${result}\n${SEPERATOR}\n`);
 }
 
 async function editResult(result, options) {
@@ -966,7 +1102,7 @@ async function generateAndReviewLoop({ label, factory, parser, noResult, agent,
         })
             .stopTimer();
         if (options?.interactive) {
-            logResult(result);
+            logResult(label, result);
             const reviewAnswer = await getUserReviewDecision();
             if (reviewAnswer === 'cancel') {
                 process.exit(0);
@@ -1056,7 +1192,7 @@ const getRepo = () => {
     return git;
 };
 
-async function handler$1(argv) {
+async function handler$2(argv) {
     const tokenizer = getTokenizer();
     const git = getRepo();
     const options = loadConfig(argv);
@@ -1118,7 +1254,7 @@ async function handler$1(argv) {
 /**
  * Command line options via yargs
  */
-const options$1 = {
+const options$2 = {
     model: { type: 'string', description: 'LLM/Model-Name' },
     openAIApiKey: {
         type: 'string',
@@ -1164,16 +1300,16 @@ const options$1 = {
         description: 'Ignored extensions',
     },
 };
-const builder$1 = (yargs) => {
-    return yargs.options(options$1);
+const builder$2 = (yargs) => {
+    return yargs.options(options$2);
 };
 
 var commit = {
     command: 'commit',
     desc: 'Generate commit message',
-    builder: builder$1,
-    handler: handler$1,
-    options: options$1,
+    builder: builder$2,
+    handler: handler$2,
+    options: options$2,
 };
 
 const template = `Write informative git changelog, in the imperative, based on a series of individual messages.
@@ -1249,7 +1385,7 @@ async function getCommitLogCurrentBranch({ git, logger, comparisonBranch = 'main
     return [];
 }
 
-async function handler(argv) {
+async function handler$1(argv) {
     const options = loadConfig(argv);
     const logger = new Logger(options);
     const git = getRepo();
@@ -1320,7 +1456,7 @@ async function handler(argv) {
 /**
  * Command line options via yargs
  */
-const options = {
+const options$1 = {
     range: {
         type: 'string',
         alias: 'r',
@@ -1366,13 +1502,240 @@ const options = {
         description: 'Ignored extensions',
     },
 };
-const builder = (yargs) => {
-    return yargs.options(options);
+const builder$1 = (yargs) => {
+    return yargs.options(options$1);
 };
 
 var changelog = {
     command: 'changelog',
     desc: 'Generate a changelog from a commit range',
+    builder: builder$1,
+    handler: handler$1,
+    options: options$1,
+};
+
+const handleProjectLevelConfig = async () => {
+    const projectConfiguration = await prompts$1.select({
+        message: 'select type project level configuration:',
+        choices: [
+            {
+                name: '.coco.config.json',
+                value: '.coco.config.json',
+            },
+            {
+                name: '.env',
+                value: '.env',
+            },
+        ],
+    });
+    let configFile = '.coco.config.json';
+    if (projectConfiguration === '.env') {
+        configFile = '.env';
+        if (!fs.existsSync('.env')) {
+            fs.writeFileSync('.env', '');
+        }
+    }
+    return configFile;
+};
+const handleSystemLevelConfig = () => {
+    return path.join(os.homedir(), '.gitconfig');
+};
+async function handler(argv) {
+    const options = loadConfig(argv);
+    const logger = new Logger(options);
+    const level = await prompts$1.select({
+        message: 'configure coco at the system or project level:',
+        choices: [
+            {
+                name: 'system',
+                value: 'system',
+                description: 'add coco config to your global git config',
+            },
+            {
+                name: 'project',
+                value: 'project',
+                description: 'add coco config to existing git project',
+            },
+        ],
+    });
+    let configFilePath = '';
+    switch (level) {
+        case 'system':
+            configFilePath = await handleSystemLevelConfig();
+            break;
+        case 'project':
+            configFilePath = await handleProjectLevelConfig();
+            break;
+    }
+    // interactive v.s stdout mode
+    const mode = (await prompts$1.select({
+        message: 'select mode:',
+        choices: [
+            {
+                name: 'interactive',
+                value: 'interactive',
+                description: 'interactive prompt for creating, reviewing, and committing',
+            },
+            {
+                name: 'stdout',
+                value: 'stdout',
+                description: 'print results to stdout',
+            },
+        ],
+    }));
+    const apiKey = await prompts$1.password({
+        message: `enter your OpenAI API key:`,
+        validate(input) {
+            return input.length > 0 ? true : 'API key cannot be empty';
+        },
+    });
+    const tokenLimit = await prompts$1.input({
+        message: 'maximum number of tokens for the commit message:',
+        default: '500',
+    });
+    const defaultBranch = await prompts$1.input({
+        message: 'default branch for the repository:',
+        default: 'main',
+    });
+    const advOptions = await prompts$1.confirm({
+        message: 'would you like to configure advanced options?',
+        default: false,
+    });
+    const config = {
+        openAIApiKey: '•••••••••••••••',
+        tokenLimit: parseInt(tokenLimit),
+        defaultBranch,
+        mode,
+    };
+    /**
+     * Prompt for advanced options
+     *
+     * e.g.
+     * - temperature
+     * - verbose logging
+     * - ignored files
+     * - ignored extensions
+     * - commit message prompt
+     */
+    if (advOptions) {
+        const temperature = await prompts$1.input({
+            message: 'temperature for the model:',
+            default: '0.4',
+        });
+        config.temperature = parseFloat(temperature);
+        config.verbose = await prompts$1.confirm({
+            message: 'enable verbose logging:',
+            default: false,
+        });
+        const promptForIgnores = await prompts$1.confirm({
+            message: 'would you like to configure ignored files and extensions?',
+            default: false,
+        });
+        if (promptForIgnores) {
+            const ignoredFiles = await prompts$1.input({
+                message: 'paths of files to be excluded when generating commit messages (comma-separated):',
+                default: 'package-lock.json',
+            });
+            const ignoredExtensions = await prompts$1.input({
+                message: 'file extensions to be excluded when generating commit messages (comma-separated):',
+                default: '.map, .lock',
+            });
+            config.ignoredFiles =
+                (ignoredFiles && ignoredFiles.split(',').map((file) => file.trim())) || [];
+            config.ignoredExtensions =
+                (ignoredExtensions && ignoredExtensions.split(',').map((ext) => ext.trim())) || [];
+        }
+        const promptForCommitPrompt = await prompts$1.confirm({
+            message: 'would you like to configure the commit message prompt?',
+            default: false,
+        });
+        if (promptForCommitPrompt) {
+            const commitPrompt = await prompts$1.editor({
+                message: 'modify default commit message prompt:',
+                default: COMMIT_PROMPT.template,
+            });
+            config.prompt = commitPrompt;
+        }
+    }
+    logResult('Config', JSON.stringify(config, null, 2));
+    // add to config after logging, so that the API key is not logged
+    config.openAIApiKey = apiKey;
+    const isApproved = await prompts$1.confirm({
+        message: 'look good? (hiding API key for security)',
+    });
+    if (isApproved) {
+        if (configFilePath.endsWith('.gitconfig')) {
+            await appendToGitConfig(configFilePath, config);
+        }
+        else if (configFilePath === '.env') {
+            await appendToEnvFile(configFilePath, config);
+        }
+        else if (configFilePath === '.coco.config.json') {
+            await appendToProjectConfig(configFilePath, config);
+        }
+        logger.log(`init successful! 🦾🤖🎉`, { color: 'green' });
+    }
+    else {
+        logger.log('init cancelled.', { color: 'yellow' });
+    }
+}
+
+/**
+ * Command line options via yargs
+ */
+const options = {
+    model: { type: 'string', description: 'LLM/Model-Name' },
+    openAIApiKey: {
+        type: 'string',
+        description: 'OpenAI API Key',
+        conflicts: 'huggingFaceHubApiKey',
+    },
+    huggingFaceHubApiKey: {
+        type: 'string',
+        description: 'HuggingFace Hub API Key',
+        conflicts: 'openAIApiKey',
+    },
+    tokenLimit: { type: 'number', description: 'Token limit' },
+    prompt: {
+        type: 'string',
+        alias: 'p',
+        description: 'Commit message prompt',
+    },
+    i: {
+        type: 'boolean',
+        alias: 'interactive',
+        description: 'Toggle interactive mode',
+    },
+    s: {
+        type: 'boolean',
+        description: 'Automatically commit staged changes with generated commit message',
+        default: false,
+    },
+    e: {
+        type: 'boolean',
+        alias: 'edit',
+        description: 'Open commit message in editor before proceeding',
+    },
+    summarizePrompt: {
+        type: 'string',
+        description: 'Large file summary prompt',
+    },
+    ignoredFiles: {
+        type: 'array',
+        description: 'Ignored files',
+    },
+    ignoredExtensions: {
+        type: 'array',
+        description: 'Ignored extensions',
+    },
+};
+const builder = (yargs) => {
+    return yargs.options(options);
+};
+
+var init = {
+    command: 'init',
+    desc: 'Setup coco for a new project or system',
     builder,
     handler,
     options,
@@ -1391,5 +1754,10 @@ commit.builder, commit.handler)
 // eslint-disable-next-line @typescript-eslint/ban-ts-comment
 // @ts-ignore
 changelog.builder, changelog.handler)
+    .command(init.command, init.desc, 
+// TODO: fix type on builder
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-ignore
+init.builder, init.handler)
     .demandCommand()
     .help().argv;

package/dist/lib/config/constants.d.ts

@@ -0,0 +1,13 @@
+import { Config } from './types';
+/**
+ * Default Config
+ *
+ * @type {Config}
+ */
+export declare const DEFAULT_CONFIG: Config;
+/**
+ * Config keys
+ *
+ * @type {string[]}
+ */
+export declare const CONFIG_KEYS: (keyof Config)[];

package/dist/lib/config/loadConfig.d.ts

@@ -1,10 +1,4 @@
 import { Config } from './types';
-/**
- * Default Config
- *
- * @type {Config}
- */
-export declare const DEFAULT_CONFIG: Config;
 /**
  * Load application config
  *

package/dist/lib/config/services/env.d.ts

@@ -6,3 +6,4 @@ import { Config } from '../types';
  * @returns {Config} Updated config
  **/
 export declare function loadEnvConfig(config: Config): Config;
+export declare const appendToEnvFile: (filePath: string, config: Partial<Config>) => Promise<void>;

package/dist/lib/config/services/git.d.ts

@@ -6,3 +6,10 @@ import { Config } from '../types';
  * @returns {Config} Updated config
  **/
 export declare function loadGitConfig(config: Config): Config;
+/**
+ * Appends the provided configuration to a git config file.
+ *
+ * @param filePath - The path to the .gitconfig
+ * @param config - The configuration object to append.
+ */
+export declare const appendToGitConfig: (filePath: string, config: Partial<Config>) => Promise<void>;

package/dist/lib/config/services/project.d.ts

@@ -6,3 +6,4 @@ import { Config } from '../types';
  * @returns {Config} Updated config
  **/
 export declare function loadProjectConfig(config: Config): Config;
+export declare const appendToProjectConfig: (filePath: string, config: Partial<Config>) => void;

package/dist/lib/ui/logResult.d.ts

@@ -1 +1 @@
-export declare function logResult(result: string): void;
+export declare function logResult(label: string, result: string): void;

package/dist/lib/utils/updateFileSection.d.ts

@@ -0,0 +1 @@
+export declare function updateFileSection(filePath: string, startComment: string, endComment: string, getNewContent: () => Promise<string>, confirmUpdate?: boolean): Promise<void>;