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";
import {saveAllFount} from "./font.js";
import {saveToc} from "./toc.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}`);
    
    const fileRow = await getFileBymd5(fileId);
    
    if (!fileRow) {
        return res.status(404).send("文件查询失败");
    }
    const uploadPath = "./base_files/" + fileRow.book_id + "/image/" + fileId;
    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("/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, "");
    }
    
    // 生成作者的数据
    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数据
      5、存储章节数据
     */
    /*await saveMateInfo(epub, uploadPath, file_md5, author_id)
    logger.info('书籍的基础数据处理完毕')
    await saveImgs(epub, uploadPath, file_md5, author_id);
    logger.info('书籍的图片数据处理完毕')
    await saveAllFount(epub, uploadPath, file_md5, author_id);
    logger.info('书籍的字体数据处理完毕')
    await saveAllCSS(epub, uploadPath, file_md5, author_id);
    logger.info('书籍的css数据处理完毕')
    // 存储html数据
    await htmlParser(epub, zipEpubExtract, file_md5, author_id);
    logger.info('书籍的章节数据处理完毕')*/
    // 章节数据
    await saveToc(epub, zipEpubExtract, file_md5, author_id);
    logger.info('书籍的目录处理完毕')
    logger.info('书籍处理完毕')
    res.send("书籍处理完毕");
});

export default router;