添加直播源检测功能

package.json

@@ -32,6 +32,7 @@
     "html2canvas": "^1.0.0-rc.7",
     "iptv-playlist-parser": "^0.5.0",
     "m3u": "0.0.2",
+    "m3u8-parser": "^4.5.0",
     "memcached": "^2.2.2",
     "modern-normalize": "^1.0.0",
     "mousetrap": "^1.6.5",

src/components/IPTV.vue

@@ -4,12 +4,14 @@
         <el-switch v-model="enableBatchEdit" active-text="批处理分组"></el-switch>
         <el-button @click.stop="exportChannels" icon="el-icon-upload2" >导出</el-button>
         <el-button @click.stop="importChannels" icon="el-icon-download">导入</el-button>
+        <el-button @click="checkAllChannels" icon="el-icon-refresh" :loading="checkAllChannelsLoading">检测</el-button>
         <el-button @click.stop="removeAllChannels" icon="el-icon-delete-solid">清空</el-button>
         <el-button @click.stop="resetChannelsEvent" icon="el-icon-refresh-left">重置</el-button>
     </div>
     <div class="listpage-header" id="iptv-header" v-show="enableBatchEdit">
         <el-switch v-model="enableBatchEdit" active-text="批处理分组"></el-switch>
         <el-input placeholder="新组名" v-model="batchGroupName"></el-input>
+        <el-switch v-model="batchIsActive" :active-value="1" :inactive-value="0" active-text="启用"></el-switch>
         <el-button type="primary" icon="el-icon-edit" @click.stop="saveBatchEdit">保存</el-button>
     </div>
     <div class="listpage-body" id="iptv-table">
@@ -38,6 +40,18 @@
               </el-input>
             </template>
           </el-table-column>
+          <el-table-column
+            prop="isActive"
+            label="启用">
+            <template slot-scope="scope">
+              <el-switch
+                v-model="scope.row.isActive"
+                :active-value="1"
+                :inactive-value="0"
+                @change='isActiveChangeEvent(scope.row)'>
+              </el-switch>
+            </template>
+          </el-table-column>
           <el-table-column
             sort-by="['group', 'name']"
             sortable
@@ -51,6 +65,19 @@
               <el-button type="text">{{scope.row.group}}</el-button>
             </template>
           </el-table-column>
+          <el-table-column
+            label="状态"
+            sortable
+            :sort-by="['status']"
+            width="120">
+            <template slot-scope="scope">
+              <span v-if="scope.row.status === ' '">
+                <i class="el-icon-loading"></i>
+                检测中...
+              </span>
+              <span v-else>{{scope.row.status}}</span>
+            </template>
+          </el-table-column>
           <el-table-column
             label="操作"
             header-align="right"
@@ -60,6 +87,8 @@
             </template>
             <template slot-scope="scope">
               <el-button @click.stop="moveToTopEvent(scope.row)" type="text">置顶</el-button>
+              <!-- 检测时先强制批量检测一遍，如果不强制直接单个检测时第一次不会显示“检测中”-->
+              <el-button size="mini" v-if="iptvList.every(channel => channel.status)" v-show="!checkAllChannelsLoading" @click.stop="checkSingleChannel(scope.row)" type="text">检测</el-button>
               <el-button @click.stop="removeEvent(scope.row)" type="text">删除</el-button>
             </template>
           </el-table-column>
@@ -71,7 +100,8 @@
 <script>
 import { mapMutations } from 'vuex'
 import { iptv, iptvSearch } from '../lib/dexie'
-import { iptv as defaultSites } from '../lib/dexie/initData'
+import { iptv as defaultChannels } from '../lib/dexie/initData'
+import zy from '../lib/site/tools'
 import { remote } from 'electron'
 import fs from 'fs'
 import Sortable from 'sortablejs'
@@ -84,7 +114,9 @@ export default {
       searchRecordList: [],
       enableBatchEdit: false,
       batchGroupName: '',
+      batchIsActive: 1,
       multipleSelection: [],
+      checkAllChannelsLoading: false,
       show: {
         search: false
       }
@@ -151,11 +183,12 @@ export default {
       this.updateDatabase()
     },
     saveBatchEdit () {
-      if (this.multipleSelection && this.batchGroupName) {
-        this.multipleSelection.forEach(ele => {
+      this.multipleSelection.forEach(ele => {
+        if (this.batchGroupName) {
           ele.group = this.batchGroupName
-        })
-      }
+        }
+        ele.isActive = this.batchIsActive
+      })
       this.updateDatabase()
     },
     playEvent (e) {
@@ -179,17 +212,6 @@ export default {
         this.$message.warning('删除频道失败, 错误信息: ' + err)
       })
     },
