local cjson = require "cjson"
local http = require "resty.http"
local pl = require "pl.pretty"
local httpc = http.new()
local sudoerToken = ngx.req.get_headers().HtySudoerToken
local htyhostHeader = ngx.req.get_headers().HtyHost


local function get_access_token()

    local htyuc = ngx.var.htyuc
    local host = ngx.var.host
    local wx_domain = ngx.var.wx_domain

    ngx.log(ngx.INFO, "Request URL -> ", ngx.var.uri)
    ngx.log(ngx.INFO, "HTYUC -> ", htyuc)
    ngx.log(ngx.INFO, "Host -> ", host)
    ngx.log(ngx.INFO, 'HtyHostHeader -> ', htyhostHeader)
    ngx.log(ngx.INFO, 'HtySudoerToken -> ', sudoerToken)

    local remote_url = string.format("%s/api/v1/uc/wx_get_access_token", htyuc)
    ngx.log(ngx.INFO, 'REMOTE URL -> ', remote_url)

    ngx.log(ngx.INFO, "START REQUEST TO GET ACCESS TOKEN")

    local resp, req_err = httpc:request_uri(
            remote_url,
            {
                ssl_verify = false, -- 设置参数 ssl_verify 为false 不校验ssl证书
                method = "GET",
                headers = {
                    ["HtyHost"] = wx_domain,
                    ["HtySudoerToken"] = sudoerToken,
                }
            }
    )

    ngx.log(ngx.INFO, "RESP -> ", pl.write(resp))
    ngx.log(ngx.ERR, "ERR -> ", req_err)

    if resp == nil then
        ngx.log(ngx.ERR, "FAILED TO CONNECT TO *HTYUC*", req_err)
    end

    if 200 ~= resp.status then
        ngx.log(ngx.ERR, "GET ACCESS TOKEN *FAILED*", req_err)
        ngx.say(req_err)
        ngx.exit(resp.status)
    else
        ngx.log(ngx.ERR, "RESP IS OK -> ", resp.body)
        return cjson.decode(resp.body).d
    end
end

local function get_wx_media(media_id)
    local access_token = get_access_token()
    print("access token: " .. access_token)
    local wx_media_url = string.format("https://api.weixin.qq.com/cgi-bin/media/get?access_token=%s&media_id=%s", access_token, media_id)
    print("wx_media_url: " .. wx_media_url)
    local resp, req_err = httpc:request_uri(wx_media_url, {
        method = 'GET',
        ssl_verify = ssl_verify or false, -- 设置参数 ssl_verify 为false 不校验ssl证书
    })

    if resp == nil then
        ngx.log(ngx.ERR, "FAILED TO CONNECT TO *Weixin*", req_err)
        ngx.say(req_err)
        ngx.exit(500)
    end

    if 200 ~= resp.status then
        ngx.log(ngx.ERR, 'GET WECHET MEDIA *FAILED*', req_err)
        ngx.say(req_err)
        ngx.exit(resp.status)
    end

    local filename = ngx.re.match(resp.headers['Content-disposition'], [[filename="(.+\.([a-zA-Z0-9])+)"]], "jo")[1]
    local file_dir = ngx.var.tmp_file_dir
    local saved_file = file_dir .. "/" .. filename;
    print('saved file...' .. saved_file);
    local file, err = io.open(saved_file, 'w')
    if file == nil then
        print("Can not open file..." .. err)
    else
        file:write(resp.body)
        file:close()
    end

    return saved_file;
end

return get_wx_media;