Lua script for splash

lua_source.lua

    lua_source = """
    function main(splash)
        local host = "proxy.crawlera.com"
        local port = 8010
        local user = "<API_KEY>"
        local password = ''
        local session_header = "X-Crawlera-Session"
        local session_id = "create"

        splash:on_request(function (request)
            request:set_header("X-Crawlera-UA", "desktop")
            request:set_header(session_header, session_id)
            request:set_header("X-Crawlera-Max-Retries" : 1)
            request:set_proxy{host, port, username=user, password=password}
        end)

        splash:on_response_headers(function (response)
            if type(response.headers[session_header]) ~= nil then
                session_id = response.headers[session_header]
            end
        end)

        splash:set_result_content_type("text/html; charset=utf-8")
        print(splash.args.url)
        splash:go{splash.args.url,
                        headers=splash.args.headers,
                        http_method=splash.args.http_method,
                        body=splash.args.body,}
        assert(splash:wait(splash.args.wait))
        return splash:html()
    end
    """