import type { ClusteredEvent, VelocityMetrics, VelocityLevel, SentimentType } from '@/types';
import { mlWorker } from './ml-worker';

const HOUR_MS = 60 * 60 * 1000;
const ELEVATED_THRESHOLD = 3;
const SPIKE_THRESHOLD = 6;

const NEGATIVE_WORDS = new Set([
  'war', 'attack', 'killed', 'death', 'dead', 'crisis', 'crash', 'collapse',
  'threat', 'danger', 'escalate', 'escalation', 'conflict', 'strike', 'bomb',
  'explosion', 'casualties', 'disaster', 'emergency', 'catastrophe', 'fail',
  'failure', 'reject', 'rejected', 'sanctions', 'invasion', 'missile', 'nuclear',
  'terror', 'terrorist', 'hostage', 'assassination', 'coup', 'protest', 'riot',
  'warns', 'warning', 'fears', 'concern', 'worried', 'plunge', 'plummet', 'surge',
  'flee', 'evacuate', 'shutdown', 'layoff', 'layoffs', 'cuts', 'slump', 'recession',
]);

const POSITIVE_WORDS = new Set([
  'peace', 'deal', 'agreement', 'breakthrough', 'success', 'win', 'gains',
  'recovery', 'growth', 'rise', 'surge', 'boost', 'rally', 'soar', 'jump',
  'ceasefire', 'treaty', 'alliance', 'partnership', 'cooperation', 'progress',
  'release', 'released', 'freed', 'rescue', 'saved', 'approved', 'passes',
  'record', 'milestone', 'historic', 'landmark', 'celebrates', 'victory',
]);

function analyzeSentiment(text: string): { sentiment: SentimentType; score: number } {
  const words = text.toLowerCase().split(/\W+/);
  let score = 0;

  for (const word of words) {
    if (NEGATIVE_WORDS.has(word)) score -= 1;
    if (POSITIVE_WORDS.has(word)) score += 1;
  }

  const sentiment: SentimentType = score < -1 ? 'negative' : score > 1 ? 'positive' : 'neutral';
  return { sentiment, score };
}

function calculateVelocityLevel(sourcesPerHour: number): VelocityLevel {
  if (sourcesPerHour >= SPIKE_THRESHOLD) return 'spike';
  if (sourcesPerHour >= ELEVATED_THRESHOLD) return 'elevated';
  return 'normal';
}

export function calculateVelocity(cluster: ClusteredEvent): VelocityMetrics {
  const items = cluster.allItems;

  if (items.length <= 1) {
    const { sentiment, score } = analyzeSentiment(cluster.primaryTitle);
    return { sourcesPerHour: 0, level: 'normal', trend: 'stable', sentiment, sentimentScore: score };
  }

  const timeSpanMs = cluster.lastUpdated.getTime() - cluster.firstSeen.getTime();
  const timeSpanHours = Math.max(timeSpanMs / HOUR_MS, 0.25);
  const sourcesPerHour = items.length / timeSpanHours;

  const midpoint = cluster.firstSeen.getTime() + timeSpanMs / 2;
  const recentItems = items.filter(i => i.pubDate.getTime() > midpoint);
  const olderItems = items.filter(i => i.pubDate.getTime() <= midpoint);

  let trend: 'rising' | 'stable' | 'falling' = 'stable';
  if (recentItems.length > olderItems.length * 1.5) {
    trend = 'rising';
  } else if (olderItems.length > recentItems.length * 1.5) {
    trend = 'falling';
  }

  const allText = items.map(i => i.title).join(' ');
  const { sentiment, score } = analyzeSentiment(allText);

  return {
    sourcesPerHour: Math.round(sourcesPerHour * 10) / 10,
    level: calculateVelocityLevel(sourcesPerHour),
    trend,
    sentiment,
    sentimentScore: score,
  };
}

export function enrichWithVelocity(clusters: ClusteredEvent[]): ClusteredEvent[] {
  return clusters.map(cluster => ({
    ...cluster,
    velocity: calculateVelocity(cluster),
  }));
}

export async function calculateVelocityWithML(cluster: ClusteredEvent): Promise<VelocityMetrics> {
  const baseMetrics = calculateVelocity(cluster);

  if (!mlWorker.isAvailable) return baseMetrics;

  try {
    const results = await mlWorker.classifySentiment([cluster.primaryTitle]);
    const sentiment = results[0];
    if (!sentiment) return baseMetrics;

    const mlSentiment: SentimentType = sentiment.label === 'positive' ? 'positive' :
      sentiment.label === 'negative' ? 'negative' : 'neutral';
    const mlScore = sentiment.label === 'negative' ? -sentiment.score : sentiment.score;

    return {
      ...baseMetrics,
      sentiment: mlSentiment,
      sentimentScore: mlScore,
    };
  } catch {
    return baseMetrics;
  }
}

export async function enrichWithVelocityML(clusters: ClusteredEvent[]): Promise<ClusteredEvent[]> {
  if (!mlWorker.isAvailable) {
    return enrichWithVelocity(clusters);
  }

  try {
    const titles = clusters.map(c => c.primaryTitle);
    const sentiments = await mlWorker.classifySentiment(titles);

    return clusters.map((cluster, i) => {
      const baseMetrics = calculateVelocity(cluster);
      const sentiment = sentiments[i];
      if (!sentiment) {
        return { ...cluster, velocity: baseMetrics };
      }

      const mlSentiment: SentimentType = sentiment.label === 'positive' ? 'positive' :
        sentiment.label === 'negative' ? 'negative' : 'neutral';

      return {
        ...cluster,
        velocity: {
          ...baseMetrics,
          sentiment: mlSentiment,
          sentimentScore: sentiment.label === 'negative' ? -sentiment.score : sentiment.score,
        },
      };
    });
  } catch {
    return enrichWithVelocity(clusters);
  }
}
