From a7a535fdd4f0e8e46cc5a7ac751f945fdb42311b Mon Sep 17 00:00:00 2001
From: Dheeraj Manjunath <dheerajmanju1@gmail.com>
Date: Tue, 17 May 2022 13:25:18 -0400
Subject: [PATCH 1/2] Few more auto ugprade + storage space fixes

---
 audius-cli | 21 +++++++++++++++++++--
 setup.sh   |  3 ++-
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/audius-cli b/audius-cli
index 72d315b24..da55c095f 100755
--- a/audius-cli
+++ b/audius-cli
@@ -3,6 +3,8 @@
 import json
 import os
 import pathlib
+import random
+import shutil
 import subprocess
 import sys
 import urllib.error
@@ -15,6 +17,8 @@ import psutil
 
 RECOMMENDED_CPU_COUNT = 8
 RECOMMENDED_MEMORY = 16
+RECOMMENDED_DISCOVERY_STORAGE_GB = 250
+RECOMMENDED_CONTENT_STORAGE_GB = 2000
 
 SERVICE_PORTS = {
     "discovery-provider": "5000",
@@ -142,11 +146,24 @@ def launch(ctx, service, seed):
 
     total_memory = psutil.virtual_memory().total // (1024 ** 3)
     cpu_count = psutil.cpu_count()
+    total_storage, _, _ = shutil.disk_usage("/var/k8s")
+    total_storage_gb = (total_storage // (2**30))
+    meets_storage_requirements = False
+
 
     click.echo(f"CPUs:\t{cpu_count}\t(required: {RECOMMENDED_CPU_COUNT})")
     click.echo(f"Memory:\t{total_memory}GB\t(required: {RECOMMENDED_MEMORY}GB)")
 
-    if cpu_count < RECOMMENDED_CPU_COUNT or total_memory < RECOMMENDED_MEMORY:
+    if service == "discovery-provider":
+        click.echo(f"Storage:\t{total_storage_gb}GB\t(required: {RECOMMENDED_DISCOVERY_STORAGE_GB}GB)")
+        if total_storage_gb > RECOMMENDED_DISCOVERY_STORAGE_GB:
+            meets_storage_requirements = True
+    else:
+        click.echo(f"Storage:\t{total_storage_gb}GB\t(required: {RECOMMENDED_CONTENT_STORAGE_GB}GB)")
+        if total_storage_gb > RECOMMENDED_DISCOVERY_STORAGE_GB:
+            meets_storage_requirements = True
+
+    if cpu_count < RECOMMENDED_CPU_COUNT or total_memory < RECOMMENDED_MEMORY or not meets_storage_requirements:
         click.secho("System does not meet requirements", fg="red")
     else:
         click.secho("System meets requirements", fg="green")
@@ -360,7 +377,7 @@ def upgrade(ctx, branch):
 
 
 @cli.command()
-@click.argument("cron-expression", default="0 */6 * * *")
+@click.argument("cron-expression", default=f"{random.randint(0, 60)} * * * *") # random so nodes in the network stagger upgrades
 @click.option("--remove", is_flag=True)
 @click.pass_context
 def auto_upgrade(ctx, remove, cron_expression):
diff --git a/setup.sh b/setup.sh
index d4a215706..9e121f8e5 100755
--- a/setup.sh
+++ b/setup.sh
@@ -57,6 +57,7 @@ touch discovery-provider/.env
 
 # setup service
 if [[ "$1" != "" ]]; then
+	audius-cli auto-upgrade
 	audius-cli set-config --required "$1"
 
 	read -p "Are you using an externally managed Postgres? [Y/n] " -n 1 -r
@@ -79,7 +80,7 @@ if [[ "$1" != "" ]]; then
 	echo
 	if [[ "$REPLY" =~ ^([Yy]|)$ ]]; then
 		if [[ "$1" == "discovery-provider" ]]; then
-			read -p "Run seed job? [Y/n] " -n 1 -r
+			read -p "Seed discovery db from snapshot (takes ~1 hour)? [Y/n] " -n 1 -r
 			echo
 			if [[ "$REPLY" =~ ^([Yy]|)$ ]]; then
 				extra_args="--seed"

From 43489f4b9a78dad885ed120910716516dff8fa1c Mon Sep 17 00:00:00 2001
From: Dheeraj Manjunath <dheerajmanju1@gmail.com>
Date: Tue, 17 May 2022 17:22:00 -0400
Subject: [PATCH 2/2] Another comment

---
 audius-cli | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/audius-cli b/audius-cli
index da55c095f..7568fd81b 100755
--- a/audius-cli
+++ b/audius-cli
@@ -373,7 +373,7 @@ def pull(ctx, branch):
 def upgrade(ctx, branch):
     """Pull and restarts all running services"""
     ctx.forward(pull)
-    ctx.invoke(restart)
+    ctx.invoke(restart) # i think this should call launch (or just do docker compose up)
 
 
 @cli.command()