support NW_PRE_ARGS environment variable

rogerwang · rogerwang · commit 852f9cbba53d · 2019-01-21T00:34:52.000-07:00
The value will be prepend to chromium-args
diff --git a/src/nw_package.cc b/src/nw_package.cc
@@ -23,6 +23,7 @@
 #include <vector>
 
 #include "base/command_line.h"
+#include "base/environment.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/json/json_file_value_serializer.h"
@@ -389,6 +390,8 @@ void Package::InitWithDefault() {
   // Window should show in center by default.
   window->SetString(switches::kmPosition, "center");
   root()->Set(switches::kmWindow, std::move(window));
+
+  ReadChromiumArgs();
 }
 
 bool Package::ExtractPath(const base::FilePath& path_to_extract, 
@@ -448,13 +451,18 @@ bool Package::ExtractPackage(const FilePath& zip_file, FilePath* where) {
 }
 
 void Package::ReadChromiumArgs() {
-  if (!root()->HasKey(switches::kmChromiumArgs))
-    return;
+  std::string args, env_args;
+  std::unique_ptr<base::Environment> env(base::Environment::Create());
 
-  std::string args;
-  if (!root()->GetStringASCII(switches::kmChromiumArgs, &args))
+  bool got_env = env->GetVar("NW_PRE_ARGS", &env_args);
+  if (!got_env && !root()->HasKey(switches::kmChromiumArgs))
     return;
 
+  if (!got_env && !root()->GetStringASCII(switches::kmChromiumArgs, &args))
+      return;
+
+  args = env_args + kChromiumArgsSeparator + args;
+
   std::vector<std::string> chromium_args;
   base::StringTokenizer tokenizer(args, kChromiumArgsSeparator);
   tokenizer.set_quote_chars("\'");
