index.ts

#!/usr/bin/env node
import inquirer from 'inquirer';
import inquirerFileTreeSelection from 'inquirer-file-tree-selection-prompt';
import { CliParser } from './src/cli-parser.js';
import { ConfigStore } from './src/config.js';
import { FeedbackServer } from './src/feedback-server.js';
import { AutoReviewer } from './src/index.js';
import { Utils } from './src/utils.js';
import { ReviewOptions } from './types';
import chalk from 'chalk';


inquirer.registerPrompt('file-tree-selection', inquirerFileTreeSelection);

const selectFiles = async () => {
    try {
        enum ReviewMode {
            FILES_TO_BE_REVIEWED = "FILES_TO_BE_REVIEWED",
            REVIEW_ENTIRE_CODEBASE = "REVIEW_ENTIRE_CODEBASE",
            REVIEW_AGAINST_ANOTHER_BRANCH = "REVIEW_AGAINST_ANOTHER_BRANCH"
        }

        const selectMode = await inquirer.prompt([
            {
                type: "list",
                name: 'reviewMode',
                message: 'Select a code review mode',
                choices: [
                    { name: 'Select files to be reviewed', value: ReviewMode.FILES_TO_BE_REVIEWED },
                    { name: 'Review entire code base', value: ReviewMode.REVIEW_ENTIRE_CODEBASE },
                    { name: 'Review this branch against another branch (coming soon)', value: ReviewMode.REVIEW_AGAINST_ANOTHER_BRANCH, disabled: true }
                ],
            }
        ])

        if (selectMode.reviewMode === ReviewMode.FILES_TO_BE_REVIEWED) {
            const { files } = await inquirer.prompt([
                {
                    type: 'file-tree-selection',
                    name: 'files',
                    message: 'Select files',
                    root: '.',
                    multiple: true,
                    ignore: ['node_modules/*', '.git/*', 'dist/*', 'build/*', 'coverage/*', '.vscode/*', '.gitignore'],
                },
            ]);
            return startCodeReviewProcess({
                files
            })
        }

        if (selectMode.reviewMode === ReviewMode.REVIEW_AGAINST_ANOTHER_BRANCH) {
            //TODO: you can want to view agains another branch for a specific file
            const { branch } = await inquirer.prompt([
                {
                    type: 'input',
                    name: 'branch',
                    message: 'Enter branch name',
                },
            ]);
            return startCodeReviewProcess({
                branch
            })
        }

        return startCodeReviewProcess()

        // Perform the desired action with the selected files
    } catch (error) {
        console.error('Error:', error);
    }
};

/**
 * Displays a welcome message to the user
 */
const showWelcomeMessage = () => {
    const welcomeText = `
        =======================================================
        
            Welcome to AutoReviewer - Your Code Review Helper

        =======================================================

        1. AutoReviewer will review and analyze your code
        2. AutoReviewer will provide detailed feedback
        3. With our git integration AutoReviewer will 
        automatically help you carry out code reviews faster 

        -------------------------------------------------------
    `;
    Utils.log(chalk.greenBright.bold(welcomeText));
}

showWelcomeMessage();

const autoReviewer = new AutoReviewer();
const config = new ConfigStore();
const feedbackServer = new FeedbackServer();
new CliParser(process.argv).executeCommand();

const startCodeReviewProcess = async (option?: ReviewOptions) => {
    await autoReviewer.codeReview(option)
    // const score = await autoReviewer.getScore();
    // Utils.sleep(1000);
    Utils.log(chalk.greenBright.bold('Code Review Completed successfully \n'));
    // Utils.log(chalk.greenBright.bold(`Your code score is: ${score} / 10`));
    Utils.log(chalk.greenBright.bold(`Please visit http://localhost:3123 to view your feedback`));
    return feedbackServer.init();
}

if (config.get('openai_api_key')) {
    (async () => {
        try {
            await selectFiles();
        } catch (error: any) {
            Utils.log(chalk.red.bold(`Error occurred: ${error.message}`));
        }
    })();
} else {
    (async () => {
        try {
            await config.promptUserForApiKey()
            Utils.sleep(1000);
            Utils.log(chalk.greenBright.bold(`API Key saved successfully ☑️ 

            Please run the command again to start the code review

            `));
            await selectFiles()
        } catch (error: any) {
            Utils.log(chalk.red.bold(`Error occurred: ${error.message}`));
        }
    })();
}