// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import * as Common from '../../core/common/common.js';
import * as Host from '../../core/host/host.js';
import * as i18n from '../../core/i18n/i18n.js';
import * as Platform from '../../core/platform/platform.js';
import * as SDK from '../../core/sdk/sdk.js';
import type * as Protocol from '../../generated/protocol.js';
import * as EmulationModel from '../../models/emulation/emulation.js';

import * as Spec from './web-vitals-injected/spec/spec.js';

const UIStrings = {
  /**
   * @description Warning text indicating that the Largest Contentful Paint (LCP) performance metric was affected by the user changing the simulated device.
   */
  lcpEmulationWarning:
      'Simulating a new device after the page loads can affect LCP. Reload the page after simulating a new device for accurate LCP data.',
  /**
   * @description Warning text indicating that the Largest Contentful Paint (LCP) performance metric was affected by the page loading in the background.
   */
  lcpVisibilityWarning: 'LCP value may be inflated because the page started loading in the background.',
} as const;

const str_ = i18n.i18n.registerUIStrings('models/live-metrics/LiveMetrics.ts', UIStrings);
const i18nString = i18n.i18n.getLocalizedString.bind(undefined, str_);

const LIVE_METRICS_WORLD_NAME = 'DevTools Performance Metrics';

let liveMetricsInstance: LiveMetrics;

class InjectedScript {
  static #injectedScript?: string;
  static async get(): Promise<string> {
    if (!this.#injectedScript) {
      const url = new URL('./web-vitals-injected/web-vitals-injected.generated.js', import.meta.url);
      const result = await fetch(url);
      this.#injectedScript = await result.text();
    }
    return this.#injectedScript;
  }
}

export type InteractionMap = Map<InteractionId, Interaction>;

export class LiveMetrics extends Common.ObjectWrapper.ObjectWrapper<EventTypes> implements SDK.TargetManager.Observer {
  #enabled = false;
  #target?: SDK.Target.Target;
  #scriptIdentifier?: Protocol.Page.ScriptIdentifier;
  #lastResetContextId?: Protocol.Runtime.ExecutionContextId;
  #lcpValue?: LcpValue;
  #clsValue?: ClsValue;
  #inpValue?: InpValue;
  #interactions: InteractionMap = new Map();
  #interactionsByGroupId = new Map<Spec.InteractionEntryGroupId, Interaction[]>();
  #layoutShifts: LayoutShift[] = [];
  #lastEmulationChangeTime?: number;
  #mutex = new Common.Mutex.Mutex();
  #deviceModeModel = EmulationModel.DeviceModeModel.DeviceModeModel.tryInstance();

  private constructor() {
    super();
    const targetManager = SDK.TargetManager.TargetManager.instance();
    targetManager.observeTargets(this, {scoped: true});
    // Listen for target info changes to detect prerender activation.
    // Scoped observers don't receive events when a prerendered target becomes
    // primary because setScopeTarget() isn't called during that transition.
    targetManager.addEventListener(
        SDK.TargetManager.Events.AVAILABLE_TARGETS_CHANGED, this.#onAvailableTargetsChanged, this);
  }

  #onAvailableTargetsChanged(): void {
    const primaryTarget = SDK.TargetManager.TargetManager.instance().primaryPageTarget();
    if (primaryTarget && primaryTarget !== this.#target) {
      // Primary target changed (e.g., prerender activation). Switch to it.
      void this.#switchToTarget(primaryTarget);
    }
  }

