import type { APIRoute } from 'astro';
import { load } from 'cheerio';
import { DOUBAN_ID } from '@/consts';

// 添加服务器渲染标记
export const prerender = false;

export const GET: APIRoute = async ({ request }) => {
  const url = new URL(request.url);
  const type = url.searchParams.get('type') || 'movie';
  const start = parseInt(url.searchParams.get('start') || '0');
  
  try {
    let doubanUrl = '';
    if (type === 'book') {
      doubanUrl = `https://book.douban.com/people/${DOUBAN_ID}/collect?start=${start}&sort=time&rating=all&filter=all&mode=grid`;
    } else {
      doubanUrl = `https://movie.douban.com/people/${DOUBAN_ID}/collect?start=${start}&sort=time&rating=all&filter=all&mode=grid`;
    }

    const response = await fetch(doubanUrl, {
      headers: {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
        'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
        'Referer': 'https://movie.douban.com/'
      }
    });
    
    if (!response.ok) {
      return new Response(JSON.stringify({ error: '获取豆瓣数据失败' }), {
        status: 500,
        headers: {
          'Content-Type': 'application/json'
        }
      });
    }
    
    const html = await response.text();
    const $ = load(html);
    
    // 添加类型定义
    interface DoubanItem {
      imageUrl: string;
      title: string;
      subtitle: string;
      link: string;
      intro: string;
      rating: number;
      date: string;
    }
    
    const items: DoubanItem[] = [];
    $('.item.comment-item').each((_, element) => {
      const $element = $(element);
      
      const imageUrl = $element.find('.pic img').attr('src') || '';
      const title = $element.find('.title a em').text().trim();
      const subtitle = $element.find('.title a').text().replace(title, '').trim();
      const link = $element.find('.title a').attr('href') || '';
      const intro = $element.find('.intro').text().trim();
      
      // 获取评分，从rating1-t到rating5-t
      let rating = 0;
      for (let i = 1; i <= 5; i++) {
        if ($element.find(`.rating${i}-t`).length > 0) {
          rating = i;
          break;
        }
      }
      
      const date = $element.find('.date').text().trim();
      
      items.push({
        imageUrl,
        title,
        subtitle,
        link,
        intro,
        rating,
        date
      });
    });
    
    // 改进分页信息获取逻辑
    let currentPage = 1;
    let totalPages = 1;
    
    // 尝试从当前页码元素获取信息
    if ($('.paginator .thispage').length > 0) {
      currentPage = parseInt($('.paginator .thispage').text() || '1');
      // 豆瓣可能不直接提供总页数，需要计算
      const paginatorLinks = $('.paginator a');
      let maxPage = currentPage;
      paginatorLinks.each((_, el) => {
        const pageNum = parseInt($(el).text());
        if (!isNaN(pageNum) && pageNum > maxPage) {
          maxPage = pageNum;
        }
      });
      totalPages = maxPage;
    }
    
    const pagination = {
      current: currentPage,
      total: totalPages,
      hasNext: $('.paginator .next a').length > 0,
      hasPrev: $('.paginator .prev a').length > 0
    };
    
    return new Response(JSON.stringify({ items, pagination }), {
      headers: {
        'Content-Type': 'application/json'
      }
    });
  } catch (error) {
    return new Response(JSON.stringify({ error: '获取豆瓣数据失败' }), {
      status: 500,
      headers: {
        'Content-Type': 'application/json'
      }
    });
  }
}