Add macro_lib for AST node generation and enhance Codebase structure with new fields

Author: 0xGeorgii

Cargo.toml

@@ -6,6 +6,9 @@ edition = "2021"
 [dependencies]
 syn = { version = "2.0.87", features = ["full"] }
 thiserror = "1.0.67"
+macro_lib = { path = "macro_lib" }
 
 [dev-dependencies]
 soroban-sdk = "21.7.6"
+
+workspace = { members = [".", "macro_lib"] }

macro_lib/Cargo.toml

@@ -0,0 +1,12 @@
+[package]
+name = "macro_lib"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+proc-macro = true
+
+[dependencies]
+syn = { version = "2.0", features = ["full"] }
+quote = "1.0"
+proc-macro2 = "1.0"

macro_lib/src/lib.rs

@@ -0,0 +1,43 @@
+#![warn(clippy::pedantic)]
+extern crate proc_macro;
+
+use proc_macro::TokenStream;
+use quote::quote;
+use syn::{parse_macro_input, Field, FieldMutability, Fields, Ident, ItemStruct, Type, Visibility};
+
+#[proc_macro_attribute]
+pub fn ast_node(_attr: TokenStream, item: TokenStream) -> TokenStream {
+    let mut ast = parse_macro_input!(item as ItemStruct);
+    let fields_to_add = vec![
+        ("start_line", "u32"),
+        ("start_col", "u32"),
+        ("end_line", "u32"),
+        ("end_col", "u32"),
+    ];
+
+    match &mut ast.fields {
+        Fields::Named(fields_named) => {
+            for (field_name, field_type) in fields_to_add {
+                let field_ident = Ident::new(field_name, proc_macro2::Span::call_site());
+                let field_type: Type = syn::parse_str(field_type).unwrap();
+
+                let new_field = Field {
+                    attrs: Vec::new(),
+                    vis: Visibility::Inherited,
+                    mutability: FieldMutability::None,
+                    ident: Some(field_ident),
+                    colon_token: Some(Default::default()),
+                    ty: field_type,
+                };
+                fields_named.named.push(new_field);
+            }
+        }
+        _ => panic!("Only named fields are supported"),
+    }
+
+    let result = quote! {
+        #ast
+    };
+
+    result.into()
+}

src/ast.rs

@@ -1,5 +1,6 @@
 #![warn(clippy::pedantic)]
 use crate::errors::SDKErr;
+use macro_lib::ast_node;
 use std::collections::HashMap;
 
 fn parse_file(file_name: &str, content: &mut str) -> Result<syn::File, SDKErr> {
@@ -15,6 +16,8 @@ pub struct Codebase {
     ast_map: HashMap<String, syn::File>,
     free_functions: Vec<Function>,
     contracts: Vec<Contract>,
+    structs: Vec<Struct>,
+    enums: Vec<Enum>,
 }
 
 impl Codebase {
@@ -23,6 +26,8 @@ impl Codebase {
             ast_map: HashMap::new(),
             free_functions: Vec::new(),
             contracts: Vec::new(),
+            structs: Vec::new(),
+            enums: Vec::new(),
         }
     }
 
@@ -52,6 +57,7 @@ impl Codebase {
     }
 }
 
+#[ast_node]
 #[derive(Clone)]
 pub struct Contract {
     name: String,
@@ -61,39 +67,61 @@ pub struct Contract {
 }
 
 impl Contract {
-    pub fn new(name: &str) -> Self {
+    pub fn new(name: &str, start_line: u32, start_col: u32, end_line: u32, end_col: u32) -> Self {
         Contract {
             name: name.to_string(),
             functions: Vec::new(),
             structs: Vec::new(),
             enums: Vec::new(),
+            start_line,
+            start_col,
+            end_line,
+            end_col,
         }
     }
 }
 
+#[ast_node]
 #[derive(Clone)]
 pub struct Function {}
 
 impl Function {
-    pub fn new() -> Self {
-        Function {}
+    pub fn new(start_line: u32, start_col: u32, end_line: u32, end_col: u32) -> Self {
+        Function {
+            start_line,
+            start_col,
+            end_line,
+            end_col,
+        }
     }
 }
 
+#[ast_node]
 #[derive(Clone)]
 pub struct Struct {}
 
 impl Struct {
-    pub fn new() -> Self {
-        Struct {}
+    pub fn new(start_line: u32, start_col: u32, end_line: u32, end_col: u32) -> Self {
+        Struct {
+            start_line,
+            start_col,
+            end_line,
+            end_col,
+        }
     }
 }
 
+#[ast_node]
 #[derive(Clone)]
 pub struct Enum {}
 
 impl Enum {
-    pub fn new() -> Self {
-        Enum {}
+    pub fn new(start_line: u32, start_col: u32, end_line: u32, end_col: u32) -> Self {
+        Enum {
+            start_line,
+            start_col,
+            end_line,
+            end_col,
+        }
     }
 }