import axios from 'axios'
import type { NextApiRequest, NextApiResponse } from 'next'

import apiConfig from '../../config/api.json'

export default async function handler(req: NextApiRequest, res: NextApiResponse) {
  // 'inline' is used for previewing PDF files inside the browser directly
  const { url, inline = false } = req.query

  // Only handle urls that start with OneDrive's own direct link (or SharePoint's):
  // https://public.dm.files.1drv.com/y4m0G_0GPeS8AXGrux-lVV79eU1F38VbWxtCSi-8-aUkBLeZH1H6...
  if (!url || !(url as string).startsWith(apiConfig.directLink)) {
    res.status(400).json({ error: 'Invalid URL' })
    return
  }

  const { headers, data: stream } = await axios.get(url as string, {
    responseType: 'stream',
  })

  // Check if requested file is PDF based on content-type
  if (headers['content-type'] === 'application/pdf' && inline) {
    // Get filename from content-disposition header
    const filename = headers['content-disposition'].split(/filename[*]?=/)[1]
    // Remove original content-disposition header
    delete headers['content-disposition']
    // Add new inline content-disposition header along with filename
    headers['content-disposition'] = `inline; filename*=UTF-8''${filename}`
  }

  // Send data stream as response
  res.writeHead(200, headers)
  stream.pipe(res)
}