It appears to work

Author: Aaron1011

setup.sh

@@ -0,0 +1,2 @@
+export ZDOTDIR=/home/zshtest
+stty -tostop

src/brackets.rs

@@ -3,14 +3,14 @@
 use std::collections::HashMap;
 use std::ffi::CString;
 use linkroot;
-use std::os::raw::c_char;
-use {shfunc, getshfunc, doshfunc, Shfunc};
+use std::os::raw::{c_char, c_void};
+use {shfunc, getshfunc, doshfunc, Shfunc, newlinklist, insertlinknode, linknode};
 
-fn brackets_paint(buf: &str, widget: &str, cursor: usize) {
-    let mut style: &str;
+pub fn brackets_paint(buf: &str, widget: &str, cursor: usize) {
+    let mut style: String;
     let mut level: usize = 0;
     let mut matching_pos: usize = 0;
-    let mut bracket_color_size: usize = 0;
+    let mut bracket_color_size: usize = 5; // TODO
     let mut buf_len: usize = 0;
     let mut pos: usize = 0;
 
@@ -29,7 +29,10 @@ fn brackets_paint(buf: &str, widget: &str, cursor: usize) {
                 last_of_level.insert(level, i);
             },
             ')' | ']' | '}' => {
-                matching_pos = *last_of_level.get(&level).unwrap();
+                let matching_pos = match last_of_level.get(&level) {
+                    Some(val) => *val,
+                    None => continue
+                };
                 level_pos.insert(i, level);
                 level -= 1;
 
@@ -50,20 +53,21 @@ fn brackets_paint(buf: &str, widget: &str, cursor: usize) {
         }
     }
 
-    for pos in level_pos.values() {
+    for pos in level_pos.keys() {
        if matching.contains_key(pos) {
-           style = "fg=yellow"; // TODO
+           style = format!("bracket-level-{}", ((level_pos[pos] - 1) % bracket_color_size) + 1);
        } else {
-           style = "fg=red";
+           style = "bracket-error".to_owned();
        }
+       do_highlight(*pos, *pos + 1, &style);
 
     }
 
     if widget != "zle-line-finish" {
         pos = cursor + 1;
         if level_pos.get(&pos).is_some() && matching.get(&pos).is_some() {
             let other_pos = matching[&pos];
-            do_highlight(other_pos - 1, other_pos, "standout");
+            do_highlight(other_pos, other_pos + 1, "standout");
         }
     }
 
@@ -78,10 +82,27 @@ fn brackets_match(first: char, second: char) -> bool {
     }
 }
 
+#[cfg(not(test))]
 fn do_highlight(start: usize, end: usize, style: &str) {
-    let func_name = CString::new("_zsh_highlight_add_highlight").unwrap();
     unsafe {
-        let func: Shfunc = getshfunc(func_name.as_ptr() as *mut c_char);
-        doshfunc(func, 0 as *mut linkroot, 1);
+        let func_name = str_to_ptr("_zsh_highlight_add_highlight");
+        let func = getshfunc(func_name as *mut c_char);
+
+        let list = newlinklist();
+        insertlinknode(list, (*list).list.as_ref().last as *const linknode as *mut linknode, func_name);
+        insertlinknode(list, (*list).list.as_ref().last as *const linknode as *mut linknode, str_to_ptr(&start.to_string()));
+        insertlinknode(list, (*list).list.as_ref().last as *const linknode as *mut linknode, str_to_ptr(&end.to_string()));
+        insertlinknode(list, (*list).list.as_ref().last as *const linknode as *mut linknode, str_to_ptr(style));
+
+        doshfunc(func, list as *mut linkroot, 1);
     }
 }
+
+#[cfg(test)]
+fn do_highlight(start: usize, end: usize, style: &str) {
+}
+
+
+fn str_to_ptr(s: &str) -> *mut c_void {
+    CString::new(s.to_string()).unwrap().into_raw() as *mut c_void
+}

src/lib.rs

@@ -10,12 +10,15 @@ include!("bindings.rs");
 #[macro_use]
 extern crate lazy_static;
 
+mod brackets;
+
 use std::ptr::{null, null_mut};
 use std::os::raw::{c_int, c_char};
-use std::ffi::CString;
+use std::ffi::{CString, CStr};
 use std::mem::transmute;
+use std::slice;
 
-mod brackets;
+use brackets::brackets_paint;
 
 // https://github.com/rust-lang/rfcs/issues/400#issuecomment-274140470
 macro_rules! cstr {
@@ -38,8 +41,8 @@ pub static mut bintab: BinWrapper<'static> = BinWrapper(&mut [builtin {
         flags: 0
     },
     handlerfunc: Some(bin_fastbrackets),
-    minargs: 0,
-    maxargs: -1,
+    minargs: 2,
+    maxargs: 3,
     funcid: 0,
     optstr: null_mut(),
     defopts: null_mut()
@@ -105,14 +108,47 @@ pub extern fn enables_(m: Module, enables: *mut *mut c_int) -> c_int {
 }
 
 #[no_mangle]
-pub extern fn bin_fastbrackets(name: *mut c_char, args: *mut *mut c_char, options: Options, func: c_int) -> c_int {
-    println!("Called!");
+pub extern fn bin_fastbrackets(name: *mut c_char, mut raw_args: *mut *mut c_char, options: Options, func: c_int) -> c_int {
+    let mut args: Vec<String> = Vec::new();
+
+    unsafe {
+        while *raw_args != null_mut() {
+            args.push(CStr::from_ptr(*raw_args as *const c_char).to_str().unwrap().to_owned());
+            raw_args = raw_args.offset(1);
+        }
+    }
+
+    let cursor = 0;
+    if args.len() == 2 {
+        let cursor = 0;
+    } else {
+        let cursor = match args[2].parse::<usize>() {
+            Ok(s) => s,
+            Err(e) => {
+                unsafe { zwarnnam(name, CString::new(format!("Invalid cursor argument: {:?}", e)).unwrap().into_raw()) } ;
+                return 1
+            }
+        };
+    }
+    brackets_paint(&args[0], &args[1], cursor);
+
     0
 }
 
 #[cfg(test)]
 mod tests {
+    use brackets::brackets_paint;
+    use std::str;
+
     #[test]
-    fn it_works() {
+    fn simple_bracket() {
+        brackets_paint("[]", "lol", 0);
     }
+
+    /*#[test]
+    fn large_buffer() {
+        let data = &[b'a'; 30_000];
+        let buf = unsafe { str::from_utf8_unchecked(data) };
+        brackets_paint(&buf, "hi", 0);
+    }*/
 }

test.sh

@@ -1 +1,2 @@
 zmodload aaron/fastbrackets
+fastbrackets 1 2 3