diff --git a/src/metadata_loader.js b/src/metadata_loader.js index 437e6379b72215bd51d0a59b86b2a8c508a5a9f6..fb7cea276b49d5dde9d12aad7367b397ee7f27a1 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 de5e1e77e7f684dcfb0a862379edc44f0bdfc7ec..a9b9bd3ef963984b059f6933550567dc6e84980e 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,