puter-cli/src/commands/shell.js

Command line interface for Puter cloud platform

import readline from 'node:readline';
import chalk from 'chalk';
import Conf from 'conf';
import { execCommand, getPrompt } from '../executor.js';
import { getAuthToken, login } from './auth.js';
import { PROJECT_NAME } from '../commons.js';
import ErrorModule from '../modules/ErrorModule.js';
import putility from '@heyputer/putility';

const config = new Conf({ projectName: PROJECT_NAME });

export const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
  prompt: null
});

/**
 * Update the current shell prompt
 */
export function updatePrompt(currentPath) {
  config.set('cwd', currentPath);
  rl.setPrompt(getPrompt());
}

/**
 * Start the interactive shell
 */
export async function startShell() {
  if (!getAuthToken()) {
    console.log(chalk.cyan('Please login first (or use CTRL+C to exit):'));
    await login();
    console.log(chalk.green(`Now just type: ${chalk.cyan('puter')} to begin.`));
    process.exit(0);
  }

  const modules = [
    ErrorModule,
  ];

  const context = new putility.libs.context.Context({
    events: new putility.libs.event.Emitter(),
  });

  for ( const module of modules ) module({ context });

  try {
    console.log(chalk.green('Welcome to Puter-CLI! Type "help" for available commands.'));
    rl.setPrompt(getPrompt());
    rl.prompt();

    rl.on('line', async (line) => {
      const trimmedLine = line.trim();
      if (trimmedLine) {
        try {
          await execCommand(context, trimmedLine);
        } catch (error) {
          console.error(chalk.red(error.message));
        }
      }
      rl.prompt();
    }).on('close', () => {
      console.log(chalk.yellow('\nGoodbye!'));
      process.exit(0);
    });
  } catch (error) {
    console.error(chalk.red('Error starting shell:', error));
  }
}