'use strict';

import type { Request, Response, NextFunction } from 'express';
import * as sourceCorpusStore from '../stores/sourceCorpusStore.js';
import type { SourceCorpus } from '../stores/sourceCorpusStore.js';

export interface SourceCorpusRequest extends Request {
  sourceCorpusId?: string;
  sourceCorpus?: SourceCorpus;
}

export function requireSourceCorpusId(req: SourceCorpusRequest, res: Response, next: NextFunction): void {
  const sourceCorpusId = (req.body?.sourceCorpusId ?? req.query?.sourceCorpusId) as string | undefined;
  if (!sourceCorpusId) {
    res.status(400).json({ error: 'sourceCorpusId is required' });
    return;
  }
  req.sourceCorpusId = sourceCorpusId;
  next();
}

export function requireSourceCorpus(req: SourceCorpusRequest, res: Response, next: NextFunction): void {
  const sourceCorpus = sourceCorpusStore.getSourceCorpus(req.sourceCorpusId!);
  if (!sourceCorpus) {
    res.status(404).json({ error: 'source corpus not found' });
    return;
  }
  req.sourceCorpus = sourceCorpus;
  next();
}

export function requireFile(req: Request, res: Response, next: NextFunction): void {
  if (!req.file) {
    res.status(400).json({ error: 'file is required' });
    return;
  }
  next();
}

export function requireUrl(req: Request, res: Response, next: NextFunction): void {
  if (!req.body?.url) {
    res.status(400).json({ error: 'url is required' });
    return;
  }
  next();
}