From 79883f6a51e19a046878168cf88289ddab7b977c Mon Sep 17 00:00:00 2001
From: Maxime Coste <frrrwww@gmail.com>
Date: Wed, 6 Nov 2013 19:11:46 +0000
Subject: [PATCH] make align tab-aware

---
 README.asciidoc       |  1 +
 src/normal.cc         | 43 ++++++++++++++++++++++++++++++++++++-------
 src/option_manager.cc |  1 +
 3 files changed, 38 insertions(+), 7 deletions(-)

diff --git a/README.asciidoc b/README.asciidoc
index e0bd1b4d..acd36978 100644
--- a/README.asciidoc
+++ b/README.asciidoc
@@ -397,6 +397,7 @@ Some options are built in kakoune, and can be used to control it's behaviour:
  * +complete_prefix+ _bool_: when completing in command line, and multiple
    candidates exist, enable completion with common prefix.
  * +incsearch+ _bool_: execute search as it is typed
+ * +aligntab+ _bool_: use tabs for alignement command
  * +autoinfo+ _bool_: display automatic information box for certain commands.
  * +autoshowcompl+ _bool_: automatically display possible completions when
    editing a prompt.
diff --git a/src/normal.cc b/src/normal.cc
index 226f554b..74810f62 100644
--- a/src/normal.cc
+++ b/src/normal.cc
@@ -806,13 +806,29 @@ void save_selections(Context& context, int)
                            " selections", get_color("Information") });
 }
 
+static CharCount get_column(const Buffer& buffer,
+                            CharCount tabstop, BufferCoord coord)
+{
+    auto& line = buffer[coord.line];
+    auto col = 0_char;
+    for (auto it = line.begin();
+         it != line.end() and ByteCount{it - line.begin()} < coord.column;
+         it = utf8::next(it))
+    {
+        if (*it == '\t')
+            col = (col / tabstop + 1) * tabstop;
+        else
+            ++col;
+    }
+    return col;
+}
+
 template<bool insert_at_begin>
 void align(Context& context, int)
 {
     auto& selections = context.editor().selections();
     auto& buffer = context.buffer();
-    auto get_column = [&buffer](BufferCoord coord)
-    { return buffer[coord.line].char_count_to(coord.column); };
+    const CharCount tabstop = context.options()["tabstop"].get<int>();
 
     std::vector<std::vector<const Selection*>> columns;
     LineCount last_line = -1;
@@ -830,16 +846,29 @@ void align(Context& context, int)
         last_line = line;
     }
 
+    const bool use_tabs = context.options()["aligntab"].get<bool>();
     for (auto& col : columns)
     {
-        CharCount max_col = 0;
+        CharCount maxcol = 0;
         for (auto& sel : col)
-            max_col = std::max(get_column(sel->last()), max_col);
+            maxcol = std::max(get_column(buffer, tabstop, sel->last()), maxcol);
         for (auto& sel : col)
         {
-            CharCount padding = max_col - get_column(sel->last());
-            auto it = buffer.iterator_at(insert_at_begin ? sel->min() : sel->last());
-            buffer.insert(it, String{ ' ', padding });
+            auto insert_coord = insert_at_begin ? sel->min() : sel->last();
+            auto lastcol = get_column(buffer, tabstop, sel->last());
+            String padstr;
+            if (not use_tabs)
+                padstr = String{ ' ', maxcol - lastcol };
+            else
+            {
+                auto inscol = get_column(buffer, tabstop, insert_coord);
+                auto targetcol = maxcol - (lastcol - inscol);
+                auto tabcol = inscol - (inscol % tabstop);
+                auto tabs = (targetcol - tabcol) / tabstop;
+                auto spaces = targetcol - (tabs ? (tabcol + tabs * tabstop) : inscol);
+                padstr = String{ '\t', tabs } + String{ ' ', spaces };
+            }
+            buffer.insert(buffer.iterator_at(insert_coord), std::move(padstr));
         }
     }
 }
diff --git a/src/option_manager.cc b/src/option_manager.cc
index f472d3fa..ec3ae4fe 100644
--- a/src/option_manager.cc
+++ b/src/option_manager.cc
@@ -117,6 +117,7 @@ GlobalOptions::GlobalOptions()
     declare_option<bool>("incsearch", true);
     declare_option<bool>("autoinfo", true);
     declare_option<bool>("autoshowcompl", true);
+    declare_option<bool>("aligntab", false);
     declare_option<Regex>("ignored_files", Regex{R"(^(\..*|.*\.(o|so|a))$)"});
     declare_option<String>("filetype", "");
     declare_option<std::vector<String>>("completions", {});