DailyHotApi/routes/douban_group.js

/**
 * @author: x-dr
 * @date: 2023-12-26
 * @customEditors: imsyy
 * @lastEditTime: 2024-01-02
 */

const Router = require("koa-router");
const doubanGroupNewRouter = new Router();
const axios = require("axios");
const cheerio = require("cheerio");
const { get, set, del } = require("../utils/cacheData");

// 接口信息
const routerInfo = {
  name: "douban_group",
  title: "豆瓣讨论小组",
  subtitle: "精选",
};

// 缓存键名
const cacheKey = "doubanGroupData";

// 调用时间
let updateTime = new Date().toISOString();

const url = "https://www.douban.com/group/explore";

const headers = {
  accept:
    "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
  "accept-language": "zh-CN,zh;q=0.9,en;q=0.8",
  "cache-control": "max-age=0",
  "sec-ch-ua": '"Not_A Brand";v="8", "Chromium";v="120", "Google Chrome";v="120"',
  "sec-ch-ua-mobile": "?0",
  "sec-ch-ua-platform": '"Windows"',
  "sec-fetch-dest": "document",
  "sec-fetch-mode": "navigate",
  "sec-fetch-site": "none",
  "sec-fetch-user": "?1",
  "upgrade-insecure-requests": "1",
  // "cookie": "bid=lLpb6D1JLuw; douban-fav-remind=1; _pk_id.100001.8cb4=e7d91ae46530fd1d.1680518589.; ll=\"118281\"; _pk_ref.100001.8cb4=%5B%22%22%2C%22%22%2C1703602972%2C%22http%3A%2F%2Fnew.xianbao.fun%2F%22%5D; _pk_ses.100001.8cb4=1; ap_v=0,6.0"
};

// 数据处理
const getData = (data) => {
  if (!data) return false;
  const dataList = [];
  const $ = cheerio.load(data);
  try {
    $(`.channel-item`).each((i, e) => {
      // console.log($(e).html());
      const item = cheerio.load($(e).html());
      const title = item("h3")
        .text()
        .replace(/(^\s*)|(\s*$)/g, "");
      const url = item("h3 a").attr("href");
      const hot = item('div[class="likes"]')
        .text()
        .replace(/(^\s*)|(\s*$)/g, "");
      const desc = item('div[class="block"]')
        .text()
        .replace(/(^\s*)|(\s*$|\n)/g, "");
      const source = item('div[class="source"] a')
        .text()
        .replace(/(^\s*)|(\s*$)/g, "");
      // const excerpt = item('.channel-item-desc').text().replace(/(^\s*)|(\s*$)/g, "")
      // console.log(title);
      // console.log(url);
      dataList.push({
        title: title,
        desc: desc,
        url: url,
        mobileUrl: url,
        hot: hot,
        source: source,
      });
    });
    return dataList;
  } catch (error) {
    console.error("数据处理出错" + error);
    return false;
  }
};

// trending
doubanGroupNewRouter.get("/douban_group", async (ctx) => {
  console.log("获取豆瓣讨论小组精选");
  try {
    // 从缓存中获取数据
    let data = await get(cacheKey);
    const from = data ? "cache" : "server";
    if (!data) {
      // 如果缓存中不存在数据
      console.log("从服务端重新豆瓣讨论小组精选");
      // 从服务器拉取数据
      const response = await axios.get(url, { headers });
      // console.log(response.data);
      data = getData(response.data);

      updateTime = new Date().toISOString();
      if (!data) {
        ctx.body = {
          code: 500,
          ...routerInfo,
          message: "获取失败",
        };
        return false;
      }
      // 将数据写入缓存
      await set(cacheKey, data);
    }
    ctx.body = {
      code: 200,
      message: "获取成功",
      ...routerInfo,
      from,
      total: data.length,
      updateTime,
      data,
    };
  } catch (error) {
    console.error(error);
    ctx.body = {
      code: 500,
      ...routerInfo,
      message: "获取失败",
    };
  }
});

// 豆瓣新片榜 - 获取最新数据
doubanGroupNewRouter.get("/douban_group/new", async (ctx) => {
  console.log("获取豆瓣讨论小组精选  - 最新数据");
  try {
    // 从服务器拉取最新数据
    const response = await axios.get(url, { headers });
    const newData = getData(response.data);
    updateTime = new Date().toISOString();
    console.log("从服务端重新豆瓣讨论小组精选");

    // 返回最新数据
    ctx.body = {
      code: 200,
      message: "获取成功",
      ...routerInfo,
      updateTime,
      total: newData.length,
      data: newData,
    };

    // 删除旧数据
    await del(cacheKey);
    // 将最新数据写入缓存
    await set(cacheKey, newData);
  } catch (error) {
    // 如果拉取最新数据失败，尝试从缓存中获取数据
    console.error(error);
    const cachedData = await get(cacheKey);
    if (cachedData) {
      ctx.body = {
        code: 200,
        message: "获取成功",
        ...routerInfo,
        total: cachedData.length,
        updateTime,
        data: cachedData,
      };
    } else {
      // 如果缓存中也没有数据，则返回错误信息
      ctx.body = {
        code: 500,
        ...routerInfo,
        message: "获取失败",
      };
    }
  }
});

doubanGroupNewRouter.info = routerInfo;
module.exports = doubanGroupNewRouter;