import { Column, IDataRow, IValueColumnDesc, StringColumn, ValueColumn, toolbar } from 'lineupjs';
import isEqual from 'lodash/isEqual';

// internal function copied from lineupjs
function integrateDefaults<T>(desc: T, defaults: Partial<T> = {}) {
  Object.keys(defaults).forEach((key) => {
    const typed = key as keyof T;
    if (typeof desc[typed] === 'undefined') {
      (desc as any)[typed] = defaults[typed];
    }
  });
  return desc;
}

export interface ISMILESFilter {
  /**
   * Search string which is used to filter the column data
   */
  filter: string | null;

  /**
   * Filter out rows with missing values
   */
  filterMissing: boolean;

  /**
   * The set contains matching results that should be visible
   */
  matching: Set<string> | null;
}

@toolbar('filterSMILES', 'rename')
export class SMILESColumn extends ValueColumn<string> {
  protected structureFilter: ISMILESFilter | null = null;

  protected align: string | null = null;

  constructor(id: string, desc: Readonly<IValueColumnDesc<string>>) {
    super(
      id,
      integrateDefaults(desc, {
        summaryRenderer: 'default',
      }),
    );
  }

  protected createEventList() {
    return super.createEventList().concat([StringColumn.EVENT_FILTER_CHANGED]);
  }

  filter(row: IDataRow): boolean {
    if (!this.isFiltered() || !this.structureFilter) {
      return true;
    }

    // filter out row if no valid results found
    if (this.structureFilter.matching === null) {
      return false;
    }

    const rowLabel = this.getLabel(row);

    // filter missing values
    if (rowLabel == null || rowLabel.trim() === '') {
      return !this.structureFilter.filterMissing;
    }

    return this.structureFilter.matching.has(rowLabel) ?? false;
  }

  isFiltered(): boolean {
    return this.structureFilter != null;
  }

  getFilter() {
    return this.structureFilter;
  }

  setFilter(filter: ISMILESFilter | null) {
    if (isEqual(filter, this.structureFilter)) {
      return;
    }

    // ensure that no filter of the string column is used beyond this point
    // TODO remove once the string filter is removed from the UI
    if (filter && !filter.matching) {
      return;
    }

    this.fire([StringColumn.EVENT_FILTER_CHANGED, Column.EVENT_DIRTY_VALUES, Column.EVENT_DIRTY], this.structureFilter, (this.structureFilter = filter));
  }

  clearFilter() {
    const was = this.isFiltered();
    this.setFilter(null);
    return was;
  }

  getAlign(): string | null {
    return this.align;
  }

  setAlign(structure: string | null): void {
    if (isEqual(structure, this.align)) {
      return;
    }

    this.fire([Column.EVENT_DIRTY_VALUES, Column.EVENT_DIRTY], (this.align = structure));
  }
}