ledger-sort

ledger-sort

#!/usr/bin/env lua

-- This is a tool to sort Ledger CLI journal file.
-- Implemented to be used as a filter in (Neo)VIM.
-- Usage (CLI): ledger-sort < ledger.txt
-- Usage ((Neo)VIM): :%!ledger-sort
-- You can also visually select the lines to be sorted.

local function extend(dst, src)
  for _, v in ipairs(src) do
    table.insert(dst, v)
  end
end

local function ledgersort(lines)
  local items = {}
  local prefix = {}
  local current = nil

  for _, l in ipairs(lines) do
    local date = l:match("^[0-9/-]+")

    if date then
      current = {date = date, header = l, prefix = prefix, suffix = {}}
      table.insert(items, current)
      prefix = {}
    elseif l:match("%S") then
      table.insert(current and current.suffix or prefix, l)
    elseif current then
      current = nil
    end
  end

  if #items == 0 then
    return lines
  end

  table.sort(items, function(a, b)
    return a.date < b.date
  end)

  local newlines = {}
  for _, i in ipairs(items) do
    extend(newlines, i.prefix)
    table.insert(newlines, i.header)
    extend(newlines, i.suffix)
    table.insert(newlines, "")
  end

  extend(newlines, prefix)
  return newlines
end

local lines = {}
for line in io.lines() do
  table.insert(lines, line)
end

for _, line in ipairs(ledgersort(lines)) do
  print(line)
end