# vim:set ft= ts=4 sw=4 et fdm=marker:
use lib 'lib';
use Test::Nginx::Socket::Lua;
use Cwd qw(cwd);

#worker_connections(1014);
#master_process_enabled(1);
#log_level('warn');

repeat_each(2);

plan tests => repeat_each() * (blocks() * 4 + 9);

my $pwd = cwd();

our $HttpConfig = <<_EOC_;
    lua_package_path "$pwd/lib/?.lua;../lua-resty-lrucache/lib/?.lua;;";
    init_by_lua '
        -- local verbose = true
        local verbose = false
        local outfile = "$Test::Nginx::Util::ErrLogFile"
        -- local outfile = "/tmp/v.log"
        if verbose then
            local dump = require "jit.dump"
            dump.on(nil, outfile)
        else
            local v = require "jit.v"
            v.on(outfile)
        end

        require "resty.core"
        -- jit.opt.start("hotloop=1")
        -- jit.opt.start("loopunroll=1000000")
        -- jit.off()
    ';
_EOC_

#no_diff();
no_long_string();
check_accum_error_log();
run_tests();

__DATA__

=== TEST 1: sub, no submatch, no jit compile, regex cache
--- http_config eval: $::HttpConfig
--- config
    location = /re {
        access_log off;
        content_by_lua '
            local m, err
            local sub = ngx.re.sub
            for i = 1, 350 do
                s, n, err = sub("abcbd", "b", "B", "jo")
            end
            if not s then
                ngx.log(ngx.ERR, "failed: ", err)
                return
            end
            ngx.say("s: ", s)
            ngx.say("n: ", n)
        ';
    }
--- request
GET /re
--- response_body
s: aBcbd
n: 1
--- error_log eval
qr/\[TRACE   \d+ content_by_lua\(nginx\.conf:\d+\):4 loop\]/
--- no_error_log
[error]
bad argument type
NYI



=== TEST 2: sub, no submatch, no jit compile, no regex cache
--- http_config eval: $::HttpConfig
--- config
    location = /re {
        access_log off;
        content_by_lua '
            local m, err
            local sub = ngx.re.sub
            for i = 1, 400 do
                s, n, err = sub("abcbd", "b", "B")
            end
            if not s then
                ngx.log(ngx.ERR, "failed: ", err)
                return
            end
            ngx.say("s: ", s)
            ngx.say("n: ", n)
        ';
    }
--- request
GET /re
--- response_body
s: aBcbd
n: 1
--- error_log eval
qr/\[TRACE   \d+ content_by_lua\(nginx\.conf:\d+\):4 loop\]/
--- no_error_log
[error]
bad argument type



=== TEST 3: func + submatches
--- http_config eval: $::HttpConfig
--- config
    location = /re {
        access_log off;
        content_by_lua '
            local m, err
            local function f(m)
                return "[" .. m[0] .. "(" .. m[1] .. ")]"
            end
            local sub = ngx.re.sub
            for i = 1, 200 do
                s, n, err = sub("abcbd", "b(c)", f, "jo")
            end
            if not s then
                ngx.log(ngx.ERR, "failed: ", err)
                return
            end
            ngx.say("s: ", s)
            ngx.say("n: ", n)
        ';
    }
--- request
GET /re
--- response_body
s: a[bc(c)]bd
n: 1
--- no_error_log eval
[
"[error]",
"bad argument type",
qr/NYI (?!bytecode 51 at)/,
]



=== TEST 4: replace template + submatches
--- http_config eval: $::HttpConfig
--- config
    location = /re {
        access_log off;
        content_by_lua '
            local m, err
            local sub = ngx.re.sub
            for i = 1, 350 do
                s, n, err = sub("abcbd", "b(c)", "[$0($1)]", "jo")
            end
            if not s then
                ngx.log(ngx.ERR, "failed: ", err)
                return
            end
            ngx.say("s: ", s)
            ngx.say("n: ", n)
        ';
    }
--- request
GET /re
--- response_body
s: a[bc(c)]bd
n: 1
--- error_log eval
qr/\[TRACE   \d+ content_by_lua\(nginx\.conf:\d+\):4 loop\]/

--- no_error_log
[error]
bad argument type
NYI



