From 17ba3191572cf4bfd0e63a90d9155b186308b399 Mon Sep 17 00:00:00 2001
From: haiyangcui <cuiocean@gmail.com>
Date: Sun, 13 Sep 2020 08:44:34 +0200
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0"=E5=AF=BC=E5=85=A5"=20"?=
 =?UTF-8?q?=E5=AF=BC=E5=87=BA"=20"=E9=87=8D=E7=BD=AE"?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/IPTV.vue   | 98 ++++++++++++++++++++++++++++++++-------
 src/lib/dexie/dexie.js    |  2 +-
 src/lib/dexie/initData.js | 20 ++++----
 src/lib/dexie/iptv.js     |  3 ++
 4 files changed, 97 insertions(+), 26 deletions(-)

diff --git a/src/components/IPTV.vue b/src/components/IPTV.vue
index e21ccca..07c8c40 100644
--- a/src/components/IPTV.vue
+++ b/src/components/IPTV.vue
@@ -3,7 +3,16 @@
     <div class="detail-content">
       <div class="detail-header">
         <div class="zy-select">
-            <div class="vs-placeholder vs-noAfter" @click="openAddSite">添加新源</div>
+            <div class="vs-placeholder vs-noAfter" @click="openAddSite">添加新频道</div>
+        </div>
+        <div class="zy-select">
+          <div class="vs-placeholder vs-noAfter" @click="exportSites">导出</div>
+        </div>
+        <div class="zy-select">
+          <div class="vs-placeholder vs-noAfter" @click="importSites">导入</div>
+        </div>
+        <div class="zy-select">
+          <div class="vs-placeholder vs-noAfter" @click="resetSites">重置</div>
         </div>
       </div>
       <div class="detail-body zy-scroll">
@@ -13,7 +22,7 @@
               <ul>
                 <li >
                   <span class="name">频道名</span>
-                  <span class="name">m3u8接口</span>
+                  <span class="name">Url</span>
                   <span class="operate">
                     <span class="btn"></span>
                     <span class="btn"></span>
@@ -24,7 +33,7 @@
                     <input style="height: 30px" v-model="newSite.name">
                   </span>
                   <span class="name" style="display:inline-block;vertical-align:middle">
-                    <input style="height: 30px" v-model="newSite.api">
+                    <input style="height: 30px" v-model="newSite.url">
                   </span>
                   <span class="operate">
                     <span class="btn" @click="addNewSite">添加</span>
@@ -35,9 +44,9 @@
                </ul>
              </div>
             <ul>
-              <draggable v-model="iptv" @change="listUpdatedEvent">
+              <draggable v-model="iptvList" @change="listUpdatedEvent">
                 <transition-group>
-                  <li v-for="(i, j) in iptv" :key="j">
+                  <li v-for="(i, j) in iptvList" :key="j" @click.stop="playEvent(i)">
                     <span class="name">{{i.name}}</span>
                     <span class="operate">
                       <span class="btn" @click.stop="playEvent(i)">播放</span>
@@ -57,11 +66,14 @@
 import { mapMutations } from 'vuex'
 import { iptv } from '../lib/dexie'
 import draggable from 'vuedraggable'
