init.lua

-- cmake-tools's API
local utils = require("cmake-tools.utils")
local Types = require("cmake-tools.types")
local const = require("cmake-tools.const")
local Config = require("cmake-tools.config")
local variants = require("cmake-tools.variants")
local kits = require("cmake-tools.kits")
local presets = require("cmake-tools.presets")
local log = require("cmake-tools.log")
local hints = require("cmake-tools.hints")
local _session = require("cmake-tools.session")
local window = require("cmake-tools.window")
local environment = require("cmake-tools.environment")
local file_picker = require("cmake-tools.file_picker")
local scratch = require("cmake-tools.scratch")

local ctest = require("cmake-tools.test.ctest")

local config = Config:new(const)

local cmake = {}

--- Setup cmake-tools
function cmake.setup(values)
  const = vim.tbl_deep_extend("force", const, values)
  const.cmake_executor.opts = vim.tbl_deep_extend(
    "force",
    const.cmake_executor.default_opts[const.cmake_executor.name],
    const.cmake_executor.opts or {}
  )
  const.cmake_runner.opts = vim.tbl_deep_extend(
    "force",
    const.cmake_runner.default_opts[const.cmake_runner.name],
    const.cmake_runner.opts or {}
  )

  config = Config:new(const)

  -- auto reload previous session
  local old_config = _session.load()
  _session.update(config, old_config)

  local is_executor_installed = utils.get_executor(config.executor.name).is_installed()
  local is_runner_installed = utils.get_runner(config.runner.name).is_installed()
  if type(is_executor_installed) == "string" then
    log.error(is_executor_installed)
  end
  if type(is_runner_installed) == "string" then
    log.error(is_runner_installed)
  end

  cmake.register_telescope_function()
  cmake.register_dap_function()
  cmake.register_autocmd()
  cmake.register_autocmd_provided_by_users()
  cmake.register_scratch_buffer(config.executor.name, config.runner.name)
end

