From 10f4168ec464153705a18377717a7b3c0cc24468 Mon Sep 17 00:00:00 2001
From: Janne Koschinski <janne@kuschku.de>
Date: Fri, 25 Sep 2020 22:31:33 +0200
Subject: [PATCH] Add kind field

---
 src/metadata_loader.js | 6 ++++--
 src/storage.js         | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/metadata_loader.js b/src/metadata_loader.js
index 437e637..fb7cea2 100644
--- a/src/metadata_loader.js
+++ b/src/metadata_loader.js
@@ -34,12 +34,13 @@ class MetadataLoader {
         this.storage = storage;
     }
 
-    async transformData(ids, imdbResult, tmdbResult, tmdbContentRatings, tmdbTranslations) {
+    async transformData(ids, imdbResult, tmdbResult, tmdbContentRatings, tmdbTranslations, kind) {
         const [title] = await Title.upsert({
             id: ids.uuid,
             imdb_id: ids.imdb,
             tmdb_id: ids.tmdb,
             tvdb_id: ids.tvdb,
+            kind,
             original_language: tmdbResult.original_language,
             runtime: imdbResult.runtime,
             year_start: imdbResult.startYear,
@@ -179,6 +180,7 @@ class MetadataLoader {
         })
         const [episodeTitle] = await Title.upsert({
             id: mapping.episode_id,
+            kind: "episode",
             imdb_id: imdbResult.id,
             tmdb_id: tmdbResult.id,
             tvdb_id: null,
@@ -514,7 +516,7 @@ class MetadataLoader {
             episodes: imdbEpisodes,
         };
 
-        const title = await this.transformData(ids, imdbData, tmdbResult, tmdbContentRatings, tmdbTranslations);
+        const title = await this.transformData(ids, imdbData, tmdbResult, tmdbContentRatings, tmdbTranslations, isShow ? "show" : "movie");
 
         return {
             title: title,
diff --git a/src/storage.js b/src/storage.js
index de5e1e7..a9b9bd3 100644
--- a/src/storage.js
+++ b/src/storage.js
@@ -78,6 +78,7 @@ class Backend {
             imdb_id: sequelize.DataTypes.TEXT,
             tmdb_id: sequelize.DataTypes.INTEGER,
             tvdb_id: sequelize.DataTypes.INTEGER,
+            kind: sequelize.DataTypes.TEXT,
             original_language: sequelize.DataTypes.TEXT,
             runtime: sequelize.DataTypes.INTEGER,
             year_start: sequelize.DataTypes.INTEGER,
-- 
GitLab