From f7ab2f7903916f1b1806bf06329ad73575446782 Mon Sep 17 00:00:00 2001
From: Soonho Kong <soonhok@cs.cmu.edu>
Date: Mon, 29 Sep 2014 11:09:49 -0700
Subject: [PATCH] feat(emacs/lean-company): add auto-completion for option
 names

---
 src/emacs/lean-company.el | 45 +++++++++++++++++++++++++++++++++------
 src/emacs/lean-option.el  | 12 ++++++++---
 2 files changed, 47 insertions(+), 10 deletions(-)

diff --git a/src/emacs/lean-company.el b/src/emacs/lean-company.el
index 2654dd356..7a22a3182 100644
--- a/src/emacs/lean-company.el
+++ b/src/emacs/lean-company.el
@@ -15,6 +15,7 @@
 
 (defun company-lean-hook ()
   (set (make-local-variable 'company-backends) '(company-lean--import
+                                                 company-lean--option-name
                                                  company-lean--findg
                                                  company-lean--findp))
   (setq-local company-tooltip-limit 20)                      ; bigger popup window
@@ -26,6 +27,7 @@
 (defun company-lean--check-prefix ()
   "Check whether to use company-lean or not"
   (or (company-lean--import-prefix)
+      (company-lean--option-name-prefix)
       (company-lean--findg-prefix)
       (company-lean--findp-prefix)))
 
@@ -55,7 +57,7 @@
       (--zip-with (propertize it 'file-name other) candidates lean-files))))
 
 (defun company-lean--import-prefix ()
-  "FINDG is triggered when it looks at '_'"
+  "Import auto-completion is triggered when it looks at 'import ...'"
   (when (looking-back
          (rx "import"
              (* (+ white)
@@ -95,6 +97,26 @@
     (location   (company-lean--import-location arg))
     (sorted t)))
 
+;; OPTION
+;; ======
+(defun company-lean--option-name-prefix ()
+  "Option auto-completion is triggered when it looks at 'set-option '"
+  (when (looking-back (rx "set_option" (+ white) (* (or alnum digit "." "_"))))
+    (company-grab-symbol)))
+
+(defun company-lean--option-name-candidates (prefix)
+  (let ((candidates
+         (lean-get-options-sync-with-cont
+          (lambda (option-record-alist)
+            (-map 'car option-record-alist)))))
+    (--filter (s-starts-with? prefix it) candidates)))
+
+(defun company-lean--option-name (command &optional arg &rest ignored)
+  (case command
+    (prefix (company-lean--option-name-prefix))
+    (candidates (company-lean--option-name-candidates arg))
+    (sorted t)))
+
 ;; FINDG
 ;; =====
 
@@ -132,12 +154,21 @@
 ;; FINDP
 ;; -----
 (defun company-lean--need-autocomplete ()
-  (not (looking-back
-        (rx (or "theorem" "definition" "lemma" "axiom" "parameter"
-                "variable" "hypothesis" "conjecture"
-                "corollary" "open")
-            (+ white)
-            (* (not white))))))
+  (interactive)
+  (cond ((looking-back
+          (rx (or "theorem" "definition" "lemma" "axiom" "parameter"
+                  "variable" "hypothesis" "conjecture"
+                  "corollary" "open")
+              (+ white)
+              (* (not white))))
+         nil)
+        ((looking-back (rx "set_option"
+                           (+ white)
+                           (+ (or alnum digit "." "_"))
+                           (+ white)
+                           (* (or alnum digit "." "_"))))
+         nil)
+        (t t)))
 
 (defun company-lean--findp-prefix ()
   "Returns the symbol to complete. Also, if point is on a dot,
diff --git a/src/emacs/lean-option.el b/src/emacs/lean-option.el
index 5dea66c30..3fd3ce85b 100644
--- a/src/emacs/lean-option.el
+++ b/src/emacs/lean-option.el
@@ -57,7 +57,7 @@
 (defun lean-set-option ()
   "Set Lean option."
   (interactive)
-  (lean-get-options-with-cont 'lean-set-option-cont))
+  (lean-get-options-async-with-cont 'lean-set-option-cont))
 
 (defun lean-option-read-bool (prompt)
   (completing-read prompt'(("true" 1) ("false" 2)) nil t "" nil "true"))
@@ -166,16 +166,22 @@
               `(,(lean-option-record-name option-record) . ,option-record)))
           str-list)))
 
-(defun lean-get-options-with-cont (cont)
+(defun lean-get-options-async-with-cont (cont)
   "Get Lean option and call continuation"
   (lean-server-send-cmd-async (lean-cmd-options)
                               (lambda (option-record-alist)
                                 (funcall cont option-record-alist))))
 
+(defun lean-get-options-sync-with-cont (cont)
+  "Get Lean option and call continuation"
+  (lean-server-send-cmd-sync (lean-cmd-options)
+                              (lambda (option-record-alist)
+                                (funcall cont option-record-alist))))
+
 (defun lean-get-options ()
   "Get Lean option and call continuation"
   (interactive)
-  (lean-get-options-with-cont
+  (lean-get-options-async-with-cont
    (lambda (option-record-alist)
      (message
       (s-join "\n"