From 3e72513b759ef346157c392e4cafbef75d338c1b Mon Sep 17 00:00:00 2001
From: testuser <test@example.com>
Date: Mon, 3 Feb 2025 17:51:15 +0900
Subject: [PATCH] fix(generator): syntax for top level doccment

---
 src/generator.rs | 40 +++++++++++++++++++++-------------------
 1 file changed, 21 insertions(+), 19 deletions(-)

diff --git a/src/generator.rs b/src/generator.rs
index 8820846..20b0507 100644
--- a/src/generator.rs
+++ b/src/generator.rs
@@ -252,7 +252,7 @@ impl CodeGen for Block {
             .collect::<Result<Vec<_>, _>>()?;
 
         let out = quote! {
-            #[doc = #desc]
+            #![doc = #desc]
 
             use std::marker::PhantomData;
 
@@ -343,32 +343,34 @@ impl CodeGen for Register {
                 type_ux,
             );
 
+        let desc = self.desc.unwrap_or("".to_string());
+
         let out = quote! {
-            pub mod #snake_case_name {
-                use std::marker::PhantomData;
+            #![doc = #desc]
 
-                use crate::register_spec::{DataConversionError, Modifiable, Readable, RegisterSpec, Writable};
+            use std::marker::PhantomData;
 
-                const OFFSET: usize = #addr;
+            use crate::register_spec::{DataConversionError, Modifiable, Readable, RegisterSpec, Writable};
 
-                pub struct #reg_name<'a> {
-                    mem_ptr: *mut u32,
-                    _marker: PhantomData<&'a mut super::#parent_name<'a>>,
-                }
+            const OFFSET: usize = #addr;
 
-                impl #reg_name<'_> {
-                    pub(crate) fn new(parent_ptr: *mut u32) -> Self {
-                        #reg_name {
-                            mem_ptr: unsafe { parent_ptr.add(OFFSET) },
-                            _marker: PhantomData,
-                        }
+            pub struct #reg_name<'a> {
+                mem_ptr: *mut u32,
+                _marker: PhantomData<&'a mut super::#parent_name<'a>>,
+            }
+
+            impl #reg_name<'_> {
+                pub(crate) fn new(parent_ptr: *mut u32) -> Self {
+                    #reg_name {
+                        mem_ptr: unsafe { parent_ptr.add(OFFSET) },
+                        _marker: PhantomData,
                     }
                 }
-
-                #code_reg_def
-
-                #code_t_def
             }
+
+            #code_reg_def
+
+            #code_t_def
         };
 
         let (out_path, _next_parent_path) = mod_file_path(parent_path, &snake_case_name);