1+ from __future__ import annotations
12import argparse
23import logging
34from logging .handlers import TimedRotatingFileHandler
45import os
6+ from typing import Optional , List
57
68import toml
79
1214
1315
1416class Config :
17+ branch : str
18+ network : str
19+ home_dir : str
20+ network_dir : str
21+ disable_update : bool
22+ external_ip : Optional [str ]
23+ backup_dir : Optional [str ]
24+ restore_dir : Optional [str ]
25+ eth_providers : List [str ]
26+
1527 def __init__ (self , loader : ConfigLoader ):
1628 self .logger = logging .getLogger ("launcher.Config" )
1729
@@ -23,44 +35,49 @@ def __init__(self, loader: ConfigLoader):
2335 self .network = os .environ ["NETWORK" ]
2436
2537 self .home_dir = self .loader .ensure_home_dir (os .environ ["HOST_HOME" ])
26- self .network_dir = None
38+ self .network_dir = os . path . join ( self . home_dir , self . network )
2739 self .backup_dir = None
2840 self .restore_dir = None
2941
3042 self .eth_providers = general_config [self .network ]["eth_providers" ]
3143
3244 self .nodes = nodes_config [self .network ]
3345
34- self .args = None
46+ self ._parse_command_line_arguments ()
47+ self ._parse_general_config_file ()
48+ self .network_dir = self .loader .ensure_network_dir (self .network_dir )
49+
50+ self ._parse_network_config_file ()
51+
52+ for node in self .nodes .values ():
53+ for v in node ["volumes" ]:
54+ v ["host" ] = self .expand_vars (v ["host" ])
3555
36- self .parse ()
56+ self .dump ()
3757
38- def parse (self ):
58+ def _parse_command_line_arguments (self ) -> None :
59+ self .args = None
3960 try :
4061 self .parse_command_line_arguments ()
4162 except Exception as e :
4263 raise ConfigError (ConfigErrorScope .COMMAND_LINE_ARGS ) from e
4364
44- self .network_dir = "{}/{}" .format (self .home_dir , self .network )
45-
65+ def _parse_general_config_file (self ) -> None :
66+ filename = "xud-docker.conf"
67+ conf_file = os .path .join (self .home_dir , filename )
4668 try :
4769 self .parse_general_config ()
4870 except Exception as e :
49- conf_file = "{}/{}.conf" .format (self .home_dir , "xud-docker" )
5071 raise ConfigError (ConfigErrorScope .GENERAL_CONF , conf_file = conf_file ) from e
5172
52- self .network_dir = self .loader .ensure_network_dir (self .network_dir )
53-
73+ def _parse_network_config_file (self ) -> None :
74+ filename = "{}.conf" .format (self .network )
75+ conf_file = os .path .join (self .network_dir , filename )
5476 try :
5577 self .parse_network_config ()
5678 except Exception as e :
57- conf_file = "{}/{}.conf" .format (self .network_dir , self .network )
5879 raise ConfigError (ConfigErrorScope .NETWORK_CONF , conf_file = conf_file ) from e
5980
60- for node in self .nodes .values ():
61- for v in node ["volumes" ]:
62- v ["host" ] = self .expand_vars (v ["host" ])
63-
6481 def parse_command_line_arguments (self ):
6582 parser = ArgumentParser (argument_default = argparse .SUPPRESS , prog = "xud.sh" , usage = "bash xud.sh [OPTIONS]" )
6683 parser .add_argument (
@@ -915,8 +932,100 @@ def expand_vars(self, value):
915932 return value
916933
917934 @property
918- def logfile (self ):
919- if self .network_dir :
920- network = self .network
921- return f"{ self .network_dir } { network }
922- return None
935+ def logs_dir (self ) -> str :
936+ return os .path .join (self .network_dir , "logs" )
937+
938+ @property
939+ def logfile (self ) -> str :
940+ filename = f"{ self .network }
941+ return os .path .join (self .logs_dir , filename )
942+
943+ @property
944+ def dumpfile (self ) -> str :
945+ filename = f"config.sh"
946+ return os .path .join (self .logs_dir , filename )
947+
948+ def dump (self ) -> None :
949+ """Dump xud-docker configurations as bash key-value file in logs_dir"""
950+ prefix = "XUD_DOCKER"
951+
952+ with open ("/mnt/hostfs" + self .dumpfile , "w" ) as f :
953+ def dump_attr (attr : str ) -> None :
954+ key = f"{ prefix } { attr .upper ()}
955+ value = getattr (self , attr )
956+ if not value :
957+ value = ""
958+ if isinstance (value , bool ):
959+ value = str (value ).lower ()
960+ print ("{}=\" {}\" " .format (key , value ), file = f )
961+ dump_attr ("branch" )
962+ dump_attr ("disable_update" )
963+ dump_attr ("external_ip" )
964+ dump_attr ("network" )
965+ dump_attr ("home_dir" )
966+ dump_attr ("network_dir" )
967+ dump_attr ("backup_dir" )
968+ dump_attr ("restore_dir" )
969+
970+ # dump nodes config
971+ def dump_node_attr (node : str , attr : str ) -> None :
972+ node_config = self .nodes [node ]
973+ node_prefix = f"{ prefix } { node .upper ()}
974+ if attr == "volumes" :
975+ for volume in node_config ["volumes" ]:
976+ key = f"{ node_prefix }
977+ value = "{}:{}" .format (volume ["host" ], volume ["container" ])
978+ print ("{}=\" {}\" " .format (key , value ), file = f )
979+ elif attr == "ports" :
980+ for port in node_config ["ports" ]:
981+ key = f"{ node_prefix }
982+ value = str (port )
983+ print ("{}=\" {}\" " .format (key , value ), file = f )
984+ else :
985+ key = f"{ node_prefix } { attr .upper ()}
986+ value = ""
987+ try :
988+ value = node_config [attr ]
989+ except KeyError :
990+ if node == "arby" :
991+ value = node_config .get (attr .replace ("_" , "-" ), "" )
992+ if not value :
993+ value = ""
994+ if isinstance (value , bool ):
995+ value = str (value ).lower ()
996+ print ("{}=\" {}\" " .format (key , value ), file = f )
997+
998+ for node in self .nodes .keys ():
999+ dump_node_attr (node , "image" )
1000+ dump_node_attr (node , "volumes" )
1001+ dump_node_attr (node , "ports" )
1002+ dump_node_attr (node , "mode" )
1003+ dump_node_attr (node , "disabled" )
1004+ dump_node_attr (node , "preserve_config" )
1005+ dump_node_attr (node , "use_local_image" )
1006+
1007+ if node in ["bitcoind" , "litecoind" ]:
1008+ dump_node_attr (node , "external_rpc_host" )
1009+ dump_node_attr (node , "external_rpc_port" )
1010+ #dump_node_attr(node, "external_rpc_user")
1011+ #dump_node_attr(node, "external_rpc_password")
1012+ dump_node_attr (node , "external_zmqpubrawblock" )
1013+ dump_node_attr (node , "external_zmqpubrawtx" )
1014+ elif node == "geth" :
1015+ dump_node_attr (node , "external_rpc_host" )
1016+ dump_node_attr (node , "external_rpc_port" )
1017+ #dump_node_attr(node, "infura_project_id")
1018+ #dump_node_attr(node, "infura_project_secret")
1019+ dump_node_attr (node , "cache" )
1020+ elif node == "arby" :
1021+ dump_node_attr (node , "test_centralized_baseasset_balance" )
1022+ dump_node_attr (node , "test_centralized_quoteasset_balance" )
1023+ dump_node_attr (node , "opendex_base_asset" )
1024+ dump_node_attr (node , "opendex_quote_asset" )
1025+ dump_node_attr (node , "cex_base_asset" )
1026+ dump_node_attr (node , "cex_quote_asset" )
1027+ dump_node_attr (node , "live_cex" )
1028+ dump_node_attr (node , "cex" )
1029+ #dump_node_attr(node, "cex_api_key")
1030+ #dump_node_attr(node, "cex_api_secret")
1031+ dump_node_attr (node , "margin" )
0 commit comments