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

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

use next_custom_transforms::transforms::next_ssg::next_ssg;
use once_cell::sync::Lazy;
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: Lazy<Arc<Compiler>> = Lazy::new(|| {
    let cm = Arc::new(SourceMap::new(FilePathMapping::empty()));

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

#[test]
fn should_collect_estimated_third_part_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(),
    );
    assert!(
        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,
                    &Default::default(),
                    comments,
                    |_| next_ssg(eliminated_packages.clone()),
                    |_| noop_pass(),
                )
            })
        })
        .is_ok()
    );
    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"]);
}