From 83785ba721cba90e48ab0086d1fdbea82b029aa0 Mon Sep 17 00:00:00 2001
From: gelocraft <gelocraft@users.noreply.github.com>
Date: Fri, 13 Dec 2024 11:46:01 +0800
Subject: [PATCH] nvim(after/ftplugin): create global user_command for
 ansible-lint

---
 nvim/after/ftplugin/yaml.lua | 50 +++++++++++++++++++++++++++++++++++-
 1 file changed, 49 insertions(+), 1 deletion(-)

diff --git a/nvim/after/ftplugin/yaml.lua b/nvim/after/ftplugin/yaml.lua
index 14b65fa..ac451a9 100644
--- a/nvim/after/ftplugin/yaml.lua
+++ b/nvim/after/ftplugin/yaml.lua
@@ -2,6 +2,54 @@ local set = vim.opt_local
 set.tabstop = 2
 set.softtabstop = 2
 set.shiftwidth = 2
+
 if vim.fn.executable 'yamllint' ~= 0 then
-    set.makeprg = 'yamllint -f parsable'
+	set.makeprg = 'yamllint -f parsable'
+end
+
+if vim.fn.executable 'ansible-lint' ~= 0 then
+	---@param command string
+	local create_ansible_lint_command = function(command)
+		vim.api.nvim_create_user_command(command, function(opts)
+			local argc = #opts.fargs
+			if argc == 0 then
+				local current_buffer = vim.api.nvim_buf_get_name(0)
+				vim.cmd(
+					string.format(
+						'cexpr system("ansible-lint -p %s")',
+						current_buffer
+					)
+				)
+			else
+				vim.cmd(
+					string.format(
+						'cexpr system("ansible-lint -p %s")',
+						opts.args
+					)
+				)
+			end
+		end, { nargs = '*' })
+	end
+	local ansible_files = {
+		'ansible.cfg',
+		'inventory',
+		'inventory.ini',
+		'inventory.yaml',
+		'inventory/',
+		'playbook.yaml',
+		'playbooks/',
+		'roles/',
+		'host_vars/',
+		'group_vars/',
+	}
+
+	for _, f in ipairs(ansible_files) do
+		local ansible_file = (
+			vim.fn.filereadable(f) ~= 0 or vim.fn.isdirectory(f) ~= 0
+		)
+		if ansible_file then
+			create_ansible_lint_command 'AnsibleLint'
+			break
+		end
+	end
 end