core.lua

local exports = {}
local modes = {
  tree   = 16384, --  040000
  blob   = 33188, -- 0100644
  exec   = 33261, -- 0100755
  sym    = 40960, -- 0120000
  commit = 57344, -- 0160000
}
modes.file = modes.blob
exports.modes = modes
local encoders = {}
exports.encoders = encoders
local decoders = {}
exports.decoders = decoders

-- Given a raw character, return two hex characters
local function binToHex(c)
  return string.format("%02x", string.byte(c, 1))
end

exports.binToHex = function(bin)
  local hex = string.gsub(bin, ".", binToHex)
  return hex
end

-- Given two hex characters, return a single character
local function hexToBin(cc)
  return string.char(tonumber(cc, 16))
end

exports.hexToBin = function (hex)
  local bin = string.gsub(hex, "..", hexToBin)
  return bin
end

function modes.isBlob(mode)
  -- (mode & 0140000) == 0100000
  return bit.band(mode, 49152) == 32768
end

function modes.isFile(mode)
  -- (mode & 0160000) === 0100000
  return bit.band(mode, 57344) == 32768
end

function modes.toType(mode)
         -- 0160000
  return mode == 57344 and "commit"
         -- 040000
      or mode == 16384 and "tree"
         -- (mode & 0140000) == 0100000
      or bit.band(mode, 49152) == 32768 and "blob" or
         "unknown"
end

local function treeSort(a, b)
  return ((a.mode == modes.tree) and (a.name .. "/") or a.name)
       < ((b.mode == modes.tree) and (b.name .. "/") or b.name)
end


-- Remove illegal characters in things like emails and names
local function safe(text)
  return text:gsub("^[%.,:;\"']+", "")
             :gsub("[%.,:;\"']+$", "")
             :gsub("[%z\n<>]+", "")
end
exports.safe = safe

local function formatDate(date)
  local seconds = date.seconds
  local offset = date.offset
  assert(type(seconds) == "number", "date.seconds must be number")
  assert(type(offset) == "number", "date.offset must be number")
  return string.format("%d %+03d%02d", seconds, offset / 60, offset % 60)
end

local function formatPerson(person)
  assert(type(person.name) == "string", "person.name must be string")
  assert(type(person.email) == "string", "person.email must be string")
  assert(type(person.date) == "table", "person.date must be table")
  return safe(person.name) ..
    " <" .. safe(person.email) .. "> " ..
    formatDate(person.date)
end

local function parsePerson(raw)
  local pattern = "^([^<]+) <([^>]+)> (%d+) ([-+])(%d%d)(%d%d)$"
  local name, email, seconds, sign, hours, minutes = string.match(raw, pattern)
  local offset = tonumber(hours) * 60 + tonumber(minutes)
  if sign == "-" then offset = -offset end
  return {
    name = name,
    email = email,
    date = {
      seconds = tonumber(seconds),
      offset = offset
    }
  }
end

function encoders.blob(blob)
  assert(type(blob) == "string", "blobs must be strings")
  return blob
end

function decoders.blob(raw)
  return raw
end

function encoders.tree(tree)
  assert(type(tree) == "table", "trees must be tables")
  local parts = {}
  for i = 1, #tree do
    local value = tree[i]
    assert(type(value.name) == "string", "tree entry name must be string")
    assert(type(value.mode) == "number", "tree entry mode must be number")
    assert(type(value.hash) == "string", "tree entry hash must be string")
    parts[#parts + 1] = {
      name = value.name,
      mode = value.mode,
      hash = value.hash,
    }
  end
  table.sort(parts, treeSort)
  for i = 1, #parts do
    local entry = parts[i]
    parts[i] = string.format("%o %s\0",
      entry.mode,
      entry.name
    ) .. string.gsub(entry.hash, "..", hexToBin)
  end
  return table.concat(parts)
end

