import { trimStringEnding } from '../../../shared/strings';
import { PiHoleApiStatusChangeResponse, PiHoleApiSummaryResponse } from './piHole.type';

export class PiHoleClient {
  private readonly baseHostName: string;

  constructor(
    hostname: string,
    private readonly apiToken: string
  ) {
    this.baseHostName = trimStringEnding(hostname, ['/admin/index.php', '/admin', '/']);
  }

  async getSummary() {
    const response = await fetch(
      new URL(`${this.baseHostName}/admin/api.php?summaryRaw&auth=${this.apiToken}`)
    );

    if (response.status !== 200) {
      throw new Error(`Status code does not indicate success: ${response.status}`);
    }

    const json = await response.json();

    if (Array.isArray(json)) {
      throw new Error(
        `Response does not indicate success. Authentication is most likely invalid: ${json}`
      );
    }

    return json as PiHoleApiSummaryResponse;
  }

  async enable() {
    const response = await this.sendStatusChangeRequest('enable');
    return response.status === 'enabled';
  }

  async disable() {
    const response = await this.sendStatusChangeRequest('disable');
    return response.status === 'disabled';
  }

  private async sendStatusChangeRequest(
    action: 'enable' | 'disable'
  ): Promise<PiHoleApiStatusChangeResponse> {
    const response = await fetch(
      `${this.baseHostName}/admin/api.php?${action}&auth=${this.apiToken}`
    );

    if (response.status !== 200) {
      return Promise.reject(new Error(`Status code does not indicate success: ${response.status}`));
    }

    const json = await response.json();

    if (Array.isArray(json)) {
      return Promise.reject(
        new Error(
          `Response does not indicate success. Authentication is most likely invalid: ${json}`
        )
      );
    }

    for(let loops = 0; loops < 10; loops++){
      const summary = await this.getSummary()
      if (summary.status === action + 'd'){
        return json as PiHoleApiStatusChangeResponse;
      }
      await new Promise ((resolve) => { setTimeout(resolve, 50)});
    }

    return Promise.reject(
      new Error(
        `Although PiHole received the command, it failed to update it's status: ${json}`
      )
    )
  }
}