/**
 * NovaSonic Backend Server
 * Downloads music via yt-dlp / squid.wtf (Qobuz) and saves into your Navidrome library.
 *
 * Requirements:
 *   - Node.js 18+
 *   - yt-dlp installed (https://github.com/yt-dlp/yt-dlp)
 *   - ffmpeg installed
 */

const express   = require('express');
const { execSync } = require('child_process');
const path      = require('path');
const fs        = require('fs');
const https     = require('https');

const CONFIG    = require('./lib/config');
const { jobs, updateJob, log } = require('./lib/jobs');
const { lookupItunesTrackMeta, fetchItunesTracks, searchItunesAlbumTracks } = require('./lib/metadata');
const { downloadTrack, downloadAlbumTracks } = require('./lib/downloader');

// ─── INIT ─────────────────────────────────────────────────────────────────────
const app = express();
app.use((req, res, next) => {
  res.setHeader('Access-Control-Allow-Origin', '*');
  res.setHeader('Access-Control-Allow-Methods', 'GET,POST,PATCH,DELETE,OPTIONS');
  res.setHeader('Access-Control-Allow-Headers', 'Content-Type,Authorization,X-ABS-URL,X-ABS-Token');
  res.setHeader('Access-Control-Allow-Private-Network', 'true');
  if (req.method === 'OPTIONS') { res.sendStatus(204); return; }
  next();
});
app.use(express.json());

let ytdlpVersion = 'unknown';
try {
  ytdlpVersion = execSync(`${CONFIG.ytdlpBin} --version`, { timeout: 5000 }).toString().trim();
  console.log(`✅ yt-dlp ${ytdlpVersion} found`);
} catch {
  console.warn('⚠️  yt-dlp not found!');
}

if (!fs.existsSync(CONFIG.musicPath)) {
  console.warn(`⚠️  Music path "${CONFIG.musicPath}" does not exist — creating it`);
  fs.mkdirSync(CONFIG.musicPath, { recursive: true });
}

// ─── QUEUE ────────────────────────────────────────────────────────────────────
let activeDownloads = 0;
const downloadQueue = [];

function processQueue() {
  if (activeDownloads >= CONFIG.maxParallel || downloadQueue.length === 0) return;
  const task = downloadQueue.shift();
  activeDownloads++;
  task().finally(() => { activeDownloads--; processQueue(); });
}

function enqueue(task) {
  downloadQueue.push(task);
  processQueue();
}

// ─── SPOTIFY ──────────────────────────────────────────────────────────────────
function parseSpotifyUrl(url) {
  const m = url.match(/open\.spotify\.com\/(?:[a-z]{2,8}-[a-z]{2,8}\/)?(?:intl-[a-z-]+\/)?(track|album|playlist)\/([A-Za-z0-9]+)/);
  if (!m) throw new Error('Ungültige Spotify-URL — erwartet: open.spotify.com/track|album|playlist/...');
  return { type: m[1], id: m[2] };
}

async function fetchSpotifyEmbed(type, id) {
  const r = await fetch(`https://open.spotify.com/embed/${type}/${id}`, {
    headers: {
      'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36',
      Accept: 'text/html,application/xhtml+xml',
      'Accept-Language': 'en-US,en;q=0.9',
    },
    signal: AbortSignal.timeout(12000),
  });
  if (!r.ok) throw new Error(`Spotify-Embed nicht erreichbar (${r.status})`);
  const html = await r.text();
  const m = html.match(/<script id="__NEXT_DATA__" type="application\/json">(.+?)<\/script>/s);
  if (!m) throw new Error('Spotify-Seite nicht parsebar — Struktur geändert?');
  const nd = JSON.parse(m[1]);
  let entity = nd?.props?.pageProps?.state?.data?.entity
             || nd?.props?.pageProps?.data?.entity
             || nd?.props?.pageProps?.entity
             || nd?.props?.pageProps?.state?.data;
  if (!entity) throw new Error('Keine Metadaten in Spotify-Embed gefunden');
  if (entity.data && !entity.name && !entity.trackList) entity = entity.data;
  return entity;
}

