slidev-addon-sm/bin/addwater.mjs

slidev-addon slidev sm

#!/usr/bin/env node

import { link } from "fs";
import { $, cd, fs, path } from "zx";

const PAGE_DIR = "./pages";

const baseName = (name) => {
  let t = name.split("-");
  if (t.length > 1) return t[1];
  return name;
};

try {
  //检索pages下的md文件，并以序号排序
  let subDirs = fs.readdirSync(PAGE_DIR);

  subDirs.forEach((subDir) => {
    let subDirPath = PAGE_DIR + "/" + subDir;
    let subFiles = fs
      .readdirSync(subDirPath)
      .filter((t) => path.extname(t) === ".md");
    subFiles.forEach((subFile) => {
      let subFileName = subFile.toString();
      if (subFileName === "README.md") return;
      if (subFileName.includes("_publish.md")) return;
      if (subFileName.includes("作业")) return;
      if (subFileName.includes("小测")) return;

      let found = false;
      let idx = 1;
      let dataArray = new Array();
      let firstLine = true;
      let hadlayout = false;

      let filePath = subDirPath+"/"+subFileName;
      let orderData = fs.readFileSync(filePath);
      orderData
        .toString()
        .split("\n")
        .forEach((data) => {
          let dataStr = data.toString().trim();

          // 每页的头也加上
          if(firstLine) {
            if(dataStr.trim().length == 0) {
              return;
            }
            if(!(dataStr.trim().slice(0,3).includes("---"))) {
              dataArray[idx++] = "---"
              dataArray[idx++] = "layout: SmLogoTRLayout";
              dataArray[idx++] = "---\n";

              let subFileBaseName = baseName(subFileName);
              subFileBaseName = subFileBaseName.replace(".md", "");
              
              if(!dataStr.includes("##")) {
                dataArray[idx++] = "## "+subFileBaseName+"\n";
                dataArray[idx++] = "<Toc maxDepth=\"3\" minDepth=\"3\"/>\n";
              }

              dataArray[idx++] = dataStr;
              firstLine = false;
              return;
            }
            firstLine = false;
          }

          if(found) {
            if(!dataStr.includes("layout") && !hadlayout) {
              dataArray[idx++] = "layout: SmLogoTRLayout";
              dataArray[idx++] = "---\n";
              found = false;
              hadlayout = false;
              return;
            } else {
              if(dataStr.includes("layout")) {
                hadlayout = true;
              }

              dataArray[idx++] = dataStr;
              if(dataStr.slice(0,3).includes("---")) {
                found = false;
                hadlayout = false;
              }
            }
            return;
          }

          if (dataStr.slice(0,3).includes("---")) {
            dataArray[idx++] = dataStr;
            found = true;
          } else {
            dataArray[idx++] = dataStr;
          }
        });
        fs.writeFileSync(filePath, dataArray.join("\n").trim());
    });
  });
} catch (err) {
  console.error(err);
}