Postfix completion

pkg/analysis_server/lib/src/edit/edit_domain.dart

 // Copyright (c) 2014, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 import 'dart:async';
 
 import 'package:analysis_server/plugin/edit/assist/assist_core.dart';
 import 'package:analysis_server/plugin/edit/assist/assist_dart.dart';
 import 'package:analysis_server/plugin/edit/fix/fix_core.dart';
 import 'package:analysis_server/plugin/edit/fix/fix_dart.dart';
 import 'package:analysis_server/src/analysis_server.dart';
 import 'package:analysis_server/src/collections.dart';
 import 'package:analysis_server/src/constants.dart';
 import 'package:analysis_server/src/domain_abstract.dart';
 import 'package:analysis_server/src/plugin/plugin_manager.dart';
 import 'package:analysis_server/src/plugin/result_converter.dart';
 import 'package:analysis_server/src/protocol_server.dart' hide Element;
+import 'package:analysis_server/src/services/completion/postfix/postfix_completion.dart';
 import 'package:analysis_server/src/services/completion/statement/statement_completion.dart';
 import 'package:analysis_server/src/services/correction/assist_internal.dart';
 import 'package:analysis_server/src/services/correction/fix_internal.dart';
 import 'package:analysis_server/src/services/correction/organize_directives.dart';
 import 'package:analysis_server/src/services/correction/sort_members.dart';
 import 'package:analysis_server/src/services/correction/status.dart';
 import 'package:analysis_server/src/services/refactoring/refactoring.dart';
 import 'package:analysis_server/src/services/search/search_engine.dart';
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/ast/standard_resolution_map.dart';
@@ skipping 232 matching lines @@
       errorFixesList
           .addAll(result.fixes.map(converter.convertAnalysisErrorFixes));
     }
     //
     // Send the response.
     //
     server.sendResponse(
         new EditGetFixesResult(errorFixesList).toResponse(request.id));
   }
 
+  Future getPostfixCompletion(Request request) async {
+    var params = new EditGetPostfixCompletionParams.fromRequest(request);
+    SourceChange change;
+
+    AnalysisResult result = await server.getAnalysisResult(params.file);
+    if (result != null) {
+      CompilationUnit unit = result.unit;
+      CompilationUnitElement unitElement =
+          resolutionMap.elementDeclaredByCompilationUnit(unit);
+      if (unitElement.context != null) {
+        PostfixCompletionContext context = new PostfixCompletionContext(
+            params.file,
+            result.lineInfo,
+            params.offset,
+            params.key,
+            result.driver,
+            unit,
+            unitElement,
+            result.errors);
+        PostfixCompletionProcessor processor =
+            new PostfixCompletionProcessor(context);
+        PostfixCompletion completion = await processor.compute();
+        change = completion.change;
+      }
+    }
+    if (change == null) {
+      change = new SourceChange("", edits: []);
+    }
+
+    Response response =
+        new EditGetPostfixCompletionResult(change).toResponse(request.id);
+    server.sendResponse(response);
+  }
+
   Future getStatementCompletion(Request request) async {
     var params = new EditGetStatementCompletionParams.fromRequest(request);
     SourceChange change;
 
     AnalysisResult result = await server.getAnalysisResult(params.file);
     if (result != null) {
       CompilationUnit unit = result.unit;
       CompilationUnitElement unitElement =
           resolutionMap.elementDeclaredByCompilationUnit(unit);
       if (unitElement.context != null) {
@@ skipping 37 matching lines @@
         return _getRefactoring(request);
       } else if (requestName == EDIT_ORGANIZE_DIRECTIVES) {
         organizeDirectives(request);
         return Response.DELAYED_RESPONSE;
       } else if (requestName == EDIT_SORT_MEMBERS) {
         sortMembers(request);
         return Response.DELAYED_RESPONSE;
       } else if (requestName == EDIT_GET_STATEMENT_COMPLETION) {
         getStatementCompletion(request);
         return Response.DELAYED_RESPONSE;
+      } else if (requestName == EDIT_IS_POSTFIX_COMPLETION_APPLICABLE) {
+        isPostfixCompletionApplicable(request);
+        return Response.DELAYED_RESPONSE;
+      } else if (requestName == EDIT_GET_POSTFIX_COMPLETION) {
+        getPostfixCompletion(request);
+        return Response.DELAYED_RESPONSE;
+      } else if (requestName == EDIT_LIST_POSTFIX_COMPLETION_TEMPLATES) {
+        listPostfixCompletionTemplates(request);
+        return Response.DELAYED_RESPONSE;
       }
     } on RequestFailure catch (exception) {
       return exception.response;
     }
     return null;
   }
 
+  Future isPostfixCompletionApplicable(Request request) async {
+    var params = new EditGetPostfixCompletionParams.fromRequest(request);
+    bool value = false;
+
+    AnalysisResult result = await server.getAnalysisResult(params.file);
+    if (result != null) {
+      CompilationUnit unit = result.unit;
+      CompilationUnitElement unitElement =
+          resolutionMap.elementDeclaredByCompilationUnit(unit);
+      if (unitElement.context != null) {
+        PostfixCompletionContext context = new PostfixCompletionContext(
+            params.file,
+            result.lineInfo,
+            params.offset,
+            params.key,
+            result.driver,
+            unit,
+            unitElement,
+            result.errors);
+        PostfixCompletionProcessor processor =
+            new PostfixCompletionProcessor(context);
+        value = await processor.isApplicable();
+      }
+    }
+
+    Response response = new EditIsPostfixCompletionApplicableResult(value)
+        .toResponse(request.id);
+    server.sendResponse(response);
+  }
+
+  Future listPostfixCompletionTemplates(Request request) async {
+    var templates = DartPostfixCompletion.ALL_TEMPLATES
+        .map((pfc) =>
+            new PostfixTemplateDescriptor(pfc.name, pfc.key, pfc.example))
+        .toList();
+
+    Response response = new EditListPostfixCompletionTemplatesResult(templates)
+        .toResponse(request.id);
+    server.sendResponse(response);
+  }
+
   Future<Null> organizeDirectives(Request request) async {
     var params = new EditOrganizeDirectivesParams.fromRequest(request);
     // prepare file
     String file = params.file;
     if (!engine.AnalysisEngine.isDartFileName(file)) {
       server.sendResponse(new Response.fileNotAnalyzed(request, file));
       return;
     }
     // Prepare the file information.
     AnalysisResult result = await server.getAnalysisResult(file);
@@ skipping 593 matching lines @@
     }
     return new RefactoringStatus();
   }
 }
 
 /**
  * [_RefactoringManager] throws instances of this class internally to stop
  * processing in a manager that was reset.
  */
 class _ResetError {}