--!A cross-platform build utility based on Lua
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
--     http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
-- Copyright (C) 2015-present, Xmake Open Source Community.
--
-- @author      ruki
-- @file        remove.lua
--

-- imports
import("core.base.option")
import("core.package.addon")
import("private.action.require.impl.remove_packages", {alias = "remove_all_packages"})

-- get menu options
function menu_options()

    -- description
    local description = "Remove the given packages."

    -- menu options
    local options =
    {
        {'k', "kind",            "kv", nil, "Enable static/shared library.",
                                       values = {"static", "shared"}                            },
        {'p', "plat",            "kv", nil, "Set the given platform."                           },
        {'a', "arch",            "kv", nil, "Set the given architecture."                       },
        {'m', "mode",            "kv", nil, "Set the given mode.",
                                       values = {"release", "debug"}                            },
        {'f', "configs",         "kv", nil, "Set the given extra package configs.",
                                       "e.g.",
                                       "    - xrepo remove -f \"runtimes='MD'\" zlib",
                                       "    - xrepo remove -f \"regex=true,thread=true\" boost" },
        {nil, "toolchain",       "kv", nil, "Set the toolchain name."                           },
        {nil, "toolchain_host",  "kv", nil, "Set the host toolchain name."                      },
        {                                                                                       },
        {nil, "addon",           "k", nil,  "Remove the given installed addon packages.",
                                       "e.g.",
                                       "    - xrepo remove --addon serial-monitor"           },
        {'f', "force",           "k", nil,  "Force to remove the addon packages, even if they are depended on by the others." },
        {nil, "all",             "k", nil,  "Remove all packages and ignore extra package configs.",
                                       "If `--all` is enabled, the package name parameter will support lua pattern",
                                       "e.g.",
                                       "    - xrepo remove --all",
                                       "    - xrepo remove --all zlib boost",
                                       "    - xrepo remove --all zl* boo*"                      },
        {nil, "packages",        "vs", nil, "The packages list.",
                                       "e.g.",
                                       "    - xrepo remove zlib boost",
                                       "    - xrepo remove -p iphoneos -a arm64 \"zlib >=1.2.0\"",
                                       "    - xrepo remove -p android -m debug \"pcre2 10.x\"",
                                       "    - xrepo remove -p mingw -k shared zlib",
                                       "    - xrepo remove conan::zlib/1.2.11 vcpkg::zlib"      }
    }

    -- show menu options
    local function show_options()

        -- show usage
        cprint("${bright}Usage: $${clear cyan}xrepo remove [options] packages")

        -- show description
        print("")
        print(description)

        -- show options
        option.show_options(options, "remove")
    end
    return options, show_options, description
end

-- remove packages
function _remove_packages(packages)

    -- is package configuration file? e.g. xrepo install xxx.lua
    --
    -- xxx.lua
    --   add_requires("libpng", {system = false})
    --   add_requireconfs("libpng.*", {configs = {shared = true}})
    local packagefile
    if type(packages) == "string" or #packages == 1 then
        local filepath = table.unwrap(packages)
        if type(filepath) == "string" and filepath:endswith(".lua") and os.isfile(filepath) then
            packagefile = path.absolute(filepath)
        end
    end

    -- add includes to rcfiles
    local rcfiles = {}
    local includes = option.get("includes")
    if includes then
        for _, includefile in ipairs(path.splitenv(includes)) do
            table.insert(rcfiles, path.absolute(includefile))
        end
    end

    -- enter working project directory
    local subdir = "working"
    if packagefile then
        subdir = subdir .. "-" .. hash.uuid(packagefile):split('-')[1]
    end
    local workdir = path.join(os.tmpdir(), "xrepo", subdir)
    if not os.isdir(workdir) then
        os.mkdir(workdir)
        os.cd(workdir)
        os.vrunv(os.programfile(), {"create", "-P", "."})
    else
        os.cd(workdir)
    end
    if packagefile then
        assert(os.isfile("xmake.lua"), "xmake.lua not found!")
        io.writefile("xmake.lua", ('includes("%s")\ntarget("test", {kind = "phony"})'):format((packagefile:gsub("\\", "/"))))
    end

    -- do configure first
    local config_argv = {"f", "-c", "--require=n"}
    if option.get("diagnosis") then
        table.insert(config_argv, "-vD")
    end
    if option.get("plat") then
        table.insert(config_argv, "-p")
        table.insert(config_argv, option.get("plat"))
    end
    if option.get("arch") then
        table.insert(config_argv, "-a")
        table.insert(config_argv, option.get("arch"))
    end
    local mode  = option.get("mode")
    if mode then
        table.insert(config_argv, "-m")
        table.insert(config_argv, mode)
    end
    local kind  = option.get("kind")
    if kind then
        table.insert(config_argv, "-k")
        table.insert(config_argv, kind)
    end
    if option.get("toolchain") then
        table.insert(config_argv, "--toolchain=" .. option.get("toolchain"))
    end
    if option.get("toolchain_host") then
        table.insert(config_argv, "--toolchain_host=" .. option.get("toolchain_host"))
    end
    local envs = {}
    if #rcfiles > 0 then
        envs.XMAKE_RCFILES = path.joinenv(rcfiles)
    end
    os.vrunv(os.programfile(), config_argv, {envs = envs})

    -- do remove
    local require_argv = {"require", "--uninstall"}
    if option.get("yes") then
        table.insert(require_argv, "-y")
    end
    if option.get("verbose") then
        table.insert(require_argv, "-v")
    end
    if option.get("diagnosis") then
        table.insert(require_argv, "-D")
    end
    local extra = {system = false}
    if mode == "debug" then
        extra.debug = true
    end
    if kind then
        extra.configs = extra.configs or {}
        extra.configs.shared = kind == "shared"
    end
    local configs = option.get("configs")
    if configs then
        extra.configs = extra.configs or {}
        local extra_configs, errors = ("{" .. configs .. "}"):deserialize()
        if extra_configs then
            table.join2(extra.configs, extra_configs)
        else
            raise(errors)
        end
    end
    if not packagefile then
        -- avoid overriding extra configs in add_requires/xmake.lua
        if extra then
            local extra_str = string.serialize(extra, {indent = false, strip = true})
            table.insert(require_argv, "--extra=" .. extra_str)
        end
        table.join2(require_argv, packages)
    end
    os.vexecv(os.programfile(), require_argv)
end

-- remove the given installed addons
function _remove_addons(names)
    for _, name in ipairs(names) do
        addon.remove(name, {force = option.get("force")})
        cprint("${color.success}remove ${bright}%s${clear} ok!", name)
    end
end

-- main entry
function main()
    local packages = option.get("packages")
    if option.get("addon") then
        assert(packages, "please specify the addons to be removed.")
        _remove_addons(packages)
    elseif option.get("all") then
        remove_all_packages(packages)
    elseif packages then
        _remove_packages(packages)
    else
        raise("please specify the packages to be removed.")
    end
end
