From 0cac0bda3827ad00c9f5d5699dd0a8297dc33c85 Mon Sep 17 00:00:00 2001
From: 0song <82012629+0song@users.noreply.github.com>
Date: Thu, 3 Jun 2021 20:37:31 +0800
Subject: [PATCH] chore: update doc template (#2142)

---
 website/components/demo-block.vue | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/website/components/demo-block.vue b/website/components/demo-block.vue
index c954ebad41..55680412e2 100644
--- a/website/components/demo-block.vue
+++ b/website/components/demo-block.vue
@@ -77,7 +77,8 @@ const version = '1.0.0' // element version
 const stripTemplateAndRemoveTemplate = code => {
   const result = removeSetup(stripTemplate(code))
   if (result.indexOf('<template>') === 0) {
-    return result.replace(/^<template>/, '').replace(/<\/template>$/,'')
+    const html = result.replace(/^<template>/, '').replace(/<\/template>$/,'')
+    return html.replace(/^[\r?\n|\r]/, '').replace(/[\r?\n|\r]$/, '').trim()
   }
   return result
 }
@@ -196,12 +197,15 @@ export default {
       nextTick(() => {
         const highlight = this.$el.querySelector('.highlight')
         const hlcode = highlight.querySelector('pre code')
-        hlcode.innerHTML = sanitizeHTML(`<template>${this.codepen.html}</template>
-
-<script>
+        const innerScript = `<script>
   ${this.displayDemoCode}
 ${'</sc' + 'ript>'}
-`)
+`
+        hlcode.innerHTML = sanitizeHTML(`<template>
+  ${this.codepen.html}
+</template>
+
+${this.displayDemoCode ? innerScript : ''}`)
 
         nextTick(() => {
           if (this.$el.getElementsByClassName('description').length === 0) {