MangaCircus/scraper/daemon.js

#!/usr/bin/env node
'use strict';

const http = require('http');
const path = require('path');
const { ScraperService } = require('./service');
const { secret } = require('../runtime/config');

const HOST = process.env.SCRAPER_HOST || '127.0.0.1';
const PORT = Number(process.env.SCRAPER_PORT || 8091);
const TOKEN = secret('SCRAPER_TOKEN');
const LIBRARY = path.resolve(process.env.LIBRARY || 'library');
const DATA = path.resolve(process.env.SCRAPER_DATA || path.join(process.env.DATA || 'data', 'scraper'));

if (!['127.0.0.1', 'localhost', '::1'].includes(HOST) && !TOKEN) {
  throw new Error('SCRAPER_TOKEN is required when SCRAPER_HOST is not loopback');
}

function json(res, code, value) {
  res.writeHead(code, { 'content-type': 'application/json; charset=utf-8', 'cache-control': 'no-store' });
  res.end(JSON.stringify(value));
}

async function body(req) {
  let text = '';
  for await (const chunk of req) { text += chunk; if (text.length > 1000000) throw Object.assign(new Error('Request too large'), { code: 413 }); }
  try { return text ? JSON.parse(text) : {}; } catch { throw Object.assign(new Error('Invalid JSON'), { code: 400 }); }
}

(async () => {
  const service = await new ScraperService({ libraryRoot: LIBRARY, dataRoot: DATA }).init();
  const server = http.createServer(async (req, res) => {
    try {
      const url = new URL(req.url, 'http://localhost');
      if (url.pathname === '/health') return json(res, 200, { ok: true, status: service.state.status });
      if (TOKEN && req.headers.authorization !== `Bearer ${TOKEN}`) return json(res, 401, { error: 'Unauthorized' });
      if (req.method === 'GET' && url.pathname === '/api/state') return json(res, 200, await service.publicState());
      if (req.method === 'GET' && url.pathname === '/api/revision') return json(res, 200, service.revision());
      if (req.method !== 'POST') return json(res, 404, { error: 'Not found' });
      const input = await body(req);
      if (url.pathname === '/api/queue') return json(res, 200, { item: await service.enqueue(input), state: await service.publicState() });
      if (url.pathname === '/api/queue/remove') { await service.remove(String(input.id || '')); return json(res, 200, await service.publicState()); }
      if (url.pathname === '/api/queue/clear-completed') { const result = await service.clearCompleted(String(input.source || '')); return json(res, 200, { ...result, state: await service.publicState() }); }
      if (url.pathname === '/api/queue/reorder') { await service.reorder(input.ids); return json(res, 200, await service.publicState()); }
      if (url.pathname === '/api/control') { await service.control(String(input.action || ''), input.id); return json(res, 200, await service.publicState()); }
      if (url.pathname === '/api/settings') { await service.updateSettings(input); return json(res, 200, await service.publicState()); }
      if (url.pathname === '/api/discovery/run') { await service.enqueueDiscoveryChecks('manual'); return json(res, 200, await service.publicState()); }
      if (url.pathname === '/api/library/auto-update') { await service.setMangaAutoUpdate(String(input.id || ''), !!input.enabled); return json(res, 200, await service.publicState()); }
      if (url.pathname === '/api/legacy/publish-ready') return json(res, 200, { result: await service.publishLegacyReady(), state: await service.publicState() });
      return json(res, 404, { error: 'Not found' });
    } catch (error) {
      console.error('[scraper-daemon]', error.message);
      json(res, error.code || (/not found/i.test(error.message) ? 404 : 400), { error: error.message });
    }
  });
  server.listen(PORT, HOST, () => console.log(`MangaCircus scraper on http://${HOST}:${PORT} (library: ${LIBRARY}, data: ${DATA})`));

  let closing = false;
  const close = async () => {
    if (closing) return; closing = true;
    server.close(); await service.close(); process.exit(0);
  };
  process.on('SIGINT', close);
  process.on('SIGTERM', close);
})().catch(error => { console.error(`[scraper-daemon] ${error.message}`); process.exitCode = 1; });