SubMiner/src/main/character-dictionary-runtime/manual-selection.ts

import * as fs from 'fs';
import * as path from 'path';
import type { AnilistMediaGuess } from '../../core/services/anilist/anilist-updater';
import { ensureDir } from '../../shared/fs-utils';

export type CharacterDictionaryManualSelection = {
  seriesKey: string;
  mediaId: number;
  mediaTitle: string;
  staleMediaIds: number[];
  /**
   * Season this override was saved for. Recorded explicitly because inferring it from the
   * key text is ambiguous for titles that themselves end in a season-like token.
   */
  season?: number | null;
};

type ManualSelectionStoreFile = {
  overrides?: CharacterDictionaryManualSelection[];
};

function normalizeManualMediaId(value: unknown): number | null {
  if (typeof value !== 'number' || !Number.isFinite(value)) return null;
  const mediaId = Math.floor(value);
  return mediaId > 0 ? mediaId : null;
}

function normalizeSeason(value: unknown): number | null {
  if (typeof value !== 'number' || !Number.isInteger(value) || value <= 0) return null;
  return value;
}

function normalizeSeriesKeyPart(value: string): string {
  return value
    .normalize('NFKD')
    .replace(/[':]/g, '')
    .replace(/&/g, ' and ')
    .replace(/[^a-zA-Z0-9]+/g, '-')
    .replace(/^-+|-+$/g, '')
    .replace(/-{2,}/g, '-')
    .toLowerCase();
}

function getMediaDirectoryKey(mediaPath: string | null): string {
  const rawPath = mediaPath?.trim();
  if (!rawPath) return '';

  if (/^[a-zA-Z][a-zA-Z\d+.-]*:\/\//.test(rawPath) || rawPath.startsWith('file:')) {
    try {
      const url = new URL(rawPath);
      const directoryPath = path.posix.dirname(
        decodeURIComponent(url.pathname).replace(/\\/g, '/'),
      );
      const scopedPath = `${url.hostname}${directoryPath === '/' ? '' : directoryPath}`;
      return normalizeSeriesKeyPart(scopedPath);
    } catch {
      return '';
    }
  }

  const normalizedPath = rawPath.replace(/\\/g, '/');
  const directoryPath = path.posix.dirname(normalizedPath);
  if (!directoryPath || directoryPath === '.') return '';
  return normalizeSeriesKeyPart(directoryPath);
}

function dedupeNumbers(values: number[]): number[] {
  const seen = new Set<number>();
  const result: number[] = [];
  for (const value of values) {
    const normalized = normalizeManualMediaId(value);
    if (normalized === null || seen.has(normalized)) continue;
    seen.add(normalized);
    result.push(normalized);
  }
  return result;
}

function normalizeOverride(value: unknown): CharacterDictionaryManualSelection | null {
  if (!value || typeof value !== 'object') return null;
  const raw = value as Partial<CharacterDictionaryManualSelection>;
  const seriesKey = typeof raw.seriesKey === 'string' ? raw.seriesKey.trim() : '';
  const mediaId = normalizeManualMediaId(raw.mediaId);
  const mediaTitle = typeof raw.mediaTitle === 'string' ? raw.mediaTitle.trim() : '';
  if (!seriesKey || mediaId === null || !mediaTitle) return null;
  const season = normalizeSeason(raw.season);
  return {
    seriesKey,
    mediaId,
    mediaTitle,
    staleMediaIds: dedupeNumbers(Array.isArray(raw.staleMediaIds) ? raw.staleMediaIds : []),
    ...(season === null ? {} : { season }),
  };
}

function readOverrides(filePath: string): CharacterDictionaryManualSelection[] {
  try {
    const parsed = JSON.parse(fs.readFileSync(filePath, 'utf8')) as ManualSelectionStoreFile;
    if (!Array.isArray(parsed.overrides)) return [];
    const byKey = new Map<string, CharacterDictionaryManualSelection>();
    for (const value of parsed.overrides) {
      const normalized = normalizeOverride(value);
      if (normalized) byKey.set(normalized.seriesKey, normalized);
    }
    return [...byKey.values()];
  } catch {
    return [];
  }
}

function writeOverrides(filePath: string, overrides: CharacterDictionaryManualSelection[]): void {
  ensureDir(path.dirname(filePath));
  fs.writeFileSync(filePath, JSON.stringify({ overrides }, null, 2), 'utf8');
}

function getLegacySeriesKeyCandidates(seriesKey: string): string[] {
  const scopedSeparatorIndex = seriesKey.indexOf('--');
  if (scopedSeparatorIndex < 0) return [seriesKey];
  return [seriesKey, seriesKey.slice(scopedSeparatorIndex + 2)];
}

function getDirectoryScope(seriesKey: string): string | null {
  const scopedSeparatorIndex = seriesKey.indexOf('--');
  return scopedSeparatorIndex < 0 ? null : seriesKey.slice(0, scopedSeparatorIndex);
}

/**
 * Fallback season for override records saved before the season was stored explicitly:
 * reads the "-s<N>" segment emitted by buildCharacterDictionarySeriesKey, which sits just
 * before the optional trailing year. Season 1 keys carry no segment and parse as 1.
 */
function getSeasonScopeFromKey(seriesKey: string): number {
  const match = /-s(\d{1,2})(?:-\d{4})?$/.exec(seriesKey);
  if (!match) return 1;
  const season = Number.parseInt(match[1]!, 10);
  return Number.isInteger(season) && season > 0 ? season : 1;
}

/**
 * The subset of a parsed guess the series key is built from. Kept structural so callers
 * holding a narrower guess (the AniList post-watch runtime) can build the same key.
 */
export type CharacterDictionarySeriesKeyGuess = Pick<AnilistMediaGuess, 'title' | 'season'> &
  Partial<Omit<AnilistMediaGuess, 'title' | 'season'>>;

export function buildCharacterDictionarySeriesKey(input: {
  mediaPath: string | null;
  mediaTitle: string | null;
  guess: CharacterDictionarySeriesKeyGuess | null;
}): string {
  const guessedTitle = input.guess?.title.trim() || input.guess?.alternativeTitle?.trim() || '';
  const sourceTitle =
    guessedTitle ||
    (input.mediaTitle && input.mediaTitle.trim()) ||
    (input.mediaPath && path.basename(input.mediaPath).replace(/\.[^.]+$/, '')) ||
    'unknown';
  const withoutEpisode = sourceTitle
    .replace(/\bS\d{1,2}E\d{1,3}\b/gi, ' ')
    .replace(/\bepisode\s+\d+\b/gi, ' ')
    .trim();
  const base = normalizeSeriesKeyPart(withoutEpisode) || 'unknown';
  const directoryKey = getMediaDirectoryKey(input.mediaPath);
  const scopedBase = directoryKey ? `${directoryKey}--${base}` : base;
  // Season 1 stays unsuffixed so existing keys, snapshots and overrides keep matching.
  const season = input.guess?.season;
  const seasonSegment =
    typeof season === 'number' && Number.isInteger(season) && season > 1 ? `-s${season}` : '';
  const withSeason = `${scopedBase}${seasonSegment}`;
  return input.guess?.year ? `${withSeason}-${input.guess.year}` : withSeason;
}

/** Season an override applies to: the recorded value, else parsed from its key. */
function getOverrideSeason(entry: CharacterDictionaryManualSelection): number {
  return normalizeSeason(entry.season) ?? getSeasonScopeFromKey(entry.seriesKey);
}

export function createCharacterDictionaryManualSelectionStore(deps: { userDataPath: string }) {
  const filePath = path.join(deps.userDataPath, 'character-dictionaries', 'anilist-overrides.json');

  return {
    getOverride: async (
      seriesKey: string,
      season?: number | null,
    ): Promise<CharacterDictionaryManualSelection | null> => {
      const candidates = getLegacySeriesKeyCandidates(seriesKey);
      const overrides = readOverrides(filePath);
      const exactMatch = overrides.find((entry) => entry.seriesKey === candidates[0]);
      if (exactMatch) return exactMatch;
      const directoryScope = getDirectoryScope(seriesKey);
      if (directoryScope) {
        // Same folder is only evidence of the same show when it is also the same season;
        // a flat multi-season folder would otherwise spread one override across all of them.
        const seasonScope = normalizeSeason(season) ?? getSeasonScopeFromKey(seriesKey);
        const scopedMatches = overrides.filter(
          (entry) =>
            getDirectoryScope(entry.seriesKey) === directoryScope &&
            getOverrideSeason(entry) === seasonScope,
        );
        const selectedMediaIds = new Set(scopedMatches.map((entry) => entry.mediaId));
        if (scopedMatches.length > 0 && selectedMediaIds.size === 1) {
          const latest = scopedMatches.at(-1)!;
          return {
            ...latest,
            staleMediaIds: dedupeNumbers(scopedMatches.flatMap((entry) => entry.staleMediaIds)),
          };
        }
      }
      for (const candidate of candidates.slice(1)) {
        const match = overrides.find((entry) => entry.seriesKey === candidate);
        if (match) return match;
      }
      return null;
    },
    setOverride: async (selection: CharacterDictionaryManualSelection): Promise<void> => {
      const normalized = normalizeOverride(selection);
      if (!normalized) {
        throw new Error('Invalid character dictionary manual selection.');
      }
      const directoryScope = getDirectoryScope(normalized.seriesKey);
      const seasonScope = getOverrideSeason(normalized);
      const remaining = readOverrides(filePath).filter((entry) =>
        directoryScope
          ? getDirectoryScope(entry.seriesKey) !== directoryScope ||
            getOverrideSeason(entry) !== seasonScope
          : entry.seriesKey !== normalized.seriesKey,
      );
      writeOverrides(filePath, [...remaining, normalized]);
    },
  };
}