From 2c3b2e9b522abb2b5b8659256196ca5731967dd5 Mon Sep 17 00:00:00 2001
From: Martin Chaine <martin@casimir.me>
Date: Mon, 12 Dec 2016 21:46:48 +0100
Subject: [PATCH] fix indent options and add whitespaces ones

---
 rc/extra/editorconfig.kak | 60 ++++++++++++++++++++++++++++-----------
 1 file changed, 43 insertions(+), 17 deletions(-)

diff --git a/rc/extra/editorconfig.kak b/rc/extra/editorconfig.kak
index 4a66aff6..138d8ecb 100644
--- a/rc/extra/editorconfig.kak
+++ b/rc/extra/editorconfig.kak
@@ -1,35 +1,61 @@
-def editorconfig-load -docstring "Set indentation options according to editorconfig file" %{
+# http://editorconfig.org/#file-format-details
+# ‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾‾
+
+decl -hidden bool editorconfig_trim_trailing_whitespace false
+decl -hidden int  editorconfig_insert_final_newline     0
+
+def -hidden _editorconfig-bufwritepre-hook %{
+    hook buffer BufWritePre %val{buffile} -group editorconfig-hooks %{ %sh{
+        if [ ${kak_opt_editorconfig_trim_trailing_whitespace} = "true" ]; then
+            printf %s\\n "try %{ exec -draft %{ %s\h+$<ret>d } }"
+        fi
+        if [ ${kak_opt_editorconfig_insert_final_newline} -gt 0 ]; then
+            printf %s\\n "try %{ exec -draft %{ %s\v+\Z<ret>d } }"
+        fi
+        if [ ${kak_opt_editorconfig_insert_final_newline} -gt 1 ]; then
+            printf %s\\n "try %{ exec -draft %{ geo<esc> } }"
+        fi
+    } }
+}
+
+def editorconfig-load -docstring "Set indentation optins and whitespace behavior according to editorconfig file" %{
     %sh{
         command -v editorconfig >/dev/null 2>&1 || { echo 'echo -color Error The editorconfig tool could not be found'; exit 1; }
-        editorconfig $kak_buffile | awk -F= -- '
-            /indent_style=/ { indent_style = $2 }
-            /indent_size=/  { indent_size = $2 == "tab" ? 4 : $2  }
-            /tab_width=/    { tab_width = $2 }
-            /end_of_line=/  { end_of_line = $2 }
-            /charset=/      { charset = $2 }
+        editorconfig "${kak_buffile}" | awk -F= -- '
+            /indent_style=/            { indent_style = $2 }
+            /indent_size=/             { indent_size = $2 == "tab" ? 4 : $2  }
+            /tab_width=/               { tab_width = $2 }
+            /end_of_line=/             { end_of_line = $2 }
+            /charset=/                 { charset = $2 }
+            /trim_trailing_whitespace/ { trim_trailing_whitespace = $2 }
+            /insert_final_newline/     { insert_final_newline = $2 }
 
             END {
-                if (indent_style == "tab")
-                {
+                if (indent_style == "tab") {
                     print "set buffer indentwidth 0"
                     print "set buffer aligntab true"
                 }
-                if (indent_style == "spaces") {
+                if (indent_style == "space") {
                     print "set buffer indentwidth " (indent_size == "tab" ? 4 : indent_size)
                     print "set buffer aligntab false"
                 }
                 if (indent_size || tab_width) {
                     print "set buffer tabstop " (tab_width ? tab_width : indent_size)
                 }
-                if (end_of_line) {
-                    if (end_of_line == "lf" || end_of_line == "crlf")
-                        print "set buffer eolformat " end_of_line
-                    else
-                        print "error"
+                if (end_of_line == "lf" || end_of_line == "crlf") {
+                    print "set buffer eolformat " end_of_line
+                }
+                if (charset == "utf-8-bom") {
+                    print "set buffer BOM utf-8"
+                }
+                if (trim_trailing_whitespace == "true") {
+                    print "set buffer editorconfig_trim_trailing_whitespace true"
+                }
+                if (insert_final_newline) {
+                    print "set buffer editorconfig_insert_final_newline " (insert_final_newline == "true" ? 2 : 1)
                 }
-                if (charset)
-                    print "set buffer BOM" (charset == "utf-8-bom" ? true : false)
             }
         '
     }
+    _editorconfig-bufwritepre-hook
 }