analyzed-ra-ide 0.4.2

use analyzed_bridge as build_support;
use analyzed_bridge::ast;

use std::{
    env,
    error::Error,
    fs,
    path::{Path, PathBuf},
};

const PACKAGE: &str = "ra_ap_ide";
const GENERATED_DIR: &str = "ra_ap_ide_bridge";

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let (generated, _) = build_support::prepare_bridge_package(PACKAGE, GENERATED_DIR)?;
    patch_ide_source(&generated.join("src/lib.rs"))?;
    patch_view_crate_graph_source(&generated.join("src/view_crate_graph.rs"))?;
    patch_syntax_highlighting_benches(&generated.join("src/syntax_highlighting/tests.rs"))?;
    println!("cargo:rerun-if-changed=build.rs");
    Ok(())
}

fn patch_ide_source(lib_rs: &Path) -> Result<(), Box<dyn Error>> {
    let mut source = fs::read_to_string(lib_rs)?;

    let visibility = owned_source_path("visibility.rs");
    build_support::mount_module(&mut source, None, "visibility", &visibility);
    println!("cargo:rerun-if-changed={}", visibility.display());
    build_support::append::<ast::Struct>(
        &mut source,
        "Analysis",
        &[build_support::Field {
            vis: None,
            name: "guard",
            ty: "Option<crate::visibility::AnalysisGuard>",
        }],
    )?;
    build_support::append_record_fields(
        &mut source,
        "analysis",
        "Analysis",
        &[build_support::FieldInit {
            name: "guard",
            value: Some("None"),
        }],
    )?;
    build_support::append_record_fields(
        &mut source,
        "from_ra_fixture_with_on_cursor",
        "Analysis",
        &[build_support::FieldInit {
            name: "guard",
            value: Some("None"),
        }],
    )?;

    fs::write(lib_rs, source)?;
    Ok(())
}

fn patch_view_crate_graph_source(view_crate_graph_rs: &Path) -> Result<(), Box<dyn Error>> {
    let mut source = fs::read_to_string(view_crate_graph_rs)?;

    build_support::retarget_use(&mut source, "all_crates", "crate::visibility::all_crates")?;

    fs::write(view_crate_graph_rs, source)?;
    Ok(())
}

// The upstream skip_slow_tests helper writes a cookie into the rust-analyzer
// checkout when slow tests run, which resolves to the cargo registry source
// cache for registry packages. The benchmark tests load bench_data from the
// checkout, which the registry package does not contain.
fn patch_syntax_highlighting_benches(tests_rs: &Path) -> Result<(), Box<dyn Error>> {
    let mut source = fs::read_to_string(tests_rs)?;

    for benchmark in [
        "benchmark_syntax_highlighting_long_struct",
        "syntax_highlighting_not_quadratic",
        "benchmark_syntax_highlighting_parser",
    ] {
        build_support::add_attr::<ast::Fn>(
            &mut source,
            benchmark,
            "#[ignore = \"bench_data not available in registry packages\"]",
        )?;
    }

    fs::write(tests_rs, source)?;
    Ok(())
}

fn owned_source_path(file_name: &str) -> PathBuf {
    PathBuf::from(env::var_os("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR is set by Cargo"))
        .join("src")
        .join(file_name)
}