DailyHotApi/routes/douyin.js

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

// 缓存键名
const cacheKey = "douyinHotData";
const cacheCookieKey = "douyinCookieData";

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

// 调用路径
const url = "https://www.douyin.com/aweme/v1/web/hot/search/list/?device_platform=webapp&aid=6383&channel=channel_pc_web&detail_list=1&round_trip_time=50";

// Token获取路径
const cookisUrl = "https://www.douyin.com/passport/general/login_guiding_strategy/?aid=6383";

// 数据处理
const getData = (data) => {
  if (!data) return [];
  const dataList = [];
try {
    const jsonObject = data.data.word_list;
    jsonObject.forEach((v) => {
      dataList.push({
        title: v.word,
        pic: `${v.word_cover.url_list[0]}`,
        hot: Number(v.hot_value),
        url: `https://www.douyin.com/hot/${encodeURIComponent(v.sentence_id)}`,
        mobileUrl: `https://www.douyin.com/hot/${encodeURIComponent(v.sentence_id)}`,
      });
    });
    return dataList;
  } catch (error) {
    console.error("数据处理出错" + error);
    return false;
  }
};

// 处理Cookis
const getCookies = (data) => {
    if(!data) return null;
try {
    const pattern = /passport_csrf_token=(.*); Path/s;
    const matchResult = data.headers["set-cookie"][0].match(pattern);
    const cookieData = matchResult[1];
    return cookieData;

} catch (error) {
    console.error("获取Cookis出错" + error);
    return false;
}
}

// 抖音热点榜
douyinRouter.get("/douyin", async (ctx) => {
  console.log("获取抖音热点榜");
  try {
    // 从缓存中获取数据
    let data = await get(cacheKey);
    const cookie = await get(cacheCookieKey);
    const from = data ? "cache" : "server";
    if (!data) {
      if(!cookie) {
        // 获取Cookies
        let cookisResponse = await axios.get(cookisUrl);
        let cookie = getCookies(cookisResponse);
        
        // 将Cookie写入缓存
        console.log('--- Cookie写入缓存 ---', cookie);
        await set(cacheCookieKey, cookie);
      }

      // 如果缓存中不存在数据
      console.log("从服务端重新获取抖音热点榜");
      // 从服务器拉取数据
      const response = await axios.get(url, {
        headers: {
          'Cookie': `passport_csrf_token=${cookie}`
        }
      });

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

// 抖音热点榜 - 获取最新数据
douyinRouter.get("/douyin/new", async (ctx) => {
  console.log("获取抖音热点榜 - 最新数据");
  const cookie = await get(cacheCookieKey);
  try {
    if(!cookie) {
      // 获取Cookies
      let cookisResponse = await axios.get(cookisUrl);
      let cookie = getCookies(cookisResponse);
      
      // 将Cookie写入缓存
      console.log('--- Cookie写入缓存 ---', cookie);
      await set(cacheCookieKey, cookie);
    }

    // 从服务器拉取最新数据
    const response = await axios.get(url, {
      headers: {
        'Cookie': `passport_csrf_token=${cookie}`
      }
    });
    const newData = getData(response.data);
    updateTime = new Date().toISOString();
    console.log("从服务端重新获取抖音热点榜");

    // 返回最新数据
    ctx.body = {
      code: 200,
      message: "获取成功",
      title: "抖音热点榜",
      subtitle: "热点榜",
      total: newData.length,
      updateTime,
      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: "获取成功",
        title: "抖音热点榜",
        subtitle: "热点榜",
        total: cachedData.length,
        updateTime,
        data: cachedData,
      };
    } else {
      // 如果缓存中也没有数据，则返回错误信息
      ctx.body = {
        code: 500,
        title: "抖音热点榜",
        subtitle: "热点榜",
        message: "获取失败",
      };
    }
  }
});

module.exports = douyinRouter;