#!/usr/bin/lua5.1
-- -*- Lua -*-
-- gitano-post-receive-hook
--
-- Git (with) Augmented network operations -- Post-receive hook handler
--
-- Copyright 2012-2017 Daniel Silverstone <dsilvers@digital-scurf.org>
-- All rights reserved.
--
-- Redistribution and use in source and binary forms, with or without
-- modification, are permitted provided that the following conditions
-- are met:
-- 1. Redistributions of source code must retain the above copyright
--    notice, this list of conditions and the following disclaimer.
-- 2. Redistributions in binary form must reproduce the above copyright
--    notice, this list of conditions and the following disclaimer in the
--    documentation and/or other materials provided with the distribution.
-- 3. Neither the name of the author nor the names of their contributors
--    may be used to endorse or promote products derived from this software
--    without specific prior written permission.
--
-- THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
-- ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
-- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
-- ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
-- FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
-- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
-- OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
-- HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
-- LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
-- OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
-- SUCH DAMAGE.
--
--

--[[Gitano modules installed into /usr/share/lua/5.1/?.lua]]

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

gitano.config.lib_bin_path("/usr/lib/gitano/bin")
gitano.config.share_path("/usr/share/gitano")
gitano.i18n.set_langpack_path("/usr/share/gitano/lang") gitano.i18n.set_category()
gitano.plugins.load_plugins {"/etc/gitano/plugins", "/usr/lib/gitano/plugins"}

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"
local running   = luxio.getenv("GITANO_RUNNING")

-- Check whether we are called through gitano-auth
if not running then
   return 0
end

-- 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(gitano.i18n.expand("ERROR_NO_ADMIN_REPO"));
end

local admin_head = admin_repo:get(admin_repo.HEAD)

if not admin_head then
   gitano.log.fatal(gitano.i18n.expand("ERROR_BAD_ADMIN_REPO"));
end

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

if not config then
   gitano.log.critical(gitano.i18n.expand("ERROR_CANNOT_PARSE_ADMIN"))
   gitano.log.critical("  * " .. (msg or "No error?"))
   gitano.log.fatal(gitano.i18n.expand("ERROR_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(gitano.i18n.expand("ERROR_CANNOT_LOCATE_REPO"))
   gitano.log.critical("  * " .. (tostring(msg)))
   gitano.log.fatal(gitano.i18n.expand("ERROR_CANNOT_CONTINUE"))
end

if repo.is_nascent then
   gitano.log.fatal(gitano.i18n.expand("ERROR_REPO_IS_NASCENT", {name=repo.name}))
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(pat.GITHOOK_PARSE_CHANGESET) 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 = gitano.i18n.expand("CHANGES_TO_ADMINREF_APPLIED", {name=repo.name})

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

local function report_repo(reponame, repo, msg)
   if repo then
      local s = gitano.i18n.expand("CHANGES_TO_HOOKS_APPLIED", {name=repo.name})

      gitano.log.chat(s)
      gitano.log.syslog.info(s)
   else
      gitano.log.crit(gitano.i18n.expand("ERROR_UNABLE_TO_PROCESS", {name=repo.name, msg=msg}))
   end
end

function post_receive_core_handler(repo, updates)
   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 = gitano.i18n.expand("SCANNING_FOR_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 = gitano.i18n.expand("ALL_UPDATES_DONE")
      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(gitano.i18n.expand("ERROR_UPDATE_SSH_NOT_WORK"))
      end
   elseif repo.name ~= "gitano-admin" then
      -- Not gitano-admin at all, so run the update-server-info stuff
      gitano.log.info(gitano.i18n.expand("UPDATE_HTTP_INFO"))
      local ok, err = repo.git:update_server_info()
      if not ok then
         gitano.log.warn(err)
      end
      gitano.log.info(gitano.i18n.expand("UPDATE_LASTMOD_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
   return "continue"
end

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

      local msg = gitano.i18n.expand("RUNNING_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 gitano.i18n.expand("ERROR_NO_ERROR_FOUND"))
      end
      gitano.log.info(gitano.i18n.expand("FINISHED"))
   end
   return "continue"
end

function post_receive_check_head(repo, updates)
   -- Check that HEAD is now resolvable in the repo
   if not repo.git:get("HEAD") then
      gitano.log.warn("")
      gitano.log.warn(gitano.i18n.expand("WARN_HEAD_DANGLING"))
      gitano.log.warn("")
   end
   return "continue"
end

gitano.hooks.add(gitano.hooks.names.POST_RECEIVE, -1000,
                 post_receive_core_handler)
gitano.hooks.add(gitano.hooks.names.POST_RECEIVE, 0, post_receive_run_supple)
gitano.hooks.add(gitano.hooks.names.POST_RECEIVE, 1000, post_receive_check_head)

gitano.hooks.run(gitano.hooks.names.POST_RECEIVE, repo, updates)

gitano.log.syslog.close()

return 0