--- Generate build system for this project.
-- Think it as `cmake .`
function cmake.generate(opt, callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local clean = opt.bang
  local fargs = opt.fargs or {}
  if clean then
    return cmake.clean(function()
      -- Clear CMakeCache.txt
      if config:has_build_directory() then
        utils.rmfile(config.build_directory / "CMakeCache.txt")
      end
      cmake.generate({ fargs = fargs }, callback)
    end)
  end

  -- if exists presets, preset include all info that cmake
  -- needed to execute, so we don't use cmake-kits.json and
  -- cmake-variants.[json|yaml] event they exist.
  local presets_file = config.base_settings.use_preset and presets.check(config.cwd)
  if presets_file and not config.configure_preset then
    -- this will also set value for build type from preset.
    -- default to be "Debug"
    return cmake.select_configure_preset(function()
      cmake.generate(opt, callback)
    end)
  end

  if presets_file and config.configure_preset then
    -- if exsist preset file and set configure preset, then
    -- set build directory to the `binaryDir` option of `configurePresets`
    local build_directory, no_expand_build_directory = presets.get_build_dir(
      presets.get_preset_by_name(config.configure_preset, "configurePresets", config.cwd),
      config.cwd
    )
    if build_directory ~= "" then
      config:update_build_dir(build_directory, no_expand_build_directory)
    end
    config:generate_build_directory()

    local args = {
      "--preset",
      config.configure_preset,
    }
    vim.list_extend(args, config:generate_options())
    vim.list_extend(args, fargs)

    local env = environment.get_build_environment(config)
    local cmd = const.cmake_command
    return utils.execute(cmd, config.env_script, env, args, config.cwd, config.executor, function()
      if type(callback) == "function" then
        callback()
      end
      cmake.configure_compile_commands()
      cmake.create_regenerate_on_save_autocmd()
    end, const.cmake_notifications)
  end

  -- if exists cmake-kits.json, kit is used to set
  -- environmental variables and args.
  local kits_config = kits.parse(const.cmake_kits_path, config.cwd)
  if kits_config and not config.kit then
    return cmake.select_kit(function()
      cmake.generate(opt, callback)
    end)
  end

  -- specify build type, if exists cmake-variants.json,
  -- this will get build variant from it. Or this will
  -- get build variant from "Debug, Release, RelWithDebInfo, MinSizeRel"
  if not config.build_type or not config.variant then
    -- Use the default variant
    local defaults, _ = variants.get(const.cmake_variants_message, config.cwd)
    config.build_type = table.concat(defaults.val, " + ")
    config.variant = defaults.kv
    --[[ return cmake.select_build_type(function() ]]
    --[[   cmake.generate(opt, callback) ]]
    --[[ end) ]]
  end

  -- cmake kits, if cmake-kits.json doesn't exist, kit_option will
  -- be {env={}, args={}}, so it's okay.
  local kit_option = kits.build_env_and_args(
    config.kit,
    config.executor.name == "terminal",
    config.cwd,
    const.cmake_kits_path
  )

  config.env_script = kit_option.env_script
  -- vim.print(config.env_script)

  -- macro expansion for build directory
  local build_dir = config:prepare_build_directory(kits_config)
  config:update_build_dir(build_dir, config:no_expand_build_directory_path())
  config:generate_build_directory()

  local args = {
    "-B",
    utils.transform_path(config:build_directory_path(), config.executor.name == "quickfix"),
    "-S",
    ".",
  }
  vim.list_extend(args, variants.build_arglist(config.build_type, config.cwd))
  vim.list_extend(args, kit_option.args)
  vim.list_extend(args, config:generate_options())
  vim.list_extend(args, fargs)

  local env = environment.get_build_environment(config)
  local cmd = const.cmake_command
  env = vim.tbl_extend("keep", env, kit_option.env)
  return utils.execute(cmd, config.env_script, env, args, config.cwd, config.executor, function()
    if type(callback) == "function" then
      callback()
    end
    cmake.configure_compile_commands()
    cmake.create_regenerate_on_save_autocmd()
  end, const.cmake_notifications)
end

--- Clean targets
function cmake.clean(callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local args = {
    "--build",
    utils.transform_path(config:build_directory_path(), config.executor.name == "quickfix"),
    "--target",
    "clean",
  }

  local env = environment.get_build_environment(config)
  local cmd = const.cmake_command
  return utils.execute(cmd, config.env_script, env, args, config.cwd, config.executor, function()
    if type(callback) == "function" then
      callback()
    end
  end, const.cmake_notifications)
end

--- Build this project using the make toolchain of target platform
--- think it as `cmake --build .`
function cmake.build(opt, callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local clean = opt.bang
  local fargs = opt.fargs or {}
  if clean then
    return cmake.clean(function()
      cmake.build({ fargs = fargs }, callback)
    end)
  end

  local ct = config:get_codemodel_targets()
  if not (config:has_build_directory()) or not (ct.code == Types.SUCCESS) then
    -- configure it
    return cmake.generate({ bang = false, fargs = {} }, function()
      cmake.build(opt, callback)
    end)
  end

  if opt.target == nil and config.build_target == nil then
    return cmake.select_build_target(function()
      cmake.build(opt, callback)
    end), true
  end

  local args
  local presets_file = config.base_settings.use_preset and presets.check(config.cwd)

  if presets_file and config.build_preset then
    args = { "--build", "--preset", config.build_preset } -- preset don't need define build dir.
  else
    args = {
      "--build",
      utils.transform_path(config:build_directory_path(), config.executor.name == "quickfix"),
    }
  end

  vim.list_extend(args, config:build_options())

  if opt.target ~= nil then
    vim.list_extend(args, { "--target", opt.target })
    vim.list_extend(args, fargs)
  elseif config.build_target == "all" then
    vim.list_extend(args, { "--target", "all" })
    vim.list_extend(args, fargs)
  else
    vim.list_extend(args, { "--target", config.build_target })
    vim.list_extend(args, fargs)
  end

  local env = environment.get_build_environment(config)
  local cmd = const.cmake_command
  return utils.execute(cmd, config.env_script, env, args, config.cwd, config.executor, function()
    if type(callback) == "function" then
      callback()
    end
  end, const.cmake_notifications)
end

function cmake.quick_build(opt, callback)
  -- if no target was supplied, query via ui select
  if opt.fargs[1] == nil then
    if utils.has_active_job(config.runner, config.executor) then
      return
    end

    if not (config:has_build_directory()) then
      -- configure it
      return cmake.generate({ bang = false, fargs = {} }, function()
        cmake.quick_build(opt, callback)
      end)
    end

    local targets_res = config:build_targets()
    local targets, display_targets = targets_res.data.targets, targets_res.data.display_targets

    vim.ui.select(
      display_targets,
      { prompt = "Select target to build" },
      vim.schedule_wrap(function(_, idx)
        if not idx then
          return
        end
        cmake.build({ target = targets[idx] }, callback)
      end)
    )
  else
    cmake.build({ target = opt.fargs[1] }, callback)
  end
end

function cmake.stop_executor()
  utils.stop_executor(config.executor)
end

function cmake.stop_runner()
  utils.stop_runner(config.runner)
end

--- CMake install targets
function cmake.install(opt)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local fargs = opt.fargs

  local args = { "--install", config:build_directory_path() }
  vim.list_extend(args, fargs)
  return utils.execute(
    const.cmake_command,
    config.env_script,
    {},
    args,
    config.cwd,
    config.executor,
    nil,
    const.cmake_notifications
  )
end

function cmake.close_executor()
  utils.close_executor(config.executor)
end

function cmake.close_runner()
  utils.close_runner(config.runner)
end

function cmake.open_executor()
  utils.show_executor(config.executor)
end

function cmake.open_runner()
  utils.show_runner(config.runner)
end

function cmake.substitute_path(path, vars)
  for key, value in pairs(vars) do
    if type(value) == "string" or type(value) == "number" then
      path = path:gsub("${" .. key .. "}", value)
    else
      if next(value) then
        local prefix = key .. "."
        for innerkey, innervalue in pairs(value) do
          if type(innervalue) == "string" or type(innervalue) == "number" then
            path = path:gsub("${" .. prefix .. innerkey .. "}", innervalue)
          end
        end
      end
    end
  end

  return path
end

function cmake.get_launch_path(target)
  local model = config:get_code_model_info()[target]
  local result = config:get_launch_target_from_info(model)
  local target_path = result.data

  local launch_path = vim.fn.fnamemodify(target_path, ":h")

  if config.base_settings.working_dir and type(config.base_settings.working_dir) == "string" then
    launch_path =
      cmake.substitute_path(config.base_settings.working_dir, cmake.get_target_vars(target))
  end

  if
    config.target_settings[target]
    and config.target_settings[target].working_dir
    and type(config.target_settings[target].working_dir) == "string"
  then
    launch_path = config.target_settings[target].working_dir
    launch_path = cmake.substitute_path(launch_path, cmake.get_target_vars(target))
  end

  return launch_path
end

-- Run executable targets
function cmake.run(opt)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end
  if opt.target then
    -- explicit target requested. use that instead of the configured one
    return cmake.build({ target = opt.target }, function()
      local model = config:get_code_model_info()[opt.target]
      local result = config:get_launch_target_from_info(model)
      local target_path = result.data

      local launch_path = cmake.get_launch_path(opt.target)
      local env = environment.get_run_environment(config, opt.target)
      local _args = opt.args and opt.args or config.target_settings[opt.target].args
      local cmd = target_path
      utils.run(
        cmd,
        config.env_script,
        env,
        _args,
        launch_path,
        config.runner,
        nil,
        const.cmake_notifications
      )
    end)
  else
    local result = config:get_launch_target()
    local result_code = result.code
    if result_code == Types.NOT_CONFIGURED or result_code == Types.CANNOT_FIND_CODEMODEL_FILE then
      -- Configure it
      return cmake.generate({ bang = false, fargs = utils.deepcopy(opt.fargs) }, function()
        cmake.run(opt)
      end)
    elseif
      result_code == Types.NOT_SELECT_LAUNCH_TARGET
      or result_code == Types.NOT_A_LAUNCH_TARGET
      or result_code == Types.NOT_EXECUTABLE
    then
      -- Re Select a target that could launch
      return cmake.select_launch_target(function()
        cmake.run(opt)
      end), true
    else -- if result_code == Types.SELECTED_LAUNCH_TARGET_NOT_BUILT
      -- Build select launch target every time
      return cmake.build(
        { target = config.launch_target, fargs = utils.deepcopy(opt.fargs) },
        function()
          result = config:get_launch_target()
          local target_path = result.data

          local launch_path = cmake.get_launch_path(cmake.get_launch_target())

          local env = environment.get_run_environment(config, config.launch_target)
          local cmd = target_path
          utils.run(
            cmd,
            config.env_script,
            env,
            cmake:get_launch_args(),
            launch_path,
            config.runner,
            nil,
            const.cmake_notifications
          )
        end
      )
    end
  end
end

function cmake.quick_run(opt)
  -- if no target was supplied, query via ui select
  if opt.fargs[1] == nil then
    if utils.has_active_job(config.runner, config.executor) then
      return
    end

    if not (config:has_build_directory()) then
      -- configure it
      return cmake.generate({ bang = false, fargs = {} }, function()
        cmake.quick_run(opt)
      end)
    end

    local targets_res = config:launch_targets()
    local targets, display_targets = targets_res.data.targets, targets_res.data.display_targets

    vim.ui.select(
      display_targets,
      { prompt = "Select target to run" },
      vim.schedule_wrap(function(_, idx)
        if not idx then
          return
        end
        cmake.run({ target = targets[idx] })
      end)
    )
  else
    local target = table.remove(opt.fargs, 1)
    cmake.run({ target = target, args = opt.fargs })
  end
end

-- Set args for launch target
function cmake.launch_args(opt)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  if cmake.get_launch_target() ~= nil then
    if not config.target_settings[cmake.get_launch_target()] then
      config.target_settings[cmake.get_launch_target()] = {}
    end

    config.target_settings[cmake.get_launch_target()].args = utils.deepcopy(opt.fargs)
  end
end

function cmake.select_build_type(callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local _, types = variants.get(const.cmake_variants_message, config.cwd)
  -- Put selected build type first
  for idx, type in ipairs(types) do
    if type == config.build_type then
      table.insert(types, 1, table.remove(types, idx))
      break
    end
  end

  vim.ui.select(
    types,
    {
      prompt = "Select build type",
      format_item = function(item)
        return item.short .. item.long
      end,
    },
    vim.schedule_wrap(function(build_type)
      if not build_type then
        return
      end
      config.build_type = build_type.short
      config.variant = build_type.kv
      if type(callback) == "function" then
        callback()
      else
        cmake.generate({ bang = false, fargs = {} }, nil)
      end
    end)
  )
end

function cmake.select_kit(callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local cmake_kits = kits.get(const.cmake_kits_path, config.cwd)
  if cmake_kits then
    -- Put selected kit first
    for idx, kit in ipairs(cmake_kits) do
      if kit == config.kit then
        table.insert(cmake_kits, 1, table.remove(cmake_kits, idx))
        break
      end
    end

    vim.ui.select(
      cmake_kits,
      { prompt = "Select cmake kits" },
      vim.schedule_wrap(function(kit)
        if not kit then
          return
        end
        if config.kit ~= kit then
          config.kit = kit
        end
        if type(callback) == "function" then
          callback()
        else
          cmake.generate({ bang = false, fargs = {} }, nil)
        end
      end)
    )
  else
    log.error("Cannot find CMakeKits.[json|yaml] at Root (" .. config.cwd .. ")!!")
  end
end

function cmake.select_configure_preset(callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  -- if exists presets
  local presets_file = presets.check(config.cwd)
  if presets_file then
    local configure_preset_names =
      presets.parse("configurePresets", { include_hidden = false }, config.cwd)
    local configure_presets =
      presets.parse_name_mapped("configurePresets", { include_hidden = false }, config.cwd)
    local format_preset_name = function(p_name)
      local p = configure_presets[p_name]
      return p.displayName or p.name
    end
    vim.ui.select(
      configure_preset_names,
      {
        prompt = "Select cmake configure presets",
        format_item = format_preset_name,
      },
      vim.schedule_wrap(function(choice)
        if not choice then
          return
        end
        if config.configure_preset ~= choice then
          config.configure_preset = choice
          config.build_type = presets.get_build_type(
            presets.get_preset_by_name(choice, "configurePresets", config.cwd)
          )
        end
        if type(callback) == "function" then
          callback()
        else
          cmake.generate({ bang = false, fargs = {} }, nil)
        end
      end)
    )
  else
    log.error("Cannot find CMake[User]Presets.json at Root (" .. config.cwd .. ") !!")
  end
end

function cmake.select_build_preset(callback)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  -- if exists presets
  local presets_file = presets.check(config.cwd)
  if presets_file then
    local build_preset_names = presets.parse("buildPresets", { include_hidden = false }, config.cwd)
    local build_presets =
      presets.parse_name_mapped("buildPresets", { include_hidden = false }, config.cwd)
    build_preset_names = vim.list_extend(build_preset_names, { "None" })
    build_presets = vim.tbl_extend("keep", build_presets, { None = { displayName = "None" } })
    local format_preset_name = function(p_name)
      local p = build_presets[p_name]
      return p.displayName or p.name
    end
    vim.ui.select(
      build_preset_names,
      { prompt = "Select cmake build presets", format_item = format_preset_name },
      vim.schedule_wrap(function(choice)
        if not choice then
          return
        end
        if choice == "None" then
          config.build_preset = nil
          return
        end
        if config.build_preset ~= choice then
          config.build_preset = choice
        end
        local associated_configure_preset =
          presets.get_preset_by_name(choice, "buildPresets", config.cwd)["configurePreset"]
        local configure_preset_updated = false

        if
          associated_configure_preset and config.configure_preset ~= associated_configure_preset
        then
          config.configure_preset = associated_configure_preset
          configure_preset_updated = true
        end

        if type(callback) == "function" then
          callback()
        elseif configure_preset_updated then
          cmake.generate({ bang = true, fargs = {} }, nil)
        end
      end)
    )
  else
    log.error("Cannot find CMake[User]Presets.json at Root (" .. config.cwd .. ")!!")
  end
end

function cmake.select_build_target(callback, regenerate)
  if not (config:has_build_directory()) then
    -- configure it
    return cmake.generate({ bang = false, fargs = {} }, function()
      cmake.select_build_target(callback, true)
    end)
  end

  local targets_res = config:build_targets()

  if targets_res.code ~= Types.SUCCESS then
    -- try again
    if not regenerate then
      return
    else
      return cmake.generate({ bang = true, fargs = {} }, function()
        cmake.select_build_target(callback, false)
      end)
    end
  end
  local targets, display_targets = targets_res.data.targets, targets_res.data.display_targets
  vim.ui.select(
    display_targets,
    { prompt = "Select build target" },
    vim.schedule_wrap(function(_, idx)
      if not idx then
        return
      end
      config.build_target = targets[idx]
      if type(callback) == "function" then
        callback()
      end
    end)
  )
end

function cmake.get_cmake_launch_targets(callback)
  if not (config:has_build_directory()) then
    -- configure it
    return cmake.generate({ bang = false, fargs = {} }, function()
      cmake.get_cmake_launch_targets(callback)
    end)
  end

  if type(callback) == "function" then
    callback(config:launch_targets())
  end
end

function cmake.select_launch_target(callback, regenerate)
  if not (config:has_build_directory()) then
    -- configure it
    return cmake.generate({ bang = false, fargs = {} }, function()
      cmake.select_launch_target(callback, true)
    end)
  end

  local targets_res = config:launch_targets()

  if targets_res.code ~= Types.SUCCESS then
    -- try again
    if not regenerate then
      return
    else
      return cmake.generate({ bang = true, fargs = {} }, function()
        cmake.select_launch_target(callback, false)
      end)
    end
  end
  local targets, display_targets = targets_res.data.targets, targets_res.data.display_targets

  vim.ui.select(
    display_targets,
    { prompt = "Select launch target" },
    vim.schedule_wrap(function(_, idx)
      if not idx then
        return
      end
      config.launch_target = targets[idx]
      if type(callback) == "function" then
        callback()
      end
    end)
  )
end

function cmake.get_base_vars()
  local vars = { dir = {} }

  vars.dir.build = config:build_directory_path() .. "/"
  vars.dir.binary = "${dir.binary}"
  return vars
end

local function convert_to_table(str)
  -- do a roundtrip. this should remove unsupported stuff like function() which vim.inspect cannot convert
  local fn = loadstring(str)
  if not fn then
    return false
  end

  if pcall(function()
    vim.inspect(fn())
  end) then
    str = "return " .. vim.inspect(fn())
    fn = loadstring(str)
    if not fn then
      return false
    end

    return true, fn()
  else
    return false
  end
end

function cmake.get_target_vars(target)
  local vars = cmake.get_base_vars()

  local model = config:get_code_model_info()[target]
  local result = config:get_launch_target_from_info(model)
  vars.dir.binary = utils.get_path(result.data)
  return vars
end

function cmake.settings()
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  if not window.is_open() then
    local prev_build_dir = config.base_settings.build_dir
    local content = "local vars = " .. vim.inspect(cmake.get_base_vars())
    content = content .. "\nreturn " .. vim.inspect(config.base_settings)

    window.set_content(content)
    window.title = "CMake-Tools base settings"
    window.on_save = function(str)
      local ok, val = convert_to_table(str)
      if ok then
        config.base_settings = val
      end
    end

    window.on_exit = function(str)
      local ok, val = convert_to_table(str)
      if ok then
        config.base_settings = val
        if prev_build_dir ~= config.base_settings.build_dir then
          cmake.select_build_dir({ args = config.base_settings.build_dir })
        end
      end
    end
    window.open()
  end
end

function cmake.target_settings(opt)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end

  local result = utils.get_cmake_configuration(config.cwd)
  if result.code ~= Types.SUCCESS then
    return log.error(result.message)
  end

  local target = opt.fargs[1] or cmake.get_launch_target()

  if target == nil then
    log.warn("No launch target selected!")
    return
  end

  if not window.is_open() then
    if not config.target_settings[target] then
      config.target_settings[target] = {}
    end

    -- insert missing fields
    config.target_settings[target] = vim.tbl_deep_extend("keep", config.target_settings[target], {
      args = {},
      inherit_base_environment = true,
      env = {},
    })

    local content = "local vars = " .. vim.inspect(cmake.get_target_vars(target))
    content = content .. "\nreturn " .. vim.inspect(config.target_settings[target])

    window.set_content(content)
    window.title = "CMake-Tools settings for " .. target
    window.on_save = function(str)
      local ok, val = convert_to_table(str)
      if ok then
        config.target_settings[target] = val
      end
    end
    window.open()
  end
end

function cmake.run_test(opt)
  if utils.has_active_job(config.runner, config.executor) then
    return
  end
  local env = environment.get_build_environment(config)
  local all_tests = ctest.list_all_tests(config:build_directory_path())
  if #all_tests == 0 then
    return
  end
  table.insert(all_tests, 1, "all")
  vim.ui.select(
    all_tests,
    { prompt = "select test to run" },
    vim.schedule_wrap(function(_, idx)
      if not idx then
        return
      end
      if idx == 1 then
        ctest.run(const.ctest_command, "'.*'", config:build_directory_path(), env, config, opt)
      else
        ctest.run(
          const.ctest_command,
          all_tests[idx],
          config:build_directory_path(),
          env,
          config,
          opt
        )
      end
    end)
  )
end

function cmake.run_current_file(opt)
  local targets = {}
  local display_targets = {}
  local file = vim.fn.expand("%:p")
  local all_targets = config:launch_targets_with_sources()
  for i, target in ipairs(all_targets.data["sources"]) do
    if target.path == file then
      table.insert(targets, target.name)
      table.insert(display_targets, target.display_name)
    end
  end
  if #targets == 0 then
    return log.error("Current file is not belong to any executable.")
  end

  if #targets == 1 then
    return cmake.run({ target = targets[1], args = opt.fargs })
  else
    vim.ui.select(
      display_targets,
      { prompt = "Select launch target" },
      vim.schedule_wrap(function(_, idx)
        if not idx then
          return
        end
        return cmake.run({ target = targets[idx], args = opt.fargs })
      end)
    )
  end
end

function cmake.build_current_file(opt)
  local targets = {}
  local display_targets = {}
  local file = vim.fn.expand("%:p")
  local all_targets = config:build_targets_with_sources()
  for _, target in ipairs(all_targets.data["sources"]) do
    if target.path == file then
      table.insert(targets, target.name)
      table.insert(display_targets, target.display_name)
    end
  end
  if #targets == 0 then
    return log.error("Current file is not belong to any library.")
  end
  if #targets == 1 then
    return cmake.build({ target = targets[1], args = opt.fargs })
  else
    vim.ui.select(
      display_targets,
      { prompt = "Select build target" },
      vim.schedule_wrap(function(_, idx)
        if not idx then
          return
        end
        return cmake.build({ target = targets[idx], args = opt.fargs })
      end)
    )
  end
end

--[[ Getters ]]
function cmake.get_config()
  return config
end

function cmake.get_build_target()
  return config.build_target
end

function cmake.get_build_target_path()
  local result = config:get_build_target()
  local target_path = result.data
  return target_path
end

function cmake.get_launch_target()
  return config.launch_target
end

function cmake.get_launch_target_path()
  local result = config:get_launch_target()
  local target_path = result.data
  return target_path
end

function cmake.get_model_info()
  return config:get_code_model_info()
end

function cmake.get_launch_args()
  if cmake.get_launch_target() == nil then
    return {}
  end
  if
    config.target_settings[cmake.get_launch_target()]
    and config.target_settings[cmake.get_launch_target()].args
  then
    return config.target_settings[cmake.get_launch_target()].args
  end

  return {}
end

function cmake.get_build_environment()
  return environment.get_build_environment_table(config)
end

function cmake.get_run_environment(target)
  return environment.get_run_environment_table(
    config,
    target and target or cmake.get_launch_target()
  )
end

function cmake.get_build_type()
  return config.build_type
end

function cmake.get_kit()
  return config.kit
end

function cmake.get_configure_preset()
  return config.configure_preset
end

function cmake.get_build_preset()
  return config.build_preset
end

function cmake.get_build_directory()
  return config.build_directory
end

function cmake.is_cmake_project()
  local result = utils.get_cmake_configuration(config.cwd)
  return result.code == Types.SUCCESS
end

function cmake.has_cmake_preset()
  local presets_file = presets.check(config.cwd)
  return presets_file ~= nil
end

function cmake.get_build_targets()
  return config:build_targets()
end

function cmake.get_launch_targets()
  return config:launch_targets()
end

function cmake.get_generate_options()
  return config:generate_options()
end

function cmake.get_build_options()
  return config:build_options()
end

--[[ end ]]

function cmake.configure_compile_commands()
  if const.cmake_soft_link_compile_commands then
    cmake.compile_commands_from_soft_link()
  elseif const.cmake_compile_commands_from_lsp then
    cmake.compile_commands_from_lsp()
  end
end

function cmake.compile_commands_from_soft_link()
  if not config:has_build_directory() then
    return
  end

  local source = config:build_directory_path() .. "/compile_commands.json"
  local destination = vim.loop.cwd() .. "/compile_commands.json"
  utils.softlink(source, destination)
end

function cmake.compile_commands_from_lsp()
  if not config:has_build_directory() or not const.lsp_type then
    return
  end

  local buf = vim.api.nvim_get_current_buf()
  local clients = vim.lsp.get_active_clients({ name = const.lsp_type })
  for _, client in ipairs(clients) do
    local lspbufs = vim.lsp.get_buffers_by_client_id(client.id)
    for _, bufid in ipairs(lspbufs) do
      vim.api.nvim_set_current_buf(bufid)
      vim.cmd("LspRestart " .. tostring(client.id))
    end
  end
  vim.api.nvim_set_current_buf(buf)
end

function cmake.clangd_on_new_config(new_config)
  const.lsp_type = "clangd"

  local arg = "--compile-commands-dir=" .. config:build_directory_path()
  for i, v in ipairs(new_config.cmd) do
    if string.find(v, "%-%-compile%-commands%-dir=") ~= nil then
      table.remove(new_config.cmd, i)
    end
  end
  table.insert(new_config.cmd, arg)
end

function cmake.ccls_on_new_config(new_config)
  const.lsp_type = "ccls"

  new_config.init_options.compilationDatabaseDirectory = config:build_directory_path()
end

function cmake.select_cwd(cwd_path)
  if cwd_path.args == "" then
    vim.ui.input(
      {
        prompt = "The directory where the main CMakeLists.txt is located: \n",
        default = config.cwd,
        completion = "dir",
      },
      vim.schedule_wrap(function(input)
        if not input then
          return
        end
        --local new_path = Path:new(input)
        --if new_path:is_dir() then
        config.cwd = vim.fn.resolve(input)
        cmake.register_autocmd()
        cmake.register_autocmd_provided_by_users()
        --	end
        cmake.generate({ bang = false, fargs = {} }, nil)
      end)
    )
  elseif cwd_path.args then
    config.cwd = vim.fn.resolve(cwd_path.args)
    cmake.register_autocmd()
    cmake.register_autocmd_provided_by_users()
    cmake.generate({ bang = false, fargs = {} }, nil)
  end
end

function cmake.select_build_dir(cwd_path)
  if cwd_path.args == "" then
    vim.ui.input(
      {
        prompt = "The directory where the build files should locate: \n",
        default = config:no_expand_build_directory_path(),
        completion = "dir",
      },
      vim.schedule_wrap(function(input)
        if not input then
          return
        end
        --local new_path = Path:new(input)
        --if new_path:is_dir() then
        config:update_build_dir(vim.fn.resolve(input), vim.fn.resolve(input))
        --	end
        cmake.generate({ bang = false, fargs = {} }, nil)
      end)
    )
  elseif cwd_path.args then
    config:update_build_dir(vim.fn.resolve(cwd_path.args), vim.fn.resolve(cwd_path.args))
    cmake.generate({ bang = false, fargs = {} }, nil)
  end
end

local regenerate_id = nil
local termclose_id = nil
local vim_leave_pre_id = nil

local group = vim.api.nvim_create_augroup("cmaketools", { clear = true })

function cmake.create_regenerate_on_save_autocmd()
  if not const.cmake_regenerate_on_save then
    return
  end
  if regenerate_id then
    vim.api.nvim_del_autocmd(regenerate_id)
  end

  local cmake_files = file_picker.get_cmake_files()

  local pattern = {}
  for _, item in ipairs(cmake_files) do
    local ss = tostring(item)
    ss = ss:gsub("{", "\\{")
    ss = ss:gsub("}", "\\}")
    ss = ss:gsub("?", "\\?")
    ss = ss:gsub(",", "\\,")
    table.insert(pattern, ss)
  end

  local presets_file = config.base_settings.use_preset and presets.check(config.cwd)
  if presets_file then
    for _, item in ipairs({
      "CMakePresets.json",
      "CMakeUserPresets.json",
      "cmake-presets.json",
      "cmake-user-presets.json",
    }) do
      table.insert(pattern, config.cwd .. "/" .. item)
    end
  else
    for _, item in ipairs({
      "CMakeVariants.json",
      "CMakeVariants.yaml",
      "cmake-variants.yaml",
      "cmake-variants.json",
      "CMakeKits.json",
      "cmake-kits.json",
    }) do
      table.insert(pattern, config.cwd .. "/" .. item)
    end
  end

  if #pattern ~= 0 then
    -- for cmake files
    regenerate_id = vim.api.nvim_create_autocmd("BufWritePre", {
      group = group,
      pattern = pattern,
      callback = function()
        local buf = vim.api.nvim_get_current_buf()
        -- Check if buffer is actually modified, and only if it is modified,
        -- execute the :CMakeGenerate, otherwise return. This is to avoid unnecessary regenerattion
        local buf_modified = vim.api.nvim_buf_get_option(buf, "modified")
        if buf_modified then
          cmake.generate({ bang = false, fargs = {} }, nil)
          config:update_targets()
        end
      end,
    })
  end
end

function cmake.register_autocmd()
  -- preload the autocmd if the following option is true. only saves cmakelists.txt files
  if cmake.is_cmake_project() then
    if termclose_id then
      vim.api.nvim_del_autocmd(termclose_id)
    end
    if vim_leave_pre_id then
      vim.api.nvim_del_autocmd(vim_leave_pre_id)
    end

    cmake.create_regenerate_on_save_autocmd()

    vim_leave_pre_id = vim.api.nvim_create_autocmd("VimLeavePre", {
      group = group,
      callback = function()
        _session.save(config)
        vim.api.nvim_del_augroup_by_id(group)
      end,
    })

    if const.cmake_virtual_text_support then
      vim.api.nvim_create_autocmd({ "CursorMoved", "CursorMovedI" }, {
        group = group,
        callback = function(ev)
          local targets = {}
          local file = ev.file
          local all_targets = config:build_targets_with_sources()
          if
            all_targets
            and all_targets.data
            and type(all_targets.data) == "table"
            and all_targets.data["sources"]
          then
            for _, target in ipairs(all_targets.data["sources"]) do
              if target.path == file then
                table.insert(targets, { name = target.name, type = target.type })
              end
            end
            hints.show(ev.buf, targets)
          end
        end,
      })
    end
  end
end

function cmake.register_autocmd_provided_by_users()
  if cmake.is_cmake_project() then
    vim.api.nvim_exec_autocmds("User", { pattern = "CMakeToolsEnterProject" })
  end
end

function cmake.register_scratch_buffer(executor, runner)
  if cmake.is_cmake_project() then
    vim.schedule(function()
      scratch.create(executor, runner)
    end)
  end
end

function cmake.register_dap_function()
  local has_nvim_dap, dap = pcall(require, "dap")
  if has_nvim_dap then
    -- Debug execuable targets
    function cmake.debug(opt, callback)
      if utils.has_active_job(config.runner, config.executor) then
        return
      end

      local env = environment.get_run_environment_table(
        config,
        opt.target and opt.target or config.launch_target
      )

      -- nvim.dap expects all env vars as string
      for index, value in pairs(env) do
        env[index] = tostring(value)
      end

      if next(env) == nil then -- dap complains on empty list (env = {})
        env = nil
      end

      local can_debug_result = config:validate_for_debugging()
      if can_debug_result.code == Types.CANNOT_DEBUG_LAUNCH_TARGET then
        -- Select build type to debug
        log.info("Reselect build type to ensure it contains debug information!")
        return cmake.select_build_type(function()
          cmake.debug(opt, callback)
        end)
      end

      if opt.target then
        -- explicit target requested. use that instead of the configured one
        return cmake.build({ target = opt.target }, function()
          local model = config:get_code_model_info()[opt.target]
          local result = config:get_launch_target_from_info(model)
          local dap_config = {
            name = opt.target,
            program = result.data,
            cwd = cmake.get_launch_path(opt.target),
            args = opt.args and opt.args or config.target_settings[opt.target].args,
            env = env,
          }
          -- close cmake console
          cmake.close_executor()
          dap.run(vim.tbl_extend("force", dap_config, const.cmake_dap_configuration))
        end)
      else
        local result = config:get_launch_target()
        local result_code = result.code

        if
          result_code == Types.NOT_CONFIGURED or result_code == Types.CANNOT_FIND_CODEMODEL_FILE
        then
          -- Configure it
          return cmake.generate({ bang = false, fargs = utils.deepcopy(opt.fargs) }, function()
            cmake.debug(opt, callback)
          end)
        elseif
          result_code == Types.NOT_SELECT_LAUNCH_TARGET
          or result_code == Types.NOT_A_LAUNCH_TARGET
          or result_code == Types.NOT_EXECUTABLE
        then
          -- Re Select a target that could launch
          return cmake.select_launch_target(function()
            cmake.debug(opt, callback)
          end),
            true
        else -- if result_code == Types.SELECTED_LAUNCH_TARGET_NOT_BUILT then
          -- Build select launch target every time
          return cmake.build(
            { target = config.launch_target, fargs = utils.deepcopy(opt.fargs) },
            function()
              result = config:get_launch_target()
              local target_path = result.data
              local dap_config = {
                name = config.launch_target,
                program = target_path,
                cwd = cmake.get_launch_path(cmake.get_launch_target()),
                args = cmake:get_launch_args(),
                env = env,
              }
              -- close cmake console
              cmake.close_executor()
              dap.run(vim.tbl_extend("force", dap_config, const.cmake_dap_configuration))
            end
          )
        end
      end
    end

    function cmake.quick_debug(opt, callback)
      -- if no target was supplied, query via ui select
      if opt.fargs[1] == nil then
        if utils.has_active_job(config.runner, config.executor) then
          return
        end

        if not (config:has_build_directory()) then
          -- configure it
          return cmake.generate({ bang = false, fargs = {} }, function()
            cmake.quick_debug(opt, callback)
          end)
        end

        local targets_res = config:launch_targets()
        local targets, display_targets = targets_res.data.targets, targets_res.data.display_targets

        vim.ui.select(
          display_targets,
          { prompt = "Select target to debug" },
          vim.schedule_wrap(function(_, idx)
            if not idx then
              return
            end
            cmake.debug({ target = targets[idx] }, callback)
          end)
        )
      else
        local target = table.remove(opt.fargs, 1)
        cmake.debug({ target = target, args = opt.fargs }, callback)
      end
    end

    function cmake.debug_current_file(opt)
      local targets = {}
      local display_targets = {}
      local file = vim.fn.expand("%:p")
      local all_targets = config:launch_targets_with_sources()
      for _, target in ipairs(all_targets.data["sources"]) do
        if target.path == file then
          table.insert(targets, target.name)
          table.insert(display_targets, target.display_name)
        end
      end
      if #targets == 1 then
        return cmake.debug({ target = targets[1], args = opt.fargs })
      else
        vim.ui.select(
          display_targets,
          { prompt = "Select launch target" },
          vim.schedule_wrap(function(_, idx)
            if not idx then
              return
            end
            return cmake.debug({ target = targets[idx], args = opt.fargs })
          end)
        )
      end
    end

    --- CMake debug
    vim.api.nvim_create_user_command(
      "CMakeDebug", -- name
      cmake.debug, -- command
      { -- opts
        nargs = "*",
        desc = "CMake debug",
      }
    )

    --- CMake quick debug
    vim.api.nvim_create_user_command(
      "CMakeQuickDebug", -- name
      cmake.quick_debug, -- command
      { -- opts
        nargs = "*",
        desc = "CMake quick debug",
      }
    )

    --- CMake debug current file
    vim.api.nvim_create_user_command(
      "CMakeDebugCurrentFile", -- name
      cmake.debug_current_file, -- command
      { -- opts
        nargs = "*",
        desc = "CMake debug current file",
      }
    )
  end
end

function cmake.register_telescope_function()
  local has_telescope, telescope = pcall(require, "telescope")
  if has_telescope then
    telescope.load_extension("cmake_tools")

    function cmake.show_target_files(opt)
      -- if no target was supplied, query via ui select
      if opt.fargs[1] == nil then
        if utils.has_active_job(config.runner, config.executor) then
          return
        end

        if not (config:has_build_directory()) then
          -- configure it
          return cmake.generate({ bang = false, fargs = {} }, function()
            cmake.show(opt)
          end)
        end

        local targets_res = config:build_targets()
        local targets, display_targets = targets_res.data.targets, targets_res.data.display_targets

        for idx, v in ipairs(targets) do
          if v == "all" then -- this default target does not exist in the code model
            table.remove(targets, idx)
            table.remove(display_targets, idx)
          end
        end

        vim.ui.select(
          display_targets,
          { prompt = "Select target to inspect" },
          vim.schedule_wrap(function(_, idx)
            if not idx then
              return
            end
            file_picker.show_target_files(targets[idx])
          end)
        )
      else
        file_picker.show_target_files(opt.fargs[1])
      end
    end

    --- CMake show files
    vim.api.nvim_create_user_command(
      "CMakeShowTargetFiles", -- name
      cmake.show_target_files, -- command
      { -- opts
        nargs = "*",
        desc = "CMake show cmake model files or target",
      }
    )
  end
end

return cmake