426 lines
13 KiB
TypeScript
426 lines
13 KiB
TypeScript
import LidarrAPI from '@server/api/servarr/lidarr';
|
|
import RadarrAPI from '@server/api/servarr/radarr';
|
|
import SonarrAPI from '@server/api/servarr/sonarr';
|
|
import { MediaStatus, MediaType } from '@server/constants/media';
|
|
import { MediaServerType } from '@server/constants/server';
|
|
import { getRepository } from '@server/datasource';
|
|
import { Blacklist } from '@server/entity/Blacklist';
|
|
import type { User } from '@server/entity/User';
|
|
import { Watchlist } from '@server/entity/Watchlist';
|
|
import type { DownloadingItem } from '@server/lib/downloadtracker';
|
|
import downloadTracker from '@server/lib/downloadtracker';
|
|
import { getSettings } from '@server/lib/settings';
|
|
import logger from '@server/logger';
|
|
import { DbAwareColumn } from '@server/utils/DbColumnHelper';
|
|
import { getHostname } from '@server/utils/getHostname';
|
|
import {
|
|
AfterLoad,
|
|
Column,
|
|
Entity,
|
|
Index,
|
|
OneToMany,
|
|
OneToOne,
|
|
PrimaryGeneratedColumn,
|
|
} from 'typeorm';
|
|
import Issue from './Issue';
|
|
import { MediaRequest } from './MediaRequest';
|
|
import Season from './Season';
|
|
|
|
@Entity()
|
|
class Media {
|
|
public static async getRelatedMedia(
|
|
user: User | undefined,
|
|
ids: (number | string)[]
|
|
): Promise<Media[]> {
|
|
const mediaRepository = getRepository(Media);
|
|
|
|
try {
|
|
if (ids.length === 0) {
|
|
return [];
|
|
}
|
|
|
|
const tmdbIds = ids.filter((id): id is number => typeof id === 'number');
|
|
const mbIds = ids.filter((id): id is string => typeof id === 'string');
|
|
|
|
const queryBuilder = mediaRepository
|
|
.createQueryBuilder('media')
|
|
.leftJoinAndSelect(
|
|
'media.watchlists',
|
|
'watchlist',
|
|
'media.id = watchlist.media and watchlist.requestedBy = :userId',
|
|
{ userId: user?.id }
|
|
);
|
|
|
|
if (tmdbIds.length > 0 && mbIds.length > 0) {
|
|
queryBuilder.where(
|
|
'(media.tmdbId IN (:...tmdbIds) OR media.mbId IN (:...mbIds))',
|
|
{ tmdbIds, mbIds }
|
|
);
|
|
} else if (tmdbIds.length > 0) {
|
|
queryBuilder.where('media.tmdbId IN (:...tmdbIds)', { tmdbIds });
|
|
} else if (mbIds.length > 0) {
|
|
queryBuilder.where('media.mbId IN (:...mbIds)', { mbIds });
|
|
}
|
|
|
|
const media = await queryBuilder.getMany();
|
|
return media;
|
|
} catch (e) {
|
|
logger.error(e.message);
|
|
return [];
|
|
}
|
|
}
|
|
|
|
public static async getMedia(
|
|
id: number | string,
|
|
mediaType: MediaType
|
|
): Promise<Media | undefined> {
|
|
const mediaRepository = getRepository(Media);
|
|
|
|
try {
|
|
const whereClause =
|
|
typeof id === 'string'
|
|
? { mbId: id, mediaType }
|
|
: { tmdbId: id, mediaType };
|
|
|
|
const media = await mediaRepository.findOne({
|
|
where: whereClause,
|
|
relations: { requests: true, issues: true },
|
|
});
|
|
|
|
return media ?? undefined;
|
|
} catch (e) {
|
|
logger.error(e.message);
|
|
return undefined;
|
|
}
|
|
}
|
|
|
|
@PrimaryGeneratedColumn()
|
|
public id: number;
|
|
|
|
@Column({ type: 'varchar' })
|
|
public mediaType: MediaType;
|
|
|
|
@Column({ nullable: true })
|
|
@Index()
|
|
public tmdbId: number;
|
|
|
|
@Column({ unique: true, nullable: true })
|
|
@Index()
|
|
public tvdbId?: number;
|
|
|
|
@Column({ nullable: true })
|
|
@Index()
|
|
public imdbId?: string;
|
|
|
|
@Column({ nullable: true })
|
|
@Index()
|
|
public mbId?: string;
|
|
|
|
@Column({ type: 'int', default: MediaStatus.UNKNOWN })
|
|
public status: MediaStatus;
|
|
|
|
@Column({ type: 'int', default: MediaStatus.UNKNOWN })
|
|
public status4k: MediaStatus;
|
|
|
|
@OneToMany(() => MediaRequest, (request) => request.media, {
|
|
cascade: ['insert', 'remove'],
|
|
})
|
|
public requests: MediaRequest[];
|
|
|
|
@OneToMany(() => Watchlist, (watchlist) => watchlist.media)
|
|
public watchlists: null | Watchlist[];
|
|
|
|
@OneToMany(() => Season, (season) => season.media, {
|
|
cascade: true,
|
|
eager: true,
|
|
})
|
|
public seasons: Season[];
|
|
|
|
@OneToMany(() => Issue, (issue) => issue.media, { cascade: true })
|
|
public issues: Issue[];
|
|
|
|
@OneToOne(() => Blacklist, (blacklist) => blacklist.media)
|
|
public blacklist: Promise<Blacklist>;
|
|
|
|
@DbAwareColumn({ type: 'datetime', default: () => 'CURRENT_TIMESTAMP' })
|
|
public createdAt: Date;
|
|
|
|
@DbAwareColumn({
|
|
type: 'datetime',
|
|
default: () => 'CURRENT_TIMESTAMP',
|
|
onUpdate: 'CURRENT_TIMESTAMP',
|
|
})
|
|
public updatedAt: Date;
|
|
|
|
/**
|
|
* The `lastSeasonChange` column stores the date and time when the media was added to the library.
|
|
* It needs to be database-aware because SQLite supports `datetime` while PostgreSQL supports `timestamp with timezone (timestampz)`.
|
|
*/
|
|
@DbAwareColumn({ type: 'datetime', default: () => 'CURRENT_TIMESTAMP' })
|
|
public lastSeasonChange: Date;
|
|
|
|
/**
|
|
* The `mediaAddedAt` column stores the date and time when the media was added to the library.
|
|
* It needs to be database-aware because SQLite supports `datetime` while PostgreSQL supports `timestamp with timezone (timestampz)`.
|
|
* This column is nullable because it can be null when the media is not yet synced to the library.
|
|
*/
|
|
@DbAwareColumn({
|
|
type: 'datetime',
|
|
default: () => 'CURRENT_TIMESTAMP',
|
|
nullable: true,
|
|
})
|
|
public mediaAddedAt: Date;
|
|
|
|
@Column({ nullable: false, type: 'int', default: 0 })
|
|
public serviceId?: number | null;
|
|
|
|
@Column({ nullable: true, type: 'int' })
|
|
public serviceId4k?: number | null;
|
|
|
|
@Column({ nullable: true, type: 'int' })
|
|
public externalServiceId?: number | null;
|
|
|
|
@Column({ nullable: true, type: 'int' })
|
|
public externalServiceId4k?: number | null;
|
|
|
|
@Column({ nullable: true, type: 'varchar' })
|
|
public externalServiceSlug?: string | null;
|
|
|
|
@Column({ nullable: true, type: 'varchar' })
|
|
public externalServiceSlug4k?: string | null;
|
|
|
|
@Column({ nullable: true, type: 'varchar' })
|
|
public ratingKey?: string | null;
|
|
|
|
@Column({ nullable: true, type: 'varchar' })
|
|
public ratingKey4k?: string | null;
|
|
|
|
@Column({ nullable: true, type: 'varchar' })
|
|
public jellyfinMediaId?: string | null;
|
|
|
|
@Column({ nullable: true, type: 'varchar' })
|
|
public jellyfinMediaId4k?: string | null;
|
|
|
|
public serviceUrl?: string;
|
|
public serviceUrl4k?: string;
|
|
public downloadStatus?: DownloadingItem[] = [];
|
|
public downloadStatus4k?: DownloadingItem[] = [];
|
|
|
|
public mediaUrl?: string;
|
|
public mediaUrl4k?: string;
|
|
|
|
public iOSPlexUrl?: string;
|
|
public iOSPlexUrl4k?: string;
|
|
|
|
public tautulliUrl?: string;
|
|
public tautulliUrl4k?: string;
|
|
|
|
constructor(init?: Partial<Media>) {
|
|
Object.assign(this, init);
|
|
}
|
|
|
|
@AfterLoad()
|
|
public setPlexUrls(): void {
|
|
const { machineId, webAppUrl } = getSettings().plex;
|
|
const { externalUrl: tautulliUrl } = getSettings().tautulli;
|
|
|
|
if (getSettings().main.mediaServerType == MediaServerType.PLEX) {
|
|
if (this.ratingKey) {
|
|
this.mediaUrl = `${
|
|
webAppUrl ? webAppUrl : 'https://app.plex.tv/desktop'
|
|
}#!/server/${machineId}/details?key=%2Flibrary%2Fmetadata%2F${
|
|
this.ratingKey
|
|
}`;
|
|
|
|
this.iOSPlexUrl = `plex://preplay/?metadataKey=%2Flibrary%2Fmetadata%2F${this.ratingKey}&server=${machineId}`;
|
|
|
|
if (tautulliUrl) {
|
|
this.tautulliUrl = `${tautulliUrl}/info?rating_key=${this.ratingKey}`;
|
|
}
|
|
|
|
if (this.ratingKey4k) {
|
|
this.mediaUrl4k = `${
|
|
webAppUrl ? webAppUrl : 'https://app.plex.tv/desktop'
|
|
}#!/server/${machineId}/details?key=%2Flibrary%2Fmetadata%2F${
|
|
this.ratingKey4k
|
|
}`;
|
|
|
|
this.iOSPlexUrl4k = `plex://preplay/?metadataKey=%2Flibrary%2Fmetadata%2F${this.ratingKey4k}&server=${machineId}`;
|
|
|
|
if (tautulliUrl) {
|
|
this.tautulliUrl4k = `${tautulliUrl}/info?rating_key=${this.ratingKey4k}`;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
const pageName =
|
|
getSettings().main.mediaServerType == MediaServerType.EMBY
|
|
? 'item'
|
|
: 'details';
|
|
const { serverId, externalHostname } = getSettings().jellyfin;
|
|
const jellyfinHost =
|
|
externalHostname && externalHostname.length > 0
|
|
? externalHostname
|
|
: getHostname();
|
|
|
|
if (this.jellyfinMediaId) {
|
|
this.mediaUrl = `${jellyfinHost}/web/index.html#!/${pageName}?id=${this.jellyfinMediaId}&context=home&serverId=${serverId}`;
|
|
}
|
|
if (this.jellyfinMediaId4k) {
|
|
this.mediaUrl4k = `${jellyfinHost}/web/index.html#!/${pageName}?id=${this.jellyfinMediaId4k}&context=home&serverId=${serverId}`;
|
|
}
|
|
}
|
|
}
|
|
|
|
@AfterLoad()
|
|
public setServiceUrl(): void {
|
|
if (this.mediaType === MediaType.MOVIE) {
|
|
if (this.serviceId !== null && this.externalServiceSlug !== null) {
|
|
const settings = getSettings();
|
|
const server = settings.radarr.find(
|
|
(radarr) => radarr.id === this.serviceId
|
|
);
|
|
|
|
if (server) {
|
|
this.serviceUrl = server.externalUrl
|
|
? `${server.externalUrl}/movie/${this.externalServiceSlug}`
|
|
: RadarrAPI.buildUrl(server, `/movie/${this.externalServiceSlug}`);
|
|
}
|
|
}
|
|
|
|
if (this.serviceId4k !== null && this.externalServiceSlug4k !== null) {
|
|
const settings = getSettings();
|
|
const server = settings.radarr.find(
|
|
(radarr) => radarr.id === this.serviceId4k
|
|
);
|
|
|
|
if (server) {
|
|
this.serviceUrl4k = server.externalUrl
|
|
? `${server.externalUrl}/movie/${this.externalServiceSlug4k}`
|
|
: RadarrAPI.buildUrl(
|
|
server,
|
|
`/movie/${this.externalServiceSlug4k}`
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (this.mediaType === MediaType.TV) {
|
|
if (this.serviceId !== null && this.externalServiceSlug !== null) {
|
|
const settings = getSettings();
|
|
const server = settings.sonarr.find(
|
|
(sonarr) => sonarr.id === this.serviceId
|
|
);
|
|
|
|
if (server) {
|
|
this.serviceUrl = server.externalUrl
|
|
? `${server.externalUrl}/series/${this.externalServiceSlug}`
|
|
: SonarrAPI.buildUrl(server, `/series/${this.externalServiceSlug}`);
|
|
}
|
|
}
|
|
|
|
if (this.serviceId4k !== null && this.externalServiceSlug4k !== null) {
|
|
const settings = getSettings();
|
|
const server = settings.sonarr.find(
|
|
(sonarr) => sonarr.id === this.serviceId4k
|
|
);
|
|
|
|
if (server) {
|
|
this.serviceUrl4k = server.externalUrl
|
|
? `${server.externalUrl}/series/${this.externalServiceSlug4k}`
|
|
: SonarrAPI.buildUrl(
|
|
server,
|
|
`/series/${this.externalServiceSlug4k}`
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (this.mediaType === MediaType.MUSIC) {
|
|
if (this.serviceId !== null && this.externalServiceSlug !== null) {
|
|
const settings = getSettings();
|
|
const server = settings.lidarr.find(
|
|
(lidarr) => lidarr.id === this.serviceId
|
|
);
|
|
|
|
if (server) {
|
|
this.serviceUrl = server.externalUrl
|
|
? `${server.externalUrl}/album/${this.externalServiceSlug}`
|
|
: LidarrAPI.buildUrl(server, `/album/${this.externalServiceSlug}`);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
@AfterLoad()
|
|
public getDownloadingItem(): void {
|
|
if (this.mediaType === MediaType.MOVIE) {
|
|
if (
|
|
this.externalServiceId !== undefined &&
|
|
this.externalServiceId !== null &&
|
|
this.serviceId !== undefined &&
|
|
this.serviceId !== null
|
|
) {
|
|
this.downloadStatus = downloadTracker.getMovieProgress(
|
|
this.serviceId,
|
|
this.externalServiceId
|
|
);
|
|
}
|
|
|
|
if (
|
|
this.externalServiceId4k !== undefined &&
|
|
this.externalServiceId4k !== null &&
|
|
this.serviceId4k !== undefined &&
|
|
this.serviceId4k !== null
|
|
) {
|
|
this.downloadStatus4k = downloadTracker.getMovieProgress(
|
|
this.serviceId4k,
|
|
this.externalServiceId4k
|
|
);
|
|
}
|
|
}
|
|
|
|
if (this.mediaType === MediaType.TV) {
|
|
if (
|
|
this.externalServiceId !== undefined &&
|
|
this.externalServiceId !== null &&
|
|
this.serviceId !== undefined &&
|
|
this.serviceId !== null
|
|
) {
|
|
this.downloadStatus = downloadTracker.getSeriesProgress(
|
|
this.serviceId,
|
|
this.externalServiceId
|
|
);
|
|
}
|
|
|
|
if (
|
|
this.externalServiceId4k !== undefined &&
|
|
this.externalServiceId4k !== null &&
|
|
this.serviceId4k !== undefined &&
|
|
this.serviceId4k !== null
|
|
) {
|
|
this.downloadStatus4k = downloadTracker.getSeriesProgress(
|
|
this.serviceId4k,
|
|
this.externalServiceId4k
|
|
);
|
|
}
|
|
}
|
|
|
|
if (this.mediaType === MediaType.MUSIC) {
|
|
if (
|
|
this.externalServiceId !== undefined &&
|
|
this.externalServiceId !== null &&
|
|
this.serviceId !== undefined &&
|
|
this.serviceId !== null
|
|
) {
|
|
this.downloadStatus = downloadTracker.getMusicProgress(
|
|
this.serviceId,
|
|
this.externalServiceId
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
export default Media;
|