import { type ActiveRecord, ExperimentalConfigManager } from '@oinone/kunlun-engine';
import { ViewType } from '@oinone/kunlun-meta';
import type { OioTreeNode } from '@oinone/kunlun-shared';
import { SPI } from '@oinone/kunlun-spi';
import { isTreeViewState, type OioAnyViewState, Widget } from '@oinone/kunlun-vue-widget';
import { BaseElementWidget } from '../../basic';
import type { TreeData } from '../../typing';
import { AbstractTreeElementWidget } from './AbstractTreeElementWidget';

@SPI.ClassFactory(
  BaseElementWidget.Token({
    viewType: ViewType.Tree,
    widget: 'tree'
  })
)
export class TreeWidget extends AbstractTreeElementWidget {
  @Widget.Reactive()
  protected showContent = true;

  protected async onNodeSelected(node: OioTreeNode<TreeData>) {
    if (ExperimentalConfigManager.treeWidgetNext()) {
      const nodeModel = node.value.metadata?.model;
      if (nodeModel === this.model.model) {
        await this.onSelectedForQuery(node);
      }
    } else {
      await this.onSelectedForQuery(node);
    }
  }

  protected async onNodeUnselected(node: OioTreeNode<TreeData>) {
    await this.onUnselected();
  }

  protected async onClearSearch() {
    await this.onUnselected();
  }

  protected async onUnselected() {
    const currentValue = this.generatorEmptyValue();
    this.reloadDataSource(currentValue);
    this.reloadActiveRecords(currentValue);
  }

  protected $$mounted() {
    super.$$mounted();
    const currentValue = this.generatorEmptyValue();
    this.reloadDataSource(currentValue);
  }

  protected generatorEmptyValue(): ActiveRecord[] {
    if (this.contentViewType === ViewType.Form) {
      return [{}];
    }
    return [];
  }

  protected $$initViewState(state: OioAnyViewState): void {
    if (isTreeViewState(state) && !state.tree) {
      state.tree = this.currentHandle;
    }
  }
}