From 4ecd023960683c01d1ca65452a37f94ef14200ab Mon Sep 17 00:00:00 2001
From: Illia Kripaka <kripakaillya@gmail.com>
Date: Wed, 12 Aug 2026 14:22:40 +0300
Subject: [PATCH] fix public visibility naming

---
 crates/prettysimf/src/simplicity_fmt/doc.rs |  2 +-
 crates/prettysimf/src/simplicity_fmt/fmt.rs | 16 ++++++++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/crates/prettysimf/src/simplicity_fmt/doc.rs b/crates/prettysimf/src/simplicity_fmt/doc.rs
index 4f3ebc4..0cac75a 100644
--- a/crates/prettysimf/src/simplicity_fmt/doc.rs
+++ b/crates/prettysimf/src/simplicity_fmt/doc.rs
@@ -288,7 +288,7 @@ fn gap_before_closing_delimiter<'src>(context: &mut Context<'_>, start: usize, c
 impl Doc for Visibility {
     fn to_doc<'src>(&self, _context: &mut Context<'_>) -> Option<RcDoc<'src>> {
         match self {
-            Visibility::Public => Some(RcDoc::text("pub(super) ")),
+            Visibility::Public => Some(RcDoc::text("pub ")),
             Visibility::Private => Some(RcDoc::nil()),
         }
     }
diff --git a/crates/prettysimf/src/simplicity_fmt/fmt.rs b/crates/prettysimf/src/simplicity_fmt/fmt.rs
index 3e9792b..788c1e2 100644
--- a/crates/prettysimf/src/simplicity_fmt/fmt.rs
+++ b/crates/prettysimf/src/simplicity_fmt/fmt.rs
@@ -196,6 +196,22 @@ mod tests {
         utils::assert_comments_preserved(source, ["// leading", "/* inside */", "// between", "// trailing"]);
     }
 
+    #[test]
+    fn preserves_public_function_visibility() {
+        let source = "pub fn safe_add_64(a:u64,b:u64)->u64{unwrap(checked_add_64(a,b))}";
+        let expected = "pub fn safe_add_64(a: u64, b: u64) -> u64 {\n    unwrap(checked_add_64(a, b))\n}";
+
+        utils::assert_formatting_in_all_modes(source, expected);
+    }
+
+    #[test]
+    fn preserves_omitted_function_visibility() {
+        let source = "fn safe_add_64(a:u64,b:u64)->u64{unwrap(checked_add_64(a,b))}";
+        let expected = "fn safe_add_64(a: u64, b: u64) -> u64 {\n    unwrap(checked_add_64(a, b))\n}";
+
+        utils::assert_formatting_in_all_modes(source, expected);
+    }
+
     #[test]
     fn preserves_comments_in_match() {
         let source = "fn main() {