Merge pull request #5 from jdrupal-dev/feature/ft-config

Allow configuring file types

Author: jdrupal-dev

README.md

@@ -10,6 +10,8 @@
 
 Install this plugin as a dependency to `hrsh7th/nvim-cmp`.
 
+Default configuration can be found in `lua/css-vars/default_config.lua`.
+
 ### [lazy.nvim](https://github.com/folke/lazy.nvim)
 ```lua
 {
@@ -18,7 +20,13 @@ Install this plugin as a dependency to `hrsh7th/nvim-cmp`.
     -- other dependencies...
     {
       "jdrupal-dev/css-vars.nvim",
-      opts = {},
+      opts = {
+        -- If you use CSS-in-JS, you can add the autocompletion to JS/TS files.
+        cmp_filetypes = { "javascript", "javascriptreact", "typescript", "typescriptreact" },
+        -- WARNING: The search is not optimized to look for variables in JS files.
+        -- If you change the search_extensions you might get false positives and weird completion results.
+        search_extensions = { ".js", ".ts", ".jsx", ".tsx" }
+      },
     },
   },
   config = function()

lua/css-vars/default_config.lua

@@ -0,0 +1,6 @@
+return {
+  -- A list of filetypes to load the cmp source in.
+  cmp_filetypes = { "css", "scss", "less" },
+  -- A list of file extensions to search for CSS variables in.
+  search_extensions = { ".css", ".scss", ".less" },
+}

lua/css-vars/init.lua

@@ -3,29 +3,32 @@ local Job = require("plenary.job")
 local registered = false
 
 local M = {}
-M.setup = function()
+M.setup = function(cfg)
+  M.__conf = vim.tbl_deep_extend("keep", cfg or {}, require("css-vars.default_config"))
   if registered then
     return
   end
   registered = true
 
+  local args = {
+    "-e",
+    "[^\\w](--[^:)]*):",
+    "-r",
+    "'$1'",
+    "-o",
+    "--no-filename",
+  }
+
+  -- Only search in files that are listed in the "search_extensions" config.
+  for _, extension in pairs(M.__conf.search_extensions) do
+    table.insert(args, "-g")
+    table.insert(args, "*" .. extension)
+  end
+  table.insert(args, vim.loop.cwd())
+
   Job:new({
     command = "rg",
-    args = {
-      "-e",
-      "[^\\w](--[^:)]*):",
-      "-r",
-      "'$1'",
-      "-o",
-      "--no-filename",
-      "-g",
-      "*.css",
-      "-g",
-      "*.less",
-      "-g",
-      "*.scss",
-      vim.loop.cwd(),
-    },
+    args = args,
     env = { PATH = vim.env.PATH },
     on_exit = function(j)
       local result = j:result()
@@ -46,11 +49,7 @@ M.setup = function()
         end
 
         source.complete = function(_, request, callback)
-          if
-              request.context.filetype ~= "css"
-              and request.context.filetype ~= "less"
-              and request.context.filetype ~= "scss"
-          then
+          if not vim.tbl_contains(M.__conf.cmp_filetypes, request.context.filetype) then
             callback({ isIncomplete = true })
             return
           end