-- @@SHEBANG
-- -*- Lua -*-
-- gitano-post-receive-hook
--
-- Git (with) Augmented network operations -- Post-receive hook handler
--
-- Copyright 2012 Daniel Silverstone <dsilvers@digital-scurf.org>
--
--

-- @@GITANO_LUA_PATH

local gitano = require "gitano"
local gall = require "gall"
local luxio = require "luxio"
local sio = require "luxio.simple"
local sp = require "luxio.subprocess"

-- @@GITANO_BIN_PATH
-- @@GITANO_SHARE_PATH

local start_log_level = gitano.log.get_level()
-- Clamp level at info until we have checked if the caller
-- is an admin or not
gitano.log.cap_level(gitano.log.level.INFO)
gitano.log.syslog.open()

local repo_root = luxio.getenv("GITANO_ROOT")
local username  = luxio.getenv("GITANO_USER") or "gitano/anonymous"
local keytag    = luxio.getenv("GITANO_KEYTAG") or "unknown"
local project   = luxio.getenv("GITANO_PROJECT") or ""
local source    = luxio.getenv("GITANO_SOURCE") or "ssh"

-- Now load the administration data
gitano.config.repo_path(repo_root)
local admin_repo = gall.repository.new((repo_root or "") .. "/gitano-admin.git")

if not admin_repo then
   gitano.log.fatal("Unable to locate administration repository.  Cannot continue");
end

local admin_head = admin_repo:get(admin_repo.HEAD)

if not admin_head then
   gitano.log.fatal("Unable to find the HEAD of the administration repository.  Cannot continue");
end

local config, msg = gitano.config.parse(admin_head)

if not config then
   gitano.log.critical("Unable to parse administration repository.")
   gitano.log.critical("  * " .. (msg or "No error?"))
   gitano.log.fatal("Cannot continue")
end

-- Now, are we an admin?
if config.groups["gitano-admin"].filtered_members[username] then
   -- Yep, so blithely reset logging level
   gitano.log.set_level(start_log_level)
end

if not config.global.silent then
   -- Not silent, bump to chatty level automatically
   gitano.log.bump_level(gitano.log.level.CHAT)
end

local repo, msg = gitano.repository.find(config, project)
if not repo then
   gitano.log.critical("Unable to locate repository.")
   gitano.log.critical("  * " .. (tostring(msg)))
   gitano.log.fatal("Cannot continue")
end
   
if repo.is_nascent then
   gitano.log.fatal("Repository " .. repo.name .. " is nascent")
end

-- Post-receive is not allowed to prevent updates, instead on stdin we get
-- a list of the refs updated and we get to react to that (such as sending
-- emails, ensuring that new rules are applied, etc)

local updates = {}
for oldsha, newsha, refname in
   (sio.stdin:read("*a")):gmatch("([^ ]+) ([^ ]+) ([^\n]+)\n?") do
   gitano.log.ddebug("post-receive:", oldsha, newsha, refname)
   updates[refname] = {oldsha, newsha, oldsha=oldsha, newsha=newsha}
end

-- updates to refs/gitano/admin will *already* have been processed thanks
-- to the repository being found above.  So if that's present, just note
-- that the updates (if any) will have been applied

if updates["refs/gitano/admin"] then
   local msg = "<" .. repo.name .. ">" .. " Any changes to admin ref have been applied."

   gitano.log.chat(msg)
   gitano.log.syslog.info(msg)
end

local function report_repo(reponame, repo, msg)
   if repo then
      local s = "<" .. reponame ..">" .. " Any changes to hooks etc have been applied"

      gitano.log.chat(s)
      gitano.log.syslog.info(s)
   else
      gitano.log.crit("<" .. reponame .. ">", "Unable to process:", msg)
   end
end

if repo.name == "gitano-admin" and updates[admin_repo.HEAD] then
   -- Updating the 'master' of gitano-admin, let's iterate all the repositories

   gitano.log.syslog.info("Updating gitano-admin")

   local msg = "Scanning repositories to apply hook/rules updates..."
   gitano.log.chat(msg)
   gitano.log.syslog.info(msg)

   local ok, msg = gitano.repository.foreach(config, report_repo)
   if not ok then
      gitano.log.crit(msg)
   end

   msg = "All repositories updated where possible."
   gitano.log.chat(msg)
   gitano.log.syslog.info(msg)

   local proc = sp.spawn({
			    gitano.config.lib_bin_path() .. "/gitano-update-ssh",
			    gitano.config.repo_path()
			 })
   local how, why = proc:wait()
   if how ~= "exit" or why ~= 0 then
      gitano.log.crit("Updating SSH keys didn't work?")
   end
elseif repo.name ~= "gitano-admin" then
   -- Not gitano-admin at all, so run the update-server-info stuff
   gitano.log.info("Updating server info for dumb HTTP transport")
   local ok, err = repo.git:update_server_info()
   if not ok then
      gitano.log.warn(err)
   end
   gitano.log.info("Updating last-modified date")
   local shas = {}
   for _, t in pairs(updates) do
      shas[#shas+1] = t.newsha
   end
   local ok, err = repo:update_modified_date(shas)
   if not ok then
      gitano.log.warn(err)
   end
end

if repo:uses_hook("post-receive") then
   gitano.log.debug("Configuring for post-receive hook")
   gitano.actions.set_supple_globals("post-receive")

   local msg = "Running repository post-receive hook"
   gitano.log.info(msg)
   gitano.log.syslog.info(msg)

   local info = {
      username = username,
      keytag = keytag,
      source = source,
      realname = (config.users[username] or {}).real_name or "",
      email = (config.users[username] or {}).email_address or "",
   }
   local ok, msg = gitano.supple.run_hook("post-receive", repo, info, updates)
   if not ok then
      gitano.log.crit(msg or "No reason given, but errored somehow.")
   end
   gitano.log.info("Finished")
end

gitano.log.syslog.close()

return 0