diff --git a/src/components/d2-container/components/d2-source.vue b/src/components/d2-container/components/d2-source.vue
index 35e4353c..529b99dc 100644
--- a/src/components/d2-container/components/d2-source.vue
+++ b/src/components/d2-container/components/d2-source.vue
@@ -1,5 +1,6 @@
 <template>
   <div
+    v-if="active"
     class="d2-source"
     :class="{ 'd2-source--active': isActive }"
     @click="handleClick">
@@ -21,6 +22,13 @@ export default {
       isActive: false
     }
   },
+  computed: {
+    active () {
+      return process.env.VUE_APP_BUILD_MODE === 'TRAVIS' ||
+        process.env.NODE_ENV === 'development' &&
+        this.filename
+    }
+  },
   mounted () {
     // 一秒后显示按钮
     setTimeout(() => {
diff --git a/src/components/d2-container/index.js b/src/components/d2-container/index.js
index e03b6c59..e9e6ec23 100644
--- a/src/components/d2-container/index.js
+++ b/src/components/d2-container/index.js
@@ -54,10 +54,9 @@ export default {
           scroll: e => this.$emit('scroll', e)
         }
       }, slots),
-      // 只在预览和开发模式下显示源码按钮
-      process.env.VUE_APP_BUILD_MODE === 'TRAVIS' || process.env.NODE_ENV === 'development' ? h(d2Source, {
+      h(d2Source, {
         props: this.$attrs
-      }) : undefined
+      })
     ])
   },
   methods: {