Trim templates before pug compiling (vuejs#86)

ChenGuangjun · ChenGuangjun · commit bd9a7c4a48fa · 2017-05-10T18:31:07.000+08:00
diff --git a/src/template/pug.js b/src/template/pug.js
@@ -1,6 +1,7 @@
 export default async function (template, extras, options) {
     const pug = require('pug')
-    const compiler = pug.compile(template, { filename: extras.id, ...options.pug })
+    const trim = typeof template === 'string' ? template.trim : template
+    const compiler = pug.compile(trim, { filename: extras.id, ...options.pug })
 
     return compiler({css: extras.modules || {}})
 }
diff --git a/test/fixtures/pug.vue b/test/fixtures/pug.vue
@@ -1,5 +1,5 @@
 <template lang="pug">
-div(class=css.test class='keep-me') foo
+  div(class=css.test class='keep-me') foo
 </template>
 
 <script lang="babel">

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,7 @@`
`1`	`1`	`export default async function (template, extras, options) {`
`2`	`2`	`const pug = require('pug')`
`3`		`- const compiler = pug.compile(template, { filename: extras.id, ...options.pug })`
	`3`	`+ const trim = typeof template === 'string' ? template.trim : template`
	`4`	`+ const compiler = pug.compile(trim, { filename: extras.id, ...options.pug })`
`4`	`5`
`5`	`6`	`return compiler({css: extras.modules \|\| {}})`
`6`	`7`	`}`