Direct Download & Stream

export default {
  async fetch(request, env, ctx) {
    const url = new URL(request.url);
    
    // Handle CORS
    if (request.method === 'OPTIONS') {
      return new Response(null, {
        headers: {
          'Access-Control-Allow-Origin': '*',
          'Access-Control-Allow-Methods': 'GET, HEAD, OPTIONS',
          'Access-Control-Allow-Headers': '*',
        }
      });
    }
    
    // Route: /download/:fileId
    if (url.pathname.startsWith('/download/')) {
      const fileId = url.pathname.replace('/download/', '').split('/')[0];
      if (fileId) return handleDownload(fileId);
    }
    
    // Route: /stream/:fileId
    if (url.pathname.startsWith('/stream/')) {
      const fileId = url.pathname.replace('/stream/', '').split('/')[0];
      if (fileId) return handleStream(fileId, request);
    }
    
    // Serve static files
    return env.ASSETS.fetch(request);
  }
};

async function handleDownload(fileId) {
  try {
    // Try direct download first
    let response = await fetch(
      `https://drive.google.com/uc?export=download&id=${fileId}`,
      {
        headers: {
          'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) Chrome/91.0'
        },
        redirect: 'follow'
      }
    );
    
    const contentType = response.headers.get('content-type') || '';
    
    // If HTML (virus warning), bypass it
    if (contentType.includes('text/html')) {
      const html = await response.text();
      const confirmMatch = html.match(/confirm=([0-9A-Za-z_-]+)/);
      
      if (confirmMatch) {
        response = await fetch(
          `https://drive.google.com/uc?export=download&id=${fileId}&confirm=${confirmMatch[1]}`,
          {
            headers: {
              'User-Agent': 'Mozilla/5.0 Chrome/91.0',
              'Cookie': response.headers.get('set-cookie') || ''
            },
            redirect: 'follow'
          }
        );
      } else {
        // Fallback: use confirm=t
        response = await fetch(
          `https://drive.google.com/uc?export=download&id=${fileId}&confirm=t`,
          { redirect: 'follow' }
        );
      }
    }
    
    // Get filename
    let filename = 'download';
    const disposition = response.headers.get('content-disposition');
    if (disposition) {
      const match = disposition.match(/filename\*?=['"]?(?:UTF-8'')?([^;\r\n"']+)/i);
      if (match) filename = decodeURIComponent(match[1]);
    }
    
    return new Response(response.body, {
      headers: {
        'Content-Type': response.headers.get('content-type') || 'application/octet-stream',
        'Content-Disposition': `attachment; filename="${filename}"`,
        'Access-Control-Allow-Origin': '*',
      }
    });
  } catch (error) {
    return new Response(JSON.stringify({ error: error.message }), {
      status: 500,
      headers: { 'Content-Type': 'application/json' }
    });
  }
}

async function handleStream(fileId, request) {
  try {
    const response = await fetch(
      `https://drive.google.com/uc?export=download&id=${fileId}&confirm=t`,
      {
        headers: {
          'User-Agent': 'Mozilla/5.0 Chrome/91.0',
          'Range': request.headers.get('Range') || ''
        },
        redirect: 'follow'
      }
    );
    
    return new Response(response.body, {
      status: response.status,
      headers: {
        'Content-Type': response.headers.get('content-type') || 'video/mp4',
        'Content-Length': response.headers.get('content-length') || '',
        'Accept-Ranges': 'bytes',
        'Access-Control-Allow-Origin': '*',
      }
    });
  } catch (error) {
    return new Response(JSON.stringify({ error: error.message }), {
      status: 500,
      headers: { 'Content-Type': 'application/json' }
    });
  }
}