feat: Support for i18next plurals functionality (#37)

Author: nabekou29

lua/js-i18n/analyzer.lua

@@ -77,6 +77,7 @@ end
 --- @param stop? integer 終了位置
 --- @return TSNode | nil, string | nil
 function M.get_node_for_key(bufnr, keys, start, stop)
+  keys = { unpack(keys) }
   local key = table.concat(keys, c.config.key_separator)
   local ts = vim.treesitter
 

lua/js-i18n/lsp/protocol/request/text_document_completion.lua

@@ -4,18 +4,30 @@ local utils = require("js-i18n.utils")
 
 --- 全ての翻訳を取得する関数
 --- @param translation I18n.TranslationSource
+--- @param library? string ライブラリ
 --- @param prefix? string
 --- @param result? table<string, string>
 --- @return table<string, string>
-local function get_all_translation(translation, prefix, result)
+local function get_all_translation(translation, library, prefix, result)
   result = result or {}
   for key, value in pairs(translation) do
     local full_key = prefix and prefix .. c.config.key_separator .. key or key
     if type(value) == "table" then
-      get_all_translation(value, full_key, result)
+      get_all_translation(value, library, full_key, result)
     else
-      result[full_key] = value
+      if library == utils.Library.I18Next then
+        for _, suffix in ipairs(c.config.libraries.i18next.plural_suffixes) do
+          -- 末尾がsuffixで終わる場合
+          if full_key:sub(-#suffix) == suffix then
+            local key_without_suffix = full_key:sub(1, -#suffix - 1)
+            result[key_without_suffix] = result[key_without_suffix] or value
+            goto continue
+          end
+        end
+      end
+      result[full_key] = result[full_key] or value
     end
+    ::continue::
   end
   return result
 end
@@ -28,12 +40,13 @@ end
 local function get_completion_items(client, bufnr, t_call)
   local lang = client:get_language(bufnr)
   local t_source = client.t_source_by_workspace[utils.get_workspace_root(bufnr)]
+  local library = utils.detect_library(bufnr)
 
   local key_prefix = t_call.key_prefix or ""
 
   local translations = {}
   for _, source in pairs(t_source:get_translation_source_by_lang(lang)) do
-    for key, value in pairs(get_all_translation(source)) do
+    for key, value in pairs(get_all_translation(source, library)) do
       if key_prefix == "" then
         translations[key] = value
       else

lua/js-i18n/lsp/protocol/request/text_document_definition.lua

@@ -26,26 +26,51 @@ local function handler(params, client)
   end
 
   local key = t_call.key
+  local library = utils.detect_library(bufnr)
 
   for file, _ in pairs(t_source:get_translation_source_by_lang(lang)) do
     local bufnr = vim.api.nvim_create_buf(false, true)
     local content = Path:new(file):read()
     vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, vim.split(content, "\n"))
 
+    --- @type lsp.Location[]
+    local result = {}
+
     local keys = vim.split(key, c.config.key_separator, { plain = true })
     local node = analyzer.get_node_for_key(bufnr, keys)
-    vim.api.nvim_buf_delete(bufnr, { force = true })
-
     if node ~= nil then
       local row_start, col_start, row_end, col_end = node:range()
-      return nil,
-        {
-          uri = vim.uri_from_fname(file),
-          range = {
-            start = { line = row_start, character = col_start },
-            ["end"] = { line = row_end, character = col_end },
-          },
-        }
+      table.insert(result, {
+        uri = vim.uri_from_fname(file),
+        range = {
+          start = { line = row_start, character = col_start },
+          ["end"] = { line = row_end, character = col_end },
+        },
+      })
+    end
+
+    if library == utils.Library.I18Next then
+      for _, suffix in ipairs(c.config.libraries.i18next.plural_suffixes) do
+        local key_with_suffix = { unpack(keys) }
+        key_with_suffix[#key_with_suffix] = key_with_suffix[#key_with_suffix] .. suffix
+
+        local node = analyzer.get_node_for_key(bufnr, key_with_suffix)
+        if node ~= nil then
+          local row_start, col_start, row_end, col_end = node:range()
+          table.insert(result, {
+            uri = vim.uri_from_fname(file),
+            range = {
+              start = { line = row_start, character = col_start },
+              ["end"] = { line = row_end, character = col_end },
+            },
+          })
+        end
+      end
+    end
+
+    vim.api.nvim_buf_delete(bufnr, { force = true })
+    if #result > 0 then
+      return nil, result
     end
   end
 

lua/js-i18n/lsp/protocol/request/text_document_hover.lua

@@ -24,11 +24,13 @@ local function handler(params, client)
   local key = t_call.key
   local keys = vim.split(key, c.config.key_separator, { plain = true })
 
+  local library = utils.detect_library(bufnr)
+
   -- 各言語の翻訳を表示
   --- @type string[]
   local contents = {}
   for _, lang in ipairs(t_source:get_available_languages()) do
-    local translation = t_source:get_translation(lang, keys)
+    local translation = t_source:get_translation(lang, keys, library)
     if translation then
       if type(translation) == "string" then
         table.insert(contents, lang .. ": `" .. translation .. "`")