From ccc158103c67837ab3c7902892ad9797480c90b8 Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Wed, 24 Apr 2024 22:20:28 -0700
Subject: [PATCH] WIP

---
 lua/argonaut/types.lua | 36 ++++++++++++++++++------------------
 1 file changed, 18 insertions(+), 18 deletions(-)

diff --git a/lua/argonaut/types.lua b/lua/argonaut/types.lua
index 37d470e..793892d 100644
--- a/lua/argonaut/types.lua
+++ b/lua/argonaut/types.lua
@@ -330,22 +330,6 @@ function WrapContext:parse()
 end
 
 function WrapContext:wrap()
-    local line = self.indent .. self.prefix
-
-    for i, param in ipairs(self.params.parsed) do
-        line = line .. param.text
-        if i < #self.params.parsed then
-            line = line .. ', '
-        end
-    end
-
-    line = line .. self.suffix
-
-    vim.fn.setline(self.range.start.row, line)
-    vim.fn.execute(string.format('%d,%dd_', self.range.start.row + 1, self.range.stop.row))
-end
-
-function WrapContext:unwrap()
     vim.fn.setline(
         self.range.start.row,
         self.indent .. self.prefix
@@ -392,11 +376,27 @@ function WrapContext:unwrap()
     end
 end
 
+function WrapContext:unwrap()
+    local line = self.indent .. self.prefix
+
+    for i, param in ipairs(self.params.parsed) do
+        line = line .. param.text
+        if i < #self.params.parsed then
+            line = line .. ', '
+        end
+    end
+
+    line = line .. self.suffix
+
+    vim.fn.setline(self.range.start.row, line)
+    vim.fn.execute(string.format('%d,%dd_', self.range.start.row + 1, self.range.stop.row))
+end
+
 function WrapContext:toggle()
     if self.range:is_wrapped() then
-        self:wrap()
-    else
         self:unwrap()
+    else
+        self:wrap()
     end
 end