add torrentio catalogs addon

catalogs/lib/repository.js

@@ -0,0 +1,29 @@
+const { Sequelize, QueryTypes } = require('sequelize');
+const { Type } = require('../../addon/lib/types');
+
+const DATABASE_URI = process.env.DATABASE_URI;
+
+const database = new Sequelize(DATABASE_URI, { logging: false });
+
+async function getIds(type, startDate, endDate) {
+  const idName = type === Type.ANIME ? 'kitsuId' : 'imdbId';
+  const episodeCondition = type === Type.SERIES
+      ? 'AND files."imdbSeason" IS NOT NULL AND files."imdbEpisode" IS NOT NULL'
+      : '';
+  const dateCondition = startDate && endDate
+      ? `AND "uploadDate" BETWEEN '${startDate}' AND '${endDate}'`
+      : ''
+  const query = `SELECT files."${idName}"
+        FROM (SELECT torrents."infoHash", torrents.seeders FROM torrents
+                WHERE seeders > 0 AND type = '${type}' ${dateCondition}
+              ) as torrents
+        JOIN files ON torrents."infoHash" = files."infoHash"
+        WHERE files."${idName}" IS NOT NULL ${episodeCondition}
+        GROUP BY files."${idName}"
+        ORDER BY max(torrents.seeders) DESC
+        LIMIT 5000`
+  const results = await database.query(query, { type: QueryTypes.SELECT });
+  return results.map(result => `${result.imdbId || result.kitsuId}`);
+}
+
+module.exports = { getIds };