const os = require('os');
const fs = require('fs');
const path = require('path');
const Database = require('better-sqlite3');
const { initAutonomySchema } = require('../src/autonomy/schema');
const { enqueueJob, leaseNextJob, completeJob, failJob } = require('../src/autonomy/jobs');
const { callCoordinator } = require('../src/autonomy/llm');
const { acceptProposal, recordRejectedProposal } = require('../src/autonomy/coordinator');

function sleep(ms) { return new Promise((resolve) => setTimeout(resolve, ms)); }

function loadConfig() {
  const configPath = path.resolve(process.env.DURIIN_CONFIG || path.join(__dirname, '..', 'config.json'));
  const raw = JSON.parse(fs.readFileSync(configPath, 'utf8'));
  require('dotenv').config({ path: path.resolve(path.dirname(configPath), '.env') });
  raw.openRouter = { ...(raw.openRouter || {}) };
  if (process.env.OPEN_ROUTER_API_KEY) raw.openRouter.apiKey = process.env.OPEN_ROUTER_API_KEY;
  if (process.env.OPEN_ROUTER_LLM_MODEL) raw.openRouter.llmModel = process.env.OPEN_ROUTER_LLM_MODEL;
  return raw;
}

function articleTimeColumns(archiveDb) {
  const columns = new Set(archiveDb.prepare('PRAGMA table_info(articles)').all().map((row) => row.name));
  const candidates = ['pub_date_effective', 'pub_date', 'ingested_at'].filter((name) => columns.has(name));
  if (!candidates.length) throw new Error('archive articles need publication or ingestion timestamps for replay');
  return { columns, effective: `COALESCE(${candidates.join(', ')})` };
}

function replayPrompt(article) {
  return `Historical evidence cutoff: ${article.effective_at}\n\n` +
    `[Evidence 1] article_id=${article.id}\nTitle: ${article.title || ''}\n${String(article.content || article.description || '').slice(0, 6000)}\n\n` +
    `Return JSON only in this shape:\n${JSON.stringify({ predictions: [{
      instrument: 'NVDA', direction: 'positive|negative', event_type: 'stable_enum',
      causal_channel: 'short description', horizon_days: 10, evidence_article_ids: [123],
      invalidation_condition: 'condition',
    }] }, null, 2)}\n\n` +
    'Use only this dated evidence. Return an empty predictions array when there is no clear, tradable hypothesis. Never include probabilities, returns, confidence, position sizes, or actions.';
}

function activeRun(db, config) {
  let run = db.prepare("SELECT * FROM autonomy_replay_runs WHERE status = 'running' ORDER BY id DESC LIMIT 1").get();
  if (run) return run;
  const watermarkDays = Math.max(1, Number(process.env.AUTONOMY_REPLAY_WATERMARK_DAYS) || 7);
  const result = db.prepare(`
    INSERT INTO autonomy_replay_runs (watermark_at, strategy_version, prompt_version, coordinator_model)
    VALUES (datetime('now', ?), 'autonomy-1', 'replay-coordinator-1', ?)
  `).run(`-${watermarkDays} days`, config.openRouter.llmModel || 'unknown');
  return db.prepare('SELECT * FROM autonomy_replay_runs WHERE id = ?').get(result.lastInsertRowid);
}

function scheduleNext(db, archiveDb, run) {
  const { columns, effective } = articleTimeColumns(archiveDb);
  const content = columns.has('content') ? "content IS NOT NULL AND content != ''" : '1=1';
  const indexFilter = columns.has('is_index_page') ? 'AND (is_index_page = 0 OR is_index_page IS NULL)' : '';
  const cursorFilter = run.cursor_effective_at
    ? `AND (datetime(${effective}) > datetime(?) OR (datetime(${effective}) = datetime(?) AND id > ?))`
    : '';
  const params = [run.watermark_at];
  if (run.cursor_effective_at) params.push(run.cursor_effective_at, run.cursor_effective_at, run.cursor_article_id);
  const article = archiveDb.prepare(`
    SELECT id, title, description, content, ${effective} AS effective_at
    FROM articles
    WHERE ${content} ${indexFilter} AND datetime(${effective}) <= datetime(?) ${cursorFilter}
    ORDER BY datetime(${effective}) ASC, id ASC LIMIT 1
  `).get(...params);
  if (!article) return null;
  enqueueJob(db, {
    jobType: 'replay_article', lane: 'historical', priority: 1, entityType: 'article', entityId: article.id,
    idempotencyKey: `replay:${run.id}:article:${article.id}`,
  });
  return article;
}

async function runReplayWorker({ archivePath, intelligencePath, workerId = `replay-${os.hostname()}-${process.pid}`, pollMs = 15000 } = {}) {
  const archiveDb = new Database(archivePath, { readonly: true });
  const db = new Database(intelligencePath);
  db.pragma('journal_mode = WAL');
  db.pragma('busy_timeout = 5000');
  initAutonomySchema(db);
  const config = loadConfig();
  const dailyBudget = Math.max(1, Number(process.env.AUTONOMY_REPLAY_DAILY_BUDGET) || 100);
  while (true) {
    try {
      const completedToday = db.prepare("SELECT COUNT(*) AS count FROM autonomy_jobs WHERE job_type='replay_article' AND status='complete' AND date(completed_at) = date('now')").get().count;
      if (completedToday >= dailyBudget) { await sleep(Math.max(pollMs, 60000)); continue; }
      const run = activeRun(db, config);
      scheduleNext(db, archiveDb, run);
      const job = leaseNextJob(db, workerId, 300, ['replay_article']);
      if (!job) { await sleep(pollMs); continue; }
      try {
        const { effective } = articleTimeColumns(archiveDb);
        const article = archiveDb.prepare(`SELECT id, title, description, content, ${effective} AS effective_at FROM articles WHERE id=?`).get(job.entity_id);
        if (!article || !article.effective_at) throw new Error(`replay article ${job.entity_id} is unavailable`);
        const raw = await callCoordinator(config, replayPrompt(article));
        try {
          acceptProposal(db, archiveDb, raw, {
            informationCutoff: article.effective_at, model: config.openRouter.llmModel || 'unknown',
            promptVersion: 'replay-coordinator-1', strategyVersion: 'autonomy-1', learningEligible: false,
            origin: 'replay', replayRunId: run.id,
          });
        } catch (validationError) {
          recordRejectedProposal(db, raw, { informationCutoff: article.effective_at, model: config.openRouter.llmModel || 'unknown', promptVersion: 'replay-coordinator-1' }, validationError.message);
        }
        db.prepare(`UPDATE autonomy_replay_runs SET cursor_article_id=?, cursor_effective_at=?, processed_articles=processed_articles+1, updated_at=datetime('now') WHERE id=?`)
          .run(article.id, article.effective_at, run.id);
        completeJob(db, job.id, workerId);
      } catch (error) { failJob(db, job.id, workerId, error); }
    } catch (error) { console.error(`[${workerId}] replay error:`, error.message); }
    await sleep(pollMs);
  }
}

module.exports = { articleTimeColumns, replayPrompt, scheduleNext, runReplayWorker };