function _spArtwork(entity) {
  const sources = entity?.albumOfTrack?.coverArt?.sources
               || entity?.album?.coverArt?.sources
               || entity?.coverArt?.sources
               || [];
  if (sources.length) return sources.sort((a, b) => (b.width || 0) - (a.width || 0))[0].url;
  return entity?.albumOfTrack?.images?.[0]?.url
      || entity?.album?.images?.[0]?.url
      || entity?.images?.[0]?.url
      || '';
}

function _spAllArtists(artistsRaw) {
  if (!artistsRaw) return '';
  const arr = Array.isArray(artistsRaw) ? artistsRaw : (artistsRaw.items || []);
  return arr.map(a => a?.profile?.name || a?.name || '').filter(Boolean).join(', ');
}

function _spArtistName(artistsRaw) {
  return _spAllArtists(artistsRaw);
}

function _spTrackItem(t, defaultAlbum, defaultArt, defaultAlbumArtist) {
  let raw = t?.track || t;
  const d = raw?.data || raw;
  const trackName = d?.name || d?.title || '';
  let artist = '';
  if (d?.artists) {
    artist = _spAllArtists(d.artists);
  } else if (typeof d?.subtitle === 'string' && d.subtitle) {
    artist = d.subtitle;
  }
  artist = artist || defaultAlbumArtist || '';
  const art     = _spArtwork(d) || defaultArt || '';
  const albumObj = d?.albumOfTrack || d?.album || {};
  const album   = albumObj?.name || defaultAlbum || '';
  const date    = albumObj?.date?.isoString || albumObj?.release_date || '';
  return {
    trackName,
    artistName:        artist,
    collectionName:    album,
    artworkUrl100:     art,
    releaseDate:       date,
    trackNumber:       d?.trackNumber || d?.track_number,
    trackExplicitness: d?.contentRating?.label === 'EXPLICIT' || d?.isExplicit || d?.explicit ? 'explicit' : 'notExplicit',
    previewUrl:        d?.previewUrl || d?.preview_url || d?.audioPreview?.url || null,
    durationMs:        d?.duration_ms || d?.durationMs || null,
  };
}

// ─── ROUTES ──────────────────────────────────────────────────────────────────

app.post('/api/spotify/track-artwork', async (req, res) => {
  const tracks = req.body.tracks || [];
  if (!tracks.length) return res.json({ results: [] });
  const CONC = 8;
  const results = [];
  for (let i = 0; i < tracks.length; i += CONC) {
    const batch = tracks.slice(i, i + CONC);
    const batchRes = await Promise.all(batch.map(async t => {
      const searchArtist = (t.artistName || '').split(/\s*[,/•]\s*/)[0].trim();
      const meta = await lookupItunesTrackMeta(searchArtist, t.trackName || '');
      return { idx: t.idx, artworkUrl: meta.artworkUrl || t.existingArt || '' };
    }));
    results.push(...batchRes);
  }
  res.json({ results });
});

app.get('/api/health', (req, res) => {
  res.json({
    status: 'ok', ytdlpVersion,
    musicPath: CONFIG.musicPath,
    audioFormat: CONFIG.audioFmt,
    downloadBackend: CONFIG.downloadBackend,
    squidQuality: CONFIG.squidQuality,
    activeJobs: activeDownloads,
    queueLength: downloadQueue.length,
  });
});

