mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-24 10:05:29 -05:00
Arvid NorbergandGitHub
fe15e9ee6f
* fix type annotation in configure.py and init.py * enable mypy redundant-expr and remove redundant expressions
378 lines
16 KiB
Python
378 lines
16 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import click
|
|
import yaml
|
|
|
|
from chia.cmds.cmd_classes import ChiaCliContext
|
|
from chia.protocols.outbound_message import NodeType
|
|
from chia.server.resolve_peer_info import set_peer_info
|
|
from chia.util.config import (
|
|
initial_config_file,
|
|
load_defaults_for_missing_services,
|
|
lock_and_load_config,
|
|
save_config,
|
|
str2bool,
|
|
)
|
|
from chia.util.network import parse_host_port
|
|
|
|
|
|
def configure(
|
|
root_path: Path,
|
|
set_farmer_peer: str,
|
|
set_solver_peer: str,
|
|
set_node_introducer: str,
|
|
set_fullnode_port: str,
|
|
set_harvester_port: str,
|
|
set_log_level: str,
|
|
enable_upnp: str,
|
|
set_outbound_peer_count: str,
|
|
set_peer_count: str,
|
|
testnet: str,
|
|
peer_connect_timeout: str,
|
|
crawler_db_path: str | None,
|
|
crawler_minimum_version_count: int | None,
|
|
seeder_domain_name: str | None,
|
|
seeder_nameserver: str | None,
|
|
set_solver_trusted_peers_only: str,
|
|
set_log_systemd: str,
|
|
) -> None:
|
|
config_yaml = "config.yaml"
|
|
with lock_and_load_config(root_path, config_yaml, fill_missing_services=True) as config:
|
|
config.update(load_defaults_for_missing_services(config=config, config_name=config_yaml))
|
|
|
|
change_made = False
|
|
if set_node_introducer:
|
|
try:
|
|
host, port = parse_host_port(set_node_introducer)
|
|
config["full_node"]["introducer_peer"]["host"] = host
|
|
config["full_node"]["introducer_peer"]["port"] = port
|
|
config["introducer"]["port"] = port
|
|
print("Node introducer updated")
|
|
change_made = True
|
|
except ValueError:
|
|
print("Node introducer address must be in format [IP:Port]")
|
|
if set_farmer_peer:
|
|
try:
|
|
host, port = parse_host_port(set_farmer_peer)
|
|
set_peer_info(config["harvester"], peer_type=NodeType.FARMER, peer_host=host, peer_port=port)
|
|
print("Farmer peer updated, make sure your harvester has the proper cert installed")
|
|
change_made = True
|
|
except ValueError:
|
|
print("Farmer address must be in format [IP:Port]")
|
|
if set_solver_peer:
|
|
try:
|
|
host, port = parse_host_port(set_solver_peer)
|
|
# ensure farmer section has solver_peers field
|
|
if "solver_peers" not in config["farmer"]:
|
|
config["farmer"]["solver_peers"] = []
|
|
# Set single solver peer (overrides any existing)
|
|
config["farmer"]["solver_peers"] = [{"host": host, "port": port}]
|
|
print(f"Solver peer updated to {host}:{port}")
|
|
change_made = True
|
|
except ValueError:
|
|
print("Solver address must be in format [IP:Port]")
|
|
if set_solver_trusted_peers_only:
|
|
config["solver"]["trusted_peers_only"] = str2bool(set_solver_trusted_peers_only)
|
|
if str2bool(set_solver_trusted_peers_only):
|
|
print("Solver will only accept trusted peer connections")
|
|
else:
|
|
print("Solver will accept connections from all peers")
|
|
change_made = True
|
|
if set_fullnode_port:
|
|
config["full_node"]["port"] = int(set_fullnode_port)
|
|
config["full_node"]["introducer_peer"]["port"] = int(set_fullnode_port)
|
|
set_peer_info(config["farmer"], peer_type=NodeType.FULL_NODE, peer_port=int(set_fullnode_port))
|
|
set_peer_info(config["timelord"], peer_type=NodeType.FULL_NODE, peer_port=int(set_fullnode_port))
|
|
set_peer_info(config["wallet"], peer_type=NodeType.FULL_NODE, peer_port=int(set_fullnode_port))
|
|
config["wallet"]["introducer_peer"]["port"] = int(set_fullnode_port)
|
|
config["introducer"]["port"] = int(set_fullnode_port)
|
|
print("Default full node port updated")
|
|
change_made = True
|
|
if set_harvester_port:
|
|
config["harvester"]["port"] = int(set_harvester_port)
|
|
print("Default harvester port updated")
|
|
change_made = True
|
|
if set_log_level:
|
|
levels = ["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG", "NOTSET"]
|
|
if set_log_level in levels:
|
|
config["logging"]["log_level"] = set_log_level
|
|
for service in config:
|
|
if isinstance(config[service], dict) and "logging" in config[service]:
|
|
config[service]["logging"]["log_level"] = set_log_level
|
|
print(f"Logging level updated. Check {root_path}/log/debug.log")
|
|
change_made = True
|
|
else:
|
|
print(f"Logging level not updated. Use one of: {levels}")
|
|
if set_log_systemd:
|
|
bool_val = str2bool(set_log_systemd)
|
|
config["logging"]["log_systemd"] = bool_val
|
|
for service in config:
|
|
if isinstance(config[service], dict) and "logging" in config[service]:
|
|
config[service]["logging"]["log_systemd"] = bool_val
|
|
if bool_val:
|
|
print("Systemd logging enabled")
|
|
else:
|
|
print("Systemd logging disabled")
|
|
change_made = True
|
|
if enable_upnp:
|
|
config["full_node"]["enable_upnp"] = str2bool(enable_upnp)
|
|
if str2bool(enable_upnp):
|
|
print("uPnP enabled")
|
|
else:
|
|
print("uPnP disabled")
|
|
change_made = True
|
|
if set_outbound_peer_count:
|
|
config["full_node"]["target_outbound_peer_count"] = int(set_outbound_peer_count)
|
|
print("Target outbound peer count updated")
|
|
change_made = True
|
|
if set_peer_count:
|
|
config["full_node"]["target_peer_count"] = int(set_peer_count)
|
|
print("Target peer count updated")
|
|
change_made = True
|
|
if testnet:
|
|
if testnet in {"true", "t"}:
|
|
print("Setting Testnet")
|
|
# check if network_overrides.constants.testnet11 exists
|
|
if (
|
|
"testnet11" not in config["network_overrides"]["constants"]
|
|
or "testnet11" not in config["network_overrides"]["config"]
|
|
):
|
|
print("Testnet11 constants missing. Adding to config...")
|
|
initial_config_str: str = initial_config_file("config.yaml")
|
|
initial_config = yaml.safe_load(initial_config_str)
|
|
if "testnet11" not in config["network_overrides"]["constants"]:
|
|
config["network_overrides"]["constants"]["testnet11"] = initial_config["network_overrides"][
|
|
"constants"
|
|
]["testnet11"]
|
|
|
|
if "testnet11" not in config["network_overrides"]["config"]:
|
|
config["network_overrides"]["config"]["testnet11"] = initial_config["network_overrides"][
|
|
"config"
|
|
]["testnet11"]
|
|
|
|
testnet_port = "58444"
|
|
testnet_introducer = "introducer-testnet11.chia.net"
|
|
testnet_dns_introducer = "dns-introducer-testnet11.chia.net"
|
|
bootstrap_peers = ["testnet11-node-us-west-2.chia.net"]
|
|
testnet = "testnet11"
|
|
config["full_node"]["peers_file_path"] = "db/peers-testnet11.dat"
|
|
config["wallet"]["wallet_peers_file_path"] = "wallet/db/wallet_peers-testnet11.dat"
|
|
config["full_node"]["port"] = int(testnet_port)
|
|
if config["full_node"]["introducer_peer"] is None:
|
|
config["full_node"]["introducer_peer"] = {}
|
|
assert config["full_node"]["introducer_peer"] is not None # mypy
|
|
if config["wallet"]["introducer_peer"] is None:
|
|
config["wallet"]["introducer_peer"] = {}
|
|
assert config["wallet"]["introducer_peer"] is not None # mypy
|
|
config["full_node"]["introducer_peer"]["port"] = int(testnet_port)
|
|
set_peer_info(config["farmer"], peer_type=NodeType.FULL_NODE, peer_port=int(testnet_port))
|
|
set_peer_info(config["timelord"], peer_type=NodeType.FULL_NODE, peer_port=int(testnet_port))
|
|
set_peer_info(config["wallet"], peer_type=NodeType.FULL_NODE, peer_port=int(testnet_port))
|
|
config["wallet"]["introducer_peer"]["port"] = int(testnet_port)
|
|
config["introducer"]["port"] = int(testnet_port)
|
|
config["full_node"]["introducer_peer"]["host"] = testnet_introducer
|
|
config["full_node"]["dns_servers"] = [testnet_dns_introducer]
|
|
config["wallet"]["introducer_peer"]["host"] = testnet_introducer
|
|
config["wallet"]["dns_servers"] = [testnet_dns_introducer]
|
|
config["selected_network"] = testnet
|
|
config["harvester"]["selected_network"] = testnet
|
|
config["pool"]["selected_network"] = testnet
|
|
config["farmer"]["selected_network"] = testnet
|
|
config["timelord"]["selected_network"] = testnet
|
|
config["full_node"]["selected_network"] = testnet
|
|
config["ui"]["selected_network"] = testnet
|
|
config["introducer"]["selected_network"] = testnet
|
|
config["wallet"]["selected_network"] = testnet
|
|
config["data_layer"]["selected_network"] = testnet
|
|
|
|
if "seeder" in config:
|
|
config["seeder"]["port"] = int(testnet_port)
|
|
config["seeder"]["other_peers_port"] = int(testnet_port)
|
|
config["seeder"]["selected_network"] = testnet
|
|
config["seeder"]["bootstrap_peers"] = bootstrap_peers
|
|
|
|
print("Default full node port, introducer and network setting updated")
|
|
change_made = True
|
|
|
|
elif testnet in {"false", "f"}:
|
|
print("Setting Mainnet")
|
|
mainnet_port = "8444"
|
|
mainnet_introducer = "introducer.chia.net"
|
|
mainnet_dns_introducer = "dns-introducer.chia.net"
|
|
bootstrap_peers = ["node.chia.net"]
|
|
net = "mainnet"
|
|
config["full_node"]["peers_file_path"] = "db/peers.dat"
|
|
config["wallet"]["wallet_peers_file_path"] = "wallet/db/wallet_peers.dat"
|
|
config["full_node"]["port"] = int(mainnet_port)
|
|
config["full_node"]["introducer_peer"]["port"] = int(mainnet_port)
|
|
set_peer_info(config["farmer"], peer_type=NodeType.FULL_NODE, peer_port=int(mainnet_port))
|
|
set_peer_info(config["timelord"], peer_type=NodeType.FULL_NODE, peer_port=int(mainnet_port))
|
|
set_peer_info(config["wallet"], peer_type=NodeType.FULL_NODE, peer_port=int(mainnet_port))
|
|
config["wallet"]["introducer_peer"]["port"] = int(mainnet_port)
|
|
config["introducer"]["port"] = int(mainnet_port)
|
|
config["full_node"]["introducer_peer"]["host"] = mainnet_introducer
|
|
config["full_node"]["dns_servers"] = [mainnet_dns_introducer]
|
|
config["wallet"]["introducer_peer"]["host"] = mainnet_introducer
|
|
config["wallet"]["dns_servers"] = [mainnet_dns_introducer]
|
|
config["selected_network"] = net
|
|
config["harvester"]["selected_network"] = net
|
|
config["pool"]["selected_network"] = net
|
|
config["farmer"]["selected_network"] = net
|
|
config["timelord"]["selected_network"] = net
|
|
config["full_node"]["selected_network"] = net
|
|
config["ui"]["selected_network"] = net
|
|
config["introducer"]["selected_network"] = net
|
|
config["wallet"]["selected_network"] = net
|
|
config["data_layer"]["selected_network"] = net
|
|
|
|
if "seeder" in config:
|
|
config["seeder"]["port"] = int(mainnet_port)
|
|
config["seeder"]["other_peers_port"] = int(mainnet_port)
|
|
config["seeder"]["selected_network"] = net
|
|
config["seeder"]["bootstrap_peers"] = bootstrap_peers
|
|
|
|
print("Default full node port, introducer and network setting updated")
|
|
change_made = True
|
|
else:
|
|
print("Please choose True or False")
|
|
|
|
if peer_connect_timeout:
|
|
config["full_node"]["peer_connect_timeout"] = int(peer_connect_timeout)
|
|
change_made = True
|
|
|
|
if crawler_db_path is not None and "seeder" in config:
|
|
config["seeder"]["crawler_db_path"] = crawler_db_path
|
|
change_made = True
|
|
|
|
if crawler_minimum_version_count is not None and "seeder" in config:
|
|
config["seeder"]["minimum_version_count"] = crawler_minimum_version_count
|
|
change_made = True
|
|
|
|
if seeder_domain_name is not None and "seeder" in config:
|
|
config["seeder"]["domain_name"] = seeder_domain_name
|
|
change_made = True
|
|
|
|
if seeder_nameserver is not None and "seeder" in config:
|
|
config["seeder"]["nameserver"] = seeder_nameserver
|
|
change_made = True
|
|
|
|
if change_made:
|
|
print("Restart any running chia services for changes to take effect")
|
|
save_config(root_path, "config.yaml", config)
|
|
|
|
|
|
@click.command("configure", help="Modify configuration", no_args_is_help=True)
|
|
@click.option(
|
|
"--testnet",
|
|
"-t",
|
|
help="configures for connection to testnet",
|
|
type=click.Choice(["true", "t", "false", "f"]),
|
|
)
|
|
@click.option("--set-node-introducer", help="Set the introducer for node - IP:Port", type=str)
|
|
@click.option("--set-farmer-peer", help="Set the farmer peer for harvester - IP:Port", type=str)
|
|
@click.option("--set-solver-peer", help="Set the solver peer for farmer - IP:Port", type=str)
|
|
@click.option(
|
|
"--set-solver-trusted-peers-only",
|
|
help="Enable/disable trusted peer requirement for solver connections",
|
|
type=click.Choice(["true", "t", "false", "f"]),
|
|
)
|
|
@click.option(
|
|
"--set-fullnode-port",
|
|
help="Set the port to use for the fullnode, useful for testing",
|
|
type=str,
|
|
)
|
|
@click.option(
|
|
"--set-harvester-port",
|
|
help="Set the port to use for the harvester, useful for testing",
|
|
type=str,
|
|
)
|
|
@click.option(
|
|
"--set-log-level",
|
|
"--log-level",
|
|
"-log-level",
|
|
help="Set the instance log level",
|
|
type=click.Choice(["CRITICAL", "ERROR", "WARNING", "INFO", "DEBUG", "NOTSET"]),
|
|
)
|
|
@click.option(
|
|
"--log-systemd",
|
|
help="Enable or disable systemd logging",
|
|
type=click.Choice(["true", "t", "false", "f"]),
|
|
)
|
|
@click.option(
|
|
"--enable-upnp",
|
|
"--upnp",
|
|
"-upnp",
|
|
help="Enable or disable uPnP",
|
|
type=click.Choice(["true", "t", "false", "f"]),
|
|
)
|
|
@click.option(
|
|
"--set_outbound-peer-count",
|
|
help="Update the target outbound peer count (default 8)",
|
|
type=str,
|
|
)
|
|
@click.option("--set-peer-count", help="Update the target peer count (default 80)", type=str)
|
|
@click.option("--set-peer-connect-timeout", help="Update the peer connect timeout (default 30)", type=str)
|
|
@click.option(
|
|
"--crawler-db-path",
|
|
help="configures the path to the crawler database",
|
|
type=str,
|
|
)
|
|
@click.option(
|
|
"--crawler-minimum-version-count",
|
|
help="configures how many of a particular version must be seen to be reported in logs",
|
|
type=int,
|
|
)
|
|
@click.option(
|
|
"--seeder-domain-name",
|
|
help="configures the seeder domain_name setting. Ex: `seeder.example.com.`",
|
|
type=str,
|
|
)
|
|
@click.option(
|
|
"--seeder-nameserver",
|
|
help="configures the seeder nameserver setting. Ex: `example.com.`",
|
|
type=str,
|
|
)
|
|
@click.pass_context
|
|
def configure_cmd(
|
|
ctx: click.Context,
|
|
set_farmer_peer: str,
|
|
set_solver_peer: str,
|
|
set_node_introducer: str,
|
|
set_fullnode_port: str,
|
|
set_harvester_port: str,
|
|
set_log_level: str,
|
|
log_systemd: str,
|
|
enable_upnp: str,
|
|
set_outbound_peer_count: str,
|
|
set_peer_count: str,
|
|
testnet: str,
|
|
set_peer_connect_timeout: str,
|
|
crawler_db_path: str | None,
|
|
crawler_minimum_version_count: int | None,
|
|
seeder_domain_name: str | None,
|
|
seeder_nameserver: str | None,
|
|
set_solver_trusted_peers_only: str,
|
|
) -> None:
|
|
configure(
|
|
ChiaCliContext.set_default(ctx).root_path,
|
|
set_farmer_peer,
|
|
set_solver_peer,
|
|
set_node_introducer,
|
|
set_fullnode_port,
|
|
set_harvester_port,
|
|
set_log_level,
|
|
enable_upnp,
|
|
set_outbound_peer_count,
|
|
set_peer_count,
|
|
testnet,
|
|
set_peer_connect_timeout,
|
|
crawler_db_path,
|
|
crawler_minimum_version_count,
|
|
seeder_domain_name,
|
|
seeder_nameserver,
|
|
set_solver_trusted_peers_only,
|
|
log_systemd,
|
|
)
|