newCashBook/epub_node/router/epub/index.js

// 添加账本
import {
  getFileBymd5,
  searchFileByPath,
  author_insert,
  get_author_info,
} from "#db";
import logger from "#logger";
import path from "node:path";
import fs from "node:fs";
import express from "express";
import { EPub } from "epub2";
import { v4 as uuidv4 } from "uuid";
import { dirExists, isFileSync, isDir, waittime } from "#utils";

import { saveImgs, calculateMD5 } from "./image.js";
import { htmlParser, saveMateInfo } from "./txt.js";
import { saveAllCSS } from "./style.js";

const router = express.Router();

// middleware that is specific to this router
router.use(function timeLog(req, res, next) {
  next();
});
// define the home page route
router.get("/", async function (req, res) {
  res.send("epub types");
});

// define the about route
router.get("/about", function (req, res) {
  res.send("About types");
});

router.get("/html/:fileId", async function (req, res) {
  const fileId = req.params.fileId; // 获取 fileId 参数
  logger.info(`Found ${fileId}`);
  const fileRow = await getFileBymd5(fileId);

  if (!fileRow) {
    return res.status(404).send("文件查询失败");
  }
  const uploadPath =
    "./base_files/" + fileRow.book_id + "/Text/" + fileId + ".html";
  console.log(46, fileRow);

  const filePath = path.resolve(uploadPath);
  // 检查文件是否存在
  if (!fs.existsSync(filePath)) {
    return res.status(404).send("服务器中不存在该文件");
  }

  // 返回文件
  res.setHeader("Content-Type", fileRow.mimetype);
  res.sendFile(filePath);
});

router.get("/img/:fileId", async function (req, res) {
  const fileId = req.params.fileId; // 获取 fileId 参数
  logger.info(`Found ${fileId}`);
  console.log(6262626, fileId);

  const fileRow = await getFileBymd5(fileId);
  console.log(6565, fileRow);

  if (!fileRow) {
    return res.status(404).send("文件查询失败");
  }
  const uploadPath = "./base_files/" + fileRow.book_id + "/image/" + fileId;
  const filePath = path.resolve(uploadPath);
  console.log(727272, filePath, fs.existsSync(filePath));

  // 检查文件是否存在
  if (!fs.existsSync(filePath)) {
    return res.status(404).send("服务器中不存在该文件");
  }

  // 返回文件
  res.setHeader("Content-Type", fileRow.mimetype);
  res.sendFile(filePath);
});

router.get("/css/:fileId", async function (req, res) {
  const fileId = req.params.fileId; // 获取 fileId 参数
  logger.info(`Found ${fileId}`);
  const fileRow = await getFileBymd5(fileId);

  if (!fileRow) {
    return res.status(404).send("文件查询失败");
  }
  const uploadPath =
    "./base_files/" + fileRow.book_id + "/style/" + fileId + ".css";

  const filePath = path.resolve(uploadPath);
  // 检查文件是否存在
  if (!fs.existsSync(filePath)) {
    return res.status(404).send("服务器中不存在该文件");
  }

  // 返回文件
  res.setHeader("Content-Type", fileRow.mimetype);
  res.sendFile(filePath);
});

// define the about route
router.put("/", async function (req, res) {
  let sampleFile;
  let uploadPath;
  let epubData;
  let zipEpubExtract;
  let epubFilePath;
  let epub;

  if (!req.files || Object.keys(req.files).length === 0) {
    return res.status(400).send("No files were uploaded.");
  }

  sampleFile = req.files.file;

  let file_md5 = sampleFile.md5;
  uploadPath = `./base_files/${file_md5}/`;
  epubFilePath = uploadPath + sampleFile.md5 + ".epub";
  zipEpubExtract = uploadPath + "epub-extract/";

  dirExists(uploadPath);

  await waittime(200);

  const isFile = isFileSync(epubFilePath);

  // 移动上传文件至指定目录
  if (!isFile) {
    sampleFile.mv(epubFilePath, function (err) {
      console.log("移动上传文件至指定目录的反馈", err);
    });
    epubData = sampleFile.data;
  } else {
    epubData = fs.readFileSync(epubFilePath);
    file_md5 = await calculateMD5(epubFilePath);
  }

  /* 是否需要解压文件 */
  if (!(await isDir(zipEpubExtract))) {
    epub = await EPub.createAsync(epubData, null, "");
    dirExists(zipEpubExtract);
    epub.zip.admZip.extractAllTo(zipEpubExtract);
  } else {
    epub = await EPub.createAsync(epubData, null, "");
  }

  res.send("About types");
  console.log(137, typeof epub.metadata.creator, epub.metadata.creator);
  // 作者
  // const author_id = await author_insert()

  let authorInfo = await get_author_info(epub.metadata.creator);
  let author_id = authorInfo.author_id;
  if (!authorInfo) {
    author_id = uuidv4();
    await author_insert({ name: epub.metadata.creator, author_id: author_id });
  }
  /*
    1、读取图片信息
    2、替换文件中的图片内容
    3、存储html数据
    4、存储css数据
   */
  await saveMateInfo(epub, uploadPath, file_md5, author_id)
  await saveImgs(epub, uploadPath, file_md5, author_id);
  await saveAllCSS(epub, uploadPath, file_md5, author_id);
  // 存储html数据
  await htmlParser(epub, zipEpubExtract, file_md5, author_id);
  console.log('书籍处理完毕')
});

export default router;