app.post('/api/spotify/resolve', async (req, res) => {
  const { url } = req.body;
  if (!url) return res.status(400).json({ error: 'url required' });
  try {
    const { type, id } = parseSpotifyUrl(url);
    const entity = await fetchSpotifyEmbed(type, id);
    const artist = _spArtistName(entity.artists) || (typeof entity.subtitle === 'string' ? entity.subtitle : '');
    const artUrl = _spArtwork(entity);

    if (type === 'track') {
      return res.json({
        type: 'track',
        name: entity.name || entity.title || '',
        artist,
        artworkUrl: artUrl,
        tracks: [_spTrackItem(entity, entity.album?.name, artUrl, artist)],
      });
    }

    if (type === 'album') {
      const trackList = entity.trackList || entity.tracks?.items || [];
      return res.json({
        type: 'album',
        name: entity.name || entity.title || '',
        artist,
        artworkUrl: artUrl,
        tracks: trackList.map(t => _spTrackItem(t, entity.name || entity.title, artUrl, artist)),
      });
    }

    if (type === 'playlist') {
      const trackList = entity.trackList || entity.content?.items || entity.tracks?.items || [];
      const owner = entity.authors?.[0]?.name
                 || entity.ownerV2?.data?.name
                 || entity.owner?.displayName
                 || entity.owner?.display_name
                 || (typeof entity.subtitle === 'string' ? entity.subtitle : '');
      return res.json({
        type: 'playlist',
        name: entity.name || entity.title || '',
        artist: owner,
        artworkUrl: artUrl,
        description: entity.description || '',
        tracks: trackList.map(t => _spTrackItem(t, null, null, null)),
      });
    }

    res.status(400).json({ error: `Unbekannter Typ: ${type}` });
  } catch(e) {
    console.error('[spotify/resolve]', e.message);
    res.status(500).json({ error: e.message });
  }
});

