mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-09-21 08:31:52 +03:00
83 lines
3.4 KiB
Python
83 lines
3.4 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Any, Dict, Optional
|
|
|
|
|
|
class PeerStoreResolver:
|
|
"""
|
|
Determines the peers data file path using values from the config
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
root_path: Path,
|
|
config: Dict[str, Any],
|
|
*,
|
|
selected_network: str,
|
|
peers_file_path_key: str, # config key for the peers data file relative path
|
|
legacy_peer_db_path_key: str, # config key for the deprecated peer db path
|
|
default_peers_file_path: str, # default value for the peers data file relative path
|
|
):
|
|
self.root_path = root_path
|
|
self.config = config
|
|
self.selected_network = selected_network
|
|
self.peers_file_path_key = peers_file_path_key
|
|
self.legacy_peer_db_path_key = legacy_peer_db_path_key
|
|
self.default_peers_file_path = default_peers_file_path
|
|
|
|
def _resolve_and_update_config(self) -> Path:
|
|
"""
|
|
Resolve the peers data file path from the config, and update the config if necessary.
|
|
We leave the legacy peer db path in the config to support downgrading.
|
|
|
|
If peers_file_path_key is not found in the config, we'll attempt to derive the path
|
|
from the the config's legacy_peer_db_path_key value.
|
|
"""
|
|
peers_file_path_str: Optional[str] = self.config.get(self.peers_file_path_key)
|
|
if peers_file_path_str is None:
|
|
# Check if the legacy peer db path exists and derive a new path from it
|
|
peer_db_path: Optional[str] = self.config.get(self.legacy_peer_db_path_key)
|
|
if peer_db_path is not None:
|
|
# Use the legacy path's directory with the new peers data filename
|
|
peers_file_path_str = os.fspath(Path(peer_db_path).parent / self._peers_file_name)
|
|
else:
|
|
# Neither value is present in the config, use the default
|
|
peers_file_path_str = os.fspath(Path(self.default_peers_file_path).parent / self._peers_file_name)
|
|
|
|
# Update the config
|
|
self.config[self.peers_file_path_key] = peers_file_path_str
|
|
return self.root_path / Path(peers_file_path_str)
|
|
|
|
@property
|
|
def _peers_file_name(self) -> str:
|
|
"""
|
|
Internal property to get the name component of the peers data file path
|
|
"""
|
|
if self.selected_network == "mainnet":
|
|
return Path(self.default_peers_file_path).name
|
|
else:
|
|
# For testnets, we include the network name in the peers data filename
|
|
path = Path(self.default_peers_file_path)
|
|
return path.with_name(f"{path.stem}_{self.selected_network}{path.suffix}").name
|
|
|
|
@property
|
|
def peers_file_path(self) -> Path:
|
|
"""
|
|
Path to the peers data file, resolved using data from the config
|
|
"""
|
|
return self._resolve_and_update_config()
|
|
|
|
@property
|
|
def legacy_peer_db_path(self) -> Optional[Path]:
|
|
"""
|
|
Path to the legacy peer db file, resolved using data from the config. The legacy
|
|
peer db is only used for migration to the new format. We're only concerned about
|
|
migrating mainnet users, so we purposefully omit the testnet filename change.
|
|
"""
|
|
peer_db_path: Optional[str] = self.config.get(self.legacy_peer_db_path_key)
|
|
if peer_db_path is not None:
|
|
return self.root_path / Path(peer_db_path)
|
|
return None
|