import app from './server';
import { serve } from '@hono/node-server';
import { FirehoseWorker } from './lib/firehose';
import { logger } from './lib/observability';
import { mkdirSync, existsSync } from 'fs';
import { backfillCache } from './lib/backfill';

const PORT = process.env.PORT ? parseInt(process.env.PORT) : 3001;
const CACHE_DIR = process.env.CACHE_DIR || './cache/sites';

// Parse CLI arguments
const args = process.argv.slice(2);
const hasBackfillFlag = args.includes('--backfill');
const backfillOnStartup = hasBackfillFlag || process.env.BACKFILL_ON_STARTUP === 'true';

// Ensure cache directory exists
if (!existsSync(CACHE_DIR)) {
  mkdirSync(CACHE_DIR, { recursive: true });
  console.log('Created cache directory:', CACHE_DIR);
}

// Start firehose worker with observability logger
const firehose = new FirehoseWorker((msg, data) => {
  logger.info(msg, data);
});

firehose.start();

// Run backfill if requested
if (backfillOnStartup) {
  console.log('🔄 Backfill requested, starting cache backfill...');
  backfillCache({
    skipExisting: true,
    concurrency: 3,
  }).then((stats) => {
    console.log('✅ Cache backfill completed');
  }).catch((err) => {
    console.error('❌ Cache backfill error:', err);
  });
}

// Add health check endpoint
app.get('/health', (c) => {
  const firehoseHealth = firehose.getHealth();
  return c.json({
    status: 'ok',
    firehose: firehoseHealth,
  });
});

// Start HTTP server with Node.js adapter
const server = serve({
  fetch: app.fetch,
  port: PORT,
});

console.log(`
Wisp Hosting Service

Server:       http://localhost:${PORT}
Health:       http://localhost:${PORT}/health
Cache:        ${CACHE_DIR}
Firehose:     Connected to Firehose
`);

// Graceful shutdown
process.on('SIGINT', async () => {
  console.log('\n🛑 Shutting down...');
  firehose.stop();
  server.close();
  process.exit(0);
});

process.on('SIGTERM', async () => {
  console.log('\n🛑 Shutting down...');
  firehose.stop();
  server.close();
  process.exit(0);
});