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 (collect successes and failures)
30    // 5. generate code - one lib.rs from successfully built programs, with build status
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    // Deduplicate programs by manifest_path to avoid duplicate constants
49    let included_programs = deduplicate_programs(included_programs);
50
51    let build_result = builder::build_programs(&included_programs);
52
53    // Generate code from successfully built programs, including build status
54    let code = codegen::generate(&build_result)?;
55    codegen::save(&cargo_manifest_dir, &code)?;
56
57    // Enable incremental builds for all programs (successful and failed)
58    enable_incremental_builds(&included_programs)?;
59
60    let mode = match &config {
61        Config::Magic => "magic".to_string(),
62        Config::Pedantic { .. } => "pedantic".to_string(),
63    };
64
65    Ok(GenerationResult::new(mode, discovered_programs))
66}
67
68/// Enable incremental builds for each program
69fn enable_incremental_builds(programs: &[SolanaProgram]) -> Result<(), Error> {
70    for program in programs {
71        println!("cargo:rerun-if-changed={}", program.manifest_path.display());
72    }
73    Ok(())
74}
75
76/// Deduplicate programs by manifest_path to handle cases where multiple workspaces
77/// discover the same program (e.g., shared dependencies)
78fn deduplicate_programs(programs: Vec<SolanaProgram>) -> Vec<SolanaProgram> {
79    use std::collections::HashMap;
80
81    let mut seen: HashMap<PathBuf, SolanaProgram> = HashMap::new();
82
83    for program in programs {
84        // Use manifest_path as the key, keeping the first occurrence
85        seen.entry(program.manifest_path.clone()).or_insert(program);
86    }
87
88    seen.into_values().collect()
89}
90
91#[cfg(test)]
92mod tests {
93    use super::*;
94    use std::path::PathBuf;
95
96    #[test]
97    fn test_deduplicate_programs_removes_duplicates_by_manifest_path() {
98        // Same program appearing twice (realistic scenario from multiple workspaces)
99        let program1 = SolanaProgram {
100            package_name: "apl-token".to_string(),
101            target_name: "apl_token".to_string(),
102            manifest_path: PathBuf::from("/repo/token/Cargo.toml"),
103        };
104
105        let program2 = SolanaProgram {
106            package_name: "apl-token".to_string(),
107            target_name: "apl_token".to_string(),
108            manifest_path: PathBuf::from("/repo/token/Cargo.toml"), // Same path!
109        };
110
111        let escrow_program = SolanaProgram {
112            package_name: "escrow_program".to_string(),
113            target_name: "escrow_program".to_string(),
114            manifest_path: PathBuf::from("/repo/examples/escrow/program/Cargo.toml"),
115        };
116
117        // Input: 3 programs (with 1 duplicate)
118        let input_programs = vec![escrow_program.clone(), program1, program2];
119        assert_eq!(input_programs.len(), 3);
120
121        // After deduplication: should have 2 unique programs
122        let deduplicated = deduplicate_programs(input_programs);
123        assert_eq!(
124            deduplicated.len(),
125            2,
126            "Should deduplicate to 2 unique programs"
127        );
128
129        // Verify we have one of each program type
130        let apl_token_count = deduplicated
131            .iter()
132            .filter(|p| p.target_name == "apl_token")
133            .count();
134        assert_eq!(
135            apl_token_count, 1,
136            "Should have exactly 1 apl_token after deduplication"
137        );
138
139        let escrow_count = deduplicated
140            .iter()
141            .filter(|p| p.target_name == "escrow_program")
142            .count();
143        assert_eq!(escrow_count, 1, "Should have exactly 1 escrow_program");
144    }
145
146    #[test]
147    fn test_deduplicate_programs_preserves_unique_programs() {
148        let program1 = SolanaProgram {
149            package_name: "counter".to_string(),
150            target_name: "counter_program".to_string(),
151            manifest_path: PathBuf::from("/repo/examples/counter/program/Cargo.toml"),
152        };
153
154        let program2 = SolanaProgram {
155            package_name: "escrow".to_string(),
156            target_name: "escrow_program".to_string(),
157            manifest_path: PathBuf::from("/repo/examples/escrow/program/Cargo.toml"),
158        };
159
160        // Input: 2 unique programs
161        let input_programs = vec![program1, program2];
162        let deduplicated = deduplicate_programs(input_programs);
163
164        // Should preserve both programs
165        assert_eq!(deduplicated.len(), 2, "Should preserve all unique programs");
166    }
167}
elf_magic/lib.rs

elf_magic/
lib.rs
