From 45603a95c0963c45aabaa35a4485a3d6fff8d762 Mon Sep 17 00:00:00 2001
From: Hunlongyu <hunlongyu@gmail.com>
Date: Tue, 10 Dec 2019 14:09:00 +0800
Subject: [PATCH] database done

---
 src/App.vue                | 19 ++++++++-------
 src/background.js          |  1 +
 src/plugin/nedb/index.js   | 38 +++++++----------------------
 src/plugin/nedb/setting.js |  4 +---
 src/plugin/nedb/video.js   | 49 ++++++++++++++++++++++++++++++++++++++
 src/store.js               | 12 ++++++----
 src/views/Settings.vue     | 11 ++++++---
 7 files changed, 86 insertions(+), 48 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index 0631308..3371baf 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -1,5 +1,5 @@
 <template>
-  <div id="app" :class="getTheme">
+  <div id="app" :class="getTheme.color">
     <Layout class="box">
         <Sider class="sider" width="70"><ZYSider /></Sider>
         <Layout>
@@ -15,6 +15,7 @@
 import { mapGetters, mapActions } from 'vuex'
 import ZYSider from '@/components/zy_sider.vue'
 import ZYHeader from '@/components/zy_header.vue'
+import setting from './plugin/nedb/setting'
 export default {
   name: 'app',
   data () {
@@ -36,13 +37,15 @@ export default {
   },
   beforeCreate () {},
   created () {
-    // db.find('theme').then(e => {
-    //   if (!e) {
-    //     this.changeTheme('light')
-    //   } else {
-    //     this.changeTheme(e)
-    //   }
-    // })
+    setting.find({ $or: [{ theme: 'light' }, { theme: 'dark' }] }).then(e => {
+      if (e.length <= 0) {
+        setting.add({ theme: 'light' }).then(res => {
+          this.changeTheme({ id: res._id, color: res.theme })
+        })
+      } else {
+        this.changeTheme({ id: e[0]._id, color: e[0].theme })
+      }
+    })
   }
 }
 </script>
diff --git a/src/background.js b/src/background.js
index ee8d3ac..8429c19 100644
--- a/src/background.js
+++ b/src/background.js
@@ -26,6 +26,7 @@ function createWindow () {
     }
   })
 
