Add shouldUpload flag to docGenAndUpload command

2 files changed, 16 insertions, 2 deletions

diff --git a/packages/monorepo-scripts/src/doc_generate_and_upload.ts b/packages/monorepo-scripts/src/doc_generate_and_upload.ts
index 602247972..77dc2f323 100644
--- a/packages/monorepo-scripts/src/doc_generate_and_upload.ts
+++ b/packages/monorepo-scripts/src/doc_generate_and_upload.ts
@@ -13,12 +13,19 @@ const args = yargs
         type: 'boolean',
         demandOption: true,
     })
+    .option('shouldUpload', {
+        describe: 'Whether we wish to upload the docs to S3 or not',
+        type: 'boolean',
+        demandOption: false,
+        default: true,
+    })
     .example("$0 --package '0x.js' --isStaging true", 'Full usage example').argv;
 
 (async () => {
     const packageName = args.package;
     const isStaging = args.isStaging;
+    const shouldUploadDocs = args.shouldUpload;
 
-    const docGenerateAndUploadUtils = new DocGenerateAndUploadUtils(packageName, isStaging);
+    const docGenerateAndUploadUtils = new DocGenerateAndUploadUtils(packageName, isStaging, shouldUploadDocs);
     await docGenerateAndUploadUtils.generateAndUploadDocsAsync();
 })();
diff --git a/packages/monorepo-scripts/src/utils/doc_generate_and_upload_utils.ts b/packages/monorepo-scripts/src/utils/doc_generate_and_upload_utils.ts
index cc4d70d63..7e8bd07da 100644
--- a/packages/monorepo-scripts/src/utils/doc_generate_and_upload_utils.ts
+++ b/packages/monorepo-scripts/src/utils/doc_generate_and_upload_utils.ts
@@ -12,6 +12,7 @@ import { utils } from './utils';
 
 export class DocGenerateAndUploadUtils {
     private _isStaging: boolean;
+    private _shouldUploadDocs: boolean;
     private _packageName: string;
     private _omitExports: string[];
     private _packagePath: string;
@@ -155,9 +156,10 @@ export class DocGenerateAndUploadUtils {
         };
         return exportInfo;
     }
-    constructor(packageName: string, isStaging: boolean) {
+    constructor(packageName: string, isStaging: boolean, shouldUploadDocs: boolean) {
         this._isStaging = isStaging;
         this._packageName = packageName;
+        this._shouldUploadDocs = shouldUploadDocs;
         this._packagePath = `${constants.monorepoRootPath}/packages/${packageName}`;
 
         this._monoRepoPkgNameToPath = {};
@@ -268,6 +270,11 @@ export class DocGenerateAndUploadUtils {
         };
         writeFileSync(jsonFilePath, JSON.stringify(docJson, null, 2));
 
+        if (this._shouldUploadDocs) {
+            await this._uploadDocsAsync(jsonFilePath, cwd);
+        }
+    }
+    private async _uploadDocsAsync(jsonFilePath: string, cwd: string) {
         const fileName = `v${this._packageJson.version}.json`;
         utils.log(`GENERATE_UPLOAD_DOCS: Doc generation successful, uploading docs... as ${fileName}`);