Expose set_source service command for functions

runtime/vm/service.cc

 // Copyright (c) 2013, 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.
 
 #include "vm/service.h"
 
 #include "include/dart_api.h"
 #include "platform/globals.h"
 
 #include "vm/compiler.h"
 #include "vm/coverage.h"
 #include "vm/cpu.h"
 #include "vm/dart_api_impl.h"
 #include "vm/dart_entry.h"
 #include "vm/debugger.h"
 #include "vm/isolate.h"
 #include "vm/message.h"
 #include "vm/message_handler.h"
 #include "vm/native_entry.h"
 #include "vm/native_arguments.h"
 #include "vm/object.h"
 #include "vm/object_graph.h"
 #include "vm/object_id_ring.h"
 #include "vm/object_store.h"
+#include "vm/parser.h"
 #include "vm/port.h"
 #include "vm/profiler.h"
 #include "vm/reusable_handles.h"
 #include "vm/stack_frame.h"
 #include "vm/symbols.h"
 #include "vm/unicode.h"
 #include "vm/version.h"
 
 namespace dart {
 
@@ skipping 1234 matching lines @@
 
 
 static bool HandleClassesFunctionsCoverage(
     Isolate* isolate, const Function& func, JSONStream* js) {
   FunctionCoverageFilter filter(func);
   CodeCoverage::PrintJSON(isolate, js, &filter);
   return true;
 }
 
 
+static bool HandleFunctionSetSource(
+    Isolate* isolate, const Class& cls, const Function& func, JSONStream* js) {
+  if (js->LookupOption("source") == NULL) {
+    PrintError(js, "set_source expects a 'source' option\n");
+    return true;
+  }
+  const String& source =
+      String::Handle(String::New(js->LookupOption("source")));
+  const Object& result = Object::Handle(
+      Parser::ParseFunctionFromSource(cls, source));
+  if (result.IsError()) {
+    Error::Cast(result).PrintJSON(js, false);
+    return true;
+  }
+  ASSERT(result.IsFunction());
+
+  // Replace function.
+  cls.RemoveFunction(func);
+  cls.AddFunction(Function::Cast(result));
+
+  JSONObject jsobj(js);
+  jsobj.AddProperty("type", "Success");
+  jsobj.AddProperty("id", "");
+  return true;
+}
+
+
 static bool HandleClassesFunctions(Isolate* isolate, const Class& cls,
                                    JSONStream* js) {
   if (js->num_arguments() != 4 && js->num_arguments() != 5) {
     PrintError(js, "Command should have 4 or 5 arguments");
     return true;
   }
   const char* encoded_id = js->GetArgument(3);
   String& id = String::Handle(isolate, String::New(encoded_id));
   id = String::DecodeIRI(id);
   if (id.IsNull()) {
     PrintError(js, "Function id %s is malformed", encoded_id);
     return true;
   }
   Function& func = Function::Handle(cls.LookupFunction(id));
   if (func.IsNull()) {
     PrintError(js, "Function %s not found", encoded_id);
     return true;
   }
   if (js->num_arguments() == 4) {
     func.PrintJSON(js, false);
     return true;
   } else {
-    const char* subcollection = js->GetArgument(4);
-    if (strcmp(subcollection, "coverage") == 0) {
+    const char* subcommand = js->GetArgument(4);
+    if (strcmp(subcommand, "coverage") == 0) {
       return HandleClassesFunctionsCoverage(isolate, func, js);
+    } else if (strcmp(subcommand, "set_source") == 0) {
+      return HandleFunctionSetSource(isolate, cls, func, js);
     } else {
-      PrintError(js, "Invalid sub collection %s", subcollection);
+      PrintError(js, "Invalid sub command %s", subcommand);
       return true;
     }
   }
   UNREACHABLE();
   return true;
 }
 
 
 static bool HandleClassesImplicitClosures(Isolate* isolate, const Class& cls,
                                           JSONStream* js) {
@@ skipping 1429 matching lines @@
   while (current != NULL) {
     if (strcmp(name, current->name()) == 0) {
       return current;
     }
     current = current->next();
   }
   return NULL;
 }
 
 }  // namespace dart