  async #switchToTarget(newTarget: SDK.Target.Target): Promise<void> {
    if (this.#target) {
      await this.disable();
    }
    this.#target = newTarget;
    await this.enable();
  }

  static instance(opts: {forceNew?: boolean} = {forceNew: false}): LiveMetrics {
    const {forceNew} = opts;
    if (!liveMetricsInstance || forceNew) {
      liveMetricsInstance = new LiveMetrics();
    }

    return liveMetricsInstance;
  }

  get lcpValue(): LcpValue|undefined {
    return this.#lcpValue;
  }

  get clsValue(): ClsValue|undefined {
    return this.#clsValue;
  }

  get inpValue(): InpValue|undefined {
    return this.#inpValue;
  }

  get interactions(): InteractionMap {
    return this.#interactions;
  }

  get layoutShifts(): LayoutShift[] {
    return this.#layoutShifts;
  }

  /**
   * Will create a log message describing the interaction's LoAF scripts.
   * Returns true if the message is successfully logged.
   */
  async logInteractionScripts(interaction: Interaction): Promise<boolean> {
    if (!this.#target) {
      return false;
    }

    const executionContextId = this.#lastResetContextId;
    if (!executionContextId) {
      return false;
    }

    const scriptsTable = [];
    for (const loaf of interaction.longAnimationFrameTimings) {
      for (const script of loaf.scripts) {
        const scriptEndTime = script.startTime + script.duration;
        if (scriptEndTime < interaction.startTime) {
          continue;
        }

        const blockingDuration = Math.round(scriptEndTime - Math.max(interaction.startTime, script.startTime));

        // TODO: Use translated strings for the table
        scriptsTable.push({
          'Blocking duration': blockingDuration,
          'Invoker type': script.invokerType || null,
          Invoker: script.invoker || null,
          Function: script.sourceFunctionName || null,
          Source: script.sourceURL || null,
          'Char position': script.sourceCharPosition || null,
        });
      }
    }

    try {
      const scriptsLimit = Spec.LOAF_LIMIT * Spec.SCRIPTS_PER_LOAF_LIMIT;
      const scriptLimitText = scriptsTable.length === scriptsLimit ? ` (limited to ${scriptsLimit})` : '';
      const loafLimitText = interaction.longAnimationFrameTimings.length === Spec.LOAF_LIMIT ?
          ` (limited to last ${Spec.LOAF_LIMIT})` :
          '';
      await this.#target.runtimeAgent().invoke_evaluate({
        expression: `
          console.group('[DevTools] Long animation frames for ${interaction.duration}ms ${
            interaction.interactionType} interaction');
          console.log('Scripts${scriptLimitText}:');
          console.table(${JSON.stringify(scriptsTable)});
          console.log('Intersecting long animation frame events${loafLimitText}:', ${
            JSON.stringify(interaction.longAnimationFrameTimings)});
          console.groupEnd();
        `,
        contextId: executionContextId,
      });
    } catch {
      return false;
    }

    return true;
  }

  #onEmulationChanged(): void {
    this.#lastEmulationChangeTime = Date.now();
  }

  /**
   * DOM nodes can't be sent over a runtime binding, so we have to retrieve
   * them separately.
   */
  async #resolveNodeRef(index: number, executionContextId: Protocol.Runtime.ExecutionContextId):
      Promise<SDK.DOMModel.DOMNode|null> {
    if (!this.#target) {
      return null;
    }

    const runtimeModel = this.#target.model(SDK.RuntimeModel.RuntimeModel);
    if (!runtimeModel) {
      return null;
    }

    const domModel = this.#target.model(SDK.DOMModel.DOMModel);
    if (!domModel) {
      return null;
    }

    const {result} = await this.#target.runtimeAgent().invoke_evaluate({
      expression: `window.getNodeForIndex(${index})`,
      contextId: executionContextId,
    });

    if (!result) {
      return null;
    }

    let remoteObject;
    try {
      remoteObject = runtimeModel.createRemoteObject(result);
      const node = await domModel.pushObjectAsNodeToFrontend(remoteObject);
      if (!node) {
        return null;
      }

      return node;
    } catch {
      return null;
    } finally {
      remoteObject?.release();
    }
  }

  #sendStatusUpdate(): void {
    this.dispatchEventToListeners(Events.STATUS, {
      lcp: this.#lcpValue,
      cls: this.#clsValue,
      inp: this.#inpValue,
      interactions: this.#interactions,
      layoutShifts: this.#layoutShifts,
    });
  }

  setStatusForTesting(status: StatusEvent): void {
    this.#lcpValue = status.lcp;
    this.#clsValue = status.cls;
    this.#inpValue = status.inp;
    this.#interactions = status.interactions;
    this.#layoutShifts = status.layoutShifts;
    this.#sendStatusUpdate();
  }

  /**
   * If there is a document update then any node handles we have already resolved will be invalid.
   * This function should re-resolve any relevant DOM nodes after a document update.
   */
  async #onDocumentUpdate(event: Common.EventTarget.EventTargetEvent<SDK.DOMModel.DOMModel>): Promise<void> {
    const domModel = event.data;

    const toRefresh = [
      this.#lcpValue?.nodeRef,
      ...this.#interactions.values().map(i => i.nodeRef),
      ...this.#layoutShifts.flatMap(shift => shift.affectedNodeRefs),
    ].filter(nodeRef => !!nodeRef);

    const idsToRefresh = new Set(toRefresh.map(nodeRef => nodeRef.backendNodeId()));
    const nodes = await domModel.pushNodesByBackendIdsToFrontend(idsToRefresh);
    if (!nodes) {
      return;
    }

    for (let i = 0; i < toRefresh.length; i++) {
      const refreshedNode = nodes.get(toRefresh[i].backendNodeId());

      // It is possible for the refreshed node to be undefined even though it was defined previously.
      // We should keep the affected nodes consistent from the user perspective, so we will just keep the stale node instead of removing it.
      if (!refreshedNode) {
        continue;
      }

      toRefresh[i] = refreshedNode;
    }

    this.#sendStatusUpdate();
  }

  async #handleWebVitalsEvent(
      webVitalsEvent: Spec.WebVitalsEvent, executionContextId: Protocol.Runtime.ExecutionContextId): Promise<void> {
    switch (webVitalsEvent.name) {
      case 'LCP': {
        const warnings: string[] = [];
        const lcpEvent: LcpValue = {
          value: webVitalsEvent.value,
          phases: webVitalsEvent.phases,
          warnings,
        };
        if (webVitalsEvent.nodeIndex !== undefined) {
          const nodeRef = await this.#resolveNodeRef(webVitalsEvent.nodeIndex, executionContextId);
          if (nodeRef) {
            lcpEvent.nodeRef = nodeRef;
          }
        }

        if (this.#lastEmulationChangeTime && Date.now() - this.#lastEmulationChangeTime < 500) {
          warnings.push(i18nString(UIStrings.lcpEmulationWarning));
        }

        if (webVitalsEvent.startedHidden) {
          warnings.push(i18nString(UIStrings.lcpVisibilityWarning));
        }

        this.#lcpValue = lcpEvent;
        break;
      }
      case 'CLS': {
        const event: ClsValue = {
          value: webVitalsEvent.value,
          clusterShiftIds: webVitalsEvent.clusterShiftIds,
        };
        this.#clsValue = event;
        break;
      }
      case 'INP': {
        const inpEvent: InpValue = {
          value: webVitalsEvent.value,
          phases: webVitalsEvent.phases,
          interactionId: `interaction-${webVitalsEvent.entryGroupId}-${webVitalsEvent.startTime}`,
        };
        this.#inpValue = inpEvent;
        break;
      }
      case 'InteractionEntry': {
        const groupInteractions =
            Platform.MapUtilities.getWithDefault(this.#interactionsByGroupId, webVitalsEvent.entryGroupId, () => []);

        // `nextPaintTime` uses the event duration which is rounded to the nearest 8ms. The best we can do
        // is check if the `nextPaintTime`s are within 8ms.
        // https://developer.mozilla.org/en-US/docs/Web/API/PerformanceEntry/duration#event
        let interaction = groupInteractions.find(
            interaction => Math.abs(interaction.nextPaintTime - webVitalsEvent.nextPaintTime) < 8);

        if (!interaction) {
          interaction = {
            interactionId: `interaction-${webVitalsEvent.entryGroupId}-${webVitalsEvent.startTime}`,
            interactionType: webVitalsEvent.interactionType,
            duration: webVitalsEvent.duration,
            eventNames: [],
            phases: webVitalsEvent.phases,
            startTime: webVitalsEvent.startTime,
            nextPaintTime: webVitalsEvent.nextPaintTime,
            longAnimationFrameTimings: webVitalsEvent.longAnimationFrameEntries,
          };

          groupInteractions.push(interaction);
          this.#interactions.set(interaction.interactionId, interaction);
        }

        // We can get multiple instances of the first input interaction since web-vitals.js installs
        // an extra listener for events of type `first-input`. This is a simple way to de-dupe those
        // events without adding complexity to the injected code.
        if (!interaction.eventNames.includes(webVitalsEvent.eventName)) {
          interaction.eventNames.push(webVitalsEvent.eventName);
        }

        if (webVitalsEvent.nodeIndex !== undefined) {
          const node = await this.#resolveNodeRef(webVitalsEvent.nodeIndex, executionContextId);
          if (node) {
            interaction.nodeRef = node;
          }
        }
        break;
      }
      case 'LayoutShift': {
        const nodePromises = webVitalsEvent.affectedNodeIndices.map(nodeIndex => {
          return this.#resolveNodeRef(nodeIndex, executionContextId);
        });

        const affectedNodes = (await Promise.all(nodePromises)).filter(nodeRef => !!nodeRef);

        const layoutShift: LayoutShift = {
          score: webVitalsEvent.score,
          uniqueLayoutShiftId: webVitalsEvent.uniqueLayoutShiftId,
          affectedNodeRefs: affectedNodes,
        };
        this.#layoutShifts.push(layoutShift);
        break;
      }
      case 'reset': {
        this.#lcpValue = undefined;
        this.#clsValue = undefined;
        this.#inpValue = undefined;
        this.#interactions.clear();
        this.#layoutShifts = [];
        break;
      }
    }

    this.#sendStatusUpdate();
  }

  #isPrimaryFrameExecutionContext(executionContextId: Protocol.Runtime.ExecutionContextId): boolean {
    if (!this.#target) {
      return false;
    }

    const runtimeModel = this.#target.model(SDK.RuntimeModel.RuntimeModel);
    const executionContext = runtimeModel?.executionContext(executionContextId);
    if (!executionContext?.frameId) {
      return false;
    }

    const resourceTreeModel = this.#target.model(SDK.ResourceTreeModel.ResourceTreeModel);
    const primaryFrameId = resourceTreeModel?.mainFrame?.id;
    return Boolean(primaryFrameId && executionContext.frameId === primaryFrameId);
  }

  async #onBindingCalled(event: {data: Protocol.Runtime.BindingCalledEvent}): Promise<void> {
    const {data} = event;
    if (data.name !== Spec.EVENT_BINDING_NAME) {
      return;
    }

    // Async tasks can be performed while handling an event (e.g. resolving DOM node)
    // Use a mutex here to ensure the events are handled in the order they are received.
    await this.#mutex.run(async () => {
      const webVitalsEvent = JSON.parse(data.payload) as Spec.WebVitalsEvent;

      if (webVitalsEvent.name === 'reset') {
        if (!this.#isPrimaryFrameExecutionContext(data.executionContextId)) {
          return;
        }
        this.#lastResetContextId = data.executionContextId;
      }

      if (this.#lastResetContextId !== data.executionContextId) {
        return;
      }

      await this.#handleWebVitalsEvent(webVitalsEvent, data.executionContextId);
    });
  }

  async #killAllLiveMetricContexts(): Promise<void> {
    const target = this.#target;
    if (!target) {
      return;
    }

    const runtimeModel = target.model(SDK.RuntimeModel.RuntimeModel);
    if (!runtimeModel) {
      return;
    }

    const killPromises = runtimeModel.executionContexts()
                             .filter(e => e.name === LIVE_METRICS_WORLD_NAME && !e.isDefault)
                             .map(e => target.runtimeAgent().invoke_evaluate({
                               // On the off chance something else creates execution contexts with the exact same name
                               // this expression should just be a noop.
                               expression: `window?.${Spec.INTERNAL_KILL_SWITCH}?.()`,
                               contextId: e.id,
                             }));

    await Promise.all(killPromises);
  }

  clearInteractions(): void {
    this.#interactions.clear();
    this.#sendStatusUpdate();
  }

  clearLayoutShifts(): void {
    this.#layoutShifts = [];
    this.#sendStatusUpdate();
  }

  async targetAdded(target: SDK.Target.Target): Promise<void> {
    // Scoped observers can also receive events for OOPIFs and workers.
    if (target !== SDK.TargetManager.TargetManager.instance().primaryPageTarget()) {
      return;
    }
    this.#target = target;
    await this.enable();
  }

  async targetRemoved(target: SDK.Target.Target): Promise<void> {
    // Scoped observers can also receive events for OOPIFs and workers.
    if (target !== this.#target) {
      return;
    }
    await this.disable();
    this.#target = undefined;
  }

  async enable(): Promise<void> {
    if (Host.InspectorFrontendHost.isUnderTest()) {
      // Enabling this impacts a lot of layout tests; we will work on fixing
      // them but for now it is easier to not run this page in layout tests.
      // b/360064852
      return;
    }

    if (!this.#target || this.#enabled) {
      return;
    }

    // Only frame targets will actually give us CWV
    if (this.#target.type() !== SDK.Target.Type.FRAME) {
      return;
    }

    const domModel = this.#target.model(SDK.DOMModel.DOMModel);
    if (!domModel) {
      return;
    }

    domModel.addEventListener(SDK.DOMModel.Events.DocumentUpdated, this.#onDocumentUpdate, this);

    const runtimeModel = this.#target.model(SDK.RuntimeModel.RuntimeModel);
    if (!runtimeModel) {
      return;
    }

    runtimeModel.addEventListener(SDK.RuntimeModel.Events.BindingCalled, this.#onBindingCalled, this);

    await runtimeModel.addBinding({
      name: Spec.EVENT_BINDING_NAME,
      executionContextName: LIVE_METRICS_WORLD_NAME,
    });

    // If DevTools is closed and reopened, the live metrics context from the previous
    // session will persist. We should ensure any old live metrics contexts are killed
    // before starting a new one.
    await this.#killAllLiveMetricContexts();

    const source = await InjectedScript.get();

    // Extra check in case the target was removed while we were initializing.
    // It's possible to be halfway-through enabling when the target is removed
    // eg try loading 'devtools://devtools/bundled/devtools_app.html?ws=127.0.0.1:99/blah'
    if (!this.#target) {
      return;
    }
    const {identifier} = await this.#target?.pageAgent().invoke_addScriptToEvaluateOnNewDocument({
      source,
      worldName: LIVE_METRICS_WORLD_NAME,
      runImmediately: true,
    });
    this.#scriptIdentifier = identifier;

    this.#deviceModeModel?.addEventListener(
        EmulationModel.DeviceModeModel.Events.UPDATED, this.#onEmulationChanged, this);

    this.#enabled = true;
  }

  async disable(): Promise<void> {
    if (!this.#target || !this.#enabled) {
      return;
    }

    // Reset to ensure clean state when re-enabling on a new target.
    // See crbug.com/478832430.
    this.#lastResetContextId = undefined;

    await this.#killAllLiveMetricContexts();

    const runtimeModel = this.#target.model(SDK.RuntimeModel.RuntimeModel);
    if (runtimeModel) {
      await runtimeModel.removeBinding({
        name: Spec.EVENT_BINDING_NAME,
      });

      runtimeModel.removeEventListener(SDK.RuntimeModel.Events.BindingCalled, this.#onBindingCalled, this);
    }

    const domModel = this.#target.model(SDK.DOMModel.DOMModel);
    if (domModel) {
      domModel.removeEventListener(SDK.DOMModel.Events.DocumentUpdated, this.#onDocumentUpdate, this);
    }

    if (this.#scriptIdentifier) {
      await this.#target.pageAgent().invoke_removeScriptToEvaluateOnNewDocument({
        identifier: this.#scriptIdentifier,
      });
    }
    this.#scriptIdentifier = undefined;

    this.#deviceModeModel?.removeEventListener(
        EmulationModel.DeviceModeModel.Events.UPDATED, this.#onEmulationChanged, this);

    this.#enabled = false;
  }
}

