/*
 * Copyright 2025 coze-dev Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
import { devtools } from 'zustand/middleware';
import { create } from 'zustand';
import { type CSpan } from '@coze-devops/common-modules/query-trace';
import {
  type GetTracesMetaInfoData,
  type Span,
} from '@coze-arch/bot-api/ob_query_api';

import {
  type BasicInfo,
  type QueryFilterItemId,
  type TargetOverallSpanInfo,
} from '../typings';
import { IS_DEV_MODE } from '../consts/env';
import { FILTERING_OPTION_ALL } from '../consts';

export type SpanCategory = GetTracesMetaInfoData['span_category'];

interface DebugPanelStore {
  isPanelShow: boolean;
  basicInfo: BasicInfo;
  /**
   * 当前选中的Query LogID
   */
  entranceMessageLogId?: string;
  /**
   * 日期筛选结果
   */
  targetDateId?: QueryFilterItemId;
  /**
   * 状态筛选结果
   */
  targetExecuteStatusId?: QueryFilterItemId;
  /**
   * 当前选中的Trace节点信息
   */
  targetOverallSpanInfo?: TargetOverallSpanInfo;
  /**
   * 当前计算后的Trace列表
   */
  enhancedOverallSpans: CSpan[];
  /**
   * 某条Trace下Span节点列表
   */
  orgDetailSpans?: Span[];
  /**
   * 额外Span类型信息（服务端提供）
   */
  spanCategory?: SpanCategory;
  /**
   * 当前选中的Span节点信息
   */
  targetDetailSpan?: CSpan;
  curBatchPage?: number;
}

interface DebugPanelAction {
  setIsPanelShow: (isPanelShow: boolean) => void;
  setBasicInfo: (basicInfo: BasicInfo) => void;
  setEntranceMessageLogId: (entranceMessageLogId: string) => void;
  setTargetOverallSpanInfo: (overallSpanInfo: TargetOverallSpanInfo) => void;
  onSelectDate: (dateId: QueryFilterItemId) => void;
  onSelectExecuteStatus: (executeStatusId: QueryFilterItemId) => void;
  setEnhancedOverallSpans: (enhancedOverallSpans: CSpan[]) => void;
  setOrgDetailSpans: (orgDetailSpans: Span[]) => void;
  setSpanCategory: (spanCategory?: SpanCategory) => void;
  setTargetDetailSpan: (targetDetailSpan?: CSpan) => void;
  setCurBatchPage: (curBatchPage: number) => void;
  resetStore: () => void;
}

const initialStore: DebugPanelStore = {
  isPanelShow: false,
  basicInfo: {
    placement: 'left',
  },
  entranceMessageLogId: undefined,
  targetDateId: FILTERING_OPTION_ALL,
  targetExecuteStatusId: FILTERING_OPTION_ALL,
  enhancedOverallSpans: [],
  targetOverallSpanInfo: undefined,
  orgDetailSpans: undefined,
  targetDetailSpan: undefined,
};

export const useDebugPanelStore = create<DebugPanelStore & DebugPanelAction>()(
  devtools(
    set => ({
      ...initialStore,
      setIsPanelShow: (isPanelShow: boolean) => {
        set({ isPanelShow });
      },
      setBasicInfo: basicInfo => {
        set({ basicInfo });
      },
      setTargetOverallSpanInfo: overallSpanInfo => {
        set({ targetOverallSpanInfo: overallSpanInfo });
      },
      onSelectDate: dateId => {
        set({ targetDateId: dateId });
      },
      onSelectExecuteStatus: executeStatusId => {
        set({ targetExecuteStatusId: executeStatusId });
      },
      setEnhancedOverallSpans: enhancedOverallSpans => {
        set({ enhancedOverallSpans });
      },
      setEntranceMessageLogId: entranceMessageLogId => {
        set({ entranceMessageLogId });
      },
      setOrgDetailSpans: orgDetailSpans => {
        set({ orgDetailSpans });
      },
      setSpanCategory: spanCategory => {
        set({ spanCategory });
      },
      setTargetDetailSpan: targetDetailSpan => {
        set({ targetDetailSpan });
      },
      setCurBatchPage: curBatchPage => {
        set({ curBatchPage });
      },
      resetStore: () => {
        set(initialStore);
      },
    }),
    {
      enabled: IS_DEV_MODE,
      name: 'debug.debugPanelStore',
    },
  ),
);