Mercurial > riddim / file revision

-- Plugin for querying a Github issue-tracker
--
-- Example config:
-- github = {
--   ["project@conference.example.org"] = {
--     user = "TheArchitect";
--     repo = "TheMatrix";
--   }
-- }

local url = require"socket.url";
local json = require"util.json";
local http = require"net.http";

function riddim.plugins.github(bot)
	local conf = bot.config.github;
	local base_url = url.parse("https://api.github.com/repos/x/y/issues/123");
	local base_path = url.parse_path(base_url.path);

	local ex = {
		headers = {
			Accept = "application/vnd.github.v3+json";
		};
	};

	local function get_issue_url(conf, number)
		base_path[2] = conf.user;
		base_path[3] = conf.repo or conf.project;
		base_path[5] = number;
		base_url.path = url.build_path(base_path);
		local url = url.build(base_url);
		return url;
	end

	bot:hook("commands/issue", function (command)
		local issue_id = tonumber(command.param);
		if not issue_id then return; end
		local current_conf = conf[command.room and command.room.jid] or conf;
		if not current_conf.user then return end
		assert(http.request(get_issue_url(current_conf, issue_id), ex, function (issue, code)
			if code > 400 then
				return command:reply("HTTP Error "..code.." :(");
			end
			issue = issue and json.decode(issue);
			if not issue then
				return command:reply("Got invalid JSON back :(");
			end
			command:reply(("%s #%d\n%s"):format(issue.title, issue.number, issue.html_url));
		end));
		return true;
	end);

	local function check_for_issue_id(message)
		local issue_id = message.body and message.body:match"#(%d+)";
		if issue_id then
			return bot:event("commands/issue", { param = issue_id, reply = message.reply, });
		end
	end

	bot:hook("groupchat/joining", function (room)
		room:hook("message", check_for_issue_id);
	end);
end