fix: handle missing OZ contract

Author: zachfedor

crates/stellar-scaffold-cli/src/commands/init.rs

@@ -86,19 +86,28 @@ impl Cmd {
         }
 
         // Update the project with the latest OpenZeppelin examples
-        self.update_oz_example(&absolute_project_path, "nft-enumerable", global_args)
-            .await?;
+        let example_contracts = ["nft-enumerable", "fungible-allowlist-example"];
+
+        for contract in example_contracts {
+            self.update_oz_example(&absolute_project_path, contract, global_args)
+                .await?;
+        }
 
         printer.checkln(format!("Project successfully created at {project_str}"));
         Ok(())
     }
 
+    /// Updates the project with an OpenZeppelin example contract
+    ///
+    /// This method attempts to generate a contract from OpenZeppelin
+    /// and prints a warning if it can't be found or generated.
     async fn update_oz_example(
         &self,
         absolute_project_path: &PathBuf,
-        contract_path: &'static str,
+        contract_path: &str,
         global_args: &global::Args,
     ) -> Result<(), Error> {
+        let printer = Print::new(global_args.quiet);
         let original_dir = env::current_dir()?;
         env::set_current_dir(absolute_project_path)?;
 
@@ -112,7 +121,7 @@ impl Cmd {
         let mut quiet_global_args = global_args.clone();
         quiet_global_args.quiet = true;
 
-        generate::contract::Cmd {
+        let result = generate::contract::Cmd {
             from: Some(contract_path.to_owned()),
             ls: false,
             from_wizard: false,
@@ -124,8 +133,32 @@ impl Cmd {
             ),
         }
         .run(&quiet_global_args)
-        .await?;
-        env::set_current_dir(original_dir)?;
+        .await;
+
+        // Restore directory before handling result
+        let _ = env::set_current_dir(original_dir);
+
+        match result {
+            Ok(()) => {
+                printer.infoln(format!(
+                    "Successfully added OpenZeppelin example contract: {}",
+                    contract_path
+                ));
+            }
+            Err(generate::contract::Error::ExampleNotFound(_)) => {
+                printer.infoln(format!(
+                    "Skipping OpenZeppelin example contract: {} (not found)",
+                    contract_path
+                ));
+            }
+            Err(e) => {
+                printer.warnln(format!(
+                    "Failed to generate example contract: {}\n{}",
+                    contract_path, e
+                ));
+            }
+        }
+
         Ok(())
     }
 }