From 5290f709248c73527af28a728e50004b757f31f5 Mon Sep 17 00:00:00 2001
From: lens0021 <lorentz0021@gmail.com>
Date: Mon, 16 Dec 2024 23:47:22 +0900
Subject: [PATCH] Remove the .ab extension from import statements for stdlib

---
 src/modules/function/declaration.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/modules/function/declaration.rs b/src/modules/function/declaration.rs
index 4cf93c7b..818c680f 100644
--- a/src/modules/function/declaration.rs
+++ b/src/modules/function/declaration.rs
@@ -309,6 +309,7 @@ impl FunctionDeclaration {
                 .map(Path::new)
                 .and_then(Path::file_name)
                 .and_then(OsStr::to_str)
+                .map(|s| s.trim_end_matches(".ab"))
                 .map(String::from)
                 .unwrap_or_default();
             result.push(String::from("```ab"));