+  
   if (process.env.WEBPACK_DEV_SERVER_URL) {
     // Load the url of the dev server if in development mode
     win.loadURL(process.env.WEBPACK_DEV_SERVER_URL)
diff --git a/src/plugin/nedb/index.js b/src/plugin/nedb/index.js
index 52e35d4..4bc8ebd 100644
--- a/src/plugin/nedb/index.js
+++ b/src/plugin/nedb/index.js
@@ -1,42 +1,20 @@
-import jetpack from 'fs-jetpack'
-import fs from 'fs'
 import Nedb from 'nedb'
-import { remote } from 'electron'
 
 export default class {
   constructor () {
     this.db = null
-    this.dataDir = jetpack.cwd(remote.app.getPath('appData'))
   }
 
-  createOrReadDB (db) {
-    const dirName = process.env.NODE_ENV === 'development' ? '.ZY-dev' : '.ZY'
-    const existsDir = jetpack.exists(this.dataDir.path(dirName))
-    if (!existsDir) {
-      fs.mkdir(this.dataDir.path(`${dirName}`), (err) => {
-        if (err) console.log(err)
-      })
-    }
-
-    const eSetting = fs.existsSync(this.dataDir.path(`${dirName}/${db.setting}`))
-    const eVideo = fs.existsSync(this.dataDir.path(`${dirName}/${db.video}`))
-    let database = {}
-
-    if (!eSetting) {
-      this.dataDir.write(this.dataDir.path(`${dirName}/${db.setting}`), '')
-    }
-
-    if (!eVideo) {
-      this.dataDir.write(this.dataDir.path(`${dirName}/${db.video}`), '')
-    }
+  create (db) {
+    const name = process.env.NODE_ENV === 'development' ? 'ZY-dev' : 'ZY'
+    const database = {}
 
     database.setting = new Nedb({
-      filename: this.dataDir.path(`${dirName}/${db.setting}`),
+      filename: name + db.setting,
       autoload: true
     })
-
     database.video = new Nedb({
-      filename: this.dataDir.path(`${dirName}/${db.video}`),
+      filename: name + db.video,
       autoload: true
     })
 
@@ -48,9 +26,9 @@ export default class {
       return this.db
     }
 
-    this.db = this.createOrReadDB({
-      setting: 'setting.db',
-      video: 'video.db'
+    this.db = this.create({
+      setting: '-setting',
+      video: '-video'
     })
 
     return this.db
diff --git a/src/plugin/nedb/setting.js b/src/plugin/nedb/setting.js
index 75f95e9..59fcc0e 100644
--- a/src/plugin/nedb/setting.js
+++ b/src/plugin/nedb/setting.js
@@ -4,8 +4,6 @@ const db = new DB()
 const connect = db.init()
 const setting = connect.setting
 
-console.log(setting, 'setting')
-
 export default {
   add (data) {
     return new Promise((resolve, reject) => {
@@ -25,7 +23,7 @@ export default {
   },
   update (id, data) {
     return new Promise((resolve, reject) => {
-      setting.find({ _id: id }, { $set: data }, (err, docs) => {
+      setting.update({ _id: id }, { $set: data }, (err, docs) => {
         if (err) { reject(err) }
         resolve(docs)
       })
diff --git a/src/plugin/nedb/video.js b/src/plugin/nedb/video.js
index e69de29..e93bcb8 100644
--- a/src/plugin/nedb/video.js
+++ b/src/plugin/nedb/video.js
@@ -0,0 +1,49 @@
+// import DB from './index'
+import DB from './index'
+
+const db = new DB()
+const connect = db.init()
+const video = connect.video
+
+export default {
+  add (data) {
+    return new Promise((resolve, reject) => {
+      video.insert(data, (err, docs) => {
+        if (err) { reject(err) }
+        resolve(docs)
+      })
+    })
+  },
+  find (data) {
+    return new Promise((resolve, reject) => {
+      video.find(data, (err, docs) => {
+        if (err) { reject(err) }
+        resolve(docs)
+      })
+    })
+  },
+  update (id, data) {
+    return new Promise((resolve, reject) => {
+      video.find({ _id: id }, { $set: data }, (err, docs) => {
+        if (err) { reject(err) }
+        resolve(docs)
+      })
+    })
+  },
+  remove (id) {
+    return new Promise((resolve, reject) => {
+      video.remove({ _id: id }, {}, (err, docs) => {
+        if (err) { reject(err) }
+        resolve(docs)
+      })
+    })
+  },
+  removeAll () {
+    return new Promise((resolve, reject) => {
+      video.remove({}, { multi: true }, (err, docs) => {
+        if (err) { reject(err) }
+        resolve(docs)
+      })
+    })
+  }
+}
diff --git a/src/store.js b/src/store.js
index b72f3cd..ad5bf91 100644
--- a/src/store.js
+++ b/src/store.js
@@ -1,11 +1,15 @@
 import Vue from 'vue'
 import Vuex from 'vuex'
+import setting from '@/plugin/nedb/setting'
 
 Vue.use(Vuex)
 
 export default new Vuex.Store({
   state: {
-    theme: 'light',
+    theme: {
+      id: '',
+      color: 'light'
+    },
     iconActive: 'search',
     video: {}
   },
@@ -36,9 +40,9 @@ export default new Vuex.Store({
       // localStorage.collection = payload
     },
     changeTheme: ({ commit }, payload) => {
-      // setting.set('theme', payload).then(e => {
-      //   commit('SET_THEME', e)
-      // })
+      setting.update(payload.id, { theme: payload.color }).then(res => {
+        commit('SET_THEME', payload)
+      })
     }
   }
 })
diff --git a/src/views/Settings.vue b/src/views/Settings.vue
index 3c5ff11..c89c1a7 100644
--- a/src/views/Settings.vue
+++ b/src/views/Settings.vue
@@ -1,15 +1,20 @@
 <template>
   <Row>
-    <Button type="primary" @click="changeTheme('light')">light</Button>
-    <Button type="primary" @click="changeTheme('dark')">Dark</Button>
+    <Button type="primary" @click="changeTheme({ id: getTheme.id, color: 'light' })">light</Button>
+    <Button type="primary" @click="changeTheme({ id: getTheme.id, color: 'dark' })">Dark</Button>
     <Button type="primary" @click="add()">Add</Button>
   </Row>
 </template>
 <script>
-import { mapActions } from 'vuex'
+import { mapGetters, mapActions } from 'vuex'
 import settingDB from '@/plugin/nedb/setting'
 export default {
   name: 'settings',
+  computed: {
+    ...mapGetters([
+      'getTheme'
+    ])
+  },
   methods: {
     ...mapActions([
       'changeTheme'