1 files changed, 5 insertions, 4 deletions
diff --git a/src/bin/cli.rs b/src/bin/cli.rs
index 275bd410..d0bbe989 100644
--- a/src/bin/cli.rs
+++ b/src/bin/cli.rs
@@ -1,4 +1,4 @@
-use authzd::{generate_entities_from_api, write_entities_file};
+use authzd::EntitiesRepository;
 use clap::{Parser, Subcommand};
 
 #[derive(Parser, Debug)]
@@ -50,9 +50,10 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
             token,
             host,
         } => {
-            let entities = generate_entities_from_api(token, host, project).await?;
-
-            write_entities_file(&entities, &output)?;
+            let repository = EntitiesRepository::new(token, host, project);
+            let entities = repository.generate().await?;
+            let json = serde_json::to_string_pretty(&entities)?;
+            std::fs::write(output, json)?;
         }
     }