import RNFetchBlob from 'react-native-blob-util';

import type {
  FetchBlobResponse,
  StatefulPromise,
} from 'react-native-blob-util';

import type {
  SessionTaskInterface,
  SessionTaskOptionsType,
} from '../types/type';
import { KEY_PREFIX } from '../Utils/constants';

export * from 'react-native-blob-util';

export class SimpleSessionProvider implements SessionTaskInterface {
  // current caching m3u8 playlist
  // any are session task object
  private downloadingList: {
    [key in string]?: StatefulPromise<FetchBlobResponse>;
  } = {};

  dataTask = (
    url: string,
    options: SessionTaskOptionsType,
    callback?: (data: any, res: any, error?: Error) => void
  ): StatefulPromise<FetchBlobResponse> => {
    const downloadTask = RNFetchBlob.config({
      session: KEY_PREFIX,
      ...options,
    }).fetch('GET', url, {
      'RNFB-Response': 'base64',
      ...options.headers,
    });
    // mark it as downloading
    this.downloadingList[url] = downloadTask;
    // listen response download
    downloadTask
      .then((res) => {
        // res.respInfo?.headers && console.log(res.respInfo?.headers);
        callback && callback(res.data, res, undefined);
      })
      .catch((error) => {
        callback && callback(null, null, error);
      })
      .finally(() => {
        delete this.downloadingList[url];
      });
    //
    return downloadTask;
  };

  cancelTask = (url: string) => {
    const downloadTask = this.downloadingList[url];
    if (!downloadTask) {
      return;
    }

    downloadTask.cancel();

    delete this.downloadingList[url];
  };

  cancelAllTask = () => {
    Object.entries(this.downloadingList).forEach(([url, downloadTask]) => {
      url && downloadTask?.cancel();
    });

    this.downloadingList = {};
  };
}