import figures from '@inquirer/figures';
import { ChangeType } from './types/change-type.enum.js';
import { Change } from './types/diff.type.js';
import { Kind } from './types/kind.type.js';
import chalk from 'chalk';
import { highlight } from 'cli-highlight';

export class SchemaDiffPromptItem {
  public readonly change: Change | undefined;
  public readonly indent: boolean;
  public readonly type: ChangeType | undefined;
  public readonly selectable: boolean;
  public selected: boolean;
  public showDetails: boolean;

  public constructor(
    public opts: {
      change?: Change;
      type?: ChangeType;
      indent?: boolean;
      selected?: boolean;
      selectable?: boolean;
    },
  ) {
    Object.assign(
      this,
      { indent: false, selected: false, selectable: true, showDetails: false },
      opts,
    );
  }

  public static fromCollection(collection: Change): SchemaDiffPromptItem {
    return new SchemaDiffPromptItem({
      change: collection,
      type: ChangeType.COLLECTION,
    });
  }

  public static fromField(field: Change): SchemaDiffPromptItem {
    return new SchemaDiffPromptItem({
      change: field,
      indent: true,
      type: ChangeType.FIELD,
    });
  }

  public static fromRelation(relation: Change): SchemaDiffPromptItem {
    return new SchemaDiffPromptItem({
      change: relation,
      indent: true,
      type: ChangeType.RELATION,
    });
  }

  public static getDummy(collection: string) {
    return new SchemaDiffPromptItem({
      change: { collection, diff: [] },
      selectable: false,
    });
  }

  public getDetails(): string[] {
    return this.change?.diff.map((diff) => {
      return highlight(JSON.stringify(diff), { language: 'json' });
    });
  }

  public render(active: boolean): string {
    const cursor = active ? figures.pointer : ' ';
    const checkbox = this.selected ? figures.circleFilled : figures.circle;
    const details = this.showDetails
      ? this.getDetails()
          .map((detail) => `\n${this.indent ? '  ' : ''}    ${detail}`)
          .join('')
      : '';
    return `${this.indent ? '  ' : ''}${cursor} ${checkbox} ${this.colorizeDiff(this.change)}${details}`;
  }

  private colorizeDiff(change: Change) {
    if (typeof this.type === 'undefined') {
      return change.collection;
    }

    const kind = change.diff[0].kind;
    return this.kindToColor(kind)(
      change.field ? change.field : change.collection,
    );
  }

  private kindToColor(kind: Kind) {
    const kindToColorMap = {
      E: chalk.yellow,
      N: chalk.green,
      D: chalk.red,
      A: chalk.blue,
    };

    return kindToColorMap[kind];
  }
}