circuit breaker setup

Author: Camillarhi

resources/plugins/circuitbreaker/charts/circuitbreaker/templates/_helpers.tpl

@@ -0,0 +1,7 @@
+{{- define "mychart.name" -}}
+{{- .Chart.Name | trunc 63 | trimSuffix "-" -}}
+{{- end -}}
+
+{{- define "mychart.fullname" -}}
+{{- printf "%s-%s" (include "mychart.name" .) .Release.Name | trunc 63 | trimSuffix "-" -}}
+{{- end -}}

resources/plugins/circuitbreaker/charts/circuitbreaker/templates/configmap.yaml

@@ -0,0 +1,21 @@
+apiVersion: v1
+kind: ConfigMap
+metadata:
+  name: {{ include "mychart.fullname" . }}-data
+data:
+  tls.cert: |
+    -----BEGIN CERTIFICATE-----
+    MIIB8TCCAZagAwIBAgIUJDsR6mmY+TaO9pCfjtotlbOkzJMwCgYIKoZIzj0EAwIw
+    MjEfMB0GA1UECgwWbG5kIGF1dG9nZW5lcmF0ZWQgY2VydDEPMA0GA1UEAwwGd2Fy
+    bmV0MB4XDTI0MTExMTE2NTM1MFoXDTM0MTEwOTE2NTM1MFowMjEfMB0GA1UECgwW
+    bG5kIGF1dG9nZW5lcmF0ZWQgY2VydDEPMA0GA1UEAwwGd2FybmV0MFkwEwYHKoZI
+    zj0CAQYIKoZIzj0DAQcDQgAEBVltIvaTlAQI/3FFatTqVflZuZdRJ0SmRMSJrFLP
+    tp0fxE7hmteSt6gjQriy90fP8j9OJXBNAjt915kLY4zVvqOBiTCBhjAOBgNVHQ8B
+    Af8EBAMCAqQwEwYDVR0lBAwwCgYIKwYBBQUHAwEwDwYDVR0TAQH/BAUwAwEB/zAd
+    BgNVHQ4EFgQU5d8QMrwhLgTkDjWA+eXZGz+dybUwLwYDVR0RBCgwJoIJbG9jYWxo
+    b3N0ggEqhwR/AAABhxAAAAAAAAAAAAAAAAAAAAABMAoGCCqGSM49BAMCA0kAMEYC
+    IQDPofN0fEl5gTwCYhk3nZbjMqJhZ8BsSJ6K8XRhxr7zbwIhAPsgQCFOqUWg632O
+    NEO53OQ6CIqnpxSskjsFNH4ZBQOE
+    -----END CERTIFICATE-----
+  admin.macaroon.hex: |
+    0201036c6e6402f801030a1062beabbf2a614b112128afa0c0b4fdd61201301a160a0761646472657373120472656164120577726974651a130a04696e666f120472656164120577726974651a170a08696e766f69636573120472656164120577726974651a210a086d616361726f6f6e120867656e6572617465120472656164120577726974651a160a076d657373616765120472656164120577726974651a170a086f6666636861696e120472656164120577726974651a160a076f6e636861696e120472656164120577726974651a140a057065657273120472656164120577726974651a180a067369676e6572120867656e657261746512047265616400000620b17be53e367290871681055d0de15587f6d1cd47d1248fe2662ae27f62cfbdc6

resources/plugins/circuitbreaker/charts/circuitbreaker/templates/deployment.yaml

@@ -0,0 +1,46 @@
+apiVersion: v1
+kind: Pod
+metadata:
+  name: {{ include "mychart.fullname" . }}
+  labels:
+    app: {{ include "mychart.name" . }}
+    mission: {{ .Values.name }}
+spec:
+  initContainers:
+    - name: "init"
+      image: "busybox"
+      command:
+        - "sh"
+        - "-c"
+      args:
+        - >
+          mkdir -p /shared/.lnd/data/chain/bitcoin/mainnet &&
+          cp /configmap/tls.cert /shared/.lnd/tls.cert &&
+          cat /configmap/admin.macaroon.hex | xxd -r -p > /shared/.lnd/data/chain/bitcoin/mainnet/admin.macaroon
+      volumeMounts:
+        - name: shared-volume
+          mountPath: /shared
+        - name: configmap-volume
+          mountPath: /configmap
+  containers:
+    - name: {{ .Values.name }}
+      image: "{{ .Values.image.repository }}:{{ .Values.image.tag }}"
+      imagePullPolicy: {{ .Values.image.pullPolicy }}
+      command:
+        - "sh"
+        - "-c"
+      args:
+        - >
+          mkdir -p /root/.lnd/data/chain/bitcoin/mainnet &&
+          ln -s /shared/.lnd/tls.cert /root/.lnd/tls.cert &&
+          ln -s /shared/.lnd/data/chain/bitcoin/mainnet/admin.macaroon /root/.lnd/data/chain/bitcoin/mainnet/admin.macaroon &&
+          circuitbreaker --rpcserver={{ .Values.lnd.rpcserver }} --httplisten={{ .Values.lnd.httplisten }}
+      volumeMounts:
+        - name: shared-volume
+          mountPath: /shared
+  volumes:
+    - name: configmap-volume
+      configMap:
+        name: {{ include "mychart.fullname" . }}-data
+    - name: shared-volume
+      emptyDir: {}

