backend/controller/video.js

const { PrismaClient } =  require('@prisma/client')
const prisma = new PrismaClient()

const DOMPurify = require('isomorphic-dompurify')
const metadata = require('../utils/metadata.js')
const redis = require('../utils/redis.js')

exports.getVideo = async (req, res) => {
    let info
    const cached = await redis.get(`video:${req.params.id}`)

    if (cached) {
        info = JSON.parse(cached)
    } else {
        info = await prisma.videos.findFirst({
            where: {
                id: req.params.id
            },
            select: {
                title: true,
                description: true,
                thumbnail: true,
                source: true,
                published: true,
                archived: true,
                channel: true, 
                channelId: true, 
                channelAvatar: true,
                channelVerified: true,
                disabled: true,
                hasBeenReported: true
            }
        })
        
        if (!info) return res.json({ error: '404' })
        await redis.set(`video:${req.params.id}`, JSON.stringify(info), 'EX', 3600)
    }

    res.json({
        ...info,
        description: DOMPurify.sanitize(info.description),
    })
}

exports.getChannel = async (req, res) => {
    const cached = await redis.get(`channel:${req.params.id}`)
    if (cached) return res.json(JSON.parse(cached))

    const [videos, channel] = await Promise.all([
        metadata.getChannelVideos(req.params.id),
        metadata.getChannel(req.params.id)
    ])

    if (!videos || !channel || videos.error) {
        return res.json({ error: '404' });
    }

    const archived = await prisma.videos.findMany({
        where: {
            channelId: req.params.id
        }, 
        select: {
            id: true,
            title: true,
            thumbnail: true,
            published: true,
            archived: true
        }
    })

    const processedVideos = videos.map(video => ({
        id: video.url.replace('/watch?v=', ''),
        published: new Date(video.uploaded).toISOString().slice(0, 10),
        ...video
    }));
    
    archived.forEach(v => {
        const existingVideoIndex = processedVideos.findIndex(video => video.id === v.id);
        if (existingVideoIndex !== -1) {
            processedVideos[existingVideoIndex] = v;
        } else {
            processedVideos.push({ ...v, deleted: undefined });
        }
    });
    
    processedVideos.sort((a, b) => new Date(b.published) - new Date(a.published));
    
    const json = {
        name: channel.author,
        avatar: channel.authorThumbnails[1].url,
        verified: channel.authorVerified,
        videos: processedVideos
    }
    await redis.set(`channel:${req.params.id}`, JSON.stringify(json), 'EX', 3600)
    res.json(json)
}

exports.getOnlyChannelVideos = async (req, res) => {
    const cached = await redis.get(`channelVideos:${req.params.id}`)
    if (cached) return res.json(JSON.parse(cached))

    const archived = await prisma.videos.findMany({
        where: {
            channelId: req.params.id
        }, 
        select: {
            id: true,
            title: true,
            thumbnail: true,
            published: true,
            archived: true
        },
        orderBy: {
            published: 'desc'
        }
    })

    const json = {
        videos: archived 
    }
    await redis.set(`channelVideos:${req.params.id}`, JSON.stringify(json), 'EX', 3600)
    res.json(json)
}

exports.getPlaylist = async (req, res) => {
    const cached = await redis.get(`playlist:${req.params.id}`)
    if (cached) return res.json(JSON.parse(cached))

    const playlist = await metadata.getPlaylistVideos(req.params.id)
    if (!playlist || playlist.error) return res.json({ error: '404' })

    const playlistArchived = await prisma.videos.findMany({
        where: {
            playlist: req.params.id
        }, 
        select: {
            id: true,
            title: true,
            thumbnail: true,
            published: true,
            archived: true
        }
    })

    const allVideos = playlist.relatedStreams.map(video => ({
        id: video.url.replace('/watch?v=', ''),
        published: new Date(video.uploaded).toISOString().slice(0, 10),
        ...video
    }));

    await Promise.all(playlistArchived.map(async (v) => {
        const allVideo = allVideos.find(o => o.id == v.id);
        if (allVideo) {
            const index = allVideos.findIndex(o => o.id == v.id);
            allVideos[index] = v;
        } else {
            const live = await metadata.getVideoMetadata(v.id);
            allVideos.push({
                ...v,
                deleted: live.error ? true : false
            });
        }
    }));
    
    await Promise.all(allVideos.filter(v => !v.archived).map(async (v) => {
        const video = await prisma.videos.findFirst({
            where: {
                id: v.id
            },
            select: {
                id: true,
                title: true,
                thumbnail: true,
                published: true,
                archived: true
            }
        });
        if (video) {
            const index = allVideos.findIndex(o => o.id == v.id);
            allVideos[index] = video;
        }
    }));
    
    allVideos.sort((a, b) => new Date(b.published) - new Date(a.published));
    
    const json = {
        name: playlist.name,
        channel: playlist.uploader,
        url: playlist.uploaderUrl,
        avatar: playlist.uploaderAvatar,
        videos: allVideos
    }
    await redis.set(`playlist:${req.params.id}`, JSON.stringify(json), 'EX', 3600)
    res.json(json)
}