feat: add yaml interop for require

bem · May 30, 2020 · f396323 · f396323
1 parent 1cd7905
commit f396323
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 25 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -36,7 +36,8 @@
     "ora": "4.0.4",
     "style-dictionary": "2.10.0",
     "ts-node": "8.8.2",
-    "typescript": "3.8.3"
+    "typescript": "3.8.3",
+    "yaml": "1.10.0"
   },
   "devDependencies": {
     "@commitlint/cli": "8.3.5",

diff --git a/src/core-v2/build.ts b/src/core-v2/build.ts
@@ -1,3 +1,4 @@
+import './yaml-interop'
 import StyleDictionaryApi from 'style-dictionary'
 
 import { createWhitepaperConfig } from './whitepaper-config'

diff --git a/src/core-v2/yaml-interop.ts b/src/core-v2/yaml-interop.ts
@@ -0,0 +1,13 @@
+import Module from 'module'
+import { readFileSync } from 'fs'
+import YAML from 'yaml'
+
+Module.prototype.require = new Proxy(Module.prototype.require, {
+  apply(target, thisArg, args) {
+    if (/\.ya?ml$/.test(args[0])) {
+      const file = readFileSync(args[0], 'utf8')
+      return YAML.parse(file)
+    }
+    return Reflect.apply(target, thisArg, args)
+  },
+})