import { Logger } from '@stryker-mutator/api/logging';
import { commonTokens, tokens } from '@stryker-mutator/api/plugin';
import { errorToString } from '@stryker-mutator/util';
import type {
  IRestResponse,
  RestClient,
} from 'typed-rest-client/RestClient.js';

import { PackageInfo, PackageSummary } from './package-info.js';
import { PromptOption } from './prompt-option.js';

import { initializerTokens } from './index.js';

export interface NpmSearchResult {
  total: number;
  objects: Array<{ package: PackageSummary }>;
}

const getName = (packageName: string) => {
  return packageName
    .replace('@stryker-mutator/', '')
    .replace('stryker-', '')
    .split('-')[0];
};

const mapSearchResultToPromptOption = (
  searchResults: NpmSearchResult,
): PromptOption[] =>
  searchResults.objects.map((result) => ({
    name: getName(result.package.name),
    pkg: result.package,
  }));

const handleResult =
  (from: string) =>
  <T>(response: IRestResponse<T>): T => {
    if (response.statusCode === 200 && response.result) {
      return response.result;
    } else {
      throw new Error(
        `Path ${from} resulted in http status code: ${response.statusCode}.`,
      );
    }
  };

export class NpmClient {
  public static inject = tokens(
    commonTokens.logger,
    initializerTokens.restClientNpm,
    initializerTokens.npmRegistry,
  );
  constructor(
    private readonly log: Logger,
    private readonly innerNpmClient: RestClient,
    private readonly npmRegistry: string,
  ) {}

  public getTestRunnerOptions(): Promise<PromptOption[]> {
    return this.search(
      `/-/v1/search?text=keywords:${encodeURIComponent('@stryker-mutator/test-runner-plugin')}`,
    ).then(mapSearchResultToPromptOption);
  }

  public getTestReporterOptions(): Promise<PromptOption[]> {
    return this.search(
      `/-/v1/search?text=keywords:${encodeURIComponent('@stryker-mutator/reporter-plugin')}`,
    ).then(mapSearchResultToPromptOption);
  }

  public async getAdditionalConfig(
    pkgInfo: PackageSummary,
  ): Promise<PackageInfo> {
    const path = `/${encodeURIComponent(pkgInfo.name)}/${pkgInfo.version}`;
    try {
      const response = await this.innerNpmClient.get<PackageInfo>(path);
      return handleResult(path)(response);
    } catch (err) {
      this.log.warn(
        `Could not fetch additional initialization config for dependency ${pkgInfo.name}. You might need to configure it manually`,
        err,
      );
      return pkgInfo;
    }
  }

  private async search(path: string): Promise<NpmSearchResult> {
    this.log.debug(`Searching: ${path}`);
    try {
      const response = await this.innerNpmClient.get<NpmSearchResult>(path);
      return handleResult(path)(response);
    } catch (err) {
      this.log.error(
        `Unable to reach '${this.npmRegistry}' (for query ${path}). Please check your internet connection.`,
        errorToString(err),
      );
      const result: NpmSearchResult = {
        objects: [],
        total: 0,
      };
      return result;
    }
  }
}