develop202-migu_video/app.js

import http from "node:http"
import { getAndroidURL, getAndroidURL720p } from "./utils/androidURL.js";
import { readFileSync } from "./utils/fileUtil.js";
import { host, port, rateType, token, userId } from "./config.js";
import { getDateTimeStr } from "./utils/time.js";
import update from "./updateData.js";
import { printBlue, printGreen, printGrey, printMagenta, printRed, printYellow } from "./utils/colorOut.js";
import { delay } from "./utils/fetchList.js";

// 运行时长
var hours = 0
// url缓存 降低请求频率
const urlCache = {}

let loading = false

const server = http.createServer(async (req, res) => {

  while (loading) {
    await delay(50)
  }

  loading = true

  // 获取请求方法、URL 和请求头
  const { method, url, headers } = req;

  console.log()
  printMagenta("请求地址：" + url)

  if (method != "GET") {
    res.writeHead(200, { 'Content-Type': 'application/json;charset=UTF-8' });
    res.end(JSON.stringify({
      data: '请使用GET请求',
    }));
    printRed(`使用非GET请求:${method}`)

    loading = false
    return
  }

  // 响应接口内容
  if (url == "/" || url == "/interface.txt") {
    try {
      // 读取文件内容
      const data = readFileSync(process.cwd() + "/interface.txt");

      // 设置响应头
      res.setHeader('Content-Type', 'text/plain;charset=UTF-8');
      res.statusCode = 200;
      res.end(data); // 发送文件内容

      loading = false
      return
    } catch (error) {

      res.writeHead(200, { "Content-Type": "application/json;charset=UTF-8" })
      res.end("访问异常")
      printRed("接口文件响应异常")

      loading = false
      return
    }
  }

  // 回放
  if (url == "/playback.xml") {

    try {
      // 读取文件内容
      const data = readFileSync(process.cwd() + "/playback.xml");

      // 设置响应头
      res.setHeader('Content-Type', 'text/xml;charset=UTF-8');
      res.statusCode = 200;
      res.end(data); // 发送文件内容
      loading = false
      return
    } catch (error) {

      res.writeHead(200, { "Content-Type": "application/json;charset=UTF-8" })
      res.end("访问异常")
      printRed("回放文件响应异常")
      loading = false
      return
    }

  }

  let urlSplit = url.split("/")[1]
  let pid = urlSplit
  let params = ""


  if (urlSplit.match(/\?/)) {
    // 回放
    printGreen("处理传入参数")

    const urlSplit1 = urlSplit.split("?")
    pid = urlSplit1[0]
    params = urlSplit1[1]
  } else {
    printGrey("无参数传入")
  }


  if (isNaN(pid)) {
    res.writeHead(200, { "Content-Type": "application/json;charset=UTF-8" })
    res.end("地址错误")
    printRed("地址格式错误")
    loading = false
    return
  }

  printYellow("频道ID " + pid)

  // 是否存在缓存
  if (typeof urlCache[pid] === "object") {
    const valTime = urlCache[pid].valTime - Date.now()
    // 缓存是否有效
    if (valTime >= 0) {

      printGreen(`缓存有效，使用缓存数据`)

      let playURL = urlCache[pid].url
      // 节目调整
      if (playURL == "") {
        printRed(`${pid} 节目调整，暂不提供服务`)

        res.writeHead(200, { "Content-Type": "application/json;charset=UTF-8" })
        res.end("节目调整，暂不提供服务")
        loading = false
        return
      }

      // 添加回放参数
      if (params != "") {
        const resultParams = new URLSearchParams(params);
        for (const [key, value] of resultParams) {
          playURL = `${playURL}&${key}=${value}`
        }
      }
      res.writeHead(302, {
        'Content-Type': 'application/json;charset=UTF-8',
        location: playURL
      });

      res.end()
      loading = false
      return
    }
  }

  let resObj = {}
  try {
    // 未登录请求720p
    if (rateType >= 3 && (userId == "" || token == "")) {
      resObj = await getAndroidURL720p(pid)
    } else {
      resObj = await getAndroidURL(userId, token, pid, rateType)
    }
  } catch (error) {

    res.writeHead(200, { "Content-Type": "application/json;charset=UTF-8" })
    res.end("链接请求出错，请稍后重试")
    printRed("链接请求出错")
    loading = false
    return
  }

  printGreen(`添加节目缓存 ${pid}`)
  // 加入缓存
  urlCache[pid] = {
    // 有效期2小时 节目调整改为2分钟
    valTime: Date.now() + (resObj.url == "" ? 2 * 60 * 1000 : 2 * 60 * 60 * 1000),
    url: resObj.url
  }

  if (resObj.url == "") {
    printRed(`${pid} 节目调整，暂不提供服务`)

    res.writeHead(200, { "Content-Type": "application/json;charset=UTF-8" })
    res.end("节目调整，暂不提供服务")
    loading = false
    return
  }
  let playURL = resObj.url


  // console.dir(playURL, { depth: null })

  // 添加回放参数
  if (params != "") {
    const resultParams = new URLSearchParams(params);
    for (const [key, value] of resultParams) {
      playURL = `${playURL}&${key}=${value}`
    }
  }

  printGreen("链接获取成功")


  res.writeHead(302, {
    'Content-Type': 'application/json;charset=UTF-8',
    location: playURL
  });

  res.end()

  loading = false
})

server.listen(port, async () => {

  // 设置定时器，3小时更新一次
  setInterval(async () => {
    printBlue(`\n准备更新文件 ${getDateTimeStr(new Date())}\n`)
    hours += 3
    try {
      await update(hours)
    } catch (error) {
      printRed("更新失败")
    }

    printBlue(`\n当前已运行${hours}小时`)
  }, 3 * 60 * 60 * 1000);

  try {
    // 初始化数据
    await update(hours)
  } catch (error) {
    printRed("更新失败")
  }

  console.log()

  printYellow("定时器设置完毕 每3小时更新一次")
  printYellow("Server running at port " + port)
  printYellow("访问地址:  " + host)
})