feat: colorful help

Author: Chronostasys

Cargo.lock

@@ -21,7 +21,7 @@ rowan = "0.15"
 dissimilar = "1.0"
 stderrlog = "0.5"
 log = "0.4"
-clap = { version = "4.1", features = ["derive"] }
+clap = { version = "4.1", features = ["derive", "unstable-styles"] }
 colored = "2.0"
 serde = "1.0"
 lsp-types = { branch = "master", features = ["proposed"], git = "https://github.com/Pivot-Studio/lsp-types" }
@@ -42,6 +42,7 @@ indicatif = "0.17"
 parking_lot = "0.12"
 derivative = "2.2"
 console = "0.15"
+anstyle = "1.0"
 
 [target.'cfg(target_arch = "wasm32")'.dependencies]
 wasm-bindgen = "0.2"

Cargo.toml

@@ -98,7 +98,7 @@ struct GCTestObj {
     d: *mut u64,
     e: *mut GCTestObj,
 }
-fn gctest_vtable(
+extern "C" fn gctest_vtable(
     ptr: *mut u8,
     gc: &Collector,
     mark_ptr: VisitFunc,

immix/benches/immix_bench.rs

@@ -59,7 +59,7 @@ struct GCTestObj {
     d: *mut u64,
     e: *mut GCTestObj,
 }
-fn gctest_vtable(
+extern "C" fn gctest_vtable(
     ptr: *mut u8,
     gc: &Collector,
     mark_ptr: VisitFunc,

immix/benches/immix_stress_bench.rs

@@ -47,9 +47,9 @@ struct CollectorStatus {
     collecting: bool,
 }
 
-pub type VisitFunc = unsafe fn(&Collector, *mut u8);
+pub type VisitFunc = unsafe extern "C" fn(&Collector, *mut u8);
 
-pub type VtableFunc = fn(*mut u8, &Collector, VisitFunc, VisitFunc, VisitFunc);
+pub type VtableFunc = extern "C" fn(*mut u8, &Collector, VisitFunc, VisitFunc, VisitFunc);
 
 impl Drop for Collector {
     fn drop(&mut self) {
@@ -234,7 +234,7 @@ impl Collector {
     }
 
     /// precise mark a pointer
-    unsafe fn mark_ptr(&self, ptr: *mut u8) {
+    unsafe extern "C" fn mark_ptr(&self, ptr: *mut u8) {
         let father = ptr;
         let mut ptr = *(ptr as *mut *mut u8);
         // println!("mark ptr {:p} -> {:p}", father, ptr);
@@ -330,7 +330,7 @@ impl Collector {
     ///
     /// it self does not mark the object, but mark the object's fields by calling
     /// mark_ptr
-    unsafe fn mark_complex(&self, ptr: *mut u8) {
+    unsafe extern "C" fn mark_complex(&self, ptr: *mut u8) {
         let vptr = *(ptr as *mut *mut u8);
         if vptr.is_null() {
             return;
@@ -345,7 +345,7 @@ impl Collector {
         );
     }
     /// precise mark a trait object
-    unsafe fn mark_trait(&self, ptr: *mut u8) {
+    unsafe extern "C" fn mark_trait(&self, ptr: *mut u8) {
         // if !self.thread_local_allocator.as_mut().unwrap().in_heap(ptr)
         //    &&!self.thread_local_allocator.as_mut().unwrap().in_big_heap(ptr) {
         //     return;
@@ -644,7 +644,7 @@ mod tests {
         d: *mut u64,
         e: *mut GCTestObj,
     }
-    fn gctest_vtable(
+    extern "C" fn gctest_vtable(
         ptr: *mut u8,
         gc: &Collector,
         mark_ptr: VisitFunc,
@@ -667,7 +667,7 @@ mod tests {
         d: *mut GCTestBigObj,
     }
     const BIGOBJ_ALLOC_SIZE: usize = round_n_up!(size_of::<GCTestBigObj>() + 16, 128);
-    fn gctest_vtable_big(
+    extern "C" fn gctest_vtable_big(
         ptr: *mut u8,
         gc: &Collector,
         mark_ptr: VisitFunc,

immix/src/collector.rs

@@ -310,9 +310,9 @@ impl<'a, 'ctx> LLVMBuilder<'a, 'ctx> {
         let stack_ptr = self
             .builder
             .build_alloca(llvmtp.ptr_type(AddressSpace::default()), "stack_ptr");
+        self.gc_add_root(stack_ptr.as_basic_value_enum(), obj_type);
         self.builder.position_at_end(lb);
         self.builder.build_store(stack_ptr, casted_result);
-        self.gc_add_root(stack_ptr.as_basic_value_enum(), obj_type);
         (casted_result.into_pointer_value(), stack_ptr, llvmtp)
     }
 

src/ast/builder/llvmbuilder.rs

@@ -41,7 +41,7 @@ _______   __                        __            __
                                                                                |  \__| $$
                                                                                 \$$    $$
                                                                                  \$$$$$$ 
-"#, long_about = None)]
+"#, long_about = None, styles = get_styles())]
 struct Cli {
     /// Name of the source file
     #[arg(value_parser)]
@@ -200,3 +200,18 @@ fn main() {
         Cli::command().print_help().unwrap();
     }
 }
+
+pub fn get_styles() -> clap::builder::Styles {
+    clap::builder::Styles::styled()
+        .header(
+            anstyle::Style::new()
+                .bold()
+                .underline()
+                .fg_color(Some(anstyle::Color::Ansi(anstyle::AnsiColor::Green))),
+        )
+        .literal(
+            anstyle::Style::new()
+                .bold()
+                .fg_color(Some(anstyle::Color::Ansi(anstyle::AnsiColor::BrightGreen))),
+        )
+}