import logger from '../../logger'; import ServarrBase from './base'; interface SonarrSeason { seasonNumber: number; monitored: boolean; statistics?: { previousAiring?: string; episodeFileCount: number; episodeCount: number; totalEpisodeCount: number; sizeOnDisk: number; percentOfEpisodes: number; }; } export interface SonarrSeries { title: string; sortTitle: string; seasonCount: number; status: string; overview: string; network: string; airTime: string; images: { coverType: string; url: string; }[]; remotePoster: string; seasons: SonarrSeason[]; year: number; path: string; profileId: number; languageProfileId: number; seasonFolder: boolean; monitored: boolean; useSceneNumbering: boolean; runtime: number; tvdbId: number; tvRageId: number; tvMazeId: number; firstAired: string; lastInfoSync?: string; seriesType: 'standard' | 'daily' | 'anime'; cleanTitle: string; imdbId: string; titleSlug: string; certification: string; genres: string[]; tags: number[]; added: string; ratings: { votes: number; value: number; }; qualityProfileId: number; id?: number; rootFolderPath?: string; addOptions?: { ignoreEpisodesWithFiles?: boolean; ignoreEpisodesWithoutFiles?: boolean; searchForMissingEpisodes?: boolean; }; } interface AddSeriesOptions { tvdbid: number; title: string; profileId: number; languageProfileId?: number; seasons: number[]; seasonFolder: boolean; rootFolderPath: string; tags?: number[]; seriesType: SonarrSeries['seriesType']; monitored?: boolean; searchNow?: boolean; } export interface LanguageProfile { id: number; name: string; } class SonarrAPI extends ServarrBase<{ seriesId: number; episodeId: number }> { constructor({ url, apiKey }: { url: string; apiKey: string }) { super({ url, apiKey, apiName: 'Sonarr', cacheName: 'sonarr' }); } public async getSeries(): Promise { try { const response = await this.axios.get('/series'); return response.data; } catch (e) { throw new Error(`[Sonarr] Failed to retrieve series: ${e.message}`); } } public async getSeriesByTitle(title: string): Promise { try { const response = await this.axios.get('/series/lookup', { params: { term: title, }, }); if (!response.data[0]) { throw new Error('No series found'); } return response.data; } catch (e) { logger.error('Error retrieving series by series title', { label: 'Sonarr API', errorMessage: e.message, title, }); throw new Error('No series found'); } } public async getSeriesByTvdbId(id: number): Promise { try { const response = await this.axios.get('/series/lookup', { params: { term: `tvdb:${id}`, }, }); if (!response.data[0]) { throw new Error('Series not found'); } return response.data[0]; } catch (e) { logger.error('Error retrieving series by tvdb ID', { label: 'Sonarr API', errorMessage: e.message, tvdbId: id, }); throw new Error('Series not found'); } } public async addSeries(options: AddSeriesOptions): Promise { try { const series = await this.getSeriesByTvdbId(options.tvdbid); // If the series already exists, we will simply just update it if (series.id) { series.tags = options.tags ?? series.tags; series.seasons = this.buildSeasonList(options.seasons, series.seasons); const newSeriesResponse = await this.axios.put( '/series', series ); if (newSeriesResponse.data.id) { logger.info('Updated existing series in Sonarr.', { label: 'Sonarr', seriesId: newSeriesResponse.data.id, seriesTitle: newSeriesResponse.data.title, }); logger.debug('Sonarr update details', { label: 'Sonarr', movie: newSeriesResponse.data, }); if (options.searchNow) { this.searchSeries(newSeriesResponse.data.id); } return newSeriesResponse.data; } else { logger.error('Failed to update series in Sonarr', { label: 'Sonarr', options, }); throw new Error('Failed to update series in Sonarr'); } } const createdSeriesResponse = await this.axios.post( '/series', { tvdbId: options.tvdbid, title: options.title, qualityProfileId: options.profileId, languageProfileId: options.languageProfileId, seasons: this.buildSeasonList( options.seasons, series.seasons.map((season) => ({ seasonNumber: season.seasonNumber, // We force all seasons to false if its the first request monitored: false, })) ), tags: options.tags, seasonFolder: options.seasonFolder, monitored: options.monitored, rootFolderPath: options.rootFolderPath, seriesType: options.seriesType, addOptions: { ignoreEpisodesWithFiles: true, searchForMissingEpisodes: options.searchNow, }, } as Partial ); if (createdSeriesResponse.data.id) { logger.info('Sonarr accepted request', { label: 'Sonarr' }); logger.debug('Sonarr add details', { label: 'Sonarr', movie: createdSeriesResponse.data, }); } else { logger.error('Failed to add movie to Sonarr', { label: 'Sonarr', options, }); throw new Error('Failed to add series to Sonarr'); } return createdSeriesResponse.data; } catch (e) { logger.error('Something went wrong while adding a series to Sonarr.', { label: 'Sonarr API', errorMessage: e.message, options, response: e?.response?.data, }); throw new Error('Failed to add series'); } } public async getLanguageProfiles(): Promise { try { const data = await this.getRolling( '/languageprofile', undefined, 3600 ); return data; } catch (e) { logger.error( 'Something went wrong while retrieving Sonarr language profiles.', { label: 'Sonarr API', errorMessage: e.message, } ); throw new Error('Failed to get language profiles'); } } public async searchSeries(seriesId: number): Promise { logger.info('Executing series search command.', { label: 'Sonarr API', seriesId, }); try { await this.runCommand('SeriesSearch', { seriesId }); } catch (e) { logger.error( 'Something went wrong while executing Sonarr series search.', { label: 'Sonarr API', errorMessage: e.message, seriesId, } ); } } private buildSeasonList( seasons: number[], existingSeasons?: SonarrSeason[] ): SonarrSeason[] { if (existingSeasons) { const newSeasons = existingSeasons.map((season) => { if (seasons.includes(season.seasonNumber)) { season.monitored = true; } return season; }); return newSeasons; } const newSeasons = seasons.map( (seasonNumber): SonarrSeason => ({ seasonNumber, monitored: true, }) ); return newSeasons; } } export default SonarrAPI;