ReactNativeApplicationModel.ts

// Copyright (c) Meta Platforms, Inc. and affiliates.
// Copyright 2024 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import * as Host from '../../core/host/host.js';
import type * as ProtocolProxyApi from '../../generated/protocol-proxy-api.js';
import type * as Protocol from '../../generated/protocol.js';

import {SDKModel} from './SDKModel.js';
import type {Target} from './Target.js';

export class ReactNativeApplicationModel extends SDKModel<EventTypes> implements ProtocolProxyApi.ReactNativeApplicationDispatcher {
  #enabled: boolean;
  readonly #agent: ProtocolProxyApi.ReactNativeApplicationApi;

  metadataCached: Protocol.ReactNativeApplication.MetadataUpdatedEvent | null = null;

  constructor(target: Target) {
    super(target);

    Host.rnPerfMetrics.fuseboxSetClientMetadataStarted();

    this.#enabled = false;
    this.#agent = target.reactNativeApplicationAgent();
    target.registerReactNativeApplicationDispatcher(this);

    // Auto-init. Paired with registering this model immediately in rn_fusebox.ts.
    this.ensureEnabled();
  }

  ensureEnabled(): void {
    if (this.#enabled) {
      return;
    }

    void this.#agent.invoke_enable()
      .then(result => {
        const maybeError = result.getError();
        const success = !maybeError;
        Host.rnPerfMetrics.fuseboxSetClientMetadataFinished(success, maybeError);
      })
      .catch(reason => {
        const success = false;
        Host.rnPerfMetrics.fuseboxSetClientMetadataFinished(success, reason);
      });
    this.#enabled = true;
  }

  metadataUpdated(metadata: Protocol.ReactNativeApplication.MetadataUpdatedEvent): void {
    this.metadataCached = metadata;
    this.dispatchEventToListeners(Events.METADATA_UPDATED, metadata);
  }

  traceRequested(): void {
    this.dispatchEventToListeners(Events.TRACE_REQUESTED);
  }
}

export const enum Events {
  METADATA_UPDATED = 'MetadataUpdated',
  TRACE_REQUESTED = 'TraceRequested',
}

export interface EventTypes {
  [Events.METADATA_UPDATED]: Protocol.ReactNativeApplication.MetadataUpdatedEvent;
  [Events.TRACE_REQUESTED]: void;
}