import { Disposable, ICellData, IUniverInstanceService, UniverInstanceType, Workbook } from '@univerjs/core';
import { IRenderManagerService } from '@univerjs/engine-render';
import { SheetSkeletonManagerService } from '@univerjs/sheets-ui';

export class JnpfSheetsCellService extends Disposable {
  constructor(
    @IUniverInstanceService private readonly _univerInstanceService: IUniverInstanceService,
    @IRenderManagerService private readonly _renderManagerService: IRenderManagerService,
  ) {
    super();
  }

  // 获取所有单元格数据
  public getAllCellData() {
    const sheet = this._univerInstanceService.getCurrentUnitForType<Workbook>(UniverInstanceType.UNIVER_SHEET)!.getActiveSheet();
    return sheet.getCellMatrix().getMatrix();
  }

  // 获取指定单元格数据
  public getCellData(col: number, row: number) {
    const sheet = this._univerInstanceService.getCurrentUnitForType<Workbook>(UniverInstanceType.UNIVER_SHEET)!.getActiveSheet();
    const cellMatrix = sheet.getCellMatrix().getMatrix();

    return cellMatrix?.[row]?.[col] ?? null; // 如果未找到匹配的单元格，返回 null
  }

  // 获取当前单元格的自定义属性
  public getCurrentSheetCellsCustom() {
    const sheet = this._univerInstanceService.getCurrentUnitForType<Workbook>(UniverInstanceType.UNIVER_SHEET)!.getActiveSheet();
    const cellMatrix = sheet?.getCellMatrix()?.getMatrix() ?? {};

    const result: { [row: string]: { [col: string]: any } } = {};
    Object.entries(cellMatrix)?.forEach(([row, colObj]) => {
      if (colObj && typeof colObj === 'object') {
        const rowData: { [col: string]: any } = {};

        Object.entries(colObj)?.forEach(([col, value]: any) => {
          if (value?.custom) {
            rowData[col] = { ...value };
          }
        });

        if (Object.keys(rowData)?.length) {
          result[row] = rowData;
        }
      }
    });

    return result;
  }

  // 获取激活单元格的尺寸信息
  public getTargetCellSize(col: number, row: number): null | { cellHeight: number; cellWidth: number } {
    // 获取当前的工作簿实例
    const workbook = this._univerInstanceService.getCurrentUnitForType<Workbook>(UniverInstanceType.UNIVER_SHEET);
    if (!workbook) {
      return null;
    }

    const unitId = workbook.getUnitId();

    // 获取当前激活的工作表
    const worksheet = workbook.getActiveSheet();
    if (!worksheet) {
      return null;
    }

    const subUnitId = worksheet.getSheetId();

    // 获取对应的骨架渲染服务
    const skeleton = this._renderManagerService?.getRenderById(unitId)?.with(SheetSkeletonManagerService)?.getUnitSkeleton(unitId, subUnitId)?.skeleton as any;

    if (!skeleton) {
      return null;
    }

    // 获取指定单元格的位置信息
    const cellInfo = skeleton.getCellWithCoordByIndex(row, col);
    if (!cellInfo) {
      return null;
    }

    // 解构单元格信息
    const { endX, endY, isMergedMainCell, mergeInfo, startX, startY } = cellInfo;

    let cellWidth: number;
    let cellHeight: number;

    if (isMergedMainCell && mergeInfo) {
      cellWidth = mergeInfo.endX - mergeInfo.startX;
      cellHeight = mergeInfo.endY - mergeInfo.startY;
    } else {
      cellWidth = endX - startX;
      cellHeight = endY - startY;
    }

    return {
      cellHeight: Math.floor(cellHeight) - 2,
      cellWidth: Math.floor(cellWidth) - 2,
    };
  }

  // 刷新所有单元格的视图
  public refreshRangeCellsView(range: any) {
    if (!range) {
      return;
    }

    const originalValue = range?.getValue() ?? '';
    range?.setValue(originalValue);
  }

  // 设置单元格数据
  public setCellData(col: number, row: number, cellData: ICellData) {
    const sheet = this._univerInstanceService.getCurrentUnitForType<Workbook>(UniverInstanceType.UNIVER_SHEET)!.getActiveSheet();
    const cellMatrix = sheet.getCellMatrix();

    try {
      cellMatrix.setValue(row, col, cellData);
    } catch (error) {
      console.warn(error);
    }
  }
}