-    listUpdatedEvent () {
-      iptv.clear().then(res1 => {
-        // 重新排序
-        var id = 1
-        this.iptvList.forEach(element => {
-          element.id = id
-          iptv.add(element)
-          id += 1
-        })
-      })
-    },
     exportChannels () {
       const options = {
         filters: [
@@ -291,11 +313,11 @@ export default {
       }
     },
     resetChannelsEvent () {
-      this.resetChannels(defaultSites)
+      this.resetChannels(defaultChannels)
     },
-    resetChannels (newSites) {
-      this.resetId(newSites)
-      iptv.clear().then(iptv.bulkAdd(newSites).then(this.getChannels()))
+    resetChannels (newChannels) {
+      this.resetId(newChannels)
+      iptv.clear().then(iptv.bulkAdd(newChannels).then(this.getChannels()))
     },
     removeAllChannels () {
       iptv.clear().then(res => {
@@ -362,6 +384,44 @@ export default {
           _this.updateDatabase()
         }
       })
+    },
+    isActiveChangeEvent (row) {
+      iptv.remove(row.id)
+      iptv.add(row)
+    },
+    async checkAllChannels () {
+      this.checkAllChannelsLoading = true
+      var siteList = {}
+      this.iptvList.forEach(channel => {
+        const site = channel.url.split('/')[2]
+        if (siteList[site]) {
+          siteList[site].push(channel)
+        } else {
+          siteList[site] = [channel]
+        }
+      })
+      Promise.all(Object.values(siteList).map(site => this.checkSingleSite(site))).then(res => {
+        this.checkAllChannelsLoading = false
+        this.getChannels()
+      })
+    },
+    async checkSingleSite (channelArray) {
+      for (const c of channelArray) {
+        await this.checkSingleChannel(c)
+      }
+    },
+    async checkSingleChannel (row) {
+      row.status = ' '
+      const flag = await zy.checkChannel(row.url)
+      if (flag) {
+        row.status = '可用'
+      } else {
+        row.status = '失效'
+        row.isActive = 0
+      }
+      iptv.remove(row.id)
+      iptv.add(row)
+      return row.status
     }
   },
   mounted () {

src/components/Play.vue

@@ -1051,7 +1051,7 @@ export default {
     },
     getIptvList () {
       iptv.all().then(res => {
-        this.iptvList = res
+        this.iptvList = res.filter(e => e.isActive)
       })
     },
     bindEvent () {

src/lib/dexie/dexie.js

@@ -13,7 +13,7 @@ db.version(4).stores({
   sites: '++id, key, name, api, download, isActive, group',
   history: '++id, [site+ids], name, type, year, index, time, detail',
   mini: 'id, site, ids, name, index, time',
-  iptv: '++id, name, url, group'
+  iptv: '++id, name, url, isActive, group'
 })
 
 db.on('populate', () => {

src/lib/site/tools.js

@@ -2,6 +2,7 @@ import { sites } from '../dexie'
 import axios from 'axios'
 import parser from 'fast-xml-parser'
 import cheerio from 'cheerio'
+import { Parser as M3u8Parser } from 'm3u8-parser'
 
 // axios使用系统代理  https://evandontje.com/2020/04/02/automatic-system-proxy-configuration-for-electron-applications/
 // xgplayer使用chromium代理设置，浏览器又默认使用系统代理 https://www.chromium.org/developers/design-documents/network-settings
@@ -31,7 +32,7 @@ axios.defaults.retryDelay = 1000
 axios.interceptors.response.use(function (response) {
   // 对响应数据做些事
   if (response.status && response.status === 200 && response.request.responseURL.includes('api.php') && !response.data.startsWith('<?xml')) {
-    console.log(response)
+    // console.log(response)
   }
   return response
 }, function (err) { // 请求错误时做些事
@@ -288,6 +289,29 @@ const zy = {
       return false
     }
   },
+  /**
+   * 检查直播源
+   * @param {*} channel 直播频道 url
+   * @returns boolean
+   */
+  async checkChannel (channel) {
+    return new Promise((resolve, reject) => {
+      axios.get(channel).then(res => {
+        const manifest = res.data
+        var parser = new M3u8Parser()
+        parser.push(manifest)
+        parser.end()
+        var parsedManifest = parser.manifest
+        if (parsedManifest.segments.length) {
+          resolve(true)
+        } else {
+          resolve(false)
+        }
+      }).catch(e => {
+        reject(e)
+      })
+    })
+  },
   /**
    * 获取豆瓣页面链接
    * @param {*} name 视频名称