From 1eb26d8e58efb3cf4bca2271308ccdc552250665 Mon Sep 17 00:00:00 2001
From: 07akioni <07akioni2@gmail.com>
Date: Sat, 19 Sep 2020 11:55:30 +0800
Subject: [PATCH] refactor(code): support vue3

---
 src/code/index.js                  |  2 +-
 src/code/src/{Code.vue => Code.js} | 34 ++++++++++++++++++------------
 2 files changed, 21 insertions(+), 15 deletions(-)
 rename src/code/src/{Code.vue => Code.js} (69%)

diff --git a/src/code/index.js b/src/code/index.js
index d35548c8a..b291bc3bf 100644
--- a/src/code/index.js
+++ b/src/code/index.js
@@ -1,5 +1,5 @@
 /* istanbul ignore file */
-import NCode from './src/Code.vue'
+import NCode from './src/Code.js'
 
 NCode.install = function (app, naive) {
   app.component(naive.componentPrefix + NCode.name, NCode)
diff --git a/src/code/src/Code.vue b/src/code/src/Code.js
similarity index 69%
rename from src/code/src/Code.vue
rename to src/code/src/Code.js
index 6fa4523b0..628ad41fa 100644
--- a/src/code/src/Code.vue
+++ b/src/code/src/Code.js
@@ -1,9 +1,9 @@
-<script>
-import { h } from 'vue'
+import { h, nextTick } from 'vue'
 import withapp from '../../_mixins/withapp'
 import themeable from '../../_mixins/themeable'
 import usecssr from '../../_mixins/usecssr'
 import styles from './styles/index.js'
+import { warn } from '../../_utils/naive/warn'
 
 export default {
   name: 'Code',
@@ -32,15 +32,15 @@ export default {
   },
   watch: {
     language () {
-      this.$nextTick(this.setCode)
+      nextTick(this.setCode)
     },
     code () {
-      this.$nextTick(this.setCode)
+      nextTick(this.setCode)
     }
   },
   created () {
-    if (!this.hljs && !this.$naive.hljs) {
-      console.error('[naive-ui/code]: hljs is not set.')
+    if (__DEV__ && !this.hljs && !this.$naive.hljs) {
+      warn('code', 'hljs is not set.')
     }
   },
   mounted () {
@@ -59,25 +59,32 @@ export default {
       }
     },
     setCode () {
-      if (this.language) {
+      const {
+        code,
+        language
+      } = this
+      if (language) {
         const {
           valid,
           content
-        } = this.generateCodeHTML(this.language, this.code, this.trim)
+        } = this.generateCodeHTML(language, code, this.trim)
         if (valid) {
           this.$refs.code.innerHTML = content
           return
         }
       }
-      this.$refs.code.textContent = this.code
+      this.$refs.code.textContent = code
     }
   },
   render () {
+    const { syntheticTheme } = this
     return h('pre', {
-      staticClass: 'n-code',
-      class: {
-        [`n-${this.syntheticTheme}-theme`]: this.syntheticTheme
-      }
+      class: [
+        'n-code',
+        {
+          [`n-${syntheticTheme}-theme`]: syntheticTheme
+        }
+      ]
     }, [
       h('code', {
         ref: 'code'
@@ -85,4 +92,3 @@ export default {
     ])
   }
 }
-</script>