import fs from 'fs';
import { Got } from 'got/dist/source';
import GenericOptions from '../interfaces/genericOptions';

// INSPIRED BY yar229's CODE
// https://github.com/yar229/WebDavMailRuCloud
// https://github.com/yar229/WebDavMailRuCloud/blob/master/MailRuCloud/MailRuCloudApi/Base/Repos/YandexDisk/YadWeb/Requests/

class YaAuth {
  private httpClient: Got;
  private fileLogging = false;

  private login: string;
  private password: string;

  constructor(
    login: string,
    password: string,
    httpClient: Got,
    options?: GenericOptions
  ) {
    this.httpClient = httpClient;
    this.login = login;
    this.password = password;

    if (options?.fileLogging) {
      this.fileLogging = options.fileLogging;
    }
  }

  async YadPreAuthRequest(): Promise<{ csrf: string; uuid: string }> {
    // console.log('preauth');

    const result = await this.httpClient.get('https://passport.yandex.ru/auth');

    if (this.fileLogging) {
      fs.writeFileSync('../logs/preauth.html', result.body);
    }

    const csrfRegexMatch = /"csrf":"(?<csrf>.*?)"/gm.exec(result.body);

    const processUUIDMatch = /"process_uuid":"(?<uuid>\S+?)"/gm.exec(
      result.body
    );

    if (csrfRegexMatch?.groups?.csrf && processUUIDMatch?.groups?.uuid) {
      return {
        csrf: csrfRegexMatch?.groups?.csrf,
        uuid: processUUIDMatch?.groups?.uuid,
      };
    } else {
      throw new Error('Error while parsing preauth data');
    }
  }

  async YadAuthLoginRequest(csrf: string, uuid: string) {
    // console.log('YadAuthLoginRequest');

    const result = await this.httpClient.post(
      'https://passport.yandex.ru/registration-validations/auth/multi_step/start',
      {
        headers: {
          'Sec-Fetch-Mode': 'cors',
          'Sec-Fetch-Site': 'same-origin',
          Referer:
            'https://passport.yandex.ru/auth/list?from=cloud&origin=disk_landing_web_signin_ru&retpath=https%3A%2F%2Fdisk.yandex.ru%2Fclient%2Fdisk&backpath=https%3A%2F%2Fdisk.yandex.ru&mode=edit',
        },
        form: {
          csrf_token: csrf,
          process_uuid: uuid,
          login: this.login,
          service: 'cloud',
          retpath: 'https://disk.yandex.ru?source=landing2_signin_ru',
          origin: 'disk_landing2_signin_ru',
        },
        // throwHttpErrors: false
      }
    );

    if (this.fileLogging) {
      fs.writeFileSync('../logs/YadAuthLoginRequest.html', result.body);
    }

    // console.log(result.body);

    return JSON.parse(result.body);
  }

  async YadAuthPasswordRequest(csrf: string, trackId: string) {
    // console.log('YadAuthPasswordRequest');

    const result = await this.httpClient.post(
      'https://passport.yandex.ru/registration-validations/auth/multi_step/commit_password',
      {
        body: `csrf_token=${csrf}&track_id=${trackId}&password=${encodeURIComponent(
          this.password
        )}`,
        // throwHttpErrors: false
      }
    );

    if (this.fileLogging) {
      fs.writeFileSync('../logs/YadAuthPasswordRequest.html', result.body);
    }

    // console.log(result.body);

    return JSON.parse(result.body);
  }

  async YadAuthAccountsRequest(csrf: string) {
    // console.log('YadAuthAccountsRequest');

    const result = await this.httpClient.post(
      'https://passport.yandex.ru/registration-validations/auth/accounts',
      {
        body: `csrf_token=${csrf}`,
        // throwHttpErrors: false
      }
    );

    if (this.fileLogging) {
      fs.writeFileSync('../logs/YadAuthAccountsRequest.html', result.body);
    }

    // console.log(result.body);

    return JSON.parse(result.body);
  }

  async YadAuthDiskSkRequest() {
    // console.log('YadAuthDiskSkRequest');

    const result = await this.httpClient.post(
      'https://disk.yandex.ru/client/disk',
      {
        throwHttpErrors: false,
        headers: {
          Accept:
            'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3',
          'Sec-Fetch-Mode': 'navigate',
          'Sec-Fetch-Site': 'same-site',
          'Sec-Fetch-User': '?1',
          Referer:
            'https://passport.yandex.ru/auth/list?from=cloud&origin=disk_landing_web_signin_ru&retpath=https%3A%2F%2Fdisk.yandex.ru%2Fclient%2Fdisk&backpath=https%3A%2F%2Fdisk.yandex.ru&mode=edit',
        },
      }
    );

    if (result.statusCode >= 400) {
      // console.log(result.body);
      throw new Error('Something went wrong with YadAuthDiskSkRequest');
    }

    if (this.fileLogging) {
      fs.writeFileSync('../logs/YadAuthDiskSkRequest.html', result.body);
    }

    // console.log(result.body);

    const skTokenMatch = /"sk":"(?<sk>.+?)"/gm.exec(result.body);

    const clientIdMatch = /"idClient":"(?<idClient>.+?)"/gm.exec(result.body);

    if (skTokenMatch?.groups?.sk && clientIdMatch?.groups?.idClient) {
      return {
        skToken: skTokenMatch?.groups?.sk,
        idClient: clientIdMatch?.groups?.idClient,
      };
    }

    throw new Error('Error while parsing SK token');
    // return JSON.parse(result.body);
  }
}

export default YaAuth;