vercel__next.js/crates/next-custom-transforms/tests/telemetry.rs

use std::{
    cell::RefCell,
    rc::Rc,
    sync::{Arc, LazyLock},
};

use next_custom_transforms::transforms::next_ssg::next_ssg;
use rustc_hash::FxHashSet;
use swc_core::{
    atoms::Atom,
    base::{Compiler, try_with_handler},
    common::{FileName, FilePathMapping, GLOBALS, SourceMap, comments::SingleThreadedComments},
    ecma::ast::noop_pass,
};

static COMPILER: LazyLock<Arc<Compiler>> = LazyLock::new(|| {
    let cm = Arc::new(SourceMap::new(FilePathMapping::empty()));

    Arc::new(Compiler::new(cm))
});

#[test]
fn should_collect_estimated_third_party_packages() {
    let eliminated_packages: Rc<RefCell<FxHashSet<Atom>>> = Default::default();
    let fm = COMPILER.cm.new_source_file(
        FileName::Real("fixture.js".into()).into(),
        r#"import http from 'http'
import { hash } from '@napi-rs/bcrypt'

import { omit } from '~/utils/omit'
import config from './data.json'

export default () => 'Hello World'

export function getServerSideProps() {
  console.log(http)
  console.log(config)
  return { props: { digest: hash('hello') } }
}
"#
        .to_owned(),
    );

    let swc_options = swc_core::base::config::Options {
        runtime_options: swc_core::base::config::RuntimeOptions::default()
            .plugin_runtime(Arc::new(swc_plugin_backend_wasmtime::WasmtimeRuntime)),
        ..Default::default()
    };

    try_with_handler(COMPILER.cm.clone(), Default::default(), |handler| {
        GLOBALS.set(&Default::default(), || {
            let comments = SingleThreadedComments::default();
            COMPILER.process_js_with_custom_pass(
                fm,
                None,
                handler,
                &swc_options,
                comments,
                |_| next_ssg(eliminated_packages.clone()),
                |_| noop_pass(),
            )
        })
    })
    .unwrap();

    let mut eliminated_packages_vec = Rc::into_inner(eliminated_packages)
        .expect("we should have the only remaining reference to `eliminated_packages`")
        .into_inner()
        .into_iter()
        .collect::<Vec<Atom>>();
    eliminated_packages_vec.sort_unstable(); // HashSet order is random/arbitrary
    assert_eq!(eliminated_packages_vec, vec!["@napi-rs/bcrypt", "http"]);
}