export const enum Events {
  STATUS = 'status',
}

export type InteractionId = `interaction-${number}-${number}`;

export interface MetricValue {
  value: number;
  warnings?: string[];
}

export interface LcpValue extends MetricValue {
  phases: Spec.LcpPhases;
  nodeRef?: SDK.DOMModel.DOMNode;
}

export interface InpValue extends MetricValue {
  phases: Spec.InpPhases;
  interactionId: InteractionId;
}

export interface ClsValue extends MetricValue {
  clusterShiftIds: Spec.UniqueLayoutShiftId[];
}

export interface LayoutShift {
  score: number;
  uniqueLayoutShiftId: Spec.UniqueLayoutShiftId;
  affectedNodeRefs: SDK.DOMModel.DOMNode[];
}

export interface Interaction {
  interactionId: InteractionId;
  interactionType: Spec.InteractionEntryEvent['interactionType'];
  eventNames: string[];
  duration: number;
  startTime: number;
  nextPaintTime: number;
  phases: Spec.InpPhases;
  longAnimationFrameTimings: Spec.PerformanceLongAnimationFrameTimingJSON[];
  nodeRef?: SDK.DOMModel.DOMNode;
}

export interface StatusEvent {
  lcp?: LcpValue;
  cls?: ClsValue;
  inp?: InpValue;
  interactions: InteractionMap;
  layoutShifts: LayoutShift[];
}

interface EventTypes {
  [Events.STATUS]: StatusEvent;
}