app.post('/api/download', async (req, res) => {
  const { type, artist, title, album, artworkUrl, collectionId, jobId: clientJobId, explicit,
          trackNum: bodyTrackNum, totalTracks: bodyTotalTracks, discNum: bodyDiscNum, discTotal: bodyDiscTotal } = req.body;
  if (!artist || !title) return res.status(400).json({ error: 'artist and title required' });

  const jobId = clientJobId || ('job_' + Date.now());
  jobs.set(jobId, {
    id: jobId, type, artist, title, album,
    status: 'queued', progress: 0, message: 'In Warteschlange…',
    log: [], createdAt: Date.now(), updatedAt: Date.now(),
  });
  res.json({ jobId, status: 'queued' });

  enqueue(async () => {
    updateJob(jobId, { status: 'running', progress: 5, message: 'Gestartet…' });
    try {
      if (type === 'track') {
        const sp     = req.body.tracks?.[0] || {};
        const spAlbum = sp.collectionName || album || '';
        const spArt   = sp.artworkUrl100  || artworkUrl || '';
        const spYear  = sp.releaseDate ? new Date(sp.releaseDate).getFullYear() : undefined;
        let meta = {
          album: spAlbum, year: spYear,
          trackNum:    sp.trackNumber  || bodyTrackNum    || undefined,
          totalTracks: sp.trackCount   || bodyTotalTracks || undefined,
          discNum:     sp.discNumber   || bodyDiscNum     || undefined,
          discTotal:   sp.discCount    || bodyDiscTotal   || undefined,
          artworkUrl: spArt, explicit: sp.trackExplicitness === 'explicit' || !!explicit,
          durationMs: sp.durationMs || undefined,
        };
        if (!meta.album) {
          const searchArtist = artist.split(',')[0].trim();
          log(jobId, '📋 Suche Album-Metadaten von iTunes…');
          const extra = await lookupItunesTrackMeta(searchArtist, title);
          if (extra.album) log(jobId, `📋 Album: "${extra.album}" (${extra.year || '?'})`);
          meta = { ...meta, ...extra, artworkUrl: spArt || extra.artworkUrl || '' };
        } else {
          log(jobId, `📋 Album aus Spotify: "${meta.album}"`);
        }
        const result = await downloadTrack({ jobId, artist, title, ...meta });
        if (result?.skipped) {
          updateJob(jobId, { status: 'done', progress: 100, message: 'Bereits vorhanden – übersprungen', skipped: true });
        } else {
          updateJob(jobId, { status: 'done', progress: 100, message: 'Fertig!' });
        }

      } else if (type === 'album') {
        let tracks = [];
        if (collectionId) {
          log(jobId, '📋 Lade Track-Liste von iTunes…');
          tracks = await fetchItunesTracks(collectionId);
        }
        if (!tracks.length) {
          log(jobId, '🔍 Suche Album auf iTunes…');
          tracks = await searchItunesAlbumTracks(artist, album || title);
        }
        if (!tracks.length) {
          throw new Error(`Keine einzelnen Tracks für "${album || title}" gefunden. Bitte über iTunes Global-Suche herunterladen.`);
        }
        const { done: dlDone, skipped: dlSkipped } = await downloadAlbumTracks({ jobId, artist, albumTitle: album || title, tracks, artworkUrl });
        const allSkipped  = dlSkipped === tracks.length;
        const someSkipped = dlSkipped > 0 && !allSkipped;
        const msg = allSkipped
          ? `Album bereits vorhanden – alle ${dlSkipped} Tracks übersprungen`
          : someSkipped
            ? `Album fertig! (${dlDone} geladen, ${dlSkipped} bereits vorhanden)`
            : `Album "${album || title}" fertig!`;
        updateJob(jobId, { status: 'done', progress: 100, message: msg, skipped: allSkipped });

      } else if (type === 'artist') {
        log(jobId, `🎤 Künstler-Download: ${artist}`);
        const artistData = await new Promise((resolve, reject) => {
          https.get(`https://itunes.apple.com/search?term=${encodeURIComponent(artist)}&media=music&entity=album&limit=10&country=de`, res => {
            let d = ''; res.on('data', c => d += c);
            res.on('end', () => { try { resolve(JSON.parse(d).results || []); } catch { reject(new Error('parse')); } });
          }).on('error', reject);
        });
        const artistAlbums = artistData.filter(a => a.artistName?.toLowerCase() === artist.toLowerCase()).slice(0, 10);
        log(jobId, `📀 ${artistAlbums.length} Alben gefunden`);
        let done = 0;
        for (const alb of artistAlbums) {
          const tracks = await fetchItunesTracks(alb.collectionId).catch(() => []);
          if (tracks.length) {
            await downloadAlbumTracks({ jobId, artist, albumTitle: alb.collectionName, tracks, artworkUrl: alb.artworkUrl100 });
          }
          done++;
          updateJob(jobId, { progress: Math.round((done / artistAlbums.length) * 95) });
        }
        updateJob(jobId, { status: 'done', progress: 100, message: `Künstler "${artist}" fertig!` });

      } else if (type === 'playlist') {
        const plTracks = req.body.tracks || [];
        if (!plTracks.length) throw new Error('Keine Tracks übergeben');
        log(jobId, `🎵 Playlist "${title}" — ${plTracks.length} Tracks`);
        let done = 0, skipped = 0;
        const CONC = CONFIG.maxParallel || 2;
        for (let i = 0; i < plTracks.length; i += CONC) {
          const batch = plTracks.slice(i, i + CONC);
          const batchResults = await Promise.allSettled(batch.map(async track => {
            const trackTitle  = track.trackName  || '';
            const trackArtist = track.artistName || artist || '';
            const searchArtist = trackArtist.split(',')[0].trim();
            log(jobId, `📋 iTunes: "${trackArtist} – ${trackTitle}"`);
            const itunes = await lookupItunesTrackMeta(searchArtist, trackTitle);
            return downloadTrack({
              jobId,
              artist:      trackArtist,
              albumArtist: trackArtist,
              title:       trackTitle,
              album:       itunes.album       || track.collectionName || '',
              year:        itunes.year        || (track.releaseDate ? new Date(track.releaseDate).getFullYear() : undefined),
              trackNum:    itunes.trackNum    || track.trackNumber    || undefined,
              totalTracks: itunes.totalTracks || undefined,
              genre:       itunes.genre       || track.primaryGenreName || undefined,
              artworkUrl:  itunes.artworkUrl  || track.artworkUrl100  || '',
              explicit:    track.trackExplicitness === 'explicit',
              durationMs:  itunes.durationMs  || track.durationMs    || undefined,
            });
          }));
          for (let j = 0; j < batchResults.length; j++) {
            const r = batchResults[j];
            if (r.status === 'rejected') {
              log(jobId, `⚠️  Fehler bei "${batch[j].trackName || '?'}": ${r.reason?.message || r.reason}`);
            } else if (r.value?.skipped) {
              skipped++;
            } else {
              done++;
            }
            updateJob(jobId, { progress: Math.round(((done + skipped) / plTracks.length) * 90) });
          }
        }
        const allSkipped = skipped === plTracks.length;
        updateJob(jobId, {
          status: 'done', progress: 100, skipped: allSkipped,
          message: allSkipped
            ? `Alle ${skipped} Tracks bereits vorhanden`
            : `Fertig! ${done} geladen${skipped ? `, ${skipped} übersprungen` : ''}`,
        });
      }
    } catch (err) {
      console.error(`[${jobId}] Error:`, err.message);
      updateJob(jobId, { status: 'error', message: err.message });
    }
  });
});

