import {
  XamGridColumnBaseModel,
  XamGridColumnModel,
} from './XamGridColumnModel';
import { CellBase } from './CellBase';
import { ColumnBaseCollection, XamGridCell } from './XamGridModel';
import { RowBase } from './RowBase';
import { ObservableCollection } from '../../baseframework/collections';

/**
 * Collection of CellBase for the xamGrid
 *
 * @export
 * @class CellBaseCollectionModel
 * @extends {ObservableCollection<XamGridCell>}
 * @wType Infragistics.Controls.Grids.Primitives.CellBaseCollection
 */
export class CellBaseCollectionModel extends ObservableCollection<CellBase> {
  /**
   * private collection of rows
   *
   * @private
   * @type {*}
   * @memberof CellBaseCollectionModel
   */
  private _row: any;

  /**
   * private collection of columns
   *
   * @private
   * @type {ColumnBaseCollection}
   * @memberof CellBaseCollectionModel
   */
  private _columns: ColumnBaseCollection;

  /**
   * Auxiliar collection of columns
   *
   * @private
   * @type {Map<XamGridColumnBaseModel, CellBase>}
   * @memberof CellBaseCollectionModel
   */
  private _auxColumns: Map<XamGridColumnBaseModel, CellBase>;

  /**
   * Creates an instance of CellBaseCollection.
   * @param {*} columns
   * @param {*} row
   * @memberof CellBaseCollectionModel
   */
  constructor(columns: any, row: any) {
    super();
    this._row = row;
    this._columns = columns;
    this._auxColumns = new Map<XamGridColumnBaseModel, CellBase>();
  }

  /**
   * Method to returns the Cell with a specify key
   *
   * @param {string} keyColumn
   * @return {*}
   * @memberof GroupColumnsCollection
   */
  getCellByKey(keyColumn: string | XamGridColumnModel): any {
    let column = keyColumn;
    if (typeof column === 'string') {
      let { column: columnTmp } = this._columns.getColumnByKeyDeep(column);
      column = columnTmp;
    }
    if (column instanceof XamGridColumnModel) {
      return this.getColumnFromDictionary(column);
    }
    return null;
  }

  /**
   * Gets an item from the collection by its index.
   *
   * @param {number} index
   * @type {any}
   * @memberof CellBaseCollectionModel
   * @wMethod GetItem
   */
  getItem(index: number): any {
    const column = this._columns.DataColumns.getItem(index);
    return this.getCellByKey(column);
  }

  /**
   * Gets an item from the collection by its index.
   *
   * @param {XamGridColumnModel} column
   * @type {any}
   * @memberof CellBaseCollectionModel
   */
  getColumnFromDictionary(column: XamGridColumnModel): any {
    if (this._auxColumns.has(column)) {
      return this._auxColumns.get(column);
    }
    const cellBase = column.GenerateCell(this._row);
    this.add(cellBase);
    this._auxColumns.set(column, cellBase);
    return cellBase;
  }

  /**
   * Register a new cell that comes from the UI.
   *
   * @param {any} item
   * @param {RowBase} row
   * @param {ColumnBaseCollection} Columns
   * @type {CellBase}
   * @memberof CellBaseCollectionModel
   */
  registerCellFromUI(
    item: any,
    row: RowBase,
    Columns: ColumnBaseCollection
  ): CellBase {
    const key = this.getKey(item);
    const { column, isRoot } = Columns.getColumnByKeyDeep(key);
    if (column) {
      if (this._auxColumns.has(column)) {
        return this._auxColumns.get(column);
      }
      const currentCell = row.Cells.getCellByKey(key);
      if (currentCell) {
        this._auxColumns.set(column, currentCell);
        if (isRoot) {
          this.add(currentCell);
        }
        return currentCell;
      }
      const cell = new XamGridCell(row, column);
      this._auxColumns.set(column, cell);
      if (isRoot) {
        this.add(cell);
      }
      return cell;
    }
    console.log('this is not a valid column');
  }

  /**
   * Gets the key from an item.
   *
   * @param {any} item
   * @type {string | null}
   * @memberof CellBaseCollectionModel
   */
  getKey(item: any): string | null {
    let columnString = '';
    if (item.column?.field) {
      columnString = item.column?.field;
    } else if (item._columnField) {
      columnString = item._columnField;
    }
    const key = columnString.substring(5, columnString.length);
    return key ?? '';
  }
}