// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
import type * as Common from '../../core/common/common.js';
import * as i18n from '../../core/i18n/i18n.js';
import * as Trace from '../../models/trace/trace.js';
import * as PerfUI from '../../ui/legacy/components/perf_ui/perf_ui.js';

import {buildGroupStyle, buildTrackHeader, getDurationString} from './AppenderUtils.js';
import {
  type CompatibilityTracksAppender,
  type PopoverInfo,
  type TrackAppender,
  type TrackAppenderName,
  VisualLoggingTrackName,
} from './CompatibilityTracksAppender.js';
import * as Extensions from './extensions/extensions.js';
import {TimelineFlameChartMarker} from './TimelineFlameChartView.js';
import {TimelinePanel} from './TimelinePanel.js';
import type {TimelineMarkerStyle} from './TimelineUIUtils.js';

const UIStrings = {
  /**
   * @description Text in Timeline Flame Chart Data Provider of the Performance panel
   */
  timings: 'Timings',
} as const;

const str_ = i18n.i18n.registerUIStrings('panels/timeline/TimingsTrackAppender.ts', UIStrings);
const i18nString = i18n.i18n.getLocalizedString.bind(undefined, str_);

/**
 * This defines the order these markers will be rendered if they are at the
 * same timestamp. The smaller number will be shown first - e.g. so if MarkFCP,
 * MarkDOMContent and MarkLCPCandidate have the same timestamp, visually we
 * will render [FCP][DCL][LCP] everytime.
 */
export const SORT_ORDER_PAGE_LOAD_MARKERS: Readonly<Record<string, number>> = {
  [Trace.Types.Events.Name.NAVIGATION_START]: 0,
  [Trace.Types.Events.Name.MARK_LOAD]: 1,
  [Trace.Types.Events.Name.MARK_FCP]: 2,
  [Trace.Types.Events.Name.MARK_FIRST_PAINT]: 2,
  [Trace.Types.Events.Name.MARK_DOM_CONTENT]: 3,
  [Trace.Types.Events.Name.MARK_LCP_CANDIDATE]: 4,
  [Trace.Types.Events.Name.MARK_LCP_CANDIDATE_FOR_SOFT_NAVIGATION]: 5,
};

export class TimingsTrackAppender implements TrackAppender {
  readonly appenderName: TrackAppenderName = 'Timings';

  #colorGenerator: Common.Color.Generator;
  #compatibilityBuilder: CompatibilityTracksAppender;
  #parsedTrace: Readonly<Trace.TraceModel.ParsedTrace>;
  #extensionMarkers: readonly Trace.Types.Extensions.SyntheticExtensionMarker[];
  constructor(
      compatibilityBuilder: CompatibilityTracksAppender, parsedTrace: Trace.TraceModel.ParsedTrace,
      colorGenerator: Common.Color.Generator) {
    this.#compatibilityBuilder = compatibilityBuilder;
    this.#colorGenerator = colorGenerator;
    this.#parsedTrace = parsedTrace;
    const extensionDataEnabled = TimelinePanel.extensionDataVisibilitySetting().get();
    this.#extensionMarkers = extensionDataEnabled ? this.#parsedTrace.data.ExtensionTraceData.extensionMarkers : [];
  }

