Initial commit

1 files changed, 272 insertions, 0 deletions

diff --git a/lua/tasks/module/cmake.lua b/lua/tasks/module/cmake.lua
new file mode 100644
index 0000000..4900aec
--- /dev/null
+++ b/lua/tasks/module/cmake.lua
@@ -0,0 +1,272 @@
+local Path = require('plenary.path')
+local utils = require('tasks.utils')
+local scandir = require('plenary.scandir')
+local ProjectConfig = require('tasks.project_config')
+local os = require('ffi').os:lower()
+local cmake = {}
+
+--- Parses build dir expression.
+---@param dir string: Path with expressions to replace.
+---@param build_type string
+---@return table
+local function parse_dir(dir, build_type)
+  local parsed_dir = dir:gsub('{cwd}', vim.loop.cwd())
+  parsed_dir = parsed_dir:gsub('{os}', os)
+  parsed_dir = parsed_dir:gsub('{build_type}', build_type:lower())
+  return Path:new(parsed_dir)
+end
+
+--- Returns reply directory that contains targets information.
+---@param build_dir table
+---@return unknown
+local function get_reply_dir(build_dir) return build_dir / '.cmake' / 'api' / 'v1' / 'reply' end
+
+--- Reads information about target.
+---@param codemodel_target table
+---@param reply_dir table
+---@return table
+local function get_target_info(codemodel_target, reply_dir) return vim.json.decode((reply_dir / codemodel_target['jsonFile']):read()) end
+
+--- Creates query files that to acess information about targets after CMake configuration.
+---@param build_dir table
+---@return boolean: Returns `true` on success.
+local function make_query_files(build_dir)
+  local query_dir = build_dir / '.cmake' / 'api' / 'v1' / 'query'
+  if not query_dir:mkdir({ parents = true }) then
+    utils.notify(string.format('Unable to create "%s"', query_dir.filename), vim.log.levels.ERROR)
+    return false
+  end
+
+  local codemodel_file = query_dir / 'codemodel-v2'
+  if not codemodel_file:is_file() then
+    if not codemodel_file:touch() then
+      utils.notify(string.format('Unable to create "%s"', codemodel_file.filename), vim.log.levels.ERROR)
+      return false
+    end
+  end
+  return true
+end
+
+--- Reads targets information.
+---@param reply_dir table
+---@return table?
+local function get_codemodel_targets(reply_dir)
+  local found_files = scandir.scan_dir(reply_dir.filename, { search_pattern = 'codemodel*' })
+  if #found_files == 0 then
+    utils.notify('Unable to find codemodel file', vim.log.levels.ERROR)
+    return nil
+  end
+  local codemodel = Path:new(found_files[1])
+  local codemodel_json = vim.json.decode(codemodel:read())
+  return codemodel_json['configurations'][1]['targets']
+end
+
+---@return table?
+local function get_target_names()
+  local project_config = ProjectConfig.new()
+  local build_dir = parse_dir(project_config.cmake.build_dir, project_config.cmake.build_type)
+  if not build_dir:is_dir() then
+    utils.notify(string.format('Build directory "%s" does not exist, you need to run "configure" task first', build_dir), vim.log.levels.ERROR)
+    return nil
+  end
+
+  local reply_dir = get_reply_dir(build_dir)
+  local codemodel_targets = get_codemodel_targets(reply_dir)
+  if not codemodel_targets then
+    return nil
+  end
+
+  local targets = {}
+  for _, target in ipairs(codemodel_targets) do
+    local target_info = get_target_info(target, reply_dir)
+    local target_name = target_info['name']
+    if target_name:find('_autogen') == nil then
+      table.insert(targets, target_name)
+    end
+  end
+
+  return targets
+end
+
+--- Finds path to an executable.
+---@param build_dir table
+---@param name string
+---@param reply_dir table
+---@return unknown?
+local function get_executable_path(build_dir, name, reply_dir)
+  for _, target in ipairs(get_codemodel_targets(reply_dir)) do
+    if name == target['name'] then
+      local target_info = get_target_info(target, reply_dir)
+      if target_info['type'] ~= 'EXECUTABLE' then
+        utils.notify(string.format('Specified target "%s" is not an executable', name), vim.log.levels.ERROR)
+        return nil
+      end
+
+      local target_path = Path:new(target_info['artifacts'][1]['path'])
+      if not target_path:is_absolute() then
+        target_path = build_dir / target_path
+      end
+
+      return target_path
+    end
+  end
+
+  utils.notify(string.format('Unable to find target named "%s"', name), vim.log.levels.ERROR)
+  return nil
+end
+
+--- Copies compile_commands.json file from build directory to the current working directory for LSP integration.
+local function copy_compile_commands()
+  local project_config = ProjectConfig.new()
+  local filename = 'compile_commands.json'
+  local source = parse_dir(project_config.cmake.build_dir, project_config.cmake.build_type) / filename
+  local destination = Path:new(vim.loop.cwd(), filename)
+  source:copy({ destination = destination.filename })
+end
+
+--- Task
+---@param module_config table
+---@return table?
+local function configure(module_config, _)
+  local build_dir = parse_dir(module_config.build_dir, module_config.build_type)
+  build_dir:mkdir({ parents = true })
+  if not make_query_files(build_dir) then
+    return nil
+  end
+
+  return {
+    cmd = module_config.cmd,
+    args = { '-B', build_dir.filename, '-D', 'CMAKE_BUILD_TYPE=' .. module_config.build_type },
+    after_success = copy_compile_commands,
+  }
+end
+
+--- Task
+---@param module_config table
+---@return table
+local function build(module_config, _)
+  local build_dir = parse_dir(module_config.build_dir, module_config.build_type)
+
+  local args = { '--build', build_dir.filename }
+  if module_config.target then
+    vim.list_extend(args, { '--target', module_config.target })
+  end
+
+  return {
+    cmd = module_config.cmd,
+    args = args,
+    after_success = copy_compile_commands,
+  }
+end
+
+--- Task
+---@param module_config table
+---@return table
+local function build_all(module_config, _)
+  local build_dir = parse_dir(module_config.build_dir, module_config.build_type)
+
+  return {
+    cmd = module_config.cmd,
+    args = { '--build', build_dir.filename },
+    after_success = copy_compile_commands,
+  }
+end
+
+--- Task
+---@param module_config table
+---@return table?
+local function run(module_config, _)
+  if not module_config.target then
+    utils.notify('No selected target, please set "target" parameter', vim.log.levels.ERROR)
+    return nil
+  end
+
+  local build_dir = parse_dir(module_config.build_dir, module_config.build_type)
+  if not build_dir:is_dir() then
+    utils.notify(string.format('Build directory "%s" does not exist, you need to run "configure" task first', build_dir), vim.log.levels.ERROR)
+    return nil
+  end
+
+  local target_path = get_executable_path(build_dir, module_config.target, get_reply_dir(build_dir))
+  if not target_path then
+    return
+  end
+
+  if not target_path:is_file() then
+    utils.notify(string.format('Selected target "%s" is not built', target_path.filename), vim.log.levels.ERROR)
+    return nil
+  end
+
+  return {
+    cmd = target_path.filename,
+    cwd = target_path:parent().filename,
+  }
+end
+
+--- Task
+---@param module_config table
+---@return table?
+local function debug(module_config, _)
+  if module_config.build_type ~= 'Debug' and module_config.build_type ~= 'RelWithDebInfo' then
+    utils.notify(
+      string.format('For debugging your "build_type" param should be set to "Debug" or "RelWithDebInfo", but your current build type is "%s"'),
+      module_config.build_type,
+      vim.log.levels.ERROR
+    )
+    return nil
+  end
+
+  local command = run(module_config, nil)
+  if not command then
+    return nil
+  end
+
+  command.dap_name = module_config.dap_name
+  return command
+end
+
+--- Task
+---@param module_config table
+---@return table
+local function clean(module_config, _)
+  local build_dir = parse_dir(module_config.build_dir, module_config.build_type)
+
+  return {
+    cmd = module_config.cmd,
+    args = { '--build', build_dir.filename, '--target', 'clean' },
+    after_success = copy_compile_commands,
+  }
+end
+
+--- Task
+---@param module_config table
+---@return table
+local function open_build_dir(module_config, _)
+  local build_dir = parse_dir(module_config.build_dir, module_config.build_type)
+
+  return {
+    cmd = os == 'windows' and 'start' or 'xdg-open',
+    args = { build_dir.filename },
+    ignore_stdout = true,
+    ignore_stderr = true,
+  }
+end
+
+cmake.params = {
+  target = get_target_names,
+  build_type = { 'Debug', 'Release', 'RelWithDebInfo', 'MinSizeRel' },
+  'cmake',
+  'dap_name',
+}
+cmake.condition = function() return Path:new('CMakeLists.txt'):exists() end
+cmake.tasks = {
+  configure = configure,
+  build = build,
+  build_all = build_all,
+  run = { build, run },
+  debug = debug,
+  clean = clean,
+  open_build_dir = open_build_dir,
+}
+
+return cmake