/*
 * Copyright (C) 2022 Huawei Device Co., Ltd.
 * 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 { TraceRow } from '../../component/trace/base/TraceRow';
import { BaseStruct, dataFilterHandler, drawLoadingFrame, ns2x, Rect, Render } from './ProcedureWorkerCommon';
import { ColorUtils } from '../../component/trace/base/ColorUtils';

export class HiSysEventRender extends Render {
  renderMainThread(
    req: {
      useCache: boolean;
      context: CanvasRenderingContext2D;
      type: string;
    },
    row: TraceRow<HiSysEventStruct>
  ): void {
    let hiSysEventFilter = row.dataListCache;
    let minorFilter: HiSysEventStruct[] = [];
    let criticalFilter: HiSysEventStruct[] = [];
    let minorList = hiSysEventFilter.filter((struct) => {
      return struct.depth === 0;
    });
    let criticalList = hiSysEventFilter.filter((struct) => {
      return struct.depth === 1;
    });
    dataFilterHandler(minorList, minorFilter, {
      startKey: 'ts',
      durKey: 'dur',
      startNS: TraceRow.range?.startNS ?? 0,
      endNS: TraceRow.range?.endNS ?? 0,
      totalNS: TraceRow.range?.totalNS ?? 0,
      frame: row.frame,
      paddingTop: padding * 2,
      useCache: req.useCache || !(TraceRow.range?.refresh ?? false),
    });
    dataFilterHandler(criticalList, criticalFilter, {
      startKey: 'ts',
      durKey: 'dur',
      startNS: TraceRow.range?.startNS ?? 0,
      endNS: TraceRow.range?.endNS ?? 0,
      totalNS: TraceRow.range?.totalNS ?? 0,
      frame: row.frame,
      paddingTop: rectHeight + padding * 2,
      useCache: req.useCache || !(TraceRow.range?.refresh ?? false),
    });
    hiSysEventFilter = minorFilter.concat(criticalFilter);
    drawLoadingFrame(req.context, row.dataListCache, row);
    req.context.beginPath();
    let find = false;
    for (let re of hiSysEventFilter) {
      HiSysEventStruct.draw(req.context, re);
    }
    if (!find && row.isHover) {
      HiSysEventStruct.hoverHiSysEventStruct = undefined;
    }
    req.context.closePath();
  }
}

export function hiSysEvent(
  hiSysEventList: Array<HiSysEventStruct>,
  hiSysEventFilter: Array<HiSysEventStruct>,
  startNS: number,
  endNS: number,
  totalNS: number,
  row: TraceRow<HiSysEventStruct>,
  use: boolean
): void {
  if (use && hiSysEventFilter.length > 0) {
    for (let i = 0, len = hiSysEventFilter.length; i < len; i++) {
      let item = hiSysEventFilter[i];
      if ((item.startTs || 0) + (item.dur || 0) >= startNS && (item.startTs || 0) <= endNS) {
        // @ts-ignore
        HiSysEventStruct.setSysEventFrame(item, startNS, endNS, totalNS, row.frame);
      } else {
        item.frame = undefined;
      }
    }
    return;
  }
  hiSysEventFilter.length = 0;
  if (hiSysEventList) {
    for (let index = 0; index < hiSysEventList.length; index++) {
      let item = hiSysEventList[index];
      if ((item.startTs || 0) + (item.dur || 0) >= startNS && (item.startTs || 0) <= endNS) {
        // @ts-ignore
        HiSysEventStruct.setSysEventFrame(item, startNS, endNS, totalNS, row.frame);
        hiSysEventFilter.push(item);
      }
    }
  }
}

export class HiSysEventStruct extends BaseStruct {
  static hoverHiSysEventStruct: HiSysEventStruct | undefined;
  static selectHiSysEventStruct: HiSysEventStruct | undefined;
  id: number | undefined;
  domain: string | undefined;
  eventName: string | undefined;
  eventType: string | undefined;
  startTs: number | undefined;
  tz: string | undefined;
  pid: number | undefined;
  tid: number | undefined;
  uid: number | undefined;
  info: string | undefined;
  level: string | undefined;
  seq: number | undefined;
  contents: string | undefined;
  dur: number | undefined;
  depth: number | undefined;

  static setSysEventFrame(
    sysEventNode: HiSysEventStruct,
    startNS: number,
    endNS: number,
    totalNS: number,
    frame: Rect
  ): void {
    let x1: number;
    let x2: number;
    if ((sysEventNode.startTs || 0) >= startNS && (sysEventNode.startTs || 0) <= endNS) {
      x1 = ns2x(sysEventNode.startTs || 0, startNS, endNS, totalNS, frame);
    } else {
      x1 = 0;
    }
    if (
      (sysEventNode.startTs || 0) + (sysEventNode.dur || 0) >= startNS &&
      (sysEventNode.startTs || 0) + (sysEventNode.dur || 0) <= endNS
    ) {
      x2 = ns2x((sysEventNode.startTs || 0) + (sysEventNode.dur || 0), startNS, endNS, totalNS, frame);
    } else {
      x2 = frame.width;
    }
    if (!sysEventNode.frame) {
      sysEventNode.frame = new Rect(0, 0, 0, 0);
    }
    let getV: number = x2 - x1 < 1 ? 1 : x2 - x1;
    sysEventNode.frame.x = Math.floor(x1);
    sysEventNode.frame.y = sysEventNode.depth! * rectHeight + padding * 2;
    sysEventNode.frame.width = Math.ceil(getV);
    sysEventNode.frame.height = 20;
  }

  static draw(ctx: CanvasRenderingContext2D, data: HiSysEventStruct): void {
    if (data.depth === undefined || data.depth === null) {
      return;
    }
    if (data.frame) {
      ctx.globalAlpha = 1;
      ctx.fillStyle = ColorUtils.getHisysEventColor(data.depth!);
      ctx.fillRect(data.frame.x, data.frame.y + padding * data.depth, data.frame.width, rectHeight);
    }
  }
}

const padding = 5;
const rectHeight = 10;