kr-phim/server/middleware/proxy-m3u8.ts

import { createHash } from 'crypto'
import { useRedis } from '../utils/redis'

const CACHE_TTL = 86400 // 1 ngày cho .ts files
const CACHE_PREFIX = 'hls:'

async function getCachedFile(redis: any, url: string): Promise<Buffer | null> {
  const key = CACHE_PREFIX + createHash('md5').update(url).digest('hex')
  try {
    const cached = await redis.getBuffer(key)
    return cached || null
  } catch {
    return null
  }
}

async function setCachedFile(redis: any, url: string, data: Buffer): Promise<void> {
  const key = CACHE_PREFIX + createHash('md5').update(url).digest('hex')
  try {
    await redis.set(key, data, 'EX', CACHE_TTL)
  } catch {
    // Ignore cache write errors
  }
}

export default defineEventHandler(async (event) => {
  const path = event.path || event.node?.req?.url || ''
  
  // Only handle /api/proxy-m3u8/ paths
  if (!path.startsWith('/api/proxy-m3u8/')) {
    return
  }
  
  const base64Url = path.replace('/api/proxy-m3u8/', '')
  
  if (!base64Url) {
    throw createError({ statusCode: 400, message: 'Missing url parameter' })
  }

  // Decode URL from base64
  let url: string
  try {
    url = Buffer.from(base64Url, 'base64').toString('utf-8')
  } catch {
    throw createError({ statusCode: 400, message: 'Invalid URL encoding' })
  }

  if (!url.startsWith('http://') && !url.startsWith('https://')) {
    throw createError({ statusCode: 400, message: 'Invalid URL protocol' })
  }

  try {
    const isTsFile = url.includes('.ts')
    const redis = useRedis()
    
    // Check cache for .ts files
    if (isTsFile) {
      const cached = await getCachedFile(redis, url)
      if (cached) {
        setHeader(event, 'access-control-allow-origin', '*')
        setHeader(event, 'cache-control', 'public, max-age=31536000, immutable')
        setHeader(event, 'content-type', 'video/mp2t')
        setHeader(event, 'content-length', cached.length.toString())
        setHeader(event, 'x-cache', 'HIT')
        return cached
      }
    }

    const response = await fetch(url, {
      headers: {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
        'Accept': '*/*',
        'Referer': new URL(url).origin,
        'Origin': new URL(url).origin,
      },
    })

    if (!response.ok) {
      throw createError({ statusCode: response.status, message: `Upstream error: ${response.statusText}` })
    }

    const contentType = response.headers.get('content-type') || 'application/octet-stream'
    const isM3u8 = url.includes('.m3u8') || contentType.includes('mpegurl')

    setHeader(event, 'access-control-allow-origin', '*')
    setHeader(event, 'access-control-allow-methods', 'GET, OPTIONS')

    if (isM3u8) {
      let body = await response.text()
      const baseUrl = url.substring(0, url.lastIndexOf('/') + 1)
      
      body = body.replace(/^(https?:\/\/[^\s]+)$/gm, (match) => {
        const encoded = Buffer.from(match).toString('base64')
        return `/api/proxy-m3u8/${encoded}`
      })
      
      body = body.replace(/^([^#\s][^\n]*)$/gm, (match) => {
        if (match.startsWith('http://') || match.startsWith('https://') || match.startsWith('/api/proxy-m3u8/')) {
          return match
        }
        const absoluteUrl = new URL(match, baseUrl).href
        const encoded = Buffer.from(absoluteUrl).toString('base64')
        return `/api/proxy-m3u8/${encoded}`
      })
      
      setHeader(event, 'content-type', 'application/vnd.apple.mpegurl')
      setHeader(event, 'cache-control', 'public, max-age=300')
      setHeader(event, 'x-cache', 'MISS')
      return body
    } else if (isTsFile) {
      // Cache .ts files in Redis
      const buffer = Buffer.from(await response.arrayBuffer())
      await setCachedFile(redis, url, buffer)
      
      setHeader(event, 'content-type', 'video/mp2t')
      setHeader(event, 'cache-control', 'public, max-age=31536000, immutable')
      setHeader(event, 'content-length', buffer.length.toString())
      setHeader(event, 'x-cache', 'MISS')
      return buffer
    } else {
      setHeader(event, 'content-type', contentType)
      const contentLength = response.headers.get('content-length')
      if (contentLength) {
        setHeader(event, 'content-length', contentLength)
      }
      setHeader(event, 'cache-control', 'public, max-age=3600')
      return response.body
    }
  } catch (error: any) {
    if (error.statusCode) throw error
    throw createError({ statusCode: 500, message: `Proxy error: ${error.message}` })
  }
})