use cxx_llvm_build_common::prelude::*;
use std::path::{Path, PathBuf};

pub fn project_dir() -> BoxResult<std::path::PathBuf> {
    let cargo_manifest_dir = std::env::var("CARGO_MANIFEST_DIR")?;
    let project_dir = std::path::PathBuf::from(cargo_manifest_dir);
    Ok(project_dir)
}

fn process_cxx(out_dir: &Path) -> BoxResult<()> {
    let cargo_pkg_name = "cxx-swift-auto";
    let llvm_dirs = cxx_llvm_build::Dirs::new(cargo_pkg_name)?;
    let clang_dirs = cxx_clang_build::Dirs::new(cargo_pkg_name, &llvm_dirs)?;
    let swift_dirs = cxx_swift_build::Dirs::new(cargo_pkg_name, &llvm_dirs, &clang_dirs)?;
    let includes = &[
        llvm_dirs.llvm_project.join("llvm/include"),
        llvm_dirs.llvm_cmake_build.join("include"),
        clang_dirs.clang_project.join("include"),
        clang_dirs.clang_cmake_build.join("include"),
        swift_dirs.swift_project.clone(),
        swift_dirs.swift_project.join("swift/include"),
        swift_dirs.swift_project.join("swift/stdlib/public/SwiftShims"),
        swift_dirs.swift_cmake_build.join("include"),
    ];
    cxx_build::CFG
        .exported_header_dirs
        .extend(includes.iter().map(PathBuf::as_path));
    let rust_source_files = &[
        out_dir.join("src/auto/swift/ast/ast_context.rs"),
        out_dir.join("src/auto/swift/ast/ast_walker_base.rs"),
        out_dir.join("src/auto/swift/ast/ast_walker_rust.rs"),
        out_dir.join("src/auto/swift/ast/ast_walker.rs"),
        out_dir.join("src/auto/swift/ast/ast_walker/parent_ty.rs"),
        out_dir.join("src/auto/swift/ast/ast_walker/pre_walk_result_expr.rs"),
        out_dir.join("src/auto/swift/ast/decl.rs"),
        out_dir.join("src/auto/swift/ast/dependency_tracker.rs"),
        out_dir.join("src/auto/swift/ast/diagnostic_engine.rs"),
        out_dir.join("src/auto/swift/ast/expr.rs"),
        out_dir.join("src/auto/swift/ast/identifier.rs"),
        out_dir.join("src/auto/swift/ast/identifier/small_vector.rs"),
        out_dir.join("src/auto/swift/ast/identifier/small_vector_impl.rs"),
        out_dir.join("src/auto/swift/ast/import_path/module.rs"),
        out_dir.join("src/auto/swift/ast/import_path/module/builder.rs"),
        out_dir.join("src/auto/swift/ast/module_decl.rs"),
        out_dir.join("src/auto/swift/ast/pattern.rs"),
        out_dir.join("src/auto/swift/ast/search_path_options.rs"),
        out_dir.join("src/auto/swift/ast/sil_options.rs"),
        out_dir.join("src/auto/swift/ast/stmt.rs"),
        out_dir.join("src/auto/swift/ast/type_repr.rs"),
        out_dir.join("src/auto/swift/basic/clang_importer_options.rs"),
        out_dir.join("src/auto/swift/basic/lang_options.rs"),
        out_dir.join("src/auto/swift/basic/source_loc.rs"),
        out_dir.join("src/auto/swift/basic/source_manager.rs"),
        out_dir.join("src/auto/swift/basic/type_checker_options.rs"),
        out_dir.join("src/auto/swift/clang_importer/clang_importer.rs"),
        out_dir.join("src/auto/swift/clang_importer/effective_clang_context.rs"),
        out_dir.join("src/auto/swift/clang_importer/serialized_swift_name.rs"),
        out_dir.join("src/auto/swift/clang_importer/serialized_swift_name/small_vector.rs"),
        out_dir.join("src/auto/swift/clang_importer/serialized_swift_name/small_vector_boxed.rs"),
        out_dir.join("src/auto/swift/clang_importer/serialized_swift_name/small_vector_impl.rs"),
        out_dir.join("src/auto/swift/clang_importer/swift_lookup_table.rs"),
        out_dir.join("src/auto/swift/clang_importer/swift_lookup_table/single_entry.rs"),
        out_dir.join("src/auto/swift/clang_importer/swift_lookup_table/single_entry/small_vector.rs"),
        out_dir.join("src/auto/swift/clang_importer/swift_lookup_table/single_entry/small_vector_boxed.rs"),
        out_dir.join("src/auto/swift/clang_importer/swift_lookup_table/single_entry/small_vector_impl.rs"),
        out_dir.join("src/auto/swift/symbol_graph_gen/symbol_graph_options.rs"),
        PathBuf::from("src/proxy/ast_walker_dyn.rs"),
    ];
    let files: &[&str] = &[];
    let output = "cxx-swift-auto";
    cxx_swift_build::cxx_build(&llvm_dirs, &clang_dirs, &swift_dirs, rust_source_files, files, output)?;
    Ok(())
}

fn main() -> BoxResult<()> {
    println!("cargo:rerun-if-changed=auto");
    println!("cargo:rerun-if-changed=cxx");
    let project_dir = project_dir()?;
    let out_dir = std::env::var("OUT_DIR")?;
    let out_dir = PathBuf::from(out_dir);
    let cfg_dir = project_dir.join("auto");
    cxx_auto::process_artifacts(&project_dir, &out_dir, &cfg_dir)?;
    process_cxx(&out_dir)?;
    Ok(())
}