feat: add set-local! macro (#9)

datwaft · Jul 17, 2021 · cd59b41 · cd59b41
1 parent 40c9e34
commit cd59b41
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 5 deletions.
diff --git a/fnl/core/autocmd.fnl b/fnl/core/autocmd.fnl
@@ -4,7 +4,7 @@
 ; Restore cursor on exit
 (augroup! restore-cursor-on-exit
           (autocmd! VimLeave *
-                    "set guicursor=a:ver100-blinkon0"))
+                    #(set! guicursor ["a:ver100-blinkon0"])))
 
 ; Automatically resize splits when window is resized
 (augroup! resize-splits-on-resize
@@ -32,7 +32,7 @@
 ; Disable spell in certain filetypes
 (augroup! disable-spell-on-filetypes
           (autocmd! FileType [help packer]
-                    "setlocal nospell"))
+                    #(set-local! nospell)))
 
 ; Set terminal options
 (augroup! terminal-options
@@ -41,10 +41,12 @@
                     "startinsert")
           ; Disables line number on terminal buffers
           (autocmd! TermOpen *
-                    "setlocal nonumber norelativenumber")
+                    #(do
+                       (set-local! nonumber)
+                       (set-local! norelativenumber)))
           ; Disables spell on terminal buffers
           (autocmd! TermOpen *
-                    "setlocal nospell")
+                    #(set-local! nospell))
           ; Disables sign column on terminal buffers
           (autocmd! TermOpen *
-                    "setlocal signcolumn=\"no\""))
+                    #(set-local! signcolumn :no)))
diff --git a/fnl/core/macros.fnl b/fnl/core/macros.fnl
@@ -14,6 +14,12 @@
     `(let [(ok?# value#) (pcall #(: (. vim.opt ,name) :get))]
        (if ok?# value# nil))))
 
+(fn get-local? [name]
+  (let [name (tostring name)
+        name (if (= (name:sub 1 2) "no") (name:sub 3) name)]
+    `(let [(ok?# value#) (pcall #(: (. vim.opt_local ,name) :get))]
+       (if ok?# value# nil))))
+
 (fn set! [name value]
   "Set vim option with vim.opt"
   (let [name (tostring name)
@@ -45,6 +51,37 @@
         "^" `(: (. vim.opt ,name-without-last-character) :prepend ,value)
         _ `(tset vim.opt ,name ,value)))))
 
+(fn set-local! [name value]
+  "Set vim option with vim.opt_local"
+  (let [name (tostring name)
+        value-is-nil? (nil? value)
+        name-begins-with-no? (= (name:sub 1 2) "no")
+        name (if (and value-is-nil? name-begins-with-no?)
+               (name:sub 3)
+               name)
+        value (if value-is-nil?
+                (not name-begins-with-no?)
+                value)
+        name-last-character (name:sub -1)
+        name-without-last-character (name:sub 1 -2)]
+    (if (and 
+          (list? value)
+          (or
+            (= (tostring (head value)) :hashfn)
+            (= (tostring (head value)) :fn)))
+      (let [symbol (gensym-fn!)]
+        `(tset vim.opt_local ,name (do
+                                     (global ,(sym symbol) ,value)
+                                     ,(.. "v:lua." symbol "()"))))
+      (match name-last-character
+        "?" `(get? ,name-without-last-character)
+        "!" `(tset vim.opt_local ,name-without-last-character
+                   (not (get? ,name-without-last-character)))
+        "+" `(: (. vim.opt_local ,name-without-last-character) :append ,value)
+        "-" `(: (. vim.opt_local ,name-without-last-character) :remove ,value)
+        "^" `(: (. vim.opt_local ,name-without-last-character) :prepend ,value)
+        _ `(tset vim.opt_local ,name ,value)))))
+
 (fn let! [name value]
   "Set vim variable with vim.[g b w t]"
   (let [name (tostring name)
@@ -80,7 +117,9 @@
 
 {: gensym-fn!
  : get?
+ : get-local?
  : set!
+ : set-local!
  : let!
  : augroup!
  : autocmd!}