Refactor the library tag handler to share code between standalone dart and the snapshot generator.

bin/main.cc

 // Copyright (c) 2011, 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 <stdlib.h>
 #include <string.h>
 #include <stdio.h>
 
 #include "include/dart_api.h"
 
 #include "bin/builtin.h"
+#include "bin/dartutils.h"
 #include "bin/file.h"
 #include "bin/globals.h"
-#include "bin/process_script.h"
 
 // snapshot_buffer points to a snapshot if we link in a snapshot otherwise
 // it is initialized to NULL.
 extern const uint8_t* snapshot_buffer;
 
 
 // Global state that indicates whether pprof symbol information is
 // to be generated or not.
 static const char* generate_pprof_symbols_filename = NULL;
 
@@ skipping 125 matching lines @@
     if (buffer_size > 0) {
       ASSERT(buffer != NULL);
       pprof_file->WriteFully(buffer, buffer_size);
     }
     delete pprof_file;  // Closes the file.
     Dart_ExitScope();
   }
 }
 
 
+static Dart_Handle LibraryTagHandler(Dart_LibraryTag tag,
+                                     Dart_Handle library,
+                                     Dart_Handle url) {
+  if (!Dart_IsLibrary(library)) {
+    return Dart_Error("not a library");
+  }
+  if (!Dart_IsString8(url)) {
+    return Dart_Error("url is not a string");
+  }
+  const char* url_string = NULL;
+  Dart_Handle result = Dart_StringToCString(url, &url_string);
+  if (Dart_IsError(result)) {
+    return Dart_Error("accessing url characters failed");
+  }
+  bool is_dart_scheme_url = DartUtils::IsDartSchemeURL(url_string);
+  if (tag == kCanonicalizeUrl) {
+    // If this is a Dart Scheme URL then it is not modified as it will be
+    // handled by the VM internally.
+    if (is_dart_scheme_url) {
+      return url;
+    }
+    // Create a canonical path based on the including library and current url.
+    return DartUtils::CanonicalizeURL(NULL, library, url_string);
+  }
+  if (is_dart_scheme_url) {
+    return Dart_Error("Do not know how to load '%s'", url_string);
+  }
+  result = DartUtils::LoadSource(NULL, library, url, tag, url_string);
+  if (!Dart_IsError(result) && (tag == kImportTag)) {
+    Builtin_ImportLibrary(result);
+  }
+  return result;
+}
+
+
+static Dart_Handle LoadScript(const char* script_name) {
+  Dart_Handle source = DartUtils::ReadStringFromFile(script_name);
+  if (Dart_IsError(source)) {
+    return source;
+  }
+  Dart_Handle url = Dart_NewString(script_name);
+
+  return Dart_LoadScript(url, source, LibraryTagHandler);
+}
+
+
 static void* MainIsolateInitCallback(void* data) {
   const char* script_name = reinterpret_cast<const char*>(data);
   Dart_Handle library;
   Dart_EnterScope();
 
   // Load the specified script.
   library = LoadScript(script_name);
   if (Dart_IsError(library)) {
     const char* err_msg = Dart_GetError(library);
     fprintf(stderr, "Errors encountered while loading script: %s\n", err_msg);
@@ skipping 125 matching lines @@
     return 255;  // Indicates we encountered an error.
   }
   free(canonical_script_name);
   Dart_ExitScope();
   // Dump symbol information for the profiler.
   DumpPprofSymbolInfo();
   // Shutdown the isolate.
   Dart_ShutdownIsolate();
   return 0;
 }