parser option keep filename

dd-center · Sep 22, 2020 · 4844ad1 · 4844ad1
1 parent 976b396
commit 4844ad1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/index.js b/index.js
@@ -2,11 +2,17 @@ const { readdir } = require('fs').promises
 const parse = require('./parse')
 const { UUID_NAMESPACE, linkSyntax } = require('./config')
 
-module.exports = readdir('vtbs')
+const out = (filename = false) => readdir('vtbs')
   .then(dir => parse(({
     UUID_NAMESPACE,
     linkSyntax,
     vtbs: dir
       .map(file => file.replace('.json', ''))
       .map(name => ({ name, object: require(`./vtbs/${name}`) })),
-  })))
+  }), filename))
+
+const p = out()
+
+p.withFilename = () => out(true)
+
+module.exports = p
diff --git a/parse.js b/parse.js
@@ -1,12 +1,12 @@
 const { v5: uuidv5 } = require('uuid')
 
-module.exports = ({ vtbs, linkSyntax, UUID_NAMESPACE }) => ({
+module.exports = ({ vtbs, linkSyntax, UUID_NAMESPACE }, keepFilename = false) => ({
   meta: { UUID_NAMESPACE, linkSyntax },
   vtbs: vtbs
-    .map(({ name, object }) => ({ uuid: uuidv5(name, UUID_NAMESPACE), object }))
-    .map(({ uuid, object }) => ({ uuid, object: { ...object, name: Object.entries(object.name || {}) } }))
-    .map(({ uuid, object }) => ({ uuid, object: { ...object, accounts: Object.entries(object.accounts || {}) } }))
-    .map(({ uuid, object }) => {
+    .map(({ name, object }) => ({ uuid: uuidv5(name, UUID_NAMESPACE), object, name }))
+    .map(({ object, ...extra }) => ({ object: { ...object, name: Object.entries(object.name || {}) }, ...extra }))
+    .map(({ object, ...extra }) => ({ object: { ...object, accounts: Object.entries(object.accounts || {}) }, ...extra }))
+    .map(({ uuid, object, name }) => {
       const parsed = { uuid }
       parsed.type = object.type || 'vtuber'
       parsed.bot = !!object.bot
@@ -37,6 +37,10 @@ module.exports = ({ vtbs, linkSyntax, UUID_NAMESPACE }) => ({
         parsed.group = uuid
       }
 
+      if (keepFilename) {
+        parsed.filename = name
+      }
+
       return parsed
     }),
 })