mirror of
https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git
synced 2026-03-30 13:54:01 +00:00
298 lines
12 KiB
JavaScript
Executable File
298 lines
12 KiB
JavaScript
Executable File
#!/usr/bin/env node
|
||
|
||
import { spawn } from 'child_process';
|
||
import { existsSync, mkdirSync, writeFileSync, readFileSync } from 'fs';
|
||
import { join, dirname } from 'path';
|
||
import { homedir } from 'os';
|
||
import * as readline from 'readline';
|
||
import { fileURLToPath } from 'url';
|
||
|
||
const __filename = fileURLToPath(import.meta.url);
|
||
const __dirname = dirname(__filename);
|
||
|
||
const TFCODE_DIR = join(homedir(), '.tfcode');
|
||
const TOOLS_FILE = join(TFCODE_DIR, 'tools.json');
|
||
const CREDENTIALS_FILE = join(TFCODE_DIR, 'credentials.json');
|
||
const CONFIG_FILE = join(TFCODE_DIR, 'config.json');
|
||
|
||
const COLORS = {
|
||
reset: '\x1b[0m',
|
||
bold: '\x1b[1m',
|
||
green: '\x1b[32m',
|
||
red: '\x1b[31m',
|
||
cyan: '\x1b[36m',
|
||
dim: '\x1b[90m',
|
||
yellow: '\x1b[33m',
|
||
magenta: '\x1b[35m'
|
||
};
|
||
|
||
function log(msg) { console.log(msg); }
|
||
function success(msg) { console.log(`${COLORS.green}✓${COLORS.reset} ${msg}`); }
|
||
function error(msg) { console.error(`${COLORS.red}✗${COLORS.reset} ${msg}`); }
|
||
function info(msg) { console.log(`${COLORS.cyan}ℹ${COLORS.reset} ${msg}`); }
|
||
|
||
function ensureConfigDir() {
|
||
if (!existsSync(TFCODE_DIR)) mkdirSync(TFCODE_DIR, { recursive: true });
|
||
}
|
||
|
||
function loadConfig() {
|
||
const envConfig = {
|
||
workspace_id: process.env.TF_WORKSPACE_ID,
|
||
api_key: process.env.TF_API_KEY,
|
||
region: process.env.TF_REGION
|
||
};
|
||
if (envConfig.workspace_id && envConfig.api_key) return envConfig;
|
||
if (existsSync(CONFIG_FILE)) {
|
||
try { return JSON.parse(readFileSync(CONFIG_FILE, 'utf-8')); } catch {}
|
||
}
|
||
return null;
|
||
}
|
||
|
||
function saveConfig(config) {
|
||
ensureConfigDir();
|
||
writeFileSync(CONFIG_FILE, JSON.stringify(config, null, 2));
|
||
}
|
||
|
||
function runPythonSync(method, config = null) {
|
||
const wsId = config?.workspace_id || process.env.TF_WORKSPACE_ID || '';
|
||
const apiKey = config?.api_key || process.env.TF_API_KEY || '';
|
||
const region = config?.region || process.env.TF_REGION || 'au';
|
||
|
||
const pythonCode = `
|
||
import json, sys, os
|
||
try:
|
||
os.environ["TF_WORKSPACE_ID"] = "${wsId}"
|
||
os.environ["TF_API_KEY"] = "${apiKey}"
|
||
os.environ["TF_REGION"] = "${region}"
|
||
from tf_sync.config import load_config, validate_credentials, Region
|
||
from tf_sync.tools import sync_tools
|
||
from tf_sync.config import get_region_urls
|
||
|
||
method = "${method}"
|
||
if method == "validate":
|
||
config = load_config()
|
||
result = validate_credentials(config)
|
||
urls = get_region_urls(config.region)
|
||
print(json.dumps({"success": result.success, "workspace_id": result.workspace_id, "workspace_name": result.workspace_name, "error": result.error, "base_url": urls["base_url"]}))
|
||
elif method == "sync":
|
||
config = load_config()
|
||
result = sync_tools(config)
|
||
tools_data = [{"id": t.id, "name": t.name, "description": t.description, "tool_type": t.tool_type.value, "request_type": t.request_type.value if t.request_type else None, "url": t.url, "auth_via": t.auth_via} for t in result.tools]
|
||
print(json.dumps({"success": result.success, "tools": tools_data, "by_type": result.by_type, "error": result.error}))
|
||
except Exception as e:
|
||
print(json.dumps({"success": False, "error": str(e)}))
|
||
`;
|
||
|
||
return new Promise((resolve, reject) => {
|
||
const proc = spawn(process.env.TFCODE_PYTHON_PATH || 'python3', ['-c', pythonCode], { env: { ...process.env } });
|
||
let stdout = '', stderr = '';
|
||
proc.stdout.on('data', (d) => stdout += d);
|
||
proc.stderr.on('data', (d) => stderr += d);
|
||
proc.on('close', (code) => {
|
||
if (code !== 0 && !stdout) reject(new Error(`Python failed: ${stderr}`));
|
||
else try { resolve(JSON.parse(stdout.trim())); } catch (e) { reject(new Error(`Parse error: ${stdout}`)); }
|
||
});
|
||
proc.on('error', reject);
|
||
});
|
||
}
|
||
|
||
function loadCachedTools() {
|
||
if (!existsSync(TOOLS_FILE)) return null;
|
||
try { return JSON.parse(readFileSync(TOOLS_FILE, 'utf-8')); } catch { return null; }
|
||
}
|
||
|
||
function saveToolsCache(tools) {
|
||
ensureConfigDir();
|
||
writeFileSync(TOOLS_FILE, JSON.stringify(tools, null, 2));
|
||
}
|
||
|
||
async function question(prompt) {
|
||
return new Promise((resolve) => {
|
||
const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
|
||
rl.question(prompt, (answer) => { rl.close(); resolve(answer.trim()); });
|
||
});
|
||
}
|
||
|
||
async function select(prompt, options) {
|
||
log('');
|
||
log(prompt);
|
||
log('');
|
||
options.forEach((opt, i) => log(` ${COLORS.cyan}${i + 1}.${COLORS.reset} ${opt}`));
|
||
log('');
|
||
const answer = await question('Select (1-' + options.length + '): ');
|
||
const idx = parseInt(answer) - 1;
|
||
return idx >= 0 && idx < options.length ? idx : 0;
|
||
}
|
||
|
||
async function interactiveSetup() {
|
||
log('');
|
||
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
|
||
log(`${COLORS.bold}${COLORS.magenta} tfcode Setup${COLORS.reset}`);
|
||
log(`${COLORS.bold}${COLORS.magenta}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
|
||
log('');
|
||
log('This will guide you through setting up your ToothFairyAI credentials.');
|
||
log('');
|
||
log(`${COLORS.dim}You can find your credentials at:${COLORS.reset}`);
|
||
log(`${COLORS.dim} https://app.toothfairyai.com → Settings → API Keys${COLORS.reset}`);
|
||
log('');
|
||
|
||
log(`${COLORS.bold}Step 1: Workspace ID${COLORS.reset}`);
|
||
log(`${COLORS.dim}This is your workspace UUID${COLORS.reset}`);
|
||
log('');
|
||
const workspaceId = await question('Enter your Workspace ID: ');
|
||
if (!workspaceId) { error('Workspace ID is required'); process.exit(1); }
|
||
log('');
|
||
|
||
log(`${COLORS.bold}Step 2: API Key${COLORS.reset}`);
|
||
log(`${COLORS.dim}Paste or type your API key${COLORS.reset}`);
|
||
log('');
|
||
const apiKey = await question('Enter your API Key: ');
|
||
if (!apiKey) { error('API Key is required'); process.exit(1); }
|
||
log('');
|
||
|
||
log(`${COLORS.bold}Step 3: Region${COLORS.reset}`);
|
||
const regions = ['dev (Development)', 'au (Australia)', 'eu (Europe)', 'us (United States)'];
|
||
const regionIdx = await select('Select your region:', regions);
|
||
const region = ['dev', 'au', 'eu', 'us'][regionIdx];
|
||
|
||
log('');
|
||
log(`${COLORS.bold}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
|
||
log('');
|
||
log(`${COLORS.bold}Summary:${COLORS.reset}`);
|
||
log(` Workspace ID: ${workspaceId}`);
|
||
log(` API Key: ***${apiKey.slice(-4)}`);
|
||
log(` Region: ${region}`);
|
||
log('');
|
||
|
||
const confirm = await question('Save these credentials? (Y/n): ');
|
||
if (confirm.toLowerCase() === 'n' || confirm.toLowerCase() === 'no') { log('Setup cancelled.'); return; }
|
||
|
||
const config = { workspace_id: workspaceId, api_key: apiKey, region };
|
||
saveConfig(config);
|
||
success('Credentials saved to ~/.tfcode/config.json');
|
||
log('');
|
||
|
||
const testNow = await question('Validate credentials now? (Y/n): ');
|
||
if (testNow.toLowerCase() === 'n' || testNow.toLowerCase() === 'no') return;
|
||
|
||
log('');
|
||
info('Validating credentials...');
|
||
log('');
|
||
|
||
try {
|
||
const result = await runPythonSync('validate', config);
|
||
if (result.success) {
|
||
success('Credentials valid!');
|
||
log(` API URL: ${result.base_url}`);
|
||
log(` Workspace ID: ${result.workspace_id}`);
|
||
log('');
|
||
|
||
const syncNow = await question('Sync tools now? (Y/n): ');
|
||
if (syncNow.toLowerCase() === 'n' || syncNow.toLowerCase() === 'no') return;
|
||
|
||
log('');
|
||
info('Syncing tools...');
|
||
log('');
|
||
|
||
const syncResult = await runPythonSync('sync', config);
|
||
if (syncResult.success) {
|
||
saveToolsCache(syncResult);
|
||
success(`Synced ${syncResult.tools.length} tools`);
|
||
if (syncResult.by_type && Object.keys(syncResult.by_type).length > 0) {
|
||
log('');
|
||
log('By type:');
|
||
for (const [type, count] of Object.entries(syncResult.by_type)) log(` ${type}: ${count}`);
|
||
}
|
||
} else {
|
||
error(`Sync failed: ${syncResult.error}`);
|
||
}
|
||
} else {
|
||
error(`Validation failed: ${result.error}`);
|
||
}
|
||
} catch (e) {
|
||
error(`Failed: ${e.message}`);
|
||
}
|
||
|
||
log('');
|
||
log(`${COLORS.bold}${COLORS.green}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
|
||
log(`${COLORS.bold}${COLORS.green} Setup Complete!${COLORS.reset}`);
|
||
log(`${COLORS.bold}${COLORS.green}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${COLORS.reset}`);
|
||
log('');
|
||
log('Commands:');
|
||
log(` ${COLORS.cyan}tfcode validate${COLORS.reset} Check credentials`);
|
||
log(` ${COLORS.cyan}tfcode sync${COLORS.reset} Sync tools`);
|
||
log(` ${COLORS.cyan}tfcode tools list${COLORS.reset} List tools`);
|
||
log('');
|
||
}
|
||
|
||
function showHelp() {
|
||
log('');
|
||
log(`${COLORS.bold}tfcode${COLORS.reset} - ToothFairyAI's AI coding agent`);
|
||
log('');
|
||
log('Commands:');
|
||
log(` ${COLORS.cyan}tfcode setup${COLORS.reset} Interactive credential setup`);
|
||
log(` ${COLORS.cyan}tfcode validate${COLORS.reset} Test credentials`);
|
||
log(` ${COLORS.cyan}tfcode sync${COLORS.reset} Sync tools from workspace`);
|
||
log(` ${COLORS.cyan}tfcode tools list${COLORS.reset} List synced tools`);
|
||
log(` ${COLORS.cyan}tfcode --help${COLORS.reset} Show this help`);
|
||
log(` ${COLORS.cyan}tfcode --version${COLORS.reset} Show version`);
|
||
log('');
|
||
log(`${COLORS.dim}For full TUI, run from source:${COLORS.reset}`);
|
||
log(`${COLORS.dim} bun run packages/tfcode/src/index.ts${COLORS.reset}`);
|
||
log('');
|
||
}
|
||
|
||
const args = process.argv.slice(2);
|
||
const command = args[0];
|
||
|
||
if (args.includes('--help') || args.includes('-h')) {
|
||
showHelp();
|
||
} else if (args.includes('--version') || args.includes('-v')) {
|
||
log('tfcode v1.0.0-beta.9');
|
||
} else if (command === 'setup') {
|
||
interactiveSetup();
|
||
} else if (command === 'validate') {
|
||
(async () => {
|
||
const config = loadConfig();
|
||
if (!config) { error('No credentials. Run: tfcode setup'); process.exit(1); }
|
||
info('Validating...');
|
||
try {
|
||
const result = await runPythonSync('validate', config);
|
||
if (result.success) { success('Credentials valid'); log(` API URL: ${result.base_url}`); }
|
||
else { error(`Failed: ${result.error}`); process.exit(1); }
|
||
} catch (e) { error(`Failed: ${e.message}`); process.exit(1); }
|
||
})();
|
||
} else if (command === 'sync') {
|
||
(async () => {
|
||
const config = loadConfig();
|
||
if (!config) { error('No credentials. Run: tfcode setup'); process.exit(1); }
|
||
info('Syncing tools...');
|
||
try {
|
||
const result = await runPythonSync('sync', config);
|
||
if (result.success) {
|
||
saveToolsCache(result);
|
||
success(`Synced ${result.tools.length} tools`);
|
||
if (result.by_type) { log(''); log('By type:'); for (const [t, c] of Object.entries(result.by_type)) log(` ${t}: ${c}`); }
|
||
} else { error(`Failed: ${result.error}`); process.exit(1); }
|
||
} catch (e) { error(`Failed: ${e.message}`); process.exit(1); }
|
||
})();
|
||
} else if (command === 'tools' && args[1] === 'list') {
|
||
const cached = loadCachedTools();
|
||
if (!cached?.success) { error('No tools. Run: tfcode sync'); process.exit(1); }
|
||
let tools = cached.tools;
|
||
if (args[3] === '--type' && args[4]) tools = tools.filter(t => t.tool_type === args[4]);
|
||
log(`\n${tools.length} tool(s):\n`);
|
||
for (const t of tools) {
|
||
log(` ${COLORS.cyan}${t.name}${COLORS.reset}`);
|
||
log(` Type: ${t.tool_type}`);
|
||
if (t.description) log(` ${COLORS.dim}${t.description.slice(0, 60)}${COLORS.reset}`);
|
||
log(` Auth: ${t.auth_via}\n`);
|
||
}
|
||
} else if (!command) {
|
||
// Show help instead of trying TUI (TUI requires full build)
|
||
showHelp();
|
||
} else {
|
||
error(`Unknown command: ${command}`);
|
||
showHelp();
|
||
process.exit(1);
|
||
} |