function decoders.tree(raw)
  local start, length = 1, #raw
  local tree = {}
  while start <= length do
    local pattern = "^([0-7]+) ([^%z]+)%z(....................)"
    local s, e, mode, name, hash = string.find(raw, pattern, start)
    assert(s, "Problem parsing tree")
    hash = string.gsub(hash, ".", binToHex)
    mode = tonumber(mode, 8)
    tree[#tree + 1] = {
      name = name,
      mode = mode,
      hash = hash
    }
    start = e + 1
  end
  return tree
end

function encoders.tag(tag)
  assert(type(tag) == "table", "annotated tags must be tables")
  assert(type(tag.object) == "string", "tag.object must be hash string")
  assert(type(tag.type) == "string", "tag.type must be string")
  assert(type(tag.tag) == "string", "tag.tag must be string")
  assert(type(tag.tagger) == "table", "tag.tagger must be table")
  assert(type(tag.message) == "string", "tag.message must be string")
  if tag.message[#tag.message] ~= "\n" then
    tag.message = tag.message .. "\n"
  end
  return string.format(
    "object %s\ntype %s\ntag %s\ntagger %s\n\n%s",
    tag.object, tag.type, tag.tag, formatPerson(tag.tagger), tag.message)
end

function decoders.tag(raw)
  local s, e, _, message
  s, _, message = string.find(raw, "\n\n(.*)$", 1)
  raw = string.sub(raw, 1, s)
  local start = 1
  local pattern = "^([^ ]+) ([^\n]+)\n"
  local data = {message=message}
  while true do
    local name, value
    s, e, name, value = string.find(raw, pattern, start)
    if not s then break end
    if name == "tagger" then
      value = parsePerson(value)
    end
    data[name] = value
    start = e + 1
  end
  return data
end


function encoders.commit(commit)
  assert(type(commit) == "table", "commits must be tables")
  assert(type(commit.tree) == "string", "commit.tree must be hash string")
  assert(type(commit.parents) == "table", "commit.parents must be table")
  assert(type(commit.author) == "table", "commit.author must be table")
  assert(type(commit.committer) == "table", "commit.committer must be table")
  assert(type(commit.message) == "string", "commit.message must be string")
  local parents = {}
  for i = 1, #commit.parents do
    local parent = commit.parents[i]
    assert(type(parent) == "string", "commit.parents must be hash strings")
    parents[i] = string.format("parent %s\n", parent)
  end
  return string.format(
    "tree %s\n%sauthor %s\ncommitter %s\n\n%s",
    commit.tree, table.concat(parents), formatPerson(commit.author),
    formatPerson(commit.committer), commit.message)
end

function decoders.commit(raw)
  local s, e, _, message
  s, _, message = string.find(raw, "\n\n(.*)$", 1)
  raw = string.sub(raw, 1, s)
  local start = 1
  local pattern = "^([^ ]+) ([^\n]+)\n"
  local parents = {}
  local data = {message=message,parents=parents}
  while true do
    local name, value
    s, e, name, value = string.find(raw, pattern, start)
    if not s then break end
    if name == "author" or name == "committer" then
      value = parsePerson(value)
    end
    if name == "parent" then
      parents[#parents + 1] = value
    else
      data[name] = value
    end
    start = e + 1
  end
  return data
end

function exports.frame(kind, data)
  assert(type(data) == "string", "data must be pre-encoded string")
  data = string.format("%s %d\0", kind, #data) .. data
  return data
end

function exports.deframe(raw)
  local pattern = "^([^ ]+) (%d+)%z(.*)$"
  local kind, size, body = string.match(raw, pattern)
  assert(kind, "Problem parsing framed git data")
  size = tonumber(size)
  assert(size == #body, "Body size mismatch")
  return kind, body
end

function exports.listToMap(list)
  local map = {}
  for i = 1, #list do
    local entry = list[i]
    map[entry.name] = {mode = entry.mode, hash = entry.hash}
  end
  return map
end

function exports.mapToList(map)
  local list = {}
  for name, entry in pairs(map) do
    list[#list + 1] = {name = name, mode = entry.mode, hash = entry.hash}
  end
  return list
end

return exports