hg.lua

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
module("hg", package.seeall)

__index = _M;

function open(repo_path)
	return setmetatable({ path = repo_path }, _M);
end

function exec(repo, command, ...)
	local params = {};
	for i=1,select('#', ...) do
		local param = select(i, ...);
		if param then
			--TODO: Shell escape
			params[i] = param;
		end
	end
	params = table.concat(params, " ");
	return os.execute("hg -y -q -R '"..repo.path.."' "..command.." "..params);
end

function execr(repo, command, ...)
	local params = {};
	for i=1,select('#', ...) do
		local param = select(i, ...);
		if param then
			--TODO: Shell escape
			params[i] = tostring(param);
		end
	end
	params = table.concat(params, " ");
	
	local cmd = io.popen("hg -y -R '"..repo.path.."' "..command.." "..params);
	if cmd then
		return cmd:read("*a");
	end
end

function update(repo, rev, force)
	repo:exec("up", rev and ("-r"..rev), force and "-C");
end

function identify(repo)
	local ret = repo:execr("id", "-n -i");
	if ret then
		local id, num =  ret:match("(%S+)%s(%S+)");
		return id, tonumber(num);
	end
end

function log(repo, rev, limit)
	local ret = repo:execr("log", rev and ("-r"..rev), limit and ("-l"..limit));
	if ret then
		local results, curr = {}, 1;
		for line in ret:gmatch("(.-\n)\n") do
			if not results[curr] then
				results[curr] = {};
			end
			for k,v in line:gmatch("(%w+):%s*(.-)%s*\n") do
			
				results[curr][k] = v;
			end
			curr = curr + 1;
		end
		return results;
	end
end