import express from 'express';
import { Innertube } from 'youtubei.js'
import * as fs from 'node:fs'

import { EnabledTrackTypes } from 'googlevideo/utils';
import {
  createOutputStream,
  createStreamSink,
  createSabrStream,
} from './utils/sabr-stream-factory.js';
import type { SabrPlaybackOptions } from 'googlevideo/sabr-stream';

const ffmpeg = require('fluent-ffmpeg')
const ffmpegStatic = require('ffmpeg-static')

const app = express();
require('express-ws')(app)
ffmpeg.setFfmpegPath(ffmpegStatic)

const maxRetries = 5
const platforms = ['IOS', 'ANDROID', 'YTSTUDIO_ANDROID', 'YTMUSIC_ANDROID']

app.get('/health', async (req, res) => {
  try {
    const urls = ['/video/sRMMwpDTs5k', '/channel/UCRijo3ddMTht_IHyNSNXpNQ', '/videos/UCRijo3ddMTht_IHyNSNXpNQ']

    const results = await Promise.all(urls.map(async (url) => {
      const response = await fetch(`http://localhost:8008${url}`);
      const jsonData: any = await response.json();
      const status = jsonData.error ? 'unhealthy' : 'healthy';
      return { url, status };
    }));

    console.log('Health check results:', results);

    const isHealthy = results.every(result => result.status === 'healthy');
    if (isHealthy) {
      res.status(200).json({ message: 'All endpoints are healthy', results });
    } else {
      res.status(500).json({ error: 'Health check failed', results });
      switchIps()
    }
  } catch (error:any) {
    console.error('Health check failed:', error.message);
    switchIps()
    res.status(500).json({ error: 'Health check failed', results: [], errorMessage: error.message });
  }
})

app.get('/video/:id', async (req, res) => {
  let error = ''

  for (let retries = 0; retries < maxRetries; retries++) {
    try {
      const platform = platforms[retries % platforms.length];
      const yt = await Innertube.create();
      const info = await yt.getInfo(req.params.id, { // @ts-ignore
        client: platform
      });

      if (!info) {
        error = 'ErrorCantConnectToServiceAPI'
        continue;
      }
      if (info.playability_status!.status !== 'OK') {
        error = 'ErrorYTUnavailable'
        continue;
      }
      if (info.basic_info.is_live) {
        error = 'ErrorLiveVideo'
        continue;
      }
      if (info.basic_info.title == 'Video Not Available') {
        error = 'YoutubeIsFuckingWithMe'
        continue;
      }
      return res.json(info)
    } catch (error) {
      continue
    }
  }

  res.json({ error: error || 'ErrorUnknown' })
})

app.get('/channel/:id', async (req, res) => {
  let error = ''

  for (let retries = 0; retries < maxRetries; retries++) {
    try {
      const yt = await Innertube.create();
      const info = await yt.getChannel(req.params.id);

      if (!info) {
        error = 'ErrorCantConnectToServiceAPI'
        continue;
      }
      return res.json(info)
    } catch (error) {
      continue
    }
  }

  res.json({ error: error || 'ErrorUnknown' })
})

app.get('/videos/:id', async (req, res) => {
  try {
    const videos = [];
    const yt = await Innertube.create();
    const channel = await yt.getChannel(req.params.id);
    let json = await channel.getVideos();

    videos.push(...json.videos);

    while (json.has_continuation && videos.length < 60) {
      json = await getNextPage(json);
      videos.push(...json.videos);
    }

    return res.json(videos)
  } catch (e) {
    res.json(false)
  }

  async function getNextPage(json: any) {
    const page = await json.getContinuation();
    return page;
  }
})

