1 files changed, 110 insertions, 0 deletions
diff --git a/lua/bulletjava.lua b/lua/bulletjava.lua
new file mode 100644
index 0000000..f38eec7
--- /dev/null
+++ b/lua/bulletjava.lua
@@ -0,0 +1,110 @@
+local vim = assert(vim)
+
+local M = {
+  yankmap = {}
+}
+
+-- Add imports to a java file. Attempts to sort them.
+local function add_imports(buffer, buffer_lines, qualified_import_names)
+  table.sort(qualified_import_names)
+
+  local l = 1
+  local j = 1
+
+  local changed = false
+
+  local new_lines = {}
+  local new_line = string.format("import %s;", qualified_import_names[j])
+
+  while l <= #buffer_lines and j <= #qualified_import_names  do
+    local line = buffer_lines[l]
+
+    if not string.match(line, "^package") and not string.match(line, "^%s*$") then
+      if string.match(line, "^import") then
+        if new_line <= line then
+          if new_line < line then
+            changed = true
+            table.insert(new_lines, new_line)
+          end
+          j = j + 1
+          new_line = string.format("import %s;", qualified_import_names[j] or "")
+        end
+      else
+        break
+      end
+    end
+
+    table.insert(new_lines, line)
+    l = l + 1
+  end
+
+  while j <= #qualified_import_names do
+    table.insert(new_lines, string.format("import %s;", qualified_import_names[j]))
+    j = j + 1
+  end
+
+  vim.api.nvim_buf_set_lines(buffer, 0, l - 1, 0, new_lines)
+end
+
+-- Add all the imports in qualified_import_names to the import list in a Java
+-- file.
+function M.add_imports(buffer, qualified_import_names)
+  add_imports(
+      buffer,
+      vim.api.nvim_buf_get_lines(buffer, 0, -1, 0),
+      qualified_import_names)
+end
+
+local function scrape_imported_identifiers(buffer, content_list)
+  local content = table.concat(content_list, " ")
+  local find_table = {}
+  local imports = {}
+
+  for word in string.gmatch(content, "[A-Za-z0-9_]+") do
+    find_table[word] = true
+  end
+
+  local lines = vim.api.nvim_buf_get_lines(buffer, 0, -1 ,0)
+  local l = 1
+  while l < #lines do
+    local _, _, i = string.find(lines[l], "^import.*%.(%w+);")
+    if i and find_table[i] then
+      local _, _, q = string.find(lines[l], "^import%s+(.+);")
+      table.insert(imports, q)
+    elseif not i and
+          not string.match(lines[l], "^package") and
+          not string.match(lines[l], "^%s*$") then
+      break
+    end
+    l = l + 1
+  end
+
+  return imports
+end
+
+-- Called when text is yanked
+function M.on_text_yanked()
+  local regname = vim.v.event.regname
+
+  if vim.o.ft ~= 'java' then
+    -- Clear the yankmap for this register.
+    M.yankmap[regname] = nil
+    M.yankmap['"'] = nil
+  else
+    local imports = scrape_imported_identifiers(0, vim.v.event.regcontents)
+    M.yankmap[regname] = imports
+    M.yankmap['"'] = imports
+  end
+end
+
+function M.on_text_put(regname)
+  if vim.o.ft == 'java' then
+    local imps = M.yankmap[regname] or {}
+    vim.schedule(function ()
+      vim.cmd('undojoin') -- Adding the imports should be part of the same action.
+      M.add_imports(0, imps)
+    end)
+  end
+end
+
+return M