resources/plugins/circuitbreaker/charts/circuitbreaker/templates/pod.yaml

@@ -1,7 +1,14 @@
-podName: "circuitbreaker-pod"
-rpcserver: "localhost:10009"
-httplisten: "0.0.0.0:9235"
+name: "circuitbreaker"
 image:
   repository: "camillarhi/circuitbreaker"
   tag: "latest"
-  pullPolicy: IfNotPresent
+  pullPolicy: IfNotPresent
+workingVolume:
+  name: working-volume
+  mountPath: /working
+configmapVolume:
+  name: configmap-volume
+  mountPath: /configmap
+lnd:
+  rpcserver: "host.docker.internal:10009"  # Default LND RPC server address
+  httplisten: "0.0.0.0:9235"  # Default HTTP listen address

resources/plugins/circuitbreaker/charts/circuitbreaker/templates/service.yaml

@@ -85,13 +85,11 @@ def _entrypoint(ctx, plugin_content: dict, warnet_content: dict):
 
     match hook_value:
         case HookValue.POST_DEPLOY:
-            # data = get_data(plugin_content)
-            # if data:
-            #     log.info(f"Launching circuit breaker with data: {data}")
-            # _create_secrets()
-            _launch_circuit_breaker(ctx, plugin_content)
-            # else:
-            #     _launch_circuit_breaker(ctx, install_name="circuitbreaker")
+            data = get_data(plugin_content)
+            if data:
+                _launch_pod(ctx, install_name="circuitbreaker", **data)
+            else:
+                _launch_pod(ctx, install_name="circuitbreaker")
         case _:
             log.info(f"No action required for hook {hook_value}")
 
@@ -186,40 +184,21 @@ def _create_secrets():
             log.error(f"Failed to create secrets for {node_name}: {e}")
             raise PluginError(f"Failed to create secrets for {node_name}.")
 
-def _launch_circuit_breaker(ctx, 
-                            plugin_content: dict,
-                            install_name: str="circuitbreaker", 
-                            podName: str ="circuitbreaker-pod",
-                            rpcserver: str = "localhost:10009", 
-                            httplisten: str = "0.0.0.0:9235"):
+def _launch_pod(ctx,
+                install_name: str = "circuitbreaker", 
+                podName: str = "circuitbreaker-pod", 
+                rpcserver: str = "localhost:10009", 
+                httplisten: str = "0.0.0.0:9235"):
     timestamp = int(time.time())
     # release_name = f"cb-{install_name}"
 
-    lnd_pods = subprocess.check_output(["kubectl", "get", "pods", "-l", "app=warnet", "-l", "mission=lightning", "-o", "name"]).decode().splitlines()
-    for node in lnd_pods:
-        node_name = node.split('/')[-1]
-        log.info(f"Launching Circuit Breaker for {node_name}")
-        release_name = f"circuitbreaker-{node_name}"
-        
-        command = (
-            f"helm upgrade --install {release_name} {ctx.obj[PLUGIN_DIR_TAG]}/charts/circuitbreaker "
-            f"--set podName={release_name} --set rpcserver=localhost:10009 --set httplisten=0.0.0.0:9235"
-        )
-        
-    # command = f"helm upgrade --install {release_name} {ctx.obj[PLUGIN_DIR_TAG]}/charts/circuitbreaker"
-    # command = (
-    #     f"helm upgrade --install {install_name} {ctx.obj[PLUGIN_DIR_TAG]}/charts/circuitbreaker "
-    #     f"--set podName={podName} --set rpcserver={rpcserver} --set httplisten={httplisten}"
-    # )
-        log.info(command)
-        try:
-            run_command(command)
+    command = (
+        f"helm upgrade --install {install_name} {ctx.obj[PLUGIN_DIR_TAG]}/charts/circuitbreaker "
+        f"--set podName={podName} --set rpcserver={rpcserver} --set httplisten={httplisten}"
+    )
 
-    # if(hook_value==HookValue.POST_DEPLOY):
-            wait_for_init(release_name, namespace=get_default_namespace(), quiet=True)
-        except Exception as e:
-            log.error(f"Failed to launch Circuit Breaker for {node_name}: {e}")
-
+    log.info(command)
+    log.info(run_command(command))
 
 if __name__ == "__main__":
     circuitbreaker()