aboutsummaryrefslogtreecommitdiff
path: root/lib/curl.lua
blob: bcbcfe46b4993b28bae306aeaddb364c7288ec8d (plain) (blame)
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
#!/usr/bin/env lua

local util = require 'lib/util'

local def_headr = {
	['User-Agent'] = 'cia',
	['Accept-Encoding'] = 'gzip, deflate, br'
}

local gen_head = function (t)
	local heads = ' '

	for k, v in pairs(t) do
		heads = heads .. "-H '" .. k .. ": " .. v .. "' "
	end

	return heads
end

local get = function (url, headr, args)
	local cmd, fetch, scode

	headr = util.table_merge(headr, def_headr)
	cmd = 'curl -s --compressed --write-out %{http_code} ' ..
	      url .. gen_head(headr)
	if args then
		cmd = cmd .. ' ' .. args
	end

	fetch = io.popen(cmd):read('*all')
	scode = string.match(fetch, '%d*$')
	fetch = string.gsub(fetch, '%s*%d*$', '')

	return fetch, tonumber(scode)
end

return {
	get = get,
}