remove unused types in metadata

dfinity · Jun 25, 2024 · 625483a · 625483a
1 parent 6ed5d95
commit 625483a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/rust/candid_parser/src/utils.rs b/rust/candid_parser/src/utils.rs
@@ -67,7 +67,15 @@ pub fn get_metadata(env: &TypeEnv, serv: &Option<Type>) -> Option<String> {
         TypeInner::Service(_) => serv,
         _ => unreachable!(),
     };
-    Some(candid::pretty::candid::compile(env, &Some(serv)))
+    let def_list = crate::bindings::analysis::chase_actor(env, &serv).ok()?;
+    let env = TypeEnv(
+        env.0
+            .iter()
+            .filter(|&(k, _)| def_list.contains(&k.as_str()))
+            .map(|(k, v)| (k.clone(), v.clone()))
+            .collect(),
+    );
+    Some(candid::pretty::candid::compile(&env, &Some(serv)))
 }
 
 /// Merge canister metadata candid:args and candid:service into a service constructor.