sonarr-pogdesign-importer/src/utils.ts

Import tv series from pogdesign and/or trakt into sonarr

import * as fs from 'fs';
import { IConfig } from './interfaces/config';
import SonarrApi from './sonarrapi';
import { IProfile } from './interfaces/sonarr';

const stripTrailingSlashes = (url: string) => {
    return url.endsWith('/') ? url.slice(0, -1) : url;
}

const isConfigValid = (config: IConfig) => {
    if (!config.sonarr.url.length)                   { console.log('sonarrUrl has to be defined'); return false; }
    if (!config.sonarr.apiKey.length)                { console.log('sonarrApi has to be defined'); return false; }
    if (!config.sonarr.path.length)                  { console.log('sonarrPath has to be defined'); return false; }
    if (config.sonarr.profileId <= 0)                { console.log('sonarrProfileId has to be greater or equal to 1'); return false; }
    if (!config.sonarr.path.endsWith('/'))           { console.log('sonarrPath has to end with \'/\''); return false; }
    if (config.sonarr.useSeasonFolder === undefined) { console.log('sonarrUseSeasonFolder has to be defined'); return false; }
    if (!config.genresIgnored)                       { console.log('genresIgnored has to be defined'); return false; }

    const scrapers = config.scrapers.some(scraper => scraper.type.toLocaleLowerCase() === 'pogdesign' || scraper.type.toLocaleLowerCase() === 'trakt');
    if (!scrapers) { console.log('No scrapers specified'); return false; }

    return true;
}

const getProfiles = async (config: IConfig) => {
    const sonarrApi = new SonarrApi(config);
    const res = await sonarrApi.getProfiles();
    if (!res.ok) {
        console.log(`Sonarr responded with ${res.status}: ${await res.text()}`);
        return;
    }

    const profiles: IProfile[] = await res.json();

    if (config.verbose) {
        console.log(JSON.stringify(profiles, null, 2));
    } else {
        console.log('Profile id | Profile name')
        for (const profile of profiles) {
            console.log(`${profile.id} | ${profile.name}`);
        }
    }
}

const getPaths = async (config: IConfig) => {
    const sonarrApi = new SonarrApi(config);
    const res = await sonarrApi.getPaths();
    if (!res.ok) {
        console.log(`Sonarr responded with ${res.status}: ${await res.text()}`);
        return;
    }

    const paths: IProfile[] = await res.json();

    console.log(JSON.stringify(paths, null, 2));
}

const loadConfig = async (args: {[key: string]: string}) => {
    const configPath = args.config || args.c;

    if (!configPath) {
        console.log(`Config has not been specified`);
        process.exit(1);
    }

    const fileContent = fs.readFileSync(configPath, 'utf8');

    if (!fileContent) {
        console.log(`Can not file any config located at ${configPath}`);
        process.exit(1);
    }

    let result: IConfig;
    try {
        result = JSON.parse(fileContent);
    } catch (exception) {
        console.log(exception.toString());
        console.log('\nConfig is not valid');
        process.exit(1);

        return null;
    }

    result.sonarr.url = stripTrailingSlashes(result.sonarr.url);

    if (args.profiles) {
        await getProfiles(result);
        process.exit(0);
    }

    if (args.paths) {
        await getPaths(result);
        process.exit(0);
    }

    if (!isConfigValid(result)) {
        console.log(`Config is not valid`);
        process.exit(1);
    }

    return result;
}

export {
    loadConfig,
};