Implement support for specifying string interopolation style variable substitution in the import ...

bin/main.cc

 // Copyright (c) 2012, 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 "include/dart_debugger_api.h"
 
 #include "bin/builtin.h"
 #include "bin/dartutils.h"
 #include "bin/eventhandler.h"
 #include "bin/file.h"
 #include "bin/platform.h"
 #include "bin/process.h"
 #include "platform/globals.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 stores a pointer to the application script file.
 static char* canonical_script_name = NULL;
 
 
+// Global state that stores the import URL map specified on the
+// command line.
+static CommandLineOptions* import_map_options = NULL;
+
+
 // Global state that stores a pointer to the application script snapshot.
 static bool use_script_snapshot = false;
 static uint8_t* script_snapshot_buffer = NULL;
 
 
 // Global state that indicates whether pprof symbol information is
 // to be generated or not.
 static const char* generate_pprof_symbols_filename = NULL;
 
 
@@ skipping 34 matching lines @@
   generate_pprof_symbols_filename = filename;
 }
 
 
 static void ProcessSnapshotOption(const char* snapshot) {
   ASSERT(snapshot != NULL);
   use_script_snapshot = true;
 }
 
 
+static void ProcessImportMapOption(const char* map) {
+  ASSERT(map != NULL);
+  import_map_options->AddArgument(map);
+}
+
+
 static struct {
   const char* option_name;
   void (*process)(const char* option);
 } main_options[] = {
   { "--break_at=", ProcessBreakpointOption },
   { "--compile_all", ProcessCompileAllOption },
   { "--generate_pprof_symbols=", ProcessPprofOption },
   { "--use_script_snapshot", ProcessSnapshotOption },
+  { "--import_map=", ProcessImportMapOption },
   { NULL, NULL }
 };
 
 
 static bool ProcessMainOptions(const char* option) {
   int i = 0;
   const char* name = main_options[0].option_name;
   while (name != NULL) {
     int length = strlen(name);
     if (strncmp(option, name, length) == 0) {
@@ skipping 123 matching lines @@
       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) {
+                                     Dart_Handle url,
+                                     Dart_Handle import_map) {
   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) {
     // Handle imports of dart:io.
     if (DartUtils::IsDartIOLibURL(url_string) && (tag == kImportTag)) {
       return Builtin::LoadLibrary(Builtin::kIOLibrary);
     } else {
       return Dart_Error("Do not know how to load '%s'", url_string);
     }
   }
-  result = DartUtils::LoadSource(NULL, library, url, tag, url_string);
+  result = DartUtils::LoadSource(NULL,
+                                 library,
+                                 url,
+                                 tag,
+                                 url_string,
+                                 import_map);
   if (!Dart_IsError(result) && (tag == kImportTag)) {
     Builtin::ImportLibrary(result, Builtin::kBuiltinLibrary);
   }
   return result;
 }
 
 
-static Dart_Handle LoadScript(const char* script_name) {
+static Dart_Handle LoadScript(const char* script_name,
+                              CommandLineOptions* map) {
   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);
+  intptr_t length =  (map == NULL) ? 0 : map->count();
+  Dart_Handle import_map = Dart_NewList(length * 2);
+  for (intptr_t i = 0; i < length; i++) {
+    ASSERT(map->GetArgument(i) != NULL);
+    char* name = strdup(map->GetArgument(i));
+    ASSERT(name != NULL);
+    char* map_name = strchr(name, ',');
+    intptr_t index = i * 2;
+    if (map_name != NULL) {
+      *map_name = '\0';
+      map_name += 1;
+      Dart_ListSetAt(import_map, index, Dart_NewString(name));
+      Dart_ListSetAt(import_map, (index + 1), Dart_NewString(map_name));
+    } else {
+      Dart_ListSetAt(import_map, index, Dart_NewString(name));
+      Dart_ListSetAt(import_map, (index + 1), Dart_NewString(""));
+    }
+    free(name);
+  }
+  return Dart_LoadScript(url, source, LibraryTagHandler, import_map);
 }
 
 
 static bool CreateIsolateAndSetup(const char* name_prefix,
                                   void* data, char** error) {
   Dart_Isolate isolate =
       Dart_CreateIsolate(name_prefix, snapshot_buffer, data, error);
   if (isolate == NULL) {
     return false;
   }
 
   Dart_Handle library;
   Dart_EnterScope();
 
   // Load the specified application script into the newly created isolate.
   if (script_snapshot_buffer != NULL) {
     library = Dart_LoadScriptFromSnapshot(script_snapshot_buffer);
   } else {
-    library = LoadScript(canonical_script_name);
+    library = LoadScript(canonical_script_name, import_map_options);
   }
   if (Dart_IsError(library)) {
     *error = strdup(Dart_GetError(library));
     Dart_ExitScope();
     Dart_ShutdownIsolate();
     return false;
   }
   if (!Dart_IsLibrary(library)) {
     char errbuf[256];
     snprintf(errbuf, sizeof(errbuf),
@@ skipping 123 matching lines @@
   ASSERT(buffer != NULL);
   snprintf(buffer, len, kFormat, script_name, func_name);
   return buffer;
 }
 
 
 int main(int argc, char** argv) {
   char* script_name;
   CommandLineOptions vm_options(argc);
   CommandLineOptions dart_options(argc);
+  CommandLineOptions import_map(argc);
+  import_map_options = &import_map;
 
   // Perform platform specific initialization.
   if (!Platform::Initialize()) {
     fprintf(stderr, "Initialization failed\n");
   }
 
   // Parse command line arguments.
   if (ParseArguments(argc,
                      argv,
                      &vm_options,
@@ skipping 119 matching lines @@
   DumpPprofSymbolInfo();
   // Shutdown the isolate.
   Dart_ShutdownIsolate();
   // Terminate event handler.
   EventHandler::Terminate();
   // Terminate process exit-code handler.
   Process::TerminateExitCodeHandler();
 
   return 0;
 }