codegen: Wrap unsafe functions as unsafe

Author: A6GibKm

src/codegen/function.rs

@@ -92,7 +92,13 @@ pub fn generate(
 
     let unsafe_ = if analysis.unsafe_ { "unsafe " } else { "" };
     let declaration = declaration(env, analysis);
-    let suffix = if only_declaration { ";" } else { " {" };
+    let suffix = if only_declaration {
+        ";"
+    } else if analysis.unsafe_ {
+        " { unsafe {"
+    } else {
+        " {"
+    };
 
     writeln!(w)?;
     cfg_deprecated(w, env, None, analysis.deprecated_version, commented, indent)?;
@@ -153,6 +159,9 @@ pub fn generate(
         for s in body {
             writeln!(w, "{}{}", tabs(indent), s)?;
         }
+        if analysis.unsafe_ {
+            writeln!(w, "}}")?;
+        }
     }
 
     if analysis.async_future.is_some() {