Missing imports eglot

README.md

@@ -287,6 +287,7 @@ The rustic commands can be called with prefix `C-u` if you want to modify the pa
 - `rustic-cargo-add`     Add crate to Cargo.toml using 'cargo add'
 - `rustic-cargo-rm`      Remove crate from Cargo.toml using 'cargo rm'
 - `rustic-cargo-upgrade`  Upgrade dependencies as specified in the local manifest file using 'cargo upgrade'
+- `rustic-cargo-add-missing-imports ` Add the missing imports for the current buffer to `cargo.toml`. This function requires `lsp-mode` or `eglot`.
 
 ### Test
 

rustic-cargo.el

@@ -446,6 +446,42 @@ If running with prefix command `C-u', read whole command from minibuffer."
                       (concat "cargo add " (read-from-minibuffer "Crate: ")))))
       (rustic-run-cargo-command command))))
 
+(defun rustic-cargo--eglot-missing-imports ()
+  "Find missing imports when using eglot."
+  (delete-dups (seq-reduce (lambda (missing-crates clstruct)
+                             (if (and  (flymake--diag-p clstruct)
+                                       (string-match-p
+                                        (regexp-quote "unresolved import `")
+                                        (format "%s" (flymake--diag-text clstruct))))
+                                 (cons (nth 3 (split-string (flymake--diag-text clstruct) "`")) missing-crates)
+                               missing-crates))
+                           eglot--unreported-diagnostics '())))
+
+(defun rustic-cargo--lsp-missing-imports ()
+  "Find missing imports when using lsp-mode."
+  (delete-dups
+   (seq-reduce (lambda (missing-crates errortable)
+                 (if (string= "E0432" (gethash "code" errortable))
+                     (cons (nth 3 (split-string (gethash "message" errortable) "`")) missing-crates)
+                   missing-crates))
+               (gethash (buffer-file-name) (lsp-diagnostics t)) '())))
+
+(defun rustic-cargo-add-missing-imports ()
+  "Add missing imports to Cargo.toml.
+Adds all missing imports by default.
+Use with 'prefix-arg` to select imports to add."
+  (interactive)
+  (when (rustic-cargo-edit-installed-p)
+    (if-let ((missing-imports (if (ignore-error lsp-mode)
+                                  (rustic-cargo--lsp-missing-imports)
+                                (rustic-cargo--eglot-missing-imports))))
+        (rustic-run-cargo-command (format  "cargo add %s"
+                                           (if current-prefix-arg
+                                               (completing-read "Select import to add to Cargo.toml" missing-imports)
+                                             (mapconcat 'identity  missing-imports " "))))
+      (message "Couldn't find any imports to add. If this was a mistake, make sure your language server is running properly."))))
+
+
 ;;;###autoload
 (defun rustic-cargo-rm (&optional arg)
   "Remove crate from Cargo.toml using 'cargo rm'.