=== TEST 5: replace template + submatches (exceeding buffers)
--- http_config eval: $::HttpConfig
--- config
    location = /re {
        access_log off;
        content_by_lua '
            local m, err
            local gsub = ngx.re.gsub
            local subj = string.rep("bcbd", 2048)
            for i = 1, 10 do
                s, n, err = gsub(subj, "b(c)", "[$0($1)]", "jo")
            end
            if not s then
                ngx.log(ngx.ERR, "failed: ", err)
                return
            end
            ngx.say("s: ", s)
            ngx.say("n: ", n)
        ';
    }
--- request
GET /re
--- response_body eval
"s: " . ("[bc(c)]bd" x 2048) .
"\nn: 2048\n"

--- no_error_log
[error]
bad argument type



=== TEST 6: ngx.re.gsub: use of ngx.req.get_headers in the user callback
--- http_config eval: $::HttpConfig
--- config

location = /t {
    content_by_lua '
        local data = [[
            INNER
            INNER
]]

        -- ngx.say(data)

        local res =  ngx.re.gsub(data, "INNER", function(inner_matches)
            local header = ngx.req.get_headers()["Host"]
            -- local header = ngx.var["http_HEADER"]
            return "INNER_REPLACED"
        end, "s")

        ngx.print(res)
    ';
}

--- request
GET /t
--- response_body
            INNER_REPLACED
            INNER_REPLACED

--- no_error_log
[error]
bad argument type
NYI



=== TEST 7: ngx.re.gsub: use of ngx.var in the user callback
--- http_config eval: $::HttpConfig
--- config

location = /t {
    content_by_lua '
        local data = [[
            INNER
            INNER
]]

        -- ngx.say(data)

        local res =  ngx.re.gsub(data, "INNER", function(inner_matches)
            -- local header = ngx.req.get_headers()["Host"]
            local header = ngx.var["http_HEADER"]
            return "INNER_REPLACED"
        end, "s")

        ngx.print(res)
    ';
}

--- request
GET /t
--- response_body
            INNER_REPLACED
            INNER_REPLACED

--- no_error_log
[error]
bad argument type
NYI



=== TEST 8: ngx.re.gsub: recursive calling (github openresty/lua-nginx-module#445)
--- http_config eval: $::HttpConfig
--- config

location = /t {
    content_by_lua '
        function test()
            local data = [[
                OUTER {FIRST}
]]

            local p1 = "(OUTER)(.+)"
            local p2 = "{([A-Z]+)}"

            ngx.print(data)

            local res =  ngx.re.gsub(data, p1, function(m)
                -- ngx.say("pre: m[1]: [", m[1], "]")
                -- ngx.say("pre: m[2]: [", m[2], "]")

                local res = ngx.re.gsub(m[2], p2, function(_)
                    return "REPLACED"
                end, "")

                -- ngx.say("post: m[1]: [", m[1], "]")
                -- ngx.say("post m[2]: [", m[2], "]")
                return m[1] .. res
            end, "")

            ngx.print(res)
        end

        test()
    ';
}
--- request
GET /t
--- response_body
                OUTER {FIRST}
                OUTER REPLACED
--- no_error_log
[error]
bad argument type
NYI



=== TEST 9: string replace subj is not a string type
--- http_config eval: $::HttpConfig
--- config
    location /re {
        content_by_lua '
			local newstr, n, err = ngx.re.sub(1234, "([0-9])[0-9]", 5, "jo")

			ngx.say(newstr)
        ';
    }
--- request
    GET /re
--- response_body
534
--- no_error_log
[error]
attempt to get length of local 'subj' (a number value)



=== TEST 10: func replace return is not a string type (ngx.re.sub)
--- http_config eval: $::HttpConfig
--- config
    location /re {
        content_by_lua '
			local lookup = function(m)
				-- note we are returning a number type here
				return 5
			end

			local newstr, n, err = ngx.re.sub("hello, 1234", "([0-9])[0-9]", lookup, "jo")
			ngx.say(newstr)
        ';
    }
--- request
    GET /re
--- response_body
hello, 534
--- no_error_log
[error]
attempt to get length of local 'bit' (a number value)



=== TEST 11: func replace return is not a string type (ngx.re.gsub)
--- http_config eval: $::HttpConfig
--- config
    location /re {
        content_by_lua '
			local lookup = function(m)
				-- note we are returning a number type here
				return 5
			end

			local newstr, n, err = ngx.re.gsub("hello, 1234", "([0-9])[0-9]", lookup, "jo")
			ngx.say(newstr)
        ';
    }
--- request
    GET /re
--- response_body
hello, 55
--- no_error_log
[error]
attempt to get length of local 'bit' (a number value)
