feat: Generate setup scripts from metadata.

1 files changed, 62 insertions, 0 deletions

diff --git a/docs/src/setup-script-generation-plugin/index.js b/docs/src/setup-script-generation-plugin/index.js
new file mode 100644
index 0000000..e97a4d0
--- /dev/null
+++ b/docs/src/setup-script-generation-plugin/index.js
@@ -0,0 +1,62 @@
+/*
+ * Copyright (c) 2021 The ZMK Contributors
+ *
+ * SPDX-License-Identifier: MIT
+ */
+
+var PrebuildPlugin = require("prebuild-webpack-plugin");
+const fs = require("fs");
+const glob = require("glob");
+const yaml = require("js-yaml");
+const Mustache = require("mustache");
+
+function generateSetupScripts() {
+  return glob("../app/boards/**/*.zmk.yml", (error, files) => {
+    const aggregated = files.flatMap((f) =>
+      yaml.safeLoadAll(fs.readFileSync(f, "utf8"))
+    );
+
+    const data = aggregated.reduce(
+      (agg, item) => {
+        switch (item.type) {
+          case "shield":
+            item.compatible = true;
+            item.split = item.siblings?.length > 1;
+            agg.keyboards.push(item);
+            break;
+          case "board":
+            if (!item.features?.includes("keys")) {
+              agg.boards.push(item);
+            }
+            break;
+        }
+        return agg;
+      },
+      { keyboards: [], boards: [] }
+    );
+
+    for (let script_ext of ["sh", "ps1"]) {
+      const templateBuffer = fs.readFileSync(
+        `src/templates/setup.${script_ext}.mustache`,
+        "utf8"
+      );
+      const script = Mustache.render(templateBuffer, data);
+      fs.writeFileSync(`static/setup.${script_ext}`, script);
+    }
+  });
+}
+
+module.exports = function () {
+  return {
+    name: "setup-script-generation-plugin",
+    configureWebpack() {
+      return {
+        plugins: [
+          new PrebuildPlugin({
+            build: generateSetupScripts,
+          }),
+        ],
+      };
+    },
+  };
+};