mc-pdf-studio 1.0.0

package/src/server.js

@@ -0,0 +1,250 @@
+const express = require('express');
+const multer = require('multer');
+const cors = require('cors');
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const http = require('http');
+
+let pdfParse;
+try { pdfParse = require('pdf-parse'); } catch(e) {}
+
+const app = express();
+app.use(cors());
+app.use(express.json());
+app.use(express.static(path.join(__dirname, '../public')));
+
+const upload = multer({
+  storage: multer.diskStorage({
+    destination: os.tmpdir(),
+    filename: (req, file, cb) => cb(null, `pdf-viewer-${Date.now()}-${file.originalname}`),
+  }),
+  fileFilter: (req, file, cb) => {
+    cb(null, file.mimetype === 'application/pdf' || file.originalname.endsWith('.pdf'));
+  },
+});
+
+let CONFIG = {};
+let loadedPDFs = {};  // cache: filepath -> { text, pageCount, metadata }
+
+async function extractPDFData(filePath) {
+  if (loadedPDFs[filePath]) return loadedPDFs[filePath];
+  const buf = fs.readFileSync(filePath);
+  const data = await pdfParse(buf);
+  const result = {
+    text: data.text,
+    pageCount: data.numpages,
+    metadata: data.metadata,
+    info: data.info,
+    filePath,
+    fileName: path.basename(filePath),
+  };
+  loadedPDFs[filePath] = result;
+  return result;
+}
+
+// ─── Routes ────────────────────────────────────────────────────────────────
+
+app.get('/api/config', (req, res) => {
+  res.json({
+    enableAi: CONFIG.enableAi || false,
+    aiProvider: CONFIG.aiProvider || 'anthropic',
+    aiModel: CONFIG.aiModel || null,
+    aiBaseUrl: CONFIG.aiBaseUrl || null,
+    minimap: CONFIG.minimap !== false,
+    theme: CONFIG.theme || 'dark',
+    initialFile: CONFIG.initialFile || null,
+    initialFileName: CONFIG.initialFile ? path.basename(CONFIG.initialFile) : null,
+  });
+});
+
+// Update AI provider settings at runtime (from Settings UI)
+app.post('/api/ai/settings', (req, res) => {
+  const { provider, model, baseUrl, enabled } = req.body;
+  if (provider !== undefined) CONFIG.aiProvider = provider;
+  if (model !== undefined) CONFIG.aiModel = model || null;
+  if (baseUrl !== undefined) CONFIG.aiBaseUrl = baseUrl || null;
+  if (enabled !== undefined) CONFIG.enableAi = !!enabled;
+  res.json({ ok: true });
+});
+
+// Serve the initial PDF file if provided at startup
+app.get('/api/initial-pdf', (req, res) => {
+  if (!CONFIG.initialFile || !fs.existsSync(CONFIG.initialFile)) {
+    return res.status(404).json({ error: 'No initial file' });
+  }
+  res.setHeader('Content-Type', 'application/pdf');
+  res.setHeader('Content-Disposition', `inline; filename="${path.basename(CONFIG.initialFile)}"`);
+  fs.createReadStream(CONFIG.initialFile).pipe(res);
+});
+
+// Upload a new PDF
+app.post('/api/upload', upload.single('pdf'), async (req, res) => {
+  try {
+    if (!req.file) return res.status(400).json({ error: 'No file uploaded' });
+    res.json({ filePath: req.file.path, fileName: req.file.originalname });
+  } catch (err) {
+    res.status(500).json({ error: err.message });
+  }
+});
+
+// Serve an uploaded PDF by its temp path
+app.get('/api/pdf', (req, res) => {
+  const { path: filePath } = req.query;
+  if (!filePath || !fs.existsSync(filePath)) {
+    return res.status(404).json({ error: 'File not found' });
+  }
+  res.setHeader('Content-Type', 'application/pdf');
+  res.setHeader('Content-Disposition', `inline; filename="${path.basename(filePath)}"`);
+  fs.createReadStream(filePath).pipe(res);
+});
+
+// Get PDF text & metadata (for AI context)
+app.get('/api/pdf-info', async (req, res) => {
+  try {
+    const filePath = req.query.path || CONFIG.initialFile;
+    if (!filePath || !fs.existsSync(filePath)) return res.status(404).json({ error: 'File not found' });
+    const data = await extractPDFData(filePath);
+    res.json({ pageCount: data.pageCount, metadata: data.metadata, info: data.info, fileName: data.fileName });
+  } catch (err) {
+    res.status(500).json({ error: err.message });
+  }
+});
+
+// ─── AI Chat ───────────────────────────────────────────────────────────────
+
+const PROVIDER_DEFAULTS = {
+  anthropic: { model: 'claude-sonnet-4-20250514', baseUrl: 'https://api.anthropic.com' },
+  openai:    { model: 'gpt-4o',                   baseUrl: 'https://api.openai.com' },
+  xai:       { model: 'grok-beta',                baseUrl: 'https://api.x.ai' },
+  ollama:    { model: 'llama3.2:latest',           baseUrl: 'http://localhost:11434' },
+};
+
+async function callAnthropic({ baseUrl, model, apiKey, systemPrompt, messages }) {
+  const response = await fetch(`${baseUrl}/v1/messages`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'x-api-key': apiKey,
+      'anthropic-version': '2023-06-01',
+    },
+    body: JSON.stringify({ model, max_tokens: 1024, system: systemPrompt, messages }),
+  });
+  if (!response.ok) {
+    const err = await response.json().catch(() => ({}));
+    throw new Error(err.error?.message || `HTTP ${response.status}`);
+  }
+  const data = await response.json();
+  return data.content?.[0]?.text || '';
+}
+
+async function callOpenAICompat({ baseUrl, model, apiKey, systemPrompt, messages }) {
+  const response = await fetch(`${baseUrl}/v1/chat/completions`, {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      ...(apiKey ? { 'Authorization': `Bearer ${apiKey}` } : {}),
+    },
+    body: JSON.stringify({
+      model,
+      messages: [{ role: 'system', content: systemPrompt }, ...messages],
+    }),
+  });
+  if (!response.ok) {
+    const err = await response.json().catch(() => ({}));
+    throw new Error(err.error?.message || `HTTP ${response.status}`);
+  }
+  const data = await response.json();
+  return data.choices?.[0]?.message?.content || '';
+}
+
+app.post('/api/ai/chat', async (req, res) => {
+  if (!CONFIG.enableAi) return res.status(403).json({ error: 'AI not enabled. Start with --enable-ai flag.' });
+
+  const { message, history = [], filePath, currentPage } = req.body;
+  if (!message) return res.status(400).json({ error: 'No message provided' });
+
+  const provider = CONFIG.aiProvider || 'anthropic';
+  const defaults = PROVIDER_DEFAULTS[provider] || PROVIDER_DEFAULTS.ollama;
+  const model    = CONFIG.aiModel   || defaults.model;
+  const baseUrl  = (CONFIG.aiBaseUrl || defaults.baseUrl).replace(/\/$/, '');
+
+  try {
+    let pdfContext = '';
+    const fp = filePath || CONFIG.initialFile;
+    if (fp && fs.existsSync(fp)) {
+      const data = await extractPDFData(fp);
+      const truncated = data.text.slice(0, 8000);
+      pdfContext = `\n\n<pdf_content filename="${data.fileName}" total_pages="${data.pageCount}" current_page="${currentPage || 1}">\n${truncated}${data.text.length > 8000 ? '\n... [content truncated]' : ''}\n</pdf_content>`;
+    }
+
+    const systemPrompt = `You are an AI Sidekick embedded in a PDF viewer. You help users understand, summarize, and ask questions about the PDF they are currently reading.
+
+Be concise and helpful. Format your responses in markdown when appropriate — use bullet points, bold text, and code blocks where they add clarity.
+
+When the user asks about specific pages or content, refer to the PDF context provided.${pdfContext}`;
+
+    const messages = [
+      ...history.map(h => ({ role: h.role, content: h.content })),
+      { role: 'user', content: message },
+    ];
+
+    let reply = '';
+    if (provider === 'anthropic') {
+      const apiKey = process.env.ANTHROPIC_API_KEY;
+      if (!apiKey) throw new Error('ANTHROPIC_API_KEY env var not set');
+      reply = await callAnthropic({ baseUrl, model, apiKey, systemPrompt, messages });
+    } else if (provider === 'openai') {
+      const apiKey = process.env.OPENAI_API_KEY;
+      if (!apiKey) throw new Error('OPENAI_API_KEY env var not set');
+      reply = await callOpenAICompat({ baseUrl, model, apiKey, systemPrompt, messages });
+    } else if (provider === 'xai') {
+      const apiKey = process.env.XAI_API_KEY;
+      if (!apiKey) throw new Error('XAI_API_KEY env var not set');
+      reply = await callOpenAICompat({ baseUrl, model, apiKey, systemPrompt, messages });
+    } else {
+      // ollama and any OpenAI-compatible local provider
+      reply = await callOpenAICompat({ baseUrl, model, apiKey: null, systemPrompt, messages });
+    }
+
+    res.json({ reply });
+  } catch (err) {
+    res.status(500).json({ error: err.message });
+  }
+});
+
+// ─── Server Start ──────────────────────────────────────────────────────────
+
+function start(config) {
+  CONFIG = config;
+
+  const server = http.createServer(app);
+  server.listen(config.port, () => {
+    const url = `http://localhost:${config.port}`;
+    console.log('');
+    console.log('  ╔═══════════════════════════════════════╗');
+    console.log('  ║       ✦ MC PDF Studio Ready ✦          ║');
+    console.log('  ╚═══════════════════════════════════════╝');
+    console.log('');
+    console.log(`  → Local:   ${url}`);
+    if (config.initialFile) console.log(`  → File:    ${path.basename(config.initialFile)}`);
+    if (config.enableAi) {
+      const prov = config.aiProvider || 'anthropic';
+      const mod  = config.aiModel || (PROVIDER_DEFAULTS[prov] || {}).model || '';
+      console.log(`  → AI:      Sidekick enabled ✓ (${prov} / ${mod})`);
+    }
+    console.log(`  → Theme:   ${config.theme}`);
+    console.log('');
+    console.log('  Press Ctrl+C to stop');
+    console.log('');
+
+    // Auto-open browser
+    const open = (url) => {
+      const cmd = process.platform === 'darwin' ? 'open' : process.platform === 'win32' ? 'start' : 'xdg-open';
+      require('child_process').exec(`${cmd} ${url}`);
+    };
+    try { open(url); } catch(e) {}
+  });
+}
+
+module.exports = { start };