app.get('/api/status/:jobId', (req, res) => {
  const job = jobs.get(req.params.jobId);
  if (!job) return res.status(404).json({ error: 'Job not found' });
  res.json(job);
});

app.get('/api/jobs', (req, res) => {
  res.json(Array.from(jobs.values()).sort((a, b) => b.createdAt - a.createdAt).slice(0, 50));
});

// ─── ABS PROXY ───────────────────────────────────────────────────────────────
// Forwards API calls to Audiobookshelf to avoid CORS restrictions in the browser.
// The browser sends X-ABS-URL and X-ABS-Token headers; audio streaming uses
// direct <audio src> URLs and is unaffected by CORS.
app.all('/api/abs-proxy/*path', async (req, res) => {
  const absBase  = req.headers['x-abs-url'];
  const absToken = req.headers['x-abs-token'];
  if (!absBase || !absToken) { res.status(400).json({ error: 'Missing X-ABS-URL or X-ABS-Token' }); return; }

  // Strip the proxy prefix to get the ABS path (e.g. /api/me, /api/libraries/…)
  const absPath = req.path.replace(/^\/api\/abs-proxy/, '');
  const qs      = Object.keys(req.query).length ? '?' + new URLSearchParams(req.query).toString() : '';
  const url     = `${absBase}${absPath}${qs}`;

  const fwdHeaders = { 'Authorization': `Bearer ${absToken}`, 'Content-Type': 'application/json' };
  const opts = { method: req.method, headers: fwdHeaders };
  if (req.method !== 'GET' && req.method !== 'HEAD' && req.body && Object.keys(req.body).length) {
    opts.body = JSON.stringify(req.body);
  }

  try {
    const upstream = await fetch(url, opts);
    const text = await upstream.text();
    res.status(upstream.status)
       .setHeader('Content-Type', upstream.headers.get('Content-Type') || 'application/json')
       .send(text);
  } catch (e) {
    res.status(502).json({ error: `ABS proxy error: ${e.message}` });
  }
});

// ─── START ────────────────────────────────────────────────────────────────────
app.listen(CONFIG.port, () => {
  console.log(`\n🎵 NovaSonic Backend running on port ${CONFIG.port}`);
  console.log(`   Music folder     : ${CONFIG.musicPath}`);
  console.log(`   Audio format     : ${CONFIG.audioFmt} (quality ${CONFIG.audioQuality})`);
  console.log(`   Download backend : ${CONFIG.downloadBackend}`);
  if (CONFIG.downloadBackend !== 'ytdlp') {
    console.log(`   Squid quality    : ${CONFIG.squidQuality}`);
  }
  console.log(`   yt-dlp           : ${ytdlpVersion}`);
  console.log('');
});