import("core.package.package")
import("core.base.semver")
import("core.base.hashset")
import("devel.git")
import("packages", {alias = "packages_util"})

function _load_package(packagename, packagedir, packagefile)
    local funcinfo = debug.getinfo(package.load_from_repository)
    if funcinfo and funcinfo.nparams == 3 then -- >= 2.7.8
        return package.load_from_repository(packagename, packagedir, {packagefile = packagefile})
    else
        -- deprecated
        return package.load_from_repository(packagename, nil, packagedir, packagefile)
    end
end

function _get_all_packages(pattern)
    local packages = _g.packages
    if not packages then
        packages = {}
        for _, packagedir in ipairs(os.dirs(path.join("packages", "*", "*"))) do
            local packagename = path.filename(packagedir)
            if not pattern or packagename:match(pattern) then
                local packagefile = path.join(packagedir, "xmake.lua")
                local instance = _load_package(packagename, packagedir, packagefile)
                local basename = instance:get("base")
                if instance and basename then
                    local basedir = path.join("packages", basename:sub(1, 1):lower(), basename:lower())
                    local basefile = path.join(basedir, "xmake.lua")
                    instance._BASE = _load_package(basename, basedir, basefile)
                end
                if instance then
                    table.insert(packages, instance)
                end
            end
        end
        _g.packages = packages
    end
    return packages
end

function _is_pending(instance, version)
    local branch = "autoupdate-" .. instance:name() .. "-" .. version
    local repourl = "git@github.com:xmake-io/xmake-repo.git"
    local is_pending = false
    local remote_branches = os.iorun("git ls-remote --head %s", repourl)
    if remote_branches then
        for _, remote_branch in ipairs(remote_branches:split("\n")) do
            remote_branch = remote_branch:split("%s")[2]
            if remote_branch == "refs/heads/" .. branch then
                is_pending = true
                break
            end
        end
    end
    return is_pending
end

function _update_version(instance, version, shasum)
    local branch = "autoupdate-" .. instance:name() .. "-" .. version
    local branch_current = os.iorun("git branch --show-current"):trim()
    local repourl = "git@github.com:xmake-io/xmake-repo.git"
    os.vexec("git reset --hard HEAD")
    os.vexec("git clean -fdx")
    os.execv("git", {"branch", "-D", branch}, {try = true})
    os.vexec("git checkout dev")
    os.vexec("git pull %s dev", repourl)
    os.vexec("git branch %s", branch)
    os.vexec("git checkout %s", branch)
    local inserted = false
    local scriptfile = path.join(instance:scriptdir(), "xmake.lua")
    local version_current
    if os.isfile(scriptfile) then
        io.gsub(scriptfile, "add_versions%(\"(.-)\",%s+\"(.-)\"%)", function (v, h)
            if not version_current or semver.compare(v, version_current) > 0 then
                version_current = v
            end
            if not inserted then
                inserted = true
                return string.format('add_versions("%s", "%s")\n    add_versions("%s", "%s")', version, shasum, v, h)
            end
        end)
    end
    if not inserted then
        local versionfiles = instance:get("versionfiles")
        if versionfiles then
            for _, versionfile in ipairs(table.wrap(versionfiles)) do
                if not os.isfile(versionfile) then
                    versionfile = path.join(instance:scriptdir(), versionfile)
                end
                if os.isfile(versionfile) then
                    io.insert(versionfile, 1, string.format("%s %s", version, shasum))
                    inserted = true
                end
            end
        end
    end
    if inserted then
        local body = string.format("New version of %s detected (package version: %s, last github version: %s)",
            instance:name(), version_current, version)
        os.vexec("git add .")
        os.vexec("git commit -a -m \"Update %s to %s\"", instance:name(), version)
        os.vexec("git push %s %s:%s", repourl, branch, branch)
        os.vexec("gh pr create --label \"auto-update\" --title \"Auto-update %s to %s\" --body \"%s\" -R xmake-io/xmake-repo -B dev -H %s",
            instance:name(), version, body, branch)
    end
    os.vexec("git reset --hard HEAD")
    os.vexec("git checkout %s", branch_current)
end

function main(pattern)
    local count = 0
    local maxcount = 5
    local instances = _get_all_packages(pattern)
    if #instances < maxcount then
        maxcount = #instances
    end
    math.randomseed(os.time())
    while count < maxcount and #instances > 0 do
        local idx = math.random(#instances)
        local instance = instances[idx]
        local checkupdate_filepath = path.join(instance:scriptdir(), "checkupdate.lua")
        if not os.isfile(checkupdate_filepath) then
            checkupdate_filepath = path.join(os.scriptdir(), "checkupdate.lua")
        end
        local updated = false
        if os.isfile(checkupdate_filepath) then
            local checkupdate = import("checkupdate", {rootdir = path.directory(checkupdate_filepath), anonymous = true})
            local version, shasum = checkupdate(instance)
            if version and shasum and not _is_pending(instance, version) then
                cprint("package(%s): new version ${bright}%s${clear} found, shasum: ${bright}%s", instance:name(), version, shasum)
                _update_version(instance, version, shasum)
                updated = true
            end
        end
        if updated then
            count = count + 1
        end
        table.remove(instances, idx)
    end
end