// PaddedCell Plugin for OpenClaw
// Registers pcexec and safe_restart tools

const { pcexec, pcexecSync } = require('./pcexec/dist/index.js');
const {
  safeRestart,
  createSafeRestartTool,
  StatusManager,
  createApiServer,
  startApiServer,
  SlashCommandHandler
} = require('./safe-restart/dist/index.js');

// Plugin registration function
function register(api, config) {
  const logger = api.logger || { info: console.log, error: console.error };

  logger.info('PaddedCell plugin initializing...');

  // Register pcexec tool - pass a FACTORY function that receives context
  api.registerTool((ctx) => {
    console.log(`[PaddedCell] pcexec factory called with ctx:`, JSON.stringify(ctx, null, 2));
    const agentId = ctx.agentId;
    const workspaceDir = ctx.workspaceDir;

    return {
      name: 'pcexec',
      description: 'Safe exec with password sanitization',
      parameters: {
        type: 'object',
        properties: {
          command: { type: 'string', description: 'Command to execute' },
          cwd: { type: 'string', description: 'Working directory' },
          timeout: { type: 'number', description: 'Timeout in milliseconds' },
        },
        required: ['command'],
      },
      async execute(_id, params) {
        const command = params.command;
        if (!command) {
          throw new Error('Missing required parameter: command');
        }
        console.log(`[PaddedCell] pcexec execute: agentId=${agentId}, workspaceDir=${workspaceDir}`);
        const result = await pcexec(command, {
          cwd: params.cwd || workspaceDir,
          timeout: params.timeout,
          env: {
            AGENT_ID: agentId || '',
            AGENT_WORKSPACE: workspaceDir || '',
          },
        });
        // Format output for OpenClaw tool response
        let output = result.stdout;
        if (result.stderr) {
          output += result.stderr;
        }
        return { content: [{ type: 'text', text: output }] };
      },
    };
  });

  // Register safe_restart tool - pass a FACTORY function that receives context
  api.registerTool((ctx) => {
    const agentId = ctx.agentId;
    const sessionKey = ctx.sessionKey;

    return {
      name: 'safe_restart',
      description: 'Safe coordinated restart of OpenClaw gateway',
      parameters: {
        type: 'object',
        properties: {
          rollback: { type: 'string', description: 'Rollback script path' },
          log: { type: 'string', description: 'Log file path' },
        },
      },
      async execute(_id, params) {
        return await safeRestart({
          agentId: agentId,
          sessionKey: sessionKey,
          rollback: params.rollback,
          log: params.log,
        });
      },
    };
  });

  logger.info('PaddedCell plugin initialized');
}

// Export for OpenClaw
module.exports = { register };

// Also export individual modules for direct use
module.exports.pcexec = pcexec;
module.exports.pcexecSync = pcexecSync;
module.exports.safeRestart = safeRestart;
module.exports.createSafeRestartTool = createSafeRestartTool;
module.exports.StatusManager = StatusManager;
module.exports.createApiServer = createApiServer;
module.exports.startApiServer = startApiServer;
module.exports.SlashCommandHandler = SlashCommandHandler;