fix: check-endpoints handles all exported functions (#95)

* XC-391: Add missing endpoint types

* XC-391: Rename `Callback` to `Custom`

* XC-391: Add CHANGELOG entry

* XC-391: Refactor `CanisterEndpoint`

* update changelog

---------

Co-authored-by: Linwei Shang <linwei.shang@dfinity.org>

Author: lpahlavi

CHANGELOG.md

@@ -6,6 +6,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [unreleased]
 
+* Fix: `check-endpoints` now correctly handles all exported functions, not just those prefixed with `canister_`.
+
 ## [0.9.7] - 2025-09-26
 
 * Add `check-endpoints` command to `ic-wasm`.

src/check_endpoints/mod.rs

@@ -17,48 +17,37 @@ pub enum CanisterEndpoint {
     Query(String),
     #[display("canister_composite_query:{0}")]
     CompositeQuery(String),
-    #[display("canister_heartbeat")]
-    Heartbeat,
-    #[display("canister_global_timer")]
-    GlobalTimer,
-    #[display("canister_init")]
-    Init,
-    #[display("canister_post_upgrade")]
-    PostUpgrade,
-    #[display("canister_pre_upgrade")]
-    PreUpgrade,
+    #[display("{0}")]
+    Entrypoint(String),
 }
 
 impl TryFrom<&ExportedMethodInfo> for CanisterEndpoint {
     type Error = anyhow::Error;
 
     fn try_from(method: &ExportedMethodInfo) -> Result<Self, Self::Error> {
         type EndpointConstructor = fn(&str) -> CanisterEndpoint;
-        let mappings: &[(&str, EndpointConstructor)] = &[
-            ("canister_query", |s| CanisterEndpoint::Query(s.to_string())),
+        const MAPPINGS: &[(&str, EndpointConstructor)] = &[
             ("canister_update", |s| {
                 CanisterEndpoint::Update(s.to_string())
             }),
+            ("canister_query", |s| CanisterEndpoint::Query(s.to_string())),
             ("canister_composite_query", |s| {
                 CanisterEndpoint::CompositeQuery(s.to_string())
             }),
-            ("canister_heartbeat", |_| CanisterEndpoint::Heartbeat),
-            ("canister_global_timer", |_| CanisterEndpoint::GlobalTimer),
-            ("canister_init", |_| CanisterEndpoint::Init),
-            ("canister_post_upgrade", |_| CanisterEndpoint::PostUpgrade),
-            ("canister_pre_upgrade", |_| CanisterEndpoint::PreUpgrade),
         ];
 
-        for (candid_prefix, constructor) in mappings {
+        for (candid_prefix, constructor) in MAPPINGS {
             if let Some(rest) = method.name.strip_prefix(candid_prefix) {
                 return Ok(constructor(rest.trim()));
             }
         }
 
-        Err(anyhow!(
-            "Invalid exported method in canister WASM: '{}'",
-            method.name
-        ))
+        let trimmed = method.name.trim();
+        if !trimmed.is_empty() {
+            Ok(CanisterEndpoint::Entrypoint(trimmed.to_string()))
+        } else {
+            Err(anyhow!("Exported method in canister WASM has empty name"))
+        }
     }
 }
 

tests/tests.rs

@@ -455,7 +455,7 @@ fn check_endpoints() {
         .arg("check-endpoints")
         .assert()
         .stderr(
-            "ERROR: The following endpoint is unexpected in the WASM exports section: canister_update:__motoko_async_helper\n\
+        "ERROR: The following endpoint is unexpected in the WASM exports section: canister_update:__motoko_async_helper\n\
         ERROR: The following endpoint is unexpected in the WASM exports section: canister_query:__get_candid_interface_tmp_hack\n\
         ERROR: The following endpoint is unexpected in the WASM exports section: canister_query:__motoko_stable_var_info\n\
         ERROR: The following endpoint is unexpected in the WASM exports section: canister_global_timer\n\