Top level attributes

parser/src/ast.rs

@@ -62,6 +62,7 @@ pub struct TopLevel(pub TopLevelKind, pub Vec<Attribute>);
 
 #[derive(Debug, Clone, Serialize)]
 pub enum TopLevelKind {
+    ModAttribute,
     Include(Path),
     StructDecl {
         export: bool,

parser/src/grammar.pest

@@ -23,6 +23,10 @@ attribute = {
     "#" ~ "[" ~ meta ~ "]"
 }
 
+mod_attribute = {
+    "#" ~ "!" ~ "[" ~ meta ~ "]"
+}
+
 meta = {
     static_path ~ ("=" ~ primary | "(" ~ meta_list? ~ ")")?
 }
@@ -33,7 +37,7 @@ meta_list = {
 
 attributes = { attribute* }
 
-top_level = { attributes ~ (import | function_decl | struct_decl) }
+top_level = { (attributes ~ (import | function_decl | struct_decl)) | mod_attribute }
 
 import = { sharp ~ "include" ~ "<" ~ static_path ~ ">" }
 

parser/src/lib.rs

@@ -84,9 +84,7 @@ impl From<pest::iterators::Pair<'_, Rule>> for TypeExprKind {
 impl From<pest::iterators::Pair<'_, Rule>> for Path {
     fn from(pair: pest::iterators::Pair<'_, Rule>) -> Self {
         match pair.as_rule() {
-            Rule::static_path => {
-                Path(pair.into_inner().map(|i| i.as_str().to_string()).collect())
-            }
+            Rule::static_path => Path(pair.into_inner().map(|i| i.as_str().to_string()).collect()),
             _ => unimplemented!("{pair:#?}"),
         }
     }
@@ -188,7 +186,13 @@ impl From<pest::iterators::Pair<'_, Rule>> for TopLevel {
             .map(Attribute::from)
             .collect::<Vec<_>>();
 
-        TopLevel(TopLevelKind::from(inner.next().unwrap()), attributes)
+        TopLevel(
+            inner
+                .next()
+                .map(TopLevelKind::from)
+                .unwrap_or(TopLevelKind::ModAttribute),
+            attributes,
+        )
     }
 }
 

src/codegen.rs

@@ -228,8 +228,17 @@ impl ToRust for Block {
 
 impl ToRust for TopLevel {
     fn to_rust(&self, cg: &mut RustCodegen) {
-        for attr in &self.1 {
-            cg.addln(&format!("#[{}]", attr.get_rust()));
+        match &self.0 {
+            TopLevelKind::ModAttribute => {
+                for attr in &self.1 {
+                    cg.addln(&format!("#![{}]", attr.get_rust()));
+                }
+            }
+            _ => {
+                for attr in &self.1 {
+                    cg.addln(&format!("#[{}]", attr.get_rust()));
+                }
+            }
         }
 
         self.0.to_rust(cg);
@@ -261,6 +270,7 @@ impl GetRust for Attribute {
 impl ToRust for TopLevelKind {
     fn to_rust(&self, cg: &mut RustCodegen) {
         match self {
+            Self::ModAttribute => {}
             Self::Include(path) => {
                 cg.addln(&format!("use {};", path.get_rust()));
             }

test/src/main.mist

@@ -1,3 +1,5 @@
+#![allow(unused_mut)]
+
 #[derive(Debug)]
 public struct ApiCall {
     i32 x,