  /**
   * Appends into the flame chart data the data corresponding to the
   * timings track.
   * @param trackStartLevel the horizontal level of the flame chart events where
   * the track's events will start being appended.
   * @param expanded whether the track should be rendered expanded.
   * @returns the first available level to append more data after having
   * appended the track's events.
   */
  appendTrackAtLevel(trackStartLevel: number, expanded?: boolean): number {
    const extensionMarkersAreEmpty = this.#extensionMarkers.length === 0;
    const performanceMarks = this.#parsedTrace.data.UserTimings.performanceMarks.filter(
        m => !Trace.Handlers.ModelHandlers.ExtensionTraceData.extensionDataInPerformanceTiming(m).devtoolsObj);
    const performanceMeasures = this.#parsedTrace.data.UserTimings.performanceMeasures.filter(
        m => !Trace.Handlers.ModelHandlers.ExtensionTraceData.extensionDataInPerformanceTiming(m).devtoolsObj);
    const timestampEvents = this.#parsedTrace.data.UserTimings.timestampEvents.filter(
        timeStamp =>
            !Trace.Handlers.ModelHandlers.ExtensionTraceData.extensionDataInConsoleTimeStamp(timeStamp).devtoolsObj);
    const consoleTimings = this.#parsedTrace.data.UserTimings.consoleTimings;

    // The order below is deliberate. Visually its easier to process marks when they decorate the bottom of their associated measure.
    // Therefore, we want COMPLETE events (measures) to be before INSTANT events (marks), (when they share a timestamp).
    const allTimings = [...performanceMeasures, ...consoleTimings, ...timestampEvents, ...performanceMarks].sort(
        (a, b) => a.ts - b.ts);

    if (extensionMarkersAreEmpty && allTimings.length === 0) {
      return trackStartLevel;
    }

    this.#appendTrackHeaderAtLevel(trackStartLevel, expanded);
    const newLevel = this.#appendExtensionsAtLevel(trackStartLevel);
    return this.#compatibilityBuilder.appendEventsAtLevel(allTimings, newLevel, this);
  }

  /**
   * Adds into the flame chart data the header corresponding to the
   * timings track. A header is added in the shape of a group in the
   * flame chart data. A group has a predefined style and a reference
   * to the definition of the legacy track (which should be removed
   * in the future).
   * @param currentLevel the flame chart level at which the header is
   * appended.
   */
  #appendTrackHeaderAtLevel(currentLevel: number, expanded?: boolean): void {
    const trackIsCollapsible = this.#parsedTrace.data.UserTimings.performanceMeasures.length > 0;
    const style = buildGroupStyle({
      useFirstLineForOverview: true,
      collapsible: trackIsCollapsible ? PerfUI.FlameChart.GroupCollapsibleState.IF_MULTI_ROW :
                                        PerfUI.FlameChart.GroupCollapsibleState.NEVER,
    });
    const group = buildTrackHeader(
        VisualLoggingTrackName.TIMINGS, currentLevel, i18nString(UIStrings.timings), style, /* selectable= */ true,
        expanded);
    this.#compatibilityBuilder.registerTrackForGroup(group, this);
  }
  /**
   * Adds into the flame chart data the ExtensionMarkers.
   * @param currentLevel the flame chart level from which markers will
   * be appended.
   * @returns the next level after the last occupied by the appended
   * extension markers (the first available level to append more data).
   */
  #appendExtensionsAtLevel(currentLevel: number): number {
    const markers = this.#extensionMarkers.toSorted((m1, m2) => m1.ts - m2.ts);
    if (markers.length === 0) {
      return currentLevel;
    }
    for (const marker of markers) {
      const index = this.#compatibilityBuilder.appendEventAtLevel(marker, currentLevel, this);
      // Marker events do not have a duration: rendering code in
      // FlameChart.ts relies on us setting this to NaN
      this.#compatibilityBuilder.getFlameChartTimelineData().entryTotalTimes[index] = Number.NaN;
    }

    const minTimeMs = Trace.Helpers.Timing.microToMilli(this.#parsedTrace.data.Meta.traceBounds.min);
    const flameChartMarkers = markers.map(marker => {
      // The timestamp for user timing trace events is set to the
      // start time passed by the user at the call site of the timing
      // (based on the UserTiming spec), meaning we can use event.ts
      // directly.
      // https://source.chromium.org/chromium/chromium/src/+/main:third_party/blink/renderer/core/timing/performance_user_timing.cc;l=236;drc=494419358caf690316f160a1f27d9e771a14c033
      const startTimeMs = Trace.Helpers.Timing.microToMilli(marker.ts);
      const style = this.markerStyleForExtensionMarker(marker);
      return new TimelineFlameChartMarker(startTimeMs, startTimeMs - minTimeMs, style);
    });
    this.#compatibilityBuilder.getFlameChartTimelineData().markers.push(...flameChartMarkers);
    // TODO: we would like to have markers share the level with the rest but...
    //  due to how CompatTrackAppender.appendEventsAtLevel tweaks the legacyEntryTypeByLevel array, it would take some work
    return ++currentLevel;
  }

  /*
    ------------------------------------------------------------------------------------
     The following methods  are invoked by the flame chart renderer to query features about
     events on rendering.
    ------------------------------------------------------------------------------------
  */

  /**
   * Gets the style for a page load marker event.
   * TODO(paulirish): Unify with trace/Styles.ts markerDetailsForEvent. Currently only color is read, the rest is ignored.
   */
  markerStyleForPageLoadEvent(markerEvent: Trace.Types.Events.PageLoadEvent): TimelineMarkerStyle {
    const tallMarkerDashStyle = [6, 4];
    let title = '';
    let color = 'grey';
    if (Trace.Types.Events.isMarkDOMContent(markerEvent)) {
      color = '#0867CB';
      title = Trace.Handlers.ModelHandlers.PageLoadMetrics.MetricName.DCL;
    }
    if (Trace.Types.Events.isMarkLoad(markerEvent)) {
      color = '#B31412';
      title = Trace.Handlers.ModelHandlers.PageLoadMetrics.MetricName.L;
    }
    if (Trace.Types.Events.isFirstPaint(markerEvent)) {
      color = '#228847';
      title = Trace.Handlers.ModelHandlers.PageLoadMetrics.MetricName.FP;
    }
    if (Trace.Types.Events.isFirstContentfulPaint(markerEvent)) {
      color = '#1A6937';
      title = Trace.Handlers.ModelHandlers.PageLoadMetrics.MetricName.FCP;
    }
    if (Trace.Types.Events.isAnyLargestContentfulPaintCandidate(markerEvent)) {
      color = '#1A3422';
      title = Trace.Handlers.ModelHandlers.PageLoadMetrics.MetricName.LCP;
    }
    if (Trace.Types.Events.isNavigationStart(markerEvent)) {
      color = '#FF9800';
      title = '';
    }
    return {
      title,
      dashStyle: tallMarkerDashStyle,
      lineWidth: 0.5,
      color,
      tall: true,
      lowPriority: false,
    };
  }

  markerStyleForExtensionMarker(markerEvent: Trace.Types.Extensions.SyntheticExtensionMarker): TimelineMarkerStyle {
    const tallMarkerDashStyle = [6, 4];
    const title = markerEvent.name;
    const color = Extensions.ExtensionUI.extensionEntryColor(markerEvent);
    return {
      title,
      dashStyle: tallMarkerDashStyle,
      lineWidth: 0.5,
      color,
      tall: true,
      lowPriority: false,
    };
  }

  /**
   * Gets the color an event added by this appender should be rendered with.
   */
  colorForEvent(event: Trace.Types.Events.Event): string {
    if (Trace.Types.Events.eventIsPageLoadEvent(event)) {
      return this.markerStyleForPageLoadEvent(event).color;
    }
    if (Trace.Types.Extensions.isSyntheticExtensionEntry(event)) {
      return Extensions.ExtensionUI.extensionEntryColor(event);
    }
    // Performance and console timings.
    return this.#colorGenerator.colorForID(event.name);
  }

  /**
   * Gets the title an event added by this appender should be rendered with.
   */
  titleForEvent(event: Trace.Types.Events.Event): string {
    const metricsHandler = Trace.Handlers.ModelHandlers.PageLoadMetrics;
    if (Trace.Types.Events.eventIsPageLoadEvent(event)) {
      switch (event.name) {
        case 'MarkDOMContent':
          return metricsHandler.MetricName.DCL;
        case 'MarkLoad':
          return metricsHandler.MetricName.L;
        case 'firstContentfulPaint':
          return metricsHandler.MetricName.FCP;
        case 'firstPaint':
          return metricsHandler.MetricName.FP;
        case 'largestContentfulPaint::Candidate':
          return metricsHandler.MetricName.LCP;
        case 'navigationStart':
          return '';
        default:
          return event.name;
      }
    }
    if (Trace.Types.Events.isConsoleTimeStamp(event)) {
      return `TimeStamp: ${event.args.data?.message ?? '(name unknown)'}`;
    }
    if (Trace.Types.Events.isPerformanceMark(event)) {
      return `[mark]: ${event.name}`;
    }
    // Trace.Types.Extensions.isSyntheticExtensionEntry(event) can fall through to event.name.
    return event.name;
  }

  setPopoverInfo(event: Trace.Types.Events.Event, info: PopoverInfo): void {
    // If an event is a marker event, rather than show a duration of 0, we can
    // instead show the time that the event happened, which is much more
    // useful. We do this currently for:
    // Page load events: DCL, FCP and LCP
    // performance.mark() events (including extensibility events)
    // console.timestamp() events

    const isExtensibilityMarker = Trace.Types.Extensions.isSyntheticExtensionEntry(event) &&
        Trace.Types.Extensions.isExtensionPayloadMarker(event.devtoolsObj);

    if (isExtensibilityMarker) {
      info.title = event.devtoolsObj.tooltipText || event.name;
    }

    if (Trace.Types.Events.isMarkerEvent(event) || Trace.Types.Events.isPerformanceMark(event) ||
        Trace.Types.Events.isConsoleTimeStamp(event) || isExtensibilityMarker) {
      const timeOfEvent = Trace.Helpers.Timing.timeStampForEventAdjustedByClosestNavigation(
          event,
          this.#parsedTrace.data.Meta.traceBounds,
          this.#parsedTrace.data.Meta.navigationsByNavigationId,
          this.#parsedTrace.data.Meta.softNavigationsById,
          this.#parsedTrace.data.Meta.navigationsByFrameId,
      );
      info.formattedTime = getDurationString(timeOfEvent);
    }
  }
}