From 0f49e7375e1a414f2a144694c8b1f91354a0804e Mon Sep 17 00:00:00 2001
From: Dmitry Matveyev <greenfork.lists@yandex.com>
Date: Sun, 6 Jun 2021 00:59:22 +0600
Subject: [PATCH] Add auto-inserting of "end" keyword to Elixir

fixup! Add auto-inserting of "end" keyword to Elixir
---
 rc/filetype/elixir.kak            | 23 ++++++++++++++++++-----
 test/indent/elixir/do-keyword/cmd |  1 +
 test/indent/elixir/do-keyword/in  |  1 +
 test/indent/elixir/do-keyword/out | 10 ++++++++++
 test/indent/elixir/do-keyword/rc  |  3 +++
 5 files changed, 33 insertions(+), 5 deletions(-)
 create mode 100644 test/indent/elixir/do-keyword/cmd
 create mode 100644 test/indent/elixir/do-keyword/in
 create mode 100644 test/indent/elixir/do-keyword/out
 create mode 100644 test/indent/elixir/do-keyword/rc

diff --git a/rc/filetype/elixir.kak b/rc/filetype/elixir.kak
index f06ffb1c..5b215942 100644
--- a/rc/filetype/elixir.kak
+++ b/rc/filetype/elixir.kak
@@ -19,8 +19,8 @@ hook global WinSetOption filetype=elixir %{
     require-module elixir
 
     hook window ModeChange pop:insert:.* -group elixir-trim-indent  elixir-trim-indent
-    hook window InsertChar \n -group elixir-insert elixir-insert-on-new-line
     hook window InsertChar \n -group elixir-indent elixir-indent-on-new-line
+    hook window InsertChar \n -group elixir-insert elixir-insert-on-new-line
 
     hook -once -always window WinSetOption filetype=.* %{ remove-hooks window elixir-.+ }
 }
@@ -92,12 +92,25 @@ define-command -hidden elixir-trim-indent %{
     try %{ execute-keys -draft -itersel <a-x> s \h+$ <ret> d }
 }
 
-define-command -hidden elixir-insert-on-new-line %{
-    evaluate-commands -draft -itersel %{
+define-command -hidden elixir-insert-on-new-line %[
+    evaluate-commands -no-hooks -draft -itersel %[
         # copy -- comments prefix and following white spaces
         try %{ execute-keys -draft k <a-x> s ^\h*\K--\h* <ret> y gh j P }
-    }
-}
+        # wisely add end structure
+        evaluate-commands -save-regs x %[
+            try %{ execute-keys -draft k <a-x> s ^ \h + <ret> \" x y } catch %{ reg x '' }
+            try %[
+                evaluate-commands -draft %[
+                    # Check if previous line opens a block
+                    execute-keys -draft k<a-x> <a-k>^<c-r>x(.+\bdo$)<ret>
+                    # Check that we do not already have an end for this indent level which is first set via `elixir-indent-on-new-line` hook
+                    execute-keys -draft }i J <a-x> <a-K> ^<c-r>x(end|else)[^0-9A-Za-z_!?]<ret>
+                ]
+                execute-keys -draft o<c-r>xend<esc> # insert a new line with containing end
+            ]
+        ]
+    ]
+]
 
 define-command -hidden elixir-indent-on-new-line %{
     evaluate-commands -draft -itersel %{
diff --git a/test/indent/elixir/do-keyword/cmd b/test/indent/elixir/do-keyword/cmd
new file mode 100644
index 00000000..127363ac
--- /dev/null
+++ b/test/indent/elixir/do-keyword/cmd
@@ -0,0 +1 @@
+c<ret>f()<esc>jo<ret>def f1() do<ret>1<esc>jo<ret>def f2(), do: 2<ret>
diff --git a/test/indent/elixir/do-keyword/in b/test/indent/elixir/do-keyword/in
new file mode 100644
index 00000000..5e903f7f
--- /dev/null
+++ b/test/indent/elixir/do-keyword/in
@@ -0,0 +1 @@
+test do%( )
diff --git a/test/indent/elixir/do-keyword/out b/test/indent/elixir/do-keyword/out
new file mode 100644
index 00000000..ad2373b1
--- /dev/null
+++ b/test/indent/elixir/do-keyword/out
@@ -0,0 +1,10 @@
+test do
+    f()
+end
+
+def f1() do
+    1
+end
+
+def f2(), do: 2
+
diff --git a/test/indent/elixir/do-keyword/rc b/test/indent/elixir/do-keyword/rc
new file mode 100644
index 00000000..5cc4387c
--- /dev/null
+++ b/test/indent/elixir/do-keyword/rc
@@ -0,0 +1,3 @@
+source "%val{runtime}/colors/default.kak"
+source "%val{runtime}/rc/filetype/elixir.kak"
+set buffer filetype elixir