From 7a5bc94774a50837f8c9bf8b96c8272882aca640 Mon Sep 17 00:00:00 2001
From: Aleksey Kladov <aleksey.kladov@gmail.com>
Date: Fri, 31 Aug 2018 16:30:42 +0300
Subject: complete self

---
 crates/libeditor/src/completion.rs     | 13 +++++++++++++
 crates/libeditor/src/scope/fn_scope.rs |  4 ++++
 2 files changed, 17 insertions(+)

(limited to 'crates/libeditor/src')

diff --git a/crates/libeditor/src/completion.rs b/crates/libeditor/src/completion.rs
index 06a607265..f3058c023 100644
--- a/crates/libeditor/src/completion.rs
+++ b/crates/libeditor/src/completion.rs
@@ -142,6 +142,12 @@ fn complete_fn(name_ref: ast::NameRef, scopes: &FnScopes, acc: &mut Vec<Completi
                 snippet: None,
             })
     );
+    if scopes.self_param.is_some() {
+        acc.push(CompletionItem {
+            name: "self".to_string(),
+            snippet: None,
+        })
+    }
 }
 
 #[cfg(test)]
@@ -249,6 +255,13 @@ mod tests {
                CompletionItem { name: "foo", snippet: None }]"#)
     }
 
+    #[test]
+    fn test_complete_self() {
+        check_scope_completion(r"
+            impl S { fn foo(&self) { <|> } }
+        ", r#"[CompletionItem { name: "self", snippet: None }]"#)
+    }
+
     #[test]
     fn test_completion_kewords() {
         check_snippet_completion(r"
diff --git a/crates/libeditor/src/scope/fn_scope.rs b/crates/libeditor/src/scope/fn_scope.rs
index 5c04e2f9b..78e9c061c 100644
--- a/crates/libeditor/src/scope/fn_scope.rs
+++ b/crates/libeditor/src/scope/fn_scope.rs
@@ -13,6 +13,7 @@ type ScopeId = usize;
 
 #[derive(Debug)]
 pub struct FnScopes {
+    pub self_param: Option<SyntaxNode>,
     scopes: Vec<ScopeData>,
     scope_for: HashMap<SyntaxNode, ScopeId>,
 }
@@ -20,6 +21,9 @@ pub struct FnScopes {
 impl FnScopes {
     pub fn new(fn_def: ast::FnDef) -> FnScopes {
         let mut scopes = FnScopes {
+            self_param: fn_def.param_list()
+                .and_then(|it| it.self_param())
+                .map(|it| it.syntax().owned()),
             scopes: Vec::new(),
             scope_for: HashMap::new()
         };
-- 
cgit v1.2.3