From 97b8d1b4b77273f956e014ad474bcbf716c54508 Mon Sep 17 00:00:00 2001
From: Xaekai <ribbon13@gmail.com>
Date: Wed, 26 Jan 2022 18:36:54 -0800
Subject: [PATCH] Enable caching BitChute metadata

---
 src/database.js                |  3 +++
 src/database/metadata_cache.js | 15 +++++++++++----
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/database.js b/src/database.js
index 9103ad97..31ed2017 100644
--- a/src/database.js
+++ b/src/database.js
@@ -85,6 +85,9 @@ module.exports.init = function (newDB) {
                 require('@cytube/mediaquery/lib/provider/youtube').setCache(
                     new MetadataCacheDB(db)
                 );
+                require('@cytube/mediaquery/lib/provider/bitchute').setCache(
+                    new MetadataCacheDB(db)
+                );
             }).catch(error => {
                 LOGGER.error(error.stack);
                 process.exit(1);
diff --git a/src/database/metadata_cache.js b/src/database/metadata_cache.js
index d84537e2..11eb8878 100644
--- a/src/database/metadata_cache.js
+++ b/src/database/metadata_cache.js
@@ -9,6 +9,8 @@ function mediaquery2cytube(type) {
     switch (type) {
         case 'youtube':
             return 'yt';
+        case 'bitchute':
+            return 'bc';
         default:
             throw new Error(`mediaquery2cytube: no mapping for ${type}`);
     }
@@ -18,14 +20,17 @@ function cytube2mediaquery(type) {
     switch (type) {
         case 'yt':
             return 'youtube';
+        case 'bc':
+            return 'bitchute';
         default:
             throw new Error(`cytube2mediaquery: no mapping for ${type}`);
     }
 }
 
 const cachedResultAge = new Summary({
-    name: 'cytube_yt_cache_result_age_seconds',
-    help: 'Age (in seconds) of cached record'
+    name: 'cytube_media_cache_result_age_seconds',
+    help: 'Age (in seconds) of cached record',
+    labelNames: ['source']
 });
 
 class MetadataCacheDB {
@@ -62,10 +67,11 @@ class MetadataCacheDB {
                 return null;
             }
 
+            let age = 0;
             try {
-                let age = (Date.now() - row.updated_at.getTime())/1000;
+                age = (Date.now() - row.updated_at.getTime())/1000;
                 if (age > 0) {
-                    cachedResultAge.observe(age);
+                    cachedResultAge.labels(type).observe(age);
                 }
             } catch (error) {
                 LOGGER.error('Failed to record cachedResultAge metric: %s', error.stack);
@@ -73,6 +79,7 @@ class MetadataCacheDB {
 
             let metadata = JSON.parse(row.metadata);
             metadata.type = cytube2mediaquery(metadata.type);
+            metadata.meta.cacheAge = age;
             return new Media(metadata);
         });
     }