import Axios, { AxiosInstance } from 'axios'; import logger from '../logger'; interface SonarrSeason { seasonNumber: number; monitored: boolean; } 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: string[]; added: string; ratings: { votes: number; value: number; }; qualityProfileId: number; id?: number; rootFolderPath?: string; addOptions?: { ignoreEpisodesWithFiles?: boolean; ignoreEpisodesWithoutFiles?: boolean; searchForMissingEpisodes?: boolean; }; } interface SonarrProfile { id: number; name: string; } interface SonarrRootFolder { id: number; path: string; freeSpace: number; totalSpace: number; unmappedFolders: { name: string; path: string; }[]; } interface AddSeriesOptions { tvdbid: number; title: string; profileId: number; seasons: number[]; seasonFolder: boolean; rootFolderPath: string; seriesType: SonarrSeries['seriesType']; monitored?: boolean; searchNow?: boolean; } class SonarrAPI { private axios: AxiosInstance; constructor({ url, apiKey }: { url: string; apiKey: string }) { this.axios = Axios.create({ baseURL: url, params: { apikey: apiKey, }, }); } 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', message: e.message, }); 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.seasons = this.buildSeasonList(options.seasons, series.seasons); series.addOptions = { ignoreEpisodesWithFiles: true, searchForMissingEpisodes: true, }; const newSeriesResponse = await this.axios.put( '/series', series ); return newSeriesResponse.data; } const createdSeriesResponse = await this.axios.post( '/series', { tvdbId: options.tvdbid, title: options.title, profileId: options.profileId, 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, })) ), seasonFolder: options.seasonFolder, monitored: options.monitored, rootFolderPath: options.rootFolderPath, seriesType: options.seriesType, addOptions: { ignoreEpisodesWithFiles: true, searchForMissingEpisodes: options.searchNow, }, } as Partial ); return createdSeriesResponse.data; } catch (e) { logger.error('Something went wrong adding a series to Sonarr', { label: 'Sonarr API', errorMessage: e.message, error: e, }); throw new Error('Failed to add series'); } } public async getProfiles(): Promise { try { const response = await this.axios.get('/profile'); return response.data; } catch (e) { logger.error('Something went wrong retrieving Sonarr profiles', { label: 'Sonarr API', message: e.message, }); throw new Error('Failed to get profiles'); } } public async getRootFolders(): Promise { try { const response = await this.axios.get('/rootfolder'); return response.data; } catch (e) { logger.error('Something went wrong retrieving Sonarr root folders', { label: 'Sonarr API', message: e.message, }); throw new Error('Failed to get root folders'); } } 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;