From 4bcb4a3b21f494ccfe4845230c5d8e8ad057f144 Mon Sep 17 00:00:00 2001
From: Glen Maddern <glen@cloudflare.com>
Date: Thu, 27 Jan 2022 18:57:32 +0000
Subject: [PATCH] feat: (resolves #320) disable persistence by default (adding
 --enable-persistence flag to turn back on)

---
 packages/wrangler/src/dev.tsx   | 13 ++++++++++---
 packages/wrangler/src/index.tsx |  5 +++++
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/packages/wrangler/src/dev.tsx b/packages/wrangler/src/dev.tsx
index dda43c11d2eb..1dd4a629d383 100644
--- a/packages/wrangler/src/dev.tsx
+++ b/packages/wrangler/src/dev.tsx
@@ -40,6 +40,7 @@ export type DevProps = {
   initialMode: "local" | "remote";
   jsxFactory: undefined | string;
   jsxFragment: undefined | string;
+  enablePersistence: boolean;
   bindings: CfWorkerInit["bindings"];
   public: undefined | string;
   assetPaths: undefined | AssetPaths;
@@ -102,6 +103,7 @@ function Dev(props: DevProps): JSX.Element {
           site={props.assetPaths}
           public={props.public}
           port={port}
+          enablePersistence={props.enablePersistence}
         />
       ) : (
         <Remote
@@ -184,6 +186,7 @@ function Local(props: {
   public: undefined | string;
   site: undefined | AssetPaths;
   port: number;
+  enablePersistence: boolean;
 }) {
   const { inspectorUrl } = useLocalWorker({
     name: props.name,
@@ -191,6 +194,7 @@ function Local(props: {
     format: props.format,
     bindings: props.bindings,
     port: props.port,
+    enablePersistence: props.enablePersistence,
   });
   useInspector({ inspectorUrl, port: 9229, logToTerminal: false });
   return null;
@@ -202,6 +206,7 @@ function useLocalWorker(props: {
   format: CfScriptFormat;
   bindings: CfWorkerInit["bindings"];
   port: number;
+  enablePersistence: boolean;
 }) {
   // TODO: pass vars via command line
   const { bundle, format, bindings, port } = props;
@@ -238,9 +243,11 @@ function useLocalWorker(props: {
         path.join(__dirname, "../miniflare-config-stubs/package.empty.json"),
         "--port",
         port.toString(),
-        "--kv-persist",
-        "--cache-persist",
-        "--do-persist",
+        ...(props.enablePersistence ? [
+          "--kv-persist",
+          "--cache-persist",
+          "--do-persist",
+        ] : []),
         ...Object.entries(bindings.vars || {}).flatMap(([key, value]) => {
           return ["--binding", `${key}=${value}`];
         }),
diff --git a/packages/wrangler/src/index.tsx b/packages/wrangler/src/index.tsx
index df756d168d40..2087c4c80839 100644
--- a/packages/wrangler/src/index.tsx
+++ b/packages/wrangler/src/index.tsx
@@ -541,6 +541,10 @@ export async function main(argv: string[]): Promise<void> {
         .option("jsx-fragment", {
           describe: "The function that is called for each JSX fragment",
           type: "string",
+        })
+        .option("enable-persistence", {
+          describe: "Enable persistence for this session (only for local mode)",
+          type: "boolean",
         });
     },
     async (args) => {
@@ -608,6 +612,7 @@ export async function main(argv: string[]): Promise<void> {
           initialMode={args.local ? "local" : "remote"}
           jsxFactory={args["jsx-factory"] || envRootObj?.jsx_factory}
           jsxFragment={args["jsx-fragment"] || envRootObj?.jsx_fragment}
+          enablePersistence={args["enable-persistence"] || false}
           accountId={config.account_id}
           assetPaths={getAssetPaths(
             config,