diff --git a/hugr-cli/src/extensions.rs b/hugr-cli/src/extensions.rs index 8370e5648..a6c5ff33c 100644 --- a/hugr-cli/src/extensions.rs +++ b/hugr-cli/src/extensions.rs @@ -1,5 +1,6 @@ //! Dump standard extensions in serialized form. use clap::Parser; +use hugr_core::extension::ExtensionRegistry; use std::{io::Write, path::PathBuf}; /// Dump the standard extensions. @@ -24,10 +25,10 @@ impl ExtArgs { /// Write out the standard extensions in serialized form. /// Qualified names of extensions used to generate directories under the specified output directory. /// E.g. extension "foo.bar.baz" will be written to "OUTPUT/foo/bar/baz.json". - pub fn run_dump(&self) { + pub fn run_dump(&self, registry: &ExtensionRegistry) { let base_dir = &self.outdir; - for (name, ext) in hugr_core::std_extensions::STD_REG.iter() { + for (name, ext) in registry.iter() { let mut path = base_dir.clone(); for part in name.split('.') { path.push(part); diff --git a/hugr-cli/src/main.rs b/hugr-cli/src/main.rs index c8bb0b56e..87fbab41a 100644 --- a/hugr-cli/src/main.rs +++ b/hugr-cli/src/main.rs @@ -9,7 +9,7 @@ use clap_verbosity_flag::Level; fn main() { match CliArgs::parse() { CliArgs::Validate(args) => run_validate(args), - CliArgs::GenExtensions(args) => args.run_dump(), + CliArgs::GenExtensions(args) => args.run_dump(&hugr_core::std_extensions::STD_REG), CliArgs::Mermaid(mut args) => args.run_print().unwrap(), CliArgs::External(_) => { // TODO: Implement support for external commands.