Add main logic to creating RST tables.

1 files changed, 135 insertions, 2 deletions

diff --git a/ftplugin/rst_tables.vim b/ftplugin/rst_tables.vim
index c847c70..43e6daf 100644
--- a/ftplugin/rst_tables.vim
+++ b/ftplugin/rst_tables.vim
@@ -14,11 +14,144 @@ let loaded_rst_tables_ftplugin = 1
 
 python << endpython
 import vim
-import os
-import os.path
+import re
 from vim_bridge import bridged
 
 
+@bridged  # {{{
+def get_table_bounds():
+    row, col = vim.current.window.cursor
+    upper = lower = row
+    try:
+        while vim.buffer[upper - 1].strip():
+            upper -= 1
+    except IndexError:
+        pass
+    else:
+        upper += 1
+
+    try:
+        while vim.buffer[lower - 1].strip():
+            lower += 1
+    except IndexError:
+        pass
+    else:
+        lower -= 1
+
+    return (upper, lower)
+    # }}}
+
+
+def unify_table(table):
+    max_fields = max(map(lambda row: len(row), table))
+    output = []
+    for row in table:
+        curr_len = len(row)
+        if curr_len < max_fields:
+            row += [''] * (max_fields - curr_len)
+        output.append(row)
+    return output
+
+
+def parse_table(raw_lines):
+    mkfields = lambda line: re.split('\s\s+', line)
+    output = map(mkfields, raw_lines)
+    output = unify_table(output)
+    return output
+
+
+def table_line(widths, header=False):
+    if header:
+        linechar = '='
+    else:
+        linechar = '-'
+    sep = '+'
+    parts = []
+    for width in widths:
+        parts.append(linechar * width)
+    if parts:
+        parts = [''] + parts + ['']
+    return sep.join(parts)
+
+
+def get_column_widths(table):
+    widths = []
+    for row in table:
+        num_fields = len(row)
+        # dynamically grow
+        if num_fields >= len(widths):
+            widths.extend([0] * (num_fields - len(widths)))
+        for i in range(num_fields):
+            field_text = row[i]
+            field_width = len(field_text)
+            widths[i] = max(widths[i], field_width)
+    return widths
+
+
+def pad_fields(table, widths=None):
+    """Pads fields of the table, so each row lines up nicely with the others.
+    If the widths param is None, the widths are calculated automatically.
+
+    """
+    if widths is None:
+        widths = get_column_widths(table)
+    widths = map(lambda w: ' %-' + str(w) + 's ', widths)
+
+    # Pad all fields using the calculated widths
+    output = []
+    for row in table:
+        new_row = []
+        for i in range(len(row)):
+            col = row[i]
+            col = widths[i] % col.strip()
+            new_row.append(col)
+        output.append(new_row)
+    return output
+
+
+def draw_table(table):
+    if table == []:
+        return []
+
+    col_widths = get_column_widths(table)
+    table = pad_fields(table, col_widths)
+
+    # Reserve room for the spaces
+    col_widths = map(lambda x: x + 2, col_widths)
+    header_line = table_line(col_widths, header=True)
+    normal_line = table_line(col_widths, header=False)
+
+    output = [header_line]
+    first = True
+    for row in table:
+        output.append("|".join([''] + row + ['']))
+        if first:
+            output.append(header_line)
+            first = False
+        else:
+            output.append(normal_line)
+
+    return output
+
+
+def create_table():
+    upper, lower = get_table_bounds()
+    slice = vim.buffer[upper - 1:lower]
+    slice = """\
++==========+================================================================================================+
+| Column 1 | Column 2                                                                                       |
++==========+================================================================================================+
+| Foo      | Put two (or more) spaces as a field separator.                                                 |
++----------+------------------------------------------------------------------------------------------------+
+| Bar      | Even very very long lines like these are fine, as long as you do not put in line endings here. |
++----------+------------------------------------------------------------------------------------------------+
+| Qux      | This is the last line.                                                                         |
++----------+------------------------------------------------------------------------------------------------+
+""".strip().split('\n')
+
+    #table = parse_table(slice)
+    #slice = draw_table(table)
+    vim.buffer[upper - 1:lower] = slice
 
 endpython