From 4cf7bfb04a6645e34a71ea6330b3415b1481eefc Mon Sep 17 00:00:00 2001
From: TheBeastLT <pauliox@beyond.lt>
Date: Thu, 30 Apr 2020 11:29:23 +0200
Subject: [PATCH] [scraper] create subtitle database model

---
 scraper/lib/repository.js | 45 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 43 insertions(+), 2 deletions(-)

diff --git a/scraper/lib/repository.js b/scraper/lib/repository.js
index 8e230a4..93d95bb 100644
--- a/scraper/lib/repository.js
+++ b/scraper/lib/repository.js
@@ -30,7 +30,8 @@ const Torrent = database.define('torrent',
       trackers: { type: Sequelize.STRING(4096) },
       languages: { type: Sequelize.STRING(256) },
       resolution: { type: Sequelize.STRING(16) },
-      reviewed: { type: Sequelize.BOOLEAN, allowNull: false, defaultValue: false }
+      reviewed: { type: Sequelize.BOOLEAN, allowNull: false, defaultValue: false },
+      subsChecked: { type: Sequelize.BOOLEAN, allowNull: false, defaultValue: false }
     }
 );
 
@@ -73,10 +74,45 @@ const File = database.define('file',
     }
 );
 
+const Subtitle = database.define('subtitle',
+    {
+      infoHash: {
+        type: Sequelize.STRING(64),
+        primaryKey: true,
+        allowNull: false,
+        references: { model: Torrent, key: 'infoHash' },
+        onDelete: 'CASCADE'
+      },
+      fileIndex: {
+        type: Sequelize.INTEGER,
+        primaryKey: true,
+        allowNull: false
+      },
+      fileId: {
+        type: Sequelize.BIGINT,
+        allowNull: true,
+        references: { model: File, key: 'id' },
+        onDelete: 'SET NULL'
+      },
+      title: { type: Sequelize.STRING(512), allowNull: false },
+    },
+    {
+      timestamps: false,
+      indexes: [
+        { unique: false, fields: ['fileId'] }
+      ]
+    }
+);
+
 const SkipTorrent = database.define('skip_torrent', {
   infoHash: { type: Sequelize.STRING(64), primaryKey: true },
 });
 
+Torrent.hasMany(File, { foreignKey: 'infoHash', constraints: false });
+File.belongsTo(Torrent, { foreignKey: 'infoHash', constraints: false });
+File.hasMany(Subtitle, { foreignKey: 'fileId', constraints: false });
+Subtitle.belongsTo(File, { foreignKey: 'fileId', constraints: false });
+
 function connect() {
   if (process.env.ENABLE_SYNC) {
     return database.sync({ alter: true })
@@ -138,7 +174,7 @@ function setTorrentSeeders(infoHash, seeders) {
 }
 
 function createFile(file) {
-  return File.upsert(file);
+  return File.upsert(file, { include: [Subtitle] });
 }
 
 function getFiles(torrent) {
@@ -153,6 +189,10 @@ function deleteFile(file) {
   return File.destroy({ where: { id: file.id } })
 }
 
+function createSubtitle(subtitle) {
+  return Subtitle.upsert(subtitle);
+}
+
 function getSkipTorrent(torrent) {
   return SkipTorrent.findByPk(torrent.infoHash)
       .then((result) => {
@@ -179,6 +219,7 @@ module.exports = {
   getFiles,
   getFilesBasedOnTitle,
   deleteFile,
+  createSubtitle,
   getSkipTorrent,
   createSkipTorrent,
   getTorrentsWithoutSize