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
16pub 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 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 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
63fn 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}