const fs = require('fs');
const path = require('path');
const { chatWithAI } = require('../lib/ai-helper');
const config = require('../config.json');

module.exports = async function RestructureScript() {
  console.log('[STATE] RestructureScript');


  // Replace with actual context data later.  This is place holder for getting a task script that isn't written.
  const taskName = 'example-task';
  const scriptPath = path.join(__dirname, '..', 'bot-tasks', `${taskName}.js`);

  if (!fs.existsSync(scriptPath)) {
    console.warn(`[AI] Cannot find script: ${taskName}.js`);
    return;
  }

  const original = fs.readFileSync(scriptPath, 'utf8');
  const prompt = `This script failed repeatedly:\n${original}\nRewrite it to be more reliable. Only return code.`;
  const response = await chatWithAI(prompt, config.ai);

  fs.writeFileSync(scriptPath, response);
  console.log(`[AI] Rewrote ${taskName}`);
};