tank: restore get_architecture() for docker-compose.yml creation

Author: pinheadmz

src/warnet/tank.py

@@ -7,16 +7,15 @@
 import shutil
 from copy import deepcopy
 from pathlib import Path
-from docker.api import service
 from docker.models.containers import Container
-from services.fluentd import FLUENT_IP
 from templates import TEMPLATES
 from warnet.utils import (
     exponential_backoff,
     generate_ipv4_addr,
     sanitize_tc_netem_command,
     dump_bitcoin_conf,
     SUPPORTED_TAGS,
+    get_architecture,
 )
 
 CONTAINER_PREFIX_BITCOIND = "tank"
@@ -223,7 +222,7 @@ def add_services(self, services):
                 "context": str(TEMPLATES),
                 "dockerfile": str(TEMPLATES / f"Dockerfile"),
                 "args": {
-                    "ARCH": "x86_64",
+                    "ARCH": get_architecture(),
                     "BITCOIN_URL": "https://bitcoincore.org/bin",
                     "BITCOIN_VERSION": f"{self.version}",
                 },

src/warnet/utils.py

@@ -5,6 +5,7 @@
 import os
 import random
 import re
+import subprocess
 import sys
 import time
 from io import BytesIO
@@ -63,6 +64,20 @@ def wrapper(*args, **kwargs):
     return decorator
 
 
+def get_architecture():
+    """
+    Get the architecture of the machine.
+    :return: The architecture of the machine or None if an error occurred
+    """
+    result = subprocess.run(["uname", "-m"], stdout=subprocess.PIPE)
+    arch = result.stdout.decode("utf-8").strip()
+    if arch == "arm64":
+        arch = "aarch64"
+    if arch is None:
+        raise Exception("Failed to detect architecture.")
+    return arch
+
+
 def generate_ipv4_addr(subnet):
     """
     Generate a valid random IPv4 address within the given subnet.