fixed #15;added support for downloading via bittorrent files;added new class for getting download statistics

2021-10-21 18:18:41 +08:00
parent c2ce945aad
commit 5037180a40
12 changed files with 190 additions and 36 deletions
--- a/lib/Tools/Counters.php
+++ b/lib/Tools/Counters.php
@@ -0,0 +1,65 @@
+<?php
+
+namespace OCA\NCDownloader\Tools;
+
+use OCA\NCDownloader\Tools\Aria2;
+use OCA\NCDownloader\Tools\DBConn;
+
+class Counters
+{
+    private $minmax = [0, 999];
+
+    public function __construct(Aria2 $aria2, DBConn $dbconn, $uid)
+    {
+        $this->aria2 = $aria2;
+        $this->dbconn = $dbconn;
+        $this->uid = $uid;
+    }
+    public function getCounters()
+    {
+        return [
+            'active' => $this->getCounter(),
+            'waiting' => $this->getCounter('tellWaiting'),
+            'complete' => $this->getCounter('tellStopped'),
+            'fail' => $this->getCounter('tellFail'),
+            'youtube-dl' => $this->getCounter('youtube-dl'),
+        ];
+    }
+    private function getCounter($action = 'tellActive')
+    {
+        if ($action === 'youtube-dl') {
+            $data = $this->dbconn->getYoutubeByUid($this->uid);
+        } else if ($action === 'tellActive') {
+            $data = $this->aria2->{$action}([]);
+        } else {
+            $data = $this->aria2->{$action}($this->minmax);
+        }
+
+        if (!is_array($data) && count($data) < 1) {
+            return 0;
+        }
+        if ($action !== 'youtube-dl') {
+            $data = $this->filterData($data);
+        }
+        return count($data);
+    }
+
+    private function filterData($resp)
+    {
+
+        $data = [];
+        if (empty($resp)) {
+            return $data;
+        }
+        if (isset($resp['error'])) {
+            return $resp;
+        }
+
+        $data = array_filter($resp, function ($value) {
+            $gid = $value['following'] ?? $value['gid'];
+            return (bool) ($this->dbconn->getUidByGid($gid) === $this->uid);
+        });
+
+        return $data;
+    }
+}