[earthscope, staging] feat: deploy experimental hub image for usage quotas (#7892)

* feat: add quotas requirements file

* feat: add chartpress config for experimental quotas image

* feat: add chart values for compute quotas

Author: jnywong

config/clusters/earthscope/staging.values.yaml

@@ -16,6 +16,9 @@ basehub:
           # To login to earthscope staging, now use https://staging.geolab.earthscope.cloud/hub/oauth_login
           redirect_to: https://www.earthscope.org/data/geolab/
     hub:
+      image:
+        name: quay.io/2i2c/jupyterhub-usage-quotas
+        tag: 0.0.1-0.dev.git.15310.hc76f02d9a
       config:
         GenericOAuthenticator:
           token_url: https://login-dev.earthscope.org/oauth/token
@@ -25,6 +28,41 @@ basehub:
           extra_authorize_params:
             # This isn't an actual URL, just a string. Must not have a trailing slash
             audience: https://api.dev.earthscope.org
+        UsageQuotaManager:
+          prometheus_scrape_interval: 60
+          prometheus_url: http://support-prometheus-server.support.svc.cluster.local
+          prometheus_usage_metrics:
+            memory: kube_pod_container_resource_requests{resource='memory'}
+          scope_backup_strategy:
+            empty:
+              resource: memory
+              limit:
+                value: 1000
+                unit: GiB-hours
+              window: 7
+            intersection: max
+      extraConfig:
+        11-enforce-usage-quotas: |
+          from jupyterhub_usage_quotas.manager import SpawnException, UsageQuotaManager
+
+          quota_manager = UsageQuotaManager(config=c)
+
+          async def quota_pre_spawn_hook(spawner):
+              try:
+                  output = await quota_manager.enforce(spawner)
+                  launch_flag = output["allow_server_launch"]
+              except Exception:
+                  raise SpawnException(
+                      status_code=424,
+                      log_message="Spawn failed occurred due to a failed dependency in the usage quota system. Please contact your hub admin for assistance.",
+                  )
+              if launch_flag is False:
+                  raise SpawnException(
+                      status_code=403,
+                      log_message=f"{output['error']['message']}",
+                      html_message=f"<p>Compute {output['quota']['resource']} quota limit exceeded.</p><p style='font-size:100%'>You have used <span style='color:var(--bs-red)'>{output['quota']['used']:.2f}</span> / {output['quota']['limit']['value']:.2f} {output['quota']['limit']['unit']} in the last {output['quota']['window']} days.</p><p style='font-size:100%'>Contact your JupyterHub admin if you need additional quota.</p><i style='font-size:100%;color:var(--bs-gray)'>Last updated: {output["timestamp"]} (UTC).</i>",
+                  )
+          c.KubeSpawner.pre_spawn_hook = quota_pre_spawn_hook                   
     singleuser:
       nodeSelector:
         2i2c/hub-name: staging

helm-charts/chartpress.yaml

@@ -16,3 +16,9 @@ charts:
         valuesPath: jupyterhub.hub.image
         buildArgs:
           REQUIREMENTS_FILE: requirements.txt
+      usage-quotas:
+        imageName: quay.io/2i2c/jupyterhub-usage-quotas
+        buildArgs:
+          REQUIREMENTS_FILE: quotas-requirements.txt
+        contextPath: images/hub
+        dockerfilePath: images/hub/Dockerfile

helm-charts/images/hub/quotas-requirements.txt

@@ -0,0 +1,2 @@
+git+https://github.com/2i2c-org/jupyterhub-usage-quotas@985ab1ac79e3b83ec1fd5543564347856859b0d5
+jupyterhub-fancy-profiles==0.4.1