tranche 0.2.2__tar.gz → 0.2.4__tar.gz

{tranche-0.2.2 → tranche-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tranche
-Version: 0.2.2
+Version: 0.2.4
 Summary: Command line interface used for generating local Lightning test environment
 License: MIT
 Author: Sitt Guruvanich
@@ -44,8 +44,8 @@ You can use `tranche` simply by installing via `pip` on your Terminal.
 $ pip install tranche
 > ...
 > Installing collected packages: tranche
-> Successfully installed tranche-0.2.2
-$ tranche cluster
+> Successfully installed tranche-0.2.4
+$ tranche deploy
 > Deploy specified local cluster:            ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:01
 > Generate addresses:                        ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00
 > Mine initial capital for parties:          ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00

{tranche-0.2.2 → tranche-0.2.4}/README.md

@@ -20,8 +20,8 @@ You can use `tranche` simply by installing via `pip` on your Terminal.
 $ pip install tranche
 > ...
 > Installing collected packages: tranche
-> Successfully installed tranche-0.2.2
-$ tranche cluster
+> Successfully installed tranche-0.2.4
+$ tranche deploy
 > Deploy specified local cluster:            ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:01
 > Generate addresses:                        ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00
 > Mine initial capital for parties:          ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 100% 0:00:00

{tranche-0.2.2 → tranche-0.2.4}/core.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    tranche.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 02:20
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:
@@ -14,7 +14,7 @@
 from click import group
 
 ### Local modules ###
-from src.commands import clean, cluster, flush, mine, nodekeys, ping_pong, setup
+from src.commands import clean, deploy, flush, mine, nodekeys, ping_pong, setup
 
 
 @group
@@ -23,7 +23,7 @@ def cli() -> None:
 
 
 cli.add_command(clean, "clean")
-cli.add_command(cluster, "cluster")
+cli.add_command(deploy, "deploy")
 cli.add_command(flush, "flush")
 cli.add_command(mine, "mine")
 cli.add_command(nodekeys, "nodekeys")

{tranche-0.2.2 → tranche-0.2.4}/pyproject.toml

@@ -28,7 +28,7 @@ packages = [
   {include='core.py'}
 ]
 readme = 'README.md'
-version = '0.2.2'
+version = '0.2.4'
 
 [tool.poetry.dependencies]
 APScheduler = '^3.10.4'

{tranche-0.2.2 → tranche-0.2.4}/src/__init__.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/__init__.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:
@@ -11,7 +11,7 @@
 # *************************************************************
 
 ### Local modules ###
-from src.commands import clean, cluster, flush, mine, nodekeys, ping_pong, setup
+from src.commands import clean, deploy, flush, mine, nodekeys, ping_pong, setup
 from src.schemas import (
     BlockchainInfo,
     ClusterEnum,
@@ -36,7 +36,7 @@ __all__ = [
     "Service",
     "ServiceName",
     "clean",
-    "cluster",
+    "deploy",
     "flush",
     "mine",
     "nodekeys",
@@ -44,4 +44,4 @@ __all__ = [
     "setup",
 ]
 
-__version__ = "0.2.2"
+__version__ = "0.2.4"

{tranche-0.2.2 → tranche-0.2.4}/src/commands/__init__.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/commands/__init__.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:
@@ -12,11 +12,11 @@
 
 ### Local modules ###
 from src.commands.clean import clean
-from src.commands.cluster import cluster
+from src.commands.deploy import deploy
 from src.commands.flush import flush
 from src.commands.mine import mine
 from src.commands.nodekeys import nodekeys
 from src.commands.ping_pong import ping_pong
 from src.commands.setup import setup
 
-__all__ = ["clean", "cluster", "flush", "mine", "nodekeys", "ping_pong", "setup"]
+__all__ = ["clean", "deploy", "flush", "mine", "nodekeys", "ping_pong", "setup"]

{tranche-0.2.2 → tranche-0.2.4}/src/commands/clean.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/commands/clean.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

tranche-0.2.2/src/commands/cluster.py → tranche-0.2.4/src/commands/deploy.py

@@ -1,8 +1,8 @@
 #!/usr/bin/env python3.8
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
-# FILENAME:    ~~/src/commands/cluster.py
-# VERSION: 	   0.2.2
+# FILENAME:    ~~/src/commands/deploy.py
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:
@@ -18,7 +18,7 @@ from typing import Dict, List
 ### Third-party packages ###
 from click import command, option
 from docker import DockerClient, from_env
-from docker.errors import APIError
+from docker.errors import APIError, NotFound
 from docker.models.containers import Container
 from pydantic import TypeAdapter
 from rich.progress import track
@@ -31,7 +31,7 @@ from src.schemas import MutexOption, NewAddress, Service, ServiceName
 @command
 @option("--duo", alternatives=["uno"], cls=MutexOption, is_flag=True, type=bool)
 @option("--uno", alternatives=["duo"], cls=MutexOption, is_flag=True, type=bool)
-def cluster(duo: bool, uno: bool) -> None:
+def deploy(duo: bool, uno: bool) -> None:
     """Deploy cluster, either with one or two LND nodes."""
     duo = duo or (not duo and not uno)  # defaults to duo network
     cluster: Dict[ServiceName, Service] = (CLUSTERS["duo"], CLUSTERS["uno"])[uno]
@@ -71,7 +71,12 @@ def cluster(duo: bool, uno: bool) -> None:
                     ).output
                 )
                 mining_targets.append(new_address.address)
-        bitcoind: Container = client.containers.get("tranche-bitcoind")
+        bitcoind: Container
+        try:
+            bitcoind = client.containers.get("tranche-bitcoind")
+        except NotFound:
+            print('!! Unable to find "tranche-bitcoind" container.')
+            return
         for address in track(mining_targets, "Mine initial capital for parties:".ljust(42)):
             bitcoind.exec_run(
                 """
@@ -81,4 +86,4 @@ def cluster(duo: bool, uno: bool) -> None:
             )
 
 
-__all__ = ["cluster"]
+__all__ = ["deploy"]

{tranche-0.2.2 → tranche-0.2.4}/src/commands/flush.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/commands/flush.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 06:24
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:
@@ -11,7 +11,7 @@
 # *************************************************************
 
 ### Standard packages ###
-from typing import Set
+from typing import List, Set
 
 ### Third-party packages ###
 from click import command
@@ -27,11 +27,13 @@ def flush() -> None:
     """Remove images deprecated by workspace."""
     client: DockerClient = from_env()
     if client.ping():
+        outputs: List[str] = []
         docker_images: Set[str] = {image.tags[0] for image in client.images.list()}
         for registry_id in track(DEPRECATED, "Remove deprecated images:".ljust(42)):
             if registry_id in docker_images:
                 client.images.remove(registry_id)
-                print(f"<Image: '{ registry_id }'> removed.")
+                outputs.append(f"<Image: '{ registry_id }'> removed.")
+        list(map(print, outputs))
     else:
         print("!! Unable to connect to docker daemon.")
 

{tranche-0.2.2 → tranche-0.2.4}/src/commands/mine.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/commands/nodekeys.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

tranche-0.2.4/src/commands/nodekeys.py

@@ -0,0 +1,54 @@
+#!/usr/bin/env python3.8
+# coding:utf-8
+# Copyright (C) 2022-2023 All rights reserved.
+# FILENAME:    ~~/src/commands/nodekeys.py
+# VERSION: 	   0.2.4
+# CREATED: 	   2023-12-01 05:31
+# AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
+# DESCRIPTION:
+#
+# HISTORY:
+# *************************************************************
+
+### Standard packages ###
+from re import match
+from typing import List
+
+### Third-party packages ###
+from click import command
+from docker import DockerClient, from_env
+from docker.models.containers import Container
+from pydantic import TypeAdapter
+from rich.progress import track
+
+### Local modules ###
+from src.schemas import LNDInfo
+
+
+@command
+def nodekeys() -> None:
+    """Fetch nodekeys from active LND containers."""
+    client: DockerClient = from_env()
+    if client.ping():
+        containers: List[Container] = reversed(client.containers.list())  # type: ignore[assignment]
+        lnds: List[Container] = list(
+            filter(lambda c: match(r"tranche-lnd|tranche-ping|tranche-pong", c.name), containers)
+        )
+        outputs: List[str] = []
+        for container in track(lnds, "Fetch LND nodekeys:".ljust(42)):
+            lnd_info: LNDInfo = TypeAdapter(LNDInfo).validate_json(
+                container.exec_run(
+                    """
+                    lncli
+                        --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
+                        --rpcserver=localhost:10001
+                        --tlscertpath=/home/lnd/.lnd/tls.cert
+                    getinfo
+                    """
+                ).output
+            )
+            outputs.append(f"<Nodekey: '{container.name}', '{lnd_info.identity_pubkey}'>")
+        list(map(print, outputs))
+
+
+__all__ = ["nodekeys"]

tranche-0.2.4/src/commands/ping_pong.py

@@ -0,0 +1,135 @@
+#!/usr/bin/env python3.8
+# coding:utf-8
+# Copyright (C) 2022-2023 All rights reserved.
+# FILENAME:    ~~/src/commands/ping_pong.py
+# VERSION: 	   0.2.4
+# CREATED: 	   2023-12-01 06:18
+# AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
+# DESCRIPTION:
+#
+# HISTORY:
+# *************************************************************
+
+### Standard packages ###
+from re import match
+from typing import Dict, List
+
+### Third-party packages ###
+from click import argument, command
+from docker import DockerClient, from_env
+from docker.errors import NotFound
+from docker.models.containers import Container
+from pydantic import TypeAdapter, ValidationError
+from rich.progress import track
+
+### Local modules ###
+from src.schemas import LNDInfo, NewAddress, OpenChannel
+
+
+@command
+@argument("channel_size", default=16777215)
+def ping_pong(channel_size: int) -> None:
+    """For "duo" cluster, create channels between LND nodes."""
+    client: DockerClient = from_env()
+    if client.ping():
+        bitcoind: Container
+        try:
+            bitcoind = client.containers.get("tranche-bitcoind")
+        except NotFound:
+            print('!! Unable to find "tranche-bitcoind" container.')
+            return
+        containers: List[Container] = list(reversed(client.containers.list()))
+        paddles: List[Container] = list(
+            filter(lambda c: match(r"tranche-ping|tranche-pong", c.name), containers)
+        )
+        mining_targets: Dict[str, str] = {}
+        nodekeys: Dict[str, str] = {}
+
+        ### Fetch nodekeys ###
+        for container in track(paddles, "Fetch LND nodekeys:".ljust(42)):
+            lnd_info: LNDInfo = TypeAdapter(LNDInfo).validate_json(
+                container.exec_run(
+                    """
+                    lncli
+                        --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
+                        --rpcserver=localhost:10001
+                        --tlscertpath=/home/lnd/.lnd/tls.cert
+                    getinfo
+                    """
+                ).output
+            )
+            nodekeys[container.name] = lnd_info.identity_pubkey
+            new_address: NewAddress = TypeAdapter(NewAddress).validate_json(
+                container.exec_run(
+                    """
+                    lncli
+                        --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
+                        --rpcserver=localhost:10001
+                        --tlscertpath=/home/lnd/.lnd/tls.cert
+                    newaddress p2wkh
+                    """
+                ).output
+            )
+            mining_targets[container.name] = new_address.address
+
+        ### Open channels ###
+        outputs: List[str] = []
+        for container in track(paddles, "Open channels:".ljust(42)):
+            if container.name == "tranche-ping":
+                try:
+                    open_channel: OpenChannel = TypeAdapter(OpenChannel).validate_json(
+                        container.exec_run(
+                            """
+                            lncli
+                                --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
+                                --rpcserver=localhost:10001
+                                --tlscertpath=/home/lnd/.lnd/tls.cert
+                            openchannel %d
+                                --node_key %s
+                                --connect tranche-pong:9735
+                            """
+                            % (channel_size, nodekeys.get("tranche-pong", ""))
+                        ).output
+                    )
+                    outputs.append(
+                        f"<Channel 'tranche-ping --> tranche-pong' : { open_channel.funding_txid }>"
+                    )
+                    bitcoind.exec_run(
+                        """
+                        bitcoin-cli -regtest -rpcuser=tranche -rpcpassword=tranche generatetoaddress %d %s
+                        """
+                        % (6, mining_targets.get("tranche-ping", ""))
+                    )
+                except ValidationError:
+                    outputs.append("!! Channel 'tranche-ping --> tranche-pong' already opened.")
+            elif container.name == "tranche-pong":
+                try:
+                    open_channel: OpenChannel = TypeAdapter(OpenChannel).validate_json(  # type: ignore[no-redef]
+                        container.exec_run(
+                            """
+                            lncli
+                                --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
+                                --rpcserver=localhost:10001
+                                --tlscertpath=/home/lnd/.lnd/tls.cert
+                            openchannel %d
+                                --node_key %s
+                                --connect tranche-ping:9735
+                            """
+                            % (channel_size, nodekeys.get("tranche-ping", ""))
+                        ).output
+                    )
+                    outputs.append(
+                        f"<Channel 'tranche-pong --> tranche-ping' : { open_channel.funding_txid }>"
+                    )
+                    bitcoind.exec_run(
+                        """
+                        bitcoin-cli -regtest -rpcuser=tranche -rpcpassword=tranche generatetoaddress %d %s
+                        """
+                        % (6, mining_targets.get("tranche-pong", ""))
+                    )
+                except ValidationError:
+                    outputs.append("!! Channel 'tranche-pong --> tranche-ping' already opened.")
+        list(map(print, outputs))
+
+
+__all__ = ["ping_pong"]

{tranche-0.2.2 → tranche-0.2.4}/src/commands/setup.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/commands/setup.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 06:18
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:
@@ -11,7 +11,7 @@
 # *************************************************************
 
 ### Standard packages ###
-from typing import Set
+from typing import List, Set
 
 ### Third-party packages ###
 from click import command
@@ -27,14 +27,16 @@ def setup() -> None:
     """Download docker images used by command-line interface."""
     client: DockerClient = from_env()
     if client.ping():
+        outputs: List[str] = []
         docker_images: Set[str] = {image.tags[0] for image in client.images.list()}
         for registry_id in track(IMAGES.values(), "Download required images:".ljust(42)):
             if registry_id in docker_images:
-                print(f"<Image: '{ registry_id }'> already exists in local docker images.")
+                outputs.append(f"<Image: '{ registry_id }'> already exists in local docker images.")
             else:
                 repository, tag = registry_id.split(":")
                 client.images.pull(repository=repository, tag=tag)
-                print(f"<Image: '{ registry_id }'> downloaded.")
+                outputs.append(f"<Image: '{ registry_id }'> downloaded.")
+        list(map(print, outputs))
     else:
         print("!! Unable to connect to docker daemon.")
 

{tranche-0.2.2 → tranche-0.2.4}/src/configs.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/configs.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/__init__.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/__init__.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/blockchain_info.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/blockchain_info.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/cluster_enum.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/cluster_enum.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/image_alias.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/image_alias.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/lnd_info.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/lnd_info.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/mutex_option.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/mutex_option.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/new_address.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/new_address.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/open_channel.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/open_channel.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-02 20:50
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

{tranche-0.2.2 → tranche-0.2.4}/src/schemas/service.py

@@ -2,7 +2,7 @@
 # coding:utf-8
 # Copyright (C) 2022-2023 All rights reserved.
 # FILENAME:    ~~/src/schemas/service.py
-# VERSION: 	   0.2.2
+# VERSION: 	   0.2.4
 # CREATED: 	   2023-12-01 05:31
 # AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
 # DESCRIPTION:

tranche-0.2.2/src/commands/nodekeys.py

@@ -1,47 +0,0 @@
-#!/usr/bin/env python3.8
-# coding:utf-8
-# Copyright (C) 2022-2023 All rights reserved.
-# FILENAME:    ~~/src/commands/nodekeys.py
-# VERSION: 	   0.2.2
-# CREATED: 	   2023-12-01 05:31
-# AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
-# DESCRIPTION:
-#
-# HISTORY:
-# *************************************************************
-
-### Standard packages ###
-from re import match
-
-### Third-party packages ###
-from click import command
-from docker import DockerClient, from_env
-from pydantic import TypeAdapter
-from rich.progress import track
-
-### Local modules ###
-from src.schemas import LNDInfo
-
-
-@command
-def nodekeys() -> None:
-    """Fetch nodekeys from active LND containers."""
-    client: DockerClient = from_env()
-    if client.ping():
-        for container in track(client.containers.list(), "Fetch LND nodekeys:".ljust(42)):
-            if match(r"tranche-lnd|tranche-ping|tranche-pong", container.name) is not None:
-                lnd_info: LNDInfo = TypeAdapter(LNDInfo).validate_json(
-                    container.exec_run(
-                        """
-                        lncli
-                            --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
-                            --rpcserver=localhost:10001
-                            --tlscertpath=/home/lnd/.lnd/tls.cert
-                        getinfo
-                        """
-                    ).output
-                )
-                print(f"<Nodekey: '{container.name}', '{lnd_info.identity_pubkey}'>")
-
-
-__all__ = ["nodekeys"]

tranche-0.2.2/src/commands/ping_pong.py

@@ -1,100 +0,0 @@
-#!/usr/bin/env python3.8
-# coding:utf-8
-# Copyright (C) 2022-2023 All rights reserved.
-# FILENAME:    ~~/src/commands/ping_pong.py
-# VERSION: 	   0.2.2
-# CREATED: 	   2023-12-01 06:18
-# AUTHOR: 	   Sitt Guruvanich <aekasitt.g+github@siamintech.co.th>
-# DESCRIPTION:
-#
-# HISTORY:
-# *************************************************************
-
-### Standard packages ###
-from re import match
-from typing import Dict, List
-
-### Third-party packages ###
-from click import argument, command
-from docker import DockerClient, from_env
-from docker.models.containers import Container
-from pydantic import TypeAdapter, ValidationError
-from rich.progress import track
-
-### Local modules ###
-from src.schemas import LNDInfo, OpenChannel
-
-
-@command
-@argument("channel_size", default=16777215)
-def ping_pong(channel_size: int) -> None:
-    """For "duo" cluster, create channels between LND nodes."""
-    client: DockerClient = from_env()
-    if client.ping():
-        nodekeys: Dict[str, str] = {}
-        containers: List[Container] = client.containers.list()
-
-        ### Fetch nodekeys ###
-        for container in track(containers, "Fetch LND nodekeys:".ljust(42)):
-            if match(r"tranche-ping|tranche-pong", container.name) is not None:
-                lnd_info: LNDInfo = TypeAdapter(LNDInfo).validate_json(
-                    container.exec_run(
-                        """
-                        lncli
-                            --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
-                            --rpcserver=localhost:10001
-                            --tlscertpath=/home/lnd/.lnd/tls.cert
-                        getinfo
-                        """
-                    ).output
-                )
-                nodekeys[container.name] = lnd_info.identity_pubkey
-
-        ### Open channels ###
-        errors: List[str] = []
-        funding_txids: List[str] = []
-        for container in track(containers, "Open channels:".ljust(42)):
-            if container.name == "tranche-ping":
-                try:
-                    open_channel: OpenChannel = TypeAdapter(OpenChannel).validate_json(
-                        container.exec_run(
-                            """
-                        lncli
-                            --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
-                            --rpcserver=localhost:10001
-                            --tlscertpath=/home/lnd/.lnd/tls.cert
-                        openchannel %d
-                            --node_key %s
-                            --connect tranche-pong:9735
-                        """
-                            % (channel_size, nodekeys.get("tranche-pong", ""))
-                        ).output
-                    )
-                    funding_txids.append(open_channel.funding_txid)
-                except ValidationError:
-                    errors.append("!! Channel between tranche-ping to tranche-pong already opened.")
-            elif container.name == "tranche-pong":
-                try:
-                    open_channel: OpenChannel = TypeAdapter(OpenChannel).validate_json(
-                        container.exec_run(
-                            """
-                        lncli
-                            --macaroonpath=/home/lnd/.lnd/data/chain/bitcoin/regtest/admin.macaroon
-                            --rpcserver=localhost:10001
-                            --tlscertpath=/home/lnd/.lnd/tls.cert
-                        openchannel %d
-                            --node_key %s
-                            --connect tranche-ping:9735
-                        """
-                            % (channel_size, nodekeys.get("tranche-ping", ""))
-                        ).output
-                    )
-                    funding_txids.append(open_channel.funding_txid)
-                except ValidationError:
-                    errors.append("!! Channel between tranche-pong to tranche-ping already opened.")
-        print(f"Funding transactions: { funding_txids }")
-        for error in errors:
-            print(error)
-
-
-__all__ = ["ping_pong"]