/**
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */
/// <reference lib="es2021.WeakRef" />

import {promisify} from 'node:util';
import {getHeapSnapshot, setFlagsFromString} from 'node:v8';
import {runInNewContext} from 'node:vm';
import {isPrimitive} from '@jest/get-type';
import {format as prettyFormat} from 'pretty-format';

interface LeakDetectorOptions {
  shouldGenerateV8HeapSnapshot: boolean;
}

const tick = promisify(setImmediate);

export default class LeakDetector {
  private _isReferenceBeingHeld: boolean;
  private _shouldGenerateV8HeapSnapshot: boolean;
  private readonly _finalizationRegistry?: FinalizationRegistry<undefined>;

  constructor(value: unknown, opt?: LeakDetectorOptions) {
    if (isPrimitive(value)) {
      throw new TypeError(
        [
          'Primitives cannot leak memory.',
          `You passed a ${typeof value}: <${prettyFormat(value)}>`,
        ].join(' '),
      );
    }

    // When `_finalizationRegistry` is GCed the callback we set will no longer be called,
    this._finalizationRegistry = new FinalizationRegistry(() => {
      this._isReferenceBeingHeld = false;
    });
    this._finalizationRegistry.register(value as object, undefined);

    this._isReferenceBeingHeld = true;

    this._shouldGenerateV8HeapSnapshot =
      opt?.shouldGenerateV8HeapSnapshot ?? true;

    // Ensure value is not leaked by the closure created by the "weak" callback.
    // eslint-disable-next-line no-useless-assignment
    value = null;
  }

  async isLeaking(): Promise<boolean> {
    this._runGarbageCollector();

    // wait some ticks to allow GC to run properly, see https://github.com/nodejs/node/issues/34636#issuecomment-669366235
    for (let i = 0; i < 10; i++) {
      await tick();
    }

    if (this._isReferenceBeingHeld) {
      if (this._shouldGenerateV8HeapSnapshot) {
        // Triggering a heap snapshot is more aggressive than just calling `global.gc()`,
        // but it's also quite slow. Only do it if we still think we're leaking.
        // See: https://github.com/nodejs/node/pull/48510#issuecomment-1719289759
        getHeapSnapshot();
      }

      for (let i = 0; i < 10; i++) {
        await tick();
      }
    }

    return this._isReferenceBeingHeld;
  }

  private _runGarbageCollector() {
    const isGarbageCollectorHidden = globalThis.gc == null;

    // GC is usually hidden, so we have to expose it before running.
    setFlagsFromString('--expose-gc');
    runInNewContext('gc')();

    // The GC was not initially exposed, so let's hide it again.
    if (isGarbageCollectorHidden) {
      setFlagsFromString('--no-expose-gc');
    }
  }
}
