multi-downloader-nx_mirror/gui/server/services/crunchyroll.ts

import { AuthData, CheckTokenResponse, DownloadData, EpisodeListResponse, MessageHandler, ResolveItemsData, SearchData, SearchResponse } from '../../../@types/messageHandler';
import Crunchy from '../../../crunchy';
import { getDefault } from '../../../modules/module.args';
import { languages, subtitleLanguagesFilter } from '../../../modules/module.langsData';
import WebSocketHandler from '../websocket';
import Base from './base';
import { console } from '../../../modules/log';
import * as yargs from '../../../modules/module.app-args';

class CrunchyHandler extends Base implements MessageHandler {
  private crunchy: Crunchy;
  public name = 'crunchy';
  constructor(ws: WebSocketHandler) {
    super(ws);
    this.crunchy = new Crunchy();
    this.crunchy.refreshToken();
    this.initState();
    this.getDefaults();
  }

  public getDefaults() {
    const _default = yargs.appArgv(this.crunchy.cfg.cli, true);
    this.crunchy.api = _default.crapi;
    this.crunchy.locale = _default.locale;
  }
  
  public async listEpisodes (id: string): Promise<EpisodeListResponse> {
    this.getDefaults();
    await this.crunchy.refreshToken(true);
    return { isOk: true, value: (await this.crunchy.listSeriesID(id)).list };
  }
  
  public async handleDefault(name: string) {
    return getDefault(name, this.crunchy.cfg.cli);
  }

  public async availableDubCodes(): Promise<string[]> {
    const dubLanguageCodesArray: string[] = [];
    for(const language of languages){
      if (language.cr_locale)
        dubLanguageCodesArray.push(language.code);
    }
    return [...new Set(dubLanguageCodesArray)];
  }

  public async availableSubCodes(): Promise<string[]> {
    return subtitleLanguagesFilter;
  }

  public async resolveItems(data: ResolveItemsData): Promise<boolean> {
    this.getDefaults();
    await this.crunchy.refreshToken(true);
    console.debug(`Got resolve options: ${JSON.stringify(data)}`);
    const res = await this.crunchy.downloadFromSeriesID(data.id, data);
    if (!res.isOk)
      return res.isOk;
    this.addToQueue(res.value.map(a => {
      return {
        ...data,
        
        ids: a.data.map(a => a.mediaId),
        title: a.episodeTitle,
        parent: {
          title: a.seasonTitle,
          season: a.season.toString()
        },
        e: a.e,
        image: a.image,
        episode: a.episodeNumber
      };
    }));
    return true;
  }

  public async search(data: SearchData): Promise<SearchResponse> {
    this.getDefaults();
    await this.crunchy.refreshToken(true);
    if (!data['search-type']) data['search-type'] = 'series';
    console.debug(`Got search options: ${JSON.stringify(data)}`);
    const crunchySearch = await this.crunchy.doSearch(data);
    if (!crunchySearch.isOk) {
      this.crunchy.refreshToken();
      return crunchySearch;
    }
    return { isOk: true, value: crunchySearch.value };
  }

  public async checkToken(): Promise<CheckTokenResponse> {
    if (await this.crunchy.getProfile()) {
      return { isOk: true, value: undefined };
    } else {
      return { isOk: false, reason: new Error('') };
    }
  }

  public auth(data: AuthData) {
    return this.crunchy.doAuth(data);
  }

  public async downloadItem(data: DownloadData) {
    this.getDefaults();
    await this.crunchy.refreshToken(true);
    console.debug(`Got download options: ${JSON.stringify(data)}`);
    this.setDownloading(true);
    const _default = yargs.appArgv(this.crunchy.cfg.cli, true);
    this.crunchy.api = _default.crapi;
    const res = await this.crunchy.downloadFromSeriesID(data.id, {
      dubLang: data.dubLang,
      e: data.e
    });
    if (res.isOk) {
      for (const select of res.value) {
        if (!(await this.crunchy.downloadEpisode(select, {..._default, skipsubs: false, callbackMaker: this.makeProgressHandler.bind(this), q: data.q, fileName: data.fileName, dlsubs: data.dlsubs, dlVideoOnce: data.dlVideoOnce, force: 'y', 
          novids: data.novids, noaudio: data.noaudio, hslang: data.hslang || 'none'  }))) {
          const er = new Error(`Unable to download episode ${data.e} from ${data.id}`);
          er.name = 'Download error';
          this.alertError(er);
        }
      }
    } else {
      this.alertError(res.reason);
    }
    this.sendMessage({ name: 'finish', data: undefined });
    this.setDownloading(false);
    this.onFinish();
  }
}

export default CrunchyHandler;