Add lsp-ivy-show-symbol-kind and lsp-ivy-filter-symbol-kind options

Author: orausch

lsp-ivy.el

@@ -15,8 +15,10 @@
 ;; You should have received a copy of the GNU General Public License
 ;; along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
-;; Author: Sebastian Sturm
+;; Authors: Sebastian Sturm
+;;          Oliver Rausch
 ;; Keywords: languages, debug
+;; Package-Version: 20191028.902
 ;; URL: https://github.com/emacs-lsp/lsp-ivy
 ;; Package-Requires: ((emacs "25.1") (dash "2.14.1") (lsp-mode "5.0") (ivy "0.13.0"))
 ;; Version: 0.1
@@ -33,14 +35,91 @@
 (require 'ivy)
 (require 'dash)
 (require 'lsp-mode)
+(defgroup lsp-ivy nil
+  "LSP support for ivy-based symbol completion"
+  :group 'lsp-mode)
+
+(defcustom lsp-ivy-show-symbol-kind
+  nil
+  "Whether to show the symbol's kind when showing lsp symbols"
+  :group 'lsp-ivy
+  :type 'boolean)
+
+(defcustom lsp-ivy-filter-symbol-kind
+  nil
+  "A list of LSP SymbolKind's to filter out"
+  :group 'lsp-ivy
+  :type '(repeat integer))
+
+(defcustom lsp-ivy-symbol-kind-to-string
+  [("    " . "red") ;; Unknown - 0
+   ("File" . "red") ;; File - 1
+   ("Modu" . "red") ;; Module - 2
+   ("Nmsp" . "red") ;; Namespace - 3
+   ("Pack" . "red") ;; Package - 4
+   ("Clss" . "red") ;; Class - 5
+   ("Meth" . "violet") ;; Method - 6
+   ("Prop" . "violet") ;; Property - 7
+   ("Fld " . "violet") ;; Field - 8
+   ("Cons" . "red") ;; Constructor - 9
+   ("Enum" . "red") ;; Enum - 10
+   ("Intf" . "red") ;; Interface - 11
+   ("Func" . "darkgreen") ;; Function - 12
+   ("Var " . "blue") ;; Variable - 13
+   ("Cnst" . "blue") ;; Constant - 14
+   ("Str " . "blue") ;; String - 15
+   ("Num " . "blue") ;; Number - 16
+   ("Bool " . "blue") ;; Boolean - 17
+   ("Arr " . "blue") ;; Array - 18
+   ("Obj " . "blue") ;; Object - 19
+   ("Key " . "blue") ;; Key - 20
+   ("Null" . "red") ;; Null - 21
+   ("EmMm" . "violet") ;; EnumMember - 22
+   ("Srct" . "red") ;; Struct - 23
+   ("Evnt" . "red") ;; Event - 24
+   ("Op  " . "red") ;; Operator - 25
+   ("TPar" . "red")] ;; TypeParameter - 26
+  "A vector of 26 cons cells, where the ith cons cell contains the string representation and foreground color to use for the i+1th SymbolKind (defined in the LSP)"
+  :group 'lsp-ivy
+  :type '(vector
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)
+          (cons string color)))
+
 
 (defun lsp-ivy--format-symbol-match (match)
   "Convert the (hash-valued) MATCH returned by `lsp-mode` into a candidate string."
-  (let ((container-name (gethash "containerName" match))
-        (name (gethash "name" match)))
+  (let* ((container-name (gethash "containerName" match))
+         (name (gethash "name" match))
+         (type (elt lsp-ivy-symbol-kind-to-string (gethash "kind" match) ))
+         (typestr (propertize (format "[%s]" (car type)) 'face `(:foreground ,(cdr type)))))
     (if (or (null container-name) (string-empty-p container-name))
-        name
-      (format "%s.%s" container-name name))))
+        (format "%s %s" typestr name)
+      (format "%s %s.%s" typestr container-name name))))
 
 (defun lsp-ivy--workspace-symbol-action (candidate)
   "Jump to selected CANDIDATE."
@@ -51,6 +130,9 @@
     (forward-line line)
     (forward-char character)))
 
+(defun lsp-ivy--filter-func (candidate)
+  (member (gethash "kind" candidate) lsp-ivy-filter-symbol-kind))
+
 (defun lsp-ivy--workspace-symbol (workspaces prompt initial-input)
   "Search against WORKSPACES with PROMPT and INITIAL-INPUT."
   (let ((current-request-id nil))
@@ -68,7 +150,8 @@
                  (plist-get request :id))
            (lsp-send-request-async
             request
-            #'ivy-update-candidates
+            (lambda (result)
+              (ivy-update-candidates (-remove 'lsp-ivy--filter-func result)))
             :mode 'detached)))
        0)
      :dynamic-collection t