aex-code 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Md. Abid Hasan Rafi
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the “Software”), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,71 @@
+![AEX Code Cover](aex-code-cover.jpg)
+# A E X - C O D E
+
+AEX Code is a CLI application that functions as an advanced coding agent similar to Claude Code. It is optimized for free-tier usage via OpenRouter models and provides a clean UI with both interactive and command-based workflows. This project was developed by [Md. Abid Hasan Rafi](https://abidhasanrafi.github.io) and powered by [AI Extension](https://aiextension.org).
+
+## Features
+
+- **Code Generation and Explanation:** Interactive terminal sessions within your workspace.
+- **File Management:** Create, review, and edit files with context-aware AI assistance.
+- **Safe Shell Commands:** Execute shell commands with controlled output handling.
+- **Multi-file Context Processing:** Build rich context from your codebase across multiple files.
+- **Configurable Models:** Use OpenRouter API keys to select preferred models. Defaults to `qwen/qwen3-coder:free`.
+
+## Requirements
+
+- Node.js >= 16
+
+## Setup and Configuration
+
+1. Install globally:
+   ```bash
+   npm install -g aex-code
+   ```
+
+2. Configure your OpenRouter API key:
+   ```bash
+   aex-code config --set-key "sk-or-v1-..."
+   ```
+
+3. Set your preferred model (optional):
+   ```bash
+   aex-code config --set-model "anthropic/claude-3.5-sonnet:beta"
+   ```
+
+## Usage
+
+### Interactive Shell
+
+Start an interactive session with the AI agent. The session maintains context and streams responses.
+
+```bash
+aex-code chat
+```
+
+### Analyze and Edit a File
+
+Analyze and modify local files using AI.
+
+```bash
+aex-code edit src/index.js
+```
+
+### Initialize Configuration
+
+```bash
+aex-code init
+```
+
+### Run Commands
+
+```bash
+aex-code run "npm install"
+```
+
+## Development and Contribution
+
+The project follows a modular architecture where commands are organized under `src/commands`, making it easy to extend and maintain.
+
+## License
+
+MIT

package/bin/aex-code.js

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+
+import '../src/index.js';

package/package.json

@@ -0,0 +1,42 @@
+{
+  "name": "aex-code",
+  "version": "1.0.0",
+  "description": "Next-Gen OpenRouter Coding Agent CLI - Powered by AI Extension",
+  "main": "src/index.js",
+  "scripts": {
+    "start": "node ./bin/aex-code.js"
+  },
+  "keywords": [
+    "aex-code",
+    "ai",
+    "cli",
+    "coding-agent",
+    "agent",
+    "openrouter",
+    "llm"
+  ],
+  "author": "Your Name / AI Extension",
+  "license": "MIT",
+  "type": "module",
+  "dependencies": {
+    "boxen": "^7.1.1",
+    "chalk": "^4.1.2",
+    "commander": "^14.0.3",
+    "dotenv": "^17.3.1",
+    "figlet": "^1.11.0",
+    "fs-extra": "^11.3.4",
+    "inquirer": "^8.2.7",
+    "openai": "^6.32.0",
+    "ora": "^5.4.1"
+  },
+  "bin": {
+    "aex-code": "bin/aex-code.js"
+  },
+  "engines": {
+    "node": ">=16.0.0"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/yourusername/aex-code.git"
+  }
+}

package/src/commands/chat.js

@@ -0,0 +1,91 @@
+import inquirer from 'inquirer';
+import chalk from 'chalk';
+import ora from 'ora';
+import fs from 'fs';
+import path from 'path';
+import boxen from 'boxen';
+import { createClient, getDefaultModel, debugApiKey } from '../utils/openrouter.js';
+
+const SYSTEM_PROMPT = `You are AEX Code, an advanced coding agent. You provide short, actionable coding assistance. 
+If asked to edit a file, provide the full file contents or explicitly outline what to change in the file.
+Try to use <tool_call> JSON format for file writes if possible.
+`;
+
+let conversationHistory = [];
+
+export async function chatCommand() {
+    console.clear();
+    const chatHeader = boxen(chalk.greenBright.bold('A E X   C O D E'), {
+        padding: 1, margin: 1, borderStyle: 'bold', borderColor: 'green'
+    });
+    console.log(chatHeader);
+    console.log(chalk.green(`  ■ Target Model    : ${chalk.gray(getDefaultModel())}`));
+    console.log(chalk.green(`  ■ Exit Command    : ${chalk.gray("'exit'")}\n`));
+
+    let client;
+    try {
+        client = createClient();
+    } catch (e) {
+        console.log(chalk.red(e.message));
+        process.exit(1);
+    }
+    
+    conversationHistory.push({ role: 'system', content: SYSTEM_PROMPT });
+
+    while (true) {
+        process.stdout.write('\n');
+        const { userInput } = await inquirer.prompt([
+            {
+                type: 'input',
+                name: 'userInput',
+                message: chalk.bold.green('■ USER : '),
+            }
+        ]);
+
+        if (userInput.trim().toLowerCase() === 'exit') {
+            console.log(chalk.green('\n■ SESSION TERMINATED.'));
+            break;
+        }
+
+        conversationHistory.push({ role: 'user', content: userInput });
+
+        process.stdout.write('\n');
+        const spinner = ora({
+            text: chalk.italic.green('AEX PROCESSSING...'),
+            color: 'green',
+        }).start();
+        
+        try {
+            const response = await client.chat.completions.create({
+                model: getDefaultModel(),
+                messages: conversationHistory,
+                stream: true,
+            });
+
+            spinner.stop();
+            process.stdout.write(`${chalk.bgGreen.black.bold(' ■ AEX AGENT ')} `);
+            
+            let fullContent = '';
+            for await (const chunk of response) {
+                const text = chunk.choices[0]?.delta?.content || '';
+                process.stdout.write(chalk.green(text));
+                fullContent += text;
+            }
+            console.log('\n');
+            conversationHistory.push({ role: 'assistant', content: fullContent });
+
+        } catch (e) {
+            spinner.stop();
+            console.log(chalk.bgRed.white.bold(' ■ ERROR '));
+            if (e.status === 429) {
+                console.log(chalk.red(`\nRate Limit or Provider Error (429): The active model (${getDefaultModel()}) is currently overloaded.`));
+                console.log(chalk.yellow(`Tip: Try swapping your model using:\naex-code config --set-model "google/gemini-2.0-flash-lite-preview-02-05:free"\n`));
+            } else if (e.status === 401) {
+                console.log(chalk.red(`\nUnauthorized (401): Your API key is invalid or not recognized by OpenRouter.`));
+            } else {
+                console.log(chalk.red(`\nFailed to reach OpenRouter API: ${e.message}\n`));
+            }
+            conversationHistory.pop();
+        }
+    }
+}

package/src/commands/config.js

@@ -0,0 +1,21 @@
+import chalk from 'chalk';
+import { saveConfig } from '../utils/configManager.js';
+
+export const configCommand = (options) => {
+    if (options.setKey) {
+        // Automatically strip accidental quotes just in case
+        const cleanKey = options.setKey.replace(/^['"]|['"]$/g, '');
+        saveConfig({ OPENROUTER_API_KEY: cleanKey });
+        console.log(chalk.green('✔ OpenRouter API Key configured successfully.'));
+    }
+
+    if (options.setModel) {
+        const cleanModel = options.setModel.replace(/^['"]|['"]$/g, '');
+        saveConfig({ OPENROUTER_DEFAULT_MODEL: cleanModel });
+        console.log(chalk.green(`✔ Default Model set to ${cleanModel}.`));
+    }
+
+    if (!options.setKey && !options.setModel) {
+        console.log(chalk.yellow('Usage: aex-code config --set-key <key> or --set-model <model>'));
+    }
+};

package/src/commands/edit.js

@@ -0,0 +1,33 @@
+import chalk from 'chalk';
+import fs from 'fs';
+import path from 'path';
+import ora from 'ora';
+import { createClient, getDefaultModel } from '../utils/openrouter.js';
+
+export async function editCommand(file, options) {
+    if (!fs.existsSync(file)) {
+        console.log(chalk.red(`File ${file} does not exist.`));
+        process.exit(1);
+    }
+    
+    console.log(chalk.blue(`Suggesting edits for ${file}...`));
+    const content = fs.readFileSync(file, 'utf-8');
+    
+    const client = createClient();
+    const spinner = ora('AEX Code is analyzing the file...').start();
+    
+    try {
+        const response = await client.chat.completions.create({
+            model: getDefaultModel(),
+            messages: [
+                { role: 'system', content: 'You are AEX Code. Please evaluate this code file and suggest improvements. Keep output concise.' },
+                { role: 'user', content: `File name: ${file}\n\nCode:\n\`\`\`\n${content}\n\`\`\`` }
+            ]
+        });
+        spinner.stop();
+        console.log(chalk.green(response.choices[0].message.content));
+    } catch (e) {
+        spinner.fail('Error analyzing file.');
+        console.log(chalk.red(e.message));
+    }
+}

package/src/commands/init.js

@@ -0,0 +1,16 @@
+import fs from 'fs';
+import path from 'path';
+import chalk from 'chalk';
+
+export function initCommand() {
+    console.log(chalk.green('Initializing AEX Code project...'));
+    const configPath = path.join(process.cwd(), '.aex-code');
+
+    if (!fs.existsSync(configPath)) {
+        fs.mkdirSync(configPath);
+        fs.writeFileSync(path.join(configPath, 'rules.md'), '# AEX Code Project Rules\n\nAdd project-specific rules here.');
+        console.log(chalk.green('✔ Initialized .aex-code directory securely.'));
+    } else {
+        console.log(chalk.yellow('AEX Code project already initialized in this directory.'));
+    }
+}

package/src/commands/run.js

@@ -0,0 +1,21 @@
+import { exec } from 'child_process';
+import chalk from 'chalk';
+import ora from 'ora';
+
+export function runCommand(cmdString) {
+    const spinner = ora(`Running: ${cmdString}`).start();
+    
+    exec(cmdString, (error, stdout, stderr) => {
+        spinner.stop();
+        if (error) {
+            console.error(chalk.red(`Error executing command: ${error.message}`));
+            return;
+        }
+        if (stderr) {
+            console.error(chalk.yellow(`Output:\n${stderr}`));
+        }
+        if (stdout) {
+            console.log(chalk.green(`Result:\n${stdout}`));
+        }
+    });
+}

package/src/index.js

@@ -0,0 +1,72 @@
+import { Command } from 'commander';
+import chalk from 'chalk';
+import figlet from 'figlet';
+import boxen from 'boxen';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+
+import { initCommand } from './commands/init.js';
+import { chatCommand } from './commands/chat.js';
+import { configCommand } from './commands/config.js';
+import { editCommand } from './commands/edit.js';
+import { runCommand } from './commands/run.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+const packageJson = JSON.parse(
+  fs.readFileSync(path.join(__dirname, '../package.json'), 'utf-8')
+);
+
+const program = new Command();
+
+const bannerText = figlet.textSync('AEX Code', { font: 'ANSI Shadow', horizontalLayout: 'fitted' });
+const coloredBanner = chalk.greenBright(bannerText);
+const rawTagline = ' NEXT-GEN OPENROUTER CODING AGENT - POWERED BY AI EXTENSION ';
+
+const finalBanner = boxen(`${coloredBanner}\n\n${chalk.bgGreen.black.bold(rawTagline)}`, {
+  padding: 1,
+  margin: 1,
+  borderStyle: 'bold',
+  borderColor: 'green',
+  textAlignment: 'center',
+});
+
+program
+  .name('aex-code')
+  .description(finalBanner)
+  .version(packageJson.version);
+
+program
+  .command('init')
+  .description('Initialize AEX Code in the current workspace')
+  .action(initCommand);
+
+program
+  .command('chat')
+  .description('Start an interactive chat session with AEX Code')
+  .action(chatCommand);
+
+program
+  .command('config')
+  .description('Configure API keys and default models for AEX Code')
+  .option('--set-key <key>', 'Set OpenRouter API key')
+  .option('--set-model <model>', 'Set default OpenRouter model')
+  .action(configCommand);
+
+program
+  .command('edit <file>')
+  .description('Edit or analyze a file locally')
+  .action(editCommand);
+
+program
+  .command('run <cmd>')
+  .description('Run a shell command safely')
+  .action(runCommand);
+
+program.parse(process.argv);
+
+if (!process.argv.slice(2).length) {
+  program.outputHelp();
+}

package/src/utils/configManager.js

@@ -0,0 +1,25 @@
+import fs from 'fs';
+import path from 'path';
+import os from 'os';
+
+const CONFIG_PATH = path.join(os.homedir(), '.aex-code.json');
+
+export const loadConfig = () => {
+    if (fs.existsSync(CONFIG_PATH)) {
+        try {
+            const raw = fs.readFileSync(CONFIG_PATH, 'utf-8');
+            return JSON.parse(raw);
+        } catch (e) {
+            return {};
+        }
+    }
+    return {};
+};
+
+export const saveConfig = (newConfig) => {
+    const config = { ...loadConfig(), ...newConfig };
+    fs.writeFileSync(CONFIG_PATH, JSON.stringify(config, null, 2), 'utf-8');
+    return config;
+};
+
+export const getConfig = (key) => loadConfig()[key];

package/src/utils/openrouter.js

@@ -0,0 +1,30 @@
+import OpenAI from 'openai';
+import { getConfig } from './configManager.js';
+import dotenv from 'dotenv';
+dotenv.config({ path: '.env.local', silent: true });
+
+export const createClient = () => {
+    const apiKey = process.env.OPENROUTER_API_KEY || getConfig('OPENROUTER_API_KEY');
+    if (!apiKey) {
+        throw new Error('OpenRouter API Key not found. Please run "aex-code config" to setup your key.');
+    }
+
+    return new OpenAI({
+        baseURL: 'https://openrouter.ai/api/v1',
+        apiKey: apiKey.trim(),
+        defaultHeaders: {
+            'HTTP-Referer': 'https://github.com/aex-code',
+            'X-Title': 'AEX Code CLI',
+        }
+    });
+};
+
+export const debugApiKey = () => {
+    const key = process.env.OPENROUTER_API_KEY || getConfig('OPENROUTER_API_KEY');
+    if (!key) return null;
+    return `${key.substring(0, 10)}...${key.slice(-4)}`;
+};
+
+export const getDefaultModel = () => {
+    return process.env.OPENROUTER_DEFAULT_MODEL || getConfig('OPENROUTER_DEFAULT_MODEL') || 'qwen/qwen-2.5-coder-32b-instruct:free';
+};