add WORKERD_ENABLE_ALL_AUTOGATES env variable

Author: anonrig

src/workerd/server/server.c++

@@ -25,7 +25,6 @@
 #include <workerd/io/worker-interface.h>
 #include <workerd/io/worker.h>
 #include <workerd/server/actor-id-impl.h>
-#include <workerd/util/autogate.h>
 #include <workerd/util/http-util.h>
 #include <workerd/util/mimetype.h>
 #include <workerd/util/use-perfetto-categories.h>

src/workerd/server/workerd.c++

@@ -754,6 +754,10 @@ class CliMain final: public SchemaFileImpl::ErrorReporter {
         "<const-name>", CLI_METHOD(setConstName));
   }
 
+  bool shouldEnableAllAutogates() {
+    return getenv("WORKERD_ENABLE_ALL_AUTOGATES") != nullptr;
+  }
+
   kj::MainBuilder& addServeOrTestOptions(kj::MainBuilder& builder) {
     return builder
         .addOptionWithArg({'d', "directory-path"}, CLI_METHOD(overrideDirectory), "<name>=<path>",
@@ -1077,7 +1081,11 @@ class CliMain final: public SchemaFileImpl::ErrorReporter {
     mod.setPythonModule("def test():\n pass");
     config = configBuilder.asReader();
     configOwner = kj::mv(builder);
-    util::Autogate::initAutogate(getConfig().getAutogates());
+    if (shouldEnableAllAutogates()) {
+      util::Autogate::initAllAutogates();
+    } else {
+      util::Autogate::initAutogate(getConfig().getAutogates());
+    }
   }
 
   void watch() {
@@ -1160,7 +1168,11 @@ class CliMain final: public SchemaFileImpl::ErrorReporter {
     // We'll fail at getConfig() if there are multiple top level Config objects.
     // The error message says that you have to specify which config to use, but
     // it's not clear that there is any mechanism to do that.
-    util::Autogate::initAutogate(getConfig().getAutogates());
+    if (shouldEnableAllAutogates()) {
+      util::Autogate::initAllAutogates();
+    } else {
+      util::Autogate::initAutogate(getConfig().getAutogates());
+    }
   }
 
   void setConstName(kj::StringPtr name) {

src/workerd/util/autogate.c++

@@ -66,6 +66,21 @@ void Autogate::deinitAutogate() {
   globalAutogate = kj::none;
 }
 
+void Autogate::initAllAutogates() {
+  capnp::MallocMessageBuilder message;
+  auto orphanage = message.getOrphanage();
+  auto gatesOrphan =
+      orphanage.newOrphan<capnp::List<capnp::Text>>(static_cast<size_t>(AutogateKey::NumOfKeys));
+  auto gates = gatesOrphan.get();
+
+  for (auto i = AutogateKey(0); i < AutogateKey::NumOfKeys;
+       i = AutogateKey(static_cast<size_t>(i) + 1)) {
+    gates.set(static_cast<size_t>(i), kj::str("workerd-autogate-", i));
+  }
+
+  Autogate::initAutogate(gates.asReader());
+}
+
 void Autogate::initAutogateNamesForTest(std::initializer_list<kj::StringPtr> gateNames) {
   capnp::MallocMessageBuilder message;
   auto orphanage = message.getOrphanage();

src/workerd/util/autogate.h

@@ -45,6 +45,16 @@ class Autogate {
   // Convenience method for bin-tests to invoke initAutogate() with an appropriate config.
   static void initAutogateNamesForTest(std::initializer_list<kj::StringPtr> gateNames);
 
+  // Initializes all autogates, enabling all features.
+  //
+  // This is used by the --all-autogates flag in the CLI to enable all gates regardless
+  // of what's specified in the config file. This can be useful during development to test
+  // features that are behind autogates.
+  //
+  // CAUTION: Be aware that enabling all gates may expose you to experimental features
+  // that aren't ready for production use. Use with care in production environments.
+  static void initAllAutogates();
+
   // Destroys an initialized global Autogate instance. Used only for testing.
   static void deinitAutogate();