elf_magic/
lib.rs

1mod builder;
2mod codegen;
3mod config;
4mod error;
5mod programs;
6mod workspace;
7
8use std::{env, path::PathBuf};
9
10use crate::{
11    config::Config,
12    error::Error,
13    programs::{GenerationResult, SolanaProgram},
14};
15
16/// Generate Rust bindings for Solana programs
17pub fn generate() -> Result<GenerationResult, Error> {
18    let cargo_manifest_dir = env::var("CARGO_MANIFEST_DIR")
19        .map(PathBuf::from)
20        .map_err(|e| {
21            let message = format!("CARGO_MANIFEST_DIR not set: {}", e);
22            Error::WorkspaceDiscovery(message)
23        })?;
24
25    // Core pipeline:
26    // 1. load config
27    // 2. load workspaces
28    // 3. discover programs across all workspaces
29    // 4. build included programs across all workspaces
30    // 5. generate code - one lib.rs from all included programs
31    // 6. enable incremental builds for all included programs
32
33    let config = Config::load(&cargo_manifest_dir)?;
34
35    let workspaces = workspace::load_workspaces(&config)?;
36
37    let discovered_programs = workspaces
38        .iter()
39        .map(|w| w.discover_programs())
40        .collect::<Result<Vec<_>, _>>()?;
41
42    // Extract all programs for building and code generation
43    let included_programs: Vec<SolanaProgram> = discovered_programs
44        .iter()
45        .flat_map(|w| w.included.iter().cloned())
46        .collect();
47
48    builder::build_programs(&included_programs)?;
49
50    let code = codegen::generate(&included_programs)?;
51    codegen::save(&cargo_manifest_dir, &code)?;
52
53    enable_incremental_builds(&included_programs)?;
54
55    let mode = match &config {
56        Config::Magic => "magic".to_string(),
57        Config::Pedantic { .. } => "pedantic".to_string(),
58    };
59
60    Ok(GenerationResult::new(mode, discovered_programs))
61}
62
63/// Enable incremental builds for each program
64fn enable_incremental_builds(programs: &[SolanaProgram]) -> Result<(), Error> {
65    for program in programs {
66        println!("cargo:rerun-if-changed={}", program.manifest_path.display());
67    }
68    Ok(())
69}