// @ts-ignore
app.ws('/download/:id/:quality', async (ws, req) => {
  const yt = await Innertube.create();
  const info = await yt.getInfo(req.params.id, {
    client: 'IOS'
  });

  if (info.playability_status?.status !== 'OK') {
    ws.send(`This video is not available for download (${info.playability_status?.status} ${info.playability_status?.reason}).`);
    return ws.close()
  } else if (info.basic_info.is_live) {
    ws.send('This video is live, and cannot be downloaded.');
    return ws.close()
  } else if (info.basic_info.id != req.params.id) {
    ws.send('This video is not available for download. Youtube is serving a different video.');
    return ws.close()
  } 

  const streamOptions: SabrPlaybackOptions = {
    videoQuality: req.params.quality,
    audioQuality: 'AUDIO_QUALITY_LOW',
    enabledTrackTypes: EnabledTrackTypes.VIDEO_AND_AUDIO
  };
  const { streamResults } = await createSabrStream(req.params.id, streamOptions);
  const { videoStream, audioStream, selectedFormats } = streamResults;

  const whitelistedVideos = JSON.parse(fs.readFileSync('./whitelist.json', 'utf-8'))
  const videoSizeTotal = (selectedFormats.audioFormat.contentLength || 0) 
    + (selectedFormats.videoFormat.contentLength || 0)

  if (videoSizeTotal > (1_048_576 * 150) && !whitelistedVideos.includes(req.params.id)) {
    ws.send('Is this content considered high risk? If so, please email me at admin@preservetube.com.');
    ws.send('This video is too large, and unfortunately, Preservetube does not have unlimited storage.');
    return ws.close()
  } else if (!selectedFormats.videoFormat.contentLength) {
    ws.send('Youtube isn\'t giving us enough information to be able to tell if we can process this video.')
    ws.send('Please try again later.')
    return ws.close()
  }

  const audioOutputStream = createOutputStream(req.params.id, selectedFormats.audioFormat.mimeType!);
  const videoOutputStream = createOutputStream(req.params.id, selectedFormats.videoFormat.mimeType!);

  await Promise.all([
    videoStream.pipeTo(createStreamSink(selectedFormats.videoFormat, videoOutputStream.stream, ws, 'video')),
    audioStream.pipeTo(createStreamSink(selectedFormats.audioFormat, audioOutputStream.stream, ws, 'audio'))
  ]);

  ws.send('Downloaded video and audio. Merging them together.')

  await mergeIt(audioOutputStream.filePath, videoOutputStream.filePath, `./output/${req.params.id}.mp4`, ws)
  await cleanupTempFiles([ audioOutputStream.filePath, videoOutputStream.filePath ]);

  ws.send('done')
  ws.close()
});

function mergeIt(audioPath: string, videoPath: string, outputPath: string, ws: any) {
  return new Promise((resolve, reject) => {
    ffmpeg()
      .input(videoPath)
      .input(audioPath)
      .outputOptions([ '-c:v copy', '-c:a copy', '-map 0:v:0', '-map 1:a:0' ])
      .on('progress', (progress:any) => {
        if (progress.percent) {
          ws.send(`[merging] ${progress.precent}% done`)
        }
      })
      .on('end', () => {
        resolve(outputPath);
      })
      .on('error', (err:any) => {
        reject(new Error(`Error merging files: ${err.message}`));
      })
      .save(outputPath);
  });
}

async function cleanupTempFiles(files: string[]) {
  for (const file of files) {
    try {
      fs.unlinkSync(file);
    } catch (error) {
      console.warn(`Failed to delete temp file ${file}:`, error);
    }
  }
}

async function switchIps() {
  const currentIp: any = await (await fetch('http://localhost:8000/v1/publicip/ip', {
    headers: {
      'X-API-Key': '64d1781e469965c1cdad611b0c05d313'
    }
  })).json()
  const currentDate = new Date()

  console.log(`starting switching ips. ${currentIp.public_ip}, ${currentIp.city}, ${currentIp.region}, ${currentIp.organization}`)

  const s = await fetch('http://localhost:8000/v1/vpn/status', {
    method: 'PUT',
    headers: {
      'X-API-Key': '64d1781e469965c1cdad611b0c05d313',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify({ status: 'stopped' })
  })
  console.log(`stopped vpn - ${await s.text()}`)

  const r = await fetch('http://localhost:8000/v1/vpn/status', {
    method: 'PUT',
    headers: {
      'X-API-Key': '64d1781e469965c1cdad611b0c05d313',
      'Content-Type': 'application/json'
    },
    body: JSON.stringify({ status: 'running' })
  })
  console.log(`turned on vpn - ${await r.text()}`)

  await new Promise((resolve, reject) => {
    const intervalId = setInterval(async () => {
      const newIp: any = await (await fetch('http://localhost:8000/v1/publicip/ip', {
        headers: {
          'X-API-Key': '64d1781e469965c1cdad611b0c05d313',
        },
      })).json();

      if (newIp.public_ip !== '') {
        console.log(`finished switching ips. ${newIp.public_ip}, ${newIp.city}, ${newIp.region}, ${newIp.organization}. took ${(new Date().getTime() - currentDate.getTime()) / 1000}s`)
        clearInterval(intervalId);
        resolve('done')
      }
    }, 500);
  })
}

setInterval(switchIps, 30 * 60000) // 30 minutes

app.listen(8008, () => {
  console.log('the metadata server is up.')
  switchIps()
})