/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import {
  AbstractTree,
  IAbstractTreeOptions,
} from '../../../../../vs/base/browser/ui/tree/abstractTree';
import {
  ITreeNode,
  ITreeModel,
  ITreeRenderer,
  IDataSource,
} from '../../../../../vs/base/browser/ui/tree/tree';
import { ObjectTreeModel } from '../../../../../vs/base/browser/ui/tree/objectTreeModel';
import {
  IListVirtualDelegate,
  IIdentityProvider,
} from '../../../../../vs/base/browser/ui/list/list';
import { IList } from '../../../../../vs/base/browser/ui/tree/indexTreeModel';

export interface IDataTreeOptions<T, TFilterData = void>
  extends IAbstractTreeOptions<T, TFilterData> {}

export class DataTree<TInput, T, TFilterData = void> extends AbstractTree<
  T | null,
  TFilterData,
  T | null
> {
  protected override model!: ObjectTreeModel<T, TFilterData>;

  private identityProvider: IIdentityProvider<T> | undefined;

  constructor(
    private user: string,
    container: HTMLElement,
    delegate: IListVirtualDelegate<T>,
    renderers: ITreeRenderer<T, TFilterData, any>[],
    private dataSource: IDataSource<TInput, T>,
    options: IDataTreeOptions<T, TFilterData> = {}
  ) {
    super(
      user,
      container,
      delegate,
      renderers,
      options as IDataTreeOptions<T | null, TFilterData>
    );
    this.identityProvider = options.identityProvider;
  }

  protected createModel(
    user: string,
    view: IList<ITreeNode<T, TFilterData>>,
    options: IDataTreeOptions<T, TFilterData>
  ): ITreeModel<T | null, TFilterData, T | null> {
    return new ObjectTreeModel(user, view, options);
  }
}