+import { iptv as defaultSites } from '../lib/dexie/initData'
+import { remote } from 'electron'
+import fs from 'fs'
 export default {
   name: 'iptv',
   data () {
     return {
-      iptv: [],
+      iptvList: [],
       showAddSite: false,
       newSite:
       {
@@ -88,13 +100,13 @@ export default {
   },
   watch: {
     view () {
-      this.getAllIptv()
+      this.getAllSites()
     }
   },
   methods: {
     ...mapMutations(['SET_VIEW', 'SET_DETAIL', 'SET_VIDEO', 'SET_SHARE']),
     playEvent (e) {
-      var m3u8Link = e.site
+      var m3u8Link = e.url
       const fs = require('fs')
       var externalPlayer = this.setting.externalPlayer
       if (!fs.existsSync(externalPlayer)) {
@@ -110,7 +122,7 @@ export default {
     },
     removeEvent (e) {
       iptv.remove(e.id).then(res => {
-        this.getAllIptv()
+        this.getAllSites()
       }).catch(err => {
         this.$message.warning('删除频道失败, 错误信息: ' + err)
       })
@@ -122,42 +134,94 @@ export default {
       this.showAddSite = true
     },
     addNewSite () {
-      if (!this.newSite.name || !this.newSite.site) {
+      if (!this.newSite.name || !this.newSite.url) {
         this.$message.error('名称和API接口不能为空。')
         return
       }
       var doc = {
         name: this.newSite.name,
-        site: this.newSite.site
+        url: this.newSite.url
       }
       iptv.add(doc).then(res => {
         this.newSite = {
           name: '',
-          site: ''
+          url: ''
         }
         this.$message.success('添加新源成功！')
-        this.getAllIptv()
+        this.getAllSites()
       })
     },
     listUpdatedEvent () {
       iptv.clear().then(res1 => {
         // 重新排序
         var id = 1
-        this.iptv.forEach(element => {
+        this.iptvList.forEach(element => {
           element.id = id
           iptv.add(element)
           id += 1
         })
       })
     },
-    getAllIptv () {
+    exportSites () {
+      this.getAllSites()
+      const arr = [...this.iptvList]
+      const str = JSON.stringify(arr, null, 4)
+      const options = {
+        filters: [
+          { name: 'JSON file', extensions: ['json'] },
+          { name: 'Normal text file', extensions: ['txt'] },
+          { name: 'All types', extensions: ['*'] }
+        ]
+      }
+      remote.dialog.showSaveDialog(options).then(result => {
+        if (!result.canceled) {
+          fs.writeFileSync(result.filePath, str)
+          this.$message.success('已保存成功')
+        }
+      }).catch(err => {
+        this.$message.error(err)
+      })
+    },
+    importSites () {
+      const options = {
+        filters: [
+          { name: 'JSON file', extensions: ['json'] },
+          { name: 'Normal text file', extensions: ['txt'] },
+          { name: 'All types', extensions: ['*'] }
+        ],
+        properties: ['openFile']
+      }
+      remote.dialog.showOpenDialog(options).then(result => {
+        if (!result.canceled) {
+          iptv.clear()
+          result.filePaths.forEach(file => {
+            var str = fs.readFileSync(file)
+            const json = JSON.parse(str)
+            iptv.bulkAdd(json).then(e => {
+              this.getAllSites()
+            })
+            this.$message.success('导入成功')
+          }).catch(err => {
+            this.$message.error(err)
+          })
+        }
+      })
+    },
+    resetSites () {
+      iptv.clear()
+      iptv.bulkAdd(defaultSites).then(e => {
+        this.getAllSites()
+        this.$message.success('重置成功')
+      })
+    },
+    getAllSites () {
       iptv.all().then(res => {
-        this.iptv = res
+        this.iptvList = res
       })
     }
   },
   created () {
-    this.getAllIptv()
+    this.getAllSites()
   }
 }
 </script>
diff --git a/src/lib/dexie/dexie.js b/src/lib/dexie/dexie.js
index 7aeea46..2135a2f 100644
--- a/src/lib/dexie/dexie.js
+++ b/src/lib/dexie/dexie.js
@@ -11,7 +11,7 @@ db.version(3).stores({
   sites: '++id, key, name, json, xml, down, level',
   history: '++id, site, ids, name, type, year, index, time',
   mini: 'id, site, ids, name, index, time',
-  iptv: '++id, name, site'
+  iptv: '++id, name, url'
 })
 
 db.on('populate', () => {
diff --git a/src/lib/dexie/initData.js b/src/lib/dexie/initData.js
index 03da965..1f9fc10 100644
--- a/src/lib/dexie/initData.js
+++ b/src/lib/dexie/initData.js
@@ -282,24 +282,28 @@ const getSite = (key) => {
 
 const iptv = [
   {
-    name: 'CCTV 1',
-    site: 'http://111.40.205.87/PLTV/88888888/224/3221225710/index.m3u8'
+    name: '東森新聞',
+    url: 'http://lss.line-scdn.net/cc_tw/p/live/hPFcU2yqYDmgJQRB4NlQTX2UtVl4kHRpsax1DCy8sUQ0jTx07Zh9CB38rVlgmGU05YEpIWnosVF82HRo3Z0JBDH0yUVhsH0o9fU5GEHIuWllyHUE-/720/chunklist.m3u8'
   },
   {
-    name: 'CCTV 2',
-    site: 'http://117.148.187.37/PLTV/88888888/224/3221226138/index.m3u8'
+    name: '民視新闻',
+    url: 'http://lss.line-scdn.net/cc_tw/p/live/hrw-UR8uCLFlBQTJJeFQxbit8IG1tGG0JeB1hOmcsczxtTGkMeBlkOjMoIW9qT2kGKkpqazUkcGk8WWtcJE9ra2EsbWg8A2kMLlVkOCoscW82HGsPJUs/720/chunklist.m3u8'
   },
   {
-    name: 'CCTV 13',
-    site: 'http://117.148.187.37/PLTV/88888888/224/3221226193/index.m3u8'
+    name: 'TVBS新闻',
+    url: 'http://lss.line-scdn.net/cc_tw/p/live/hN97FnM2yEBdaUg4HYkcNIDFvHyJyXwQTZQ5ddH0_T3J3BlcQPwtWdX06HydyB1ASMFlWJS83TyIuSlcSPlxXJXpoUSYmEFVCNEZadjA2TS0nDldJNw/720/chunklist.m3u8'
   },
   {
     name: 'BBC News UK',
-    site: 'http://51.52.156.22:8888/http/004'
+    url: 'http://51.52.156.22:8888/http/004'
   },
   {
     name: 'Music Channel',
-    site: 'https://edge126.rcs-rds.ro/utvedge/musicchannelhq.stream/playlist.m3u8'
+    url: 'https://edge126.rcs-rds.ro/utvedge/musicchannelhq.stream/playlist.m3u8'
+  },
+  {
+    name: 'FOX Sport HD 1',
+    url: 'https://austchannel-live.akamaized.net/hls/live/2002736/austchannel-sport/master.m3u8'
   }
 ]
 
diff --git a/src/lib/dexie/iptv.js b/src/lib/dexie/iptv.js
index 22314e8..6a49027 100644
--- a/src/lib/dexie/iptv.js
+++ b/src/lib/dexie/iptv.js
@@ -8,6 +8,9 @@ export default {
     return await iptv.clear()
   },
   async add (doc) {
+    return await iptv.add(doc)
+  },
+  async bulkAdd (doc) {
     return await iptv.bulkAdd(doc)
   },
   async find (doc) {