import {
  chQuery,
  db,
  getClientByIdCached,
  getProjectByIdCached,
} from '../index';

import { stripTrailingSlash } from '@openpanel/common';

const pickBestDomain = (domains: string[]): string | null => {
  // Filter out invalid domains
  const validDomains = domains.filter(
    (domain) =>
      domain &&
      !domain.includes('*') &&
      !domain.includes('localhost') &&
      !domain.includes('127.0.0.1'),
  );

  if (validDomains.length === 0) return null;

  // Score each domain
  const scoredDomains = validDomains.map((domain) => {
    let score = 0;

    // Prefer https (highest priority)
    if (domain.startsWith('https://')) score += 100;

    // Penalize domains from common providers like vercel, netlify, etc.
    if (
      domain.includes('vercel.app') ||
      domain.includes('netlify.app') ||
      domain.includes('herokuapp.com') ||
      domain.includes('github.io') ||
      domain.includes('gitlab.io') ||
      domain.includes('surge.sh') ||
      domain.includes('cloudfront.net') ||
      domain.includes('firebaseapp.com') ||
      domain.includes('azurestaticapps.net') ||
      domain.includes('pages.dev') ||
      domain.includes('ngrok-free.app') ||
      domain.includes('ngrok.app')
    ) {
      score -= 50;
    }

    // Penalize subdomains
    const domainParts = domain
      .replace('https://', '')
      .replace('http://', '')
      .split('.');
    if (domainParts.length <= 2) score += 50;

    // Tiebreaker: prefer shorter domains
    score -= domain.length;

    return { domain, score };
  });

  // Sort by score (highest first) and return the best domain
  const bestDomain = scoredDomains.sort((a, b) => b.score - a.score)[0];
  return bestDomain?.domain || null;
};

export const up = async () => {
  const projects = await db.project.findMany({
    include: {
      clients: true,
    },
  });

  const matches = [];
  for (const project of projects) {
    if (project.cors.length > 0 || project.domain) {
      continue;
    }

    if (project.clients.length === 0) {
      continue;
    }

    const cors = [];
    let crossDomain = false;
    for (const client of project.clients) {
      if (client.crossDomain) {
        crossDomain = true;
      }
      cors.push(
        ...(client.cors?.split(',') ?? []).map((c) =>
          stripTrailingSlash(c.trim()),
        ),
      );
      await getClientByIdCached.clear(client.id);
    }

    let domain = pickBestDomain(cors);

    if (!domain) {
      const res = await chQuery<{ origin: string }>(
        `SELECT origin FROM events_distributed WHERE project_id = '${project.id}' and origin != ''`,
      );
      if (res.length) {
        domain = pickBestDomain(res.map((r) => r.origin));
        matches.push(domain);
      }
    }

    await db.project.update({
      where: { id: project.id },
      data: {
        cors,
        crossDomain,
        domain,
      },
    });

    await getProjectByIdCached.clear(project.id);
  }
};