Provide sourcePath to generate

Author: charlybrown03

src/generate.ts

@@ -1,3 +1,6 @@
+import path from 'path'
+import fs from 'fs'
+
 import JSON5 from 'json5'
 import yaml from 'js-yaml'
 import { debug as Debug } from 'debug'
@@ -12,13 +15,17 @@ const defaultLang = 'json'
 
 const debug = Debug('vue-i18n-jest')
 
-export default function generate (blocks: SFCBlock[]) {
+export default function generate (blocks: SFCBlock[], sourcePath: string) {
   const base = `${VUE_OPTIONS}.${VUE_I18N_OPTION} = []`
   const codes = blocks.map(block => {
     if (block.type === 'i18n') {
       const lang = (block.attrs && block.attrs.lang) || defaultLang
       // const lang = block.attrs?lang ?? defaultLang
-      const data = convert(block.content, lang)
+      const content = (block.attrs && block.attrs.src)
+        ? fs.readFileSync(getAbsolutePath(block, sourcePath)).toString()
+        : block.content
+
+      const data = convert(content, lang)
       const value = JSON.stringify(JSON.parse(data))
         .replace(/\u2028/g, '\\u2028')
         .replace(/\u2029/g, '\\u2029')
@@ -45,3 +52,8 @@ function convert (source: string, lang: string): string {
       return source
   }
 }
+
+function getAbsolutePath (block: SFCBlock, sourcePath: string): string {
+  if (path.isAbsolute(block.attrs.src)) return block.attrs.src
+  return path.join(path.dirname(sourcePath), block.attrs.src)
+}

src/transform.ts

@@ -43,7 +43,7 @@ function jestProcess (
 
   let coding = ';\n'
   if (customBlocks) {
-    coding += generate(customBlocks)
+    coding += generate(customBlocks, sourcePath)
   }
 
   const retCode = code + coding

test/generate.test.ts

@@ -4,15 +4,20 @@ import { parseComponent } from 'vue-template-compiler'
 import generate from '../src/generate'
 
 const files = ['default.vue', 'json.vue', 'yaml.vue', 'json5.vue']
-const targets = files.map(file => ({
-  content: fs.readFileSync(path.resolve(__dirname, `./fixtures/${file}`)).toString(),
-  file
-}))
+const targets = files.map(file => {
+  const sourcePath = path.resolve(__dirname, `./fixtures/${file}`)
+
+  return {
+    content: fs.readFileSync(sourcePath).toString(),
+    sourcePath,
+    file
+  }
+})
 
 test('generate', () => {
   targets.forEach(target => {
     const { customBlocks } = parseComponent(target.content, { pad: true })
-    const code = generate(customBlocks)
+    const code = generate(customBlocks, target.sourcePath)
     expect(code).toMatchSnapshot(target.file)
   })
 })