mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-19 18:14:19 +01:00
109 lines
4.0 KiB
Python
109 lines
4.0 KiB
Python
from typing import Any, Dict, List, Union
|
|
|
|
from ..core.base import Method
|
|
from ..core.models import (
|
|
MeltMethodSetting,
|
|
MintMethodSetting,
|
|
)
|
|
from ..core.nuts import (
|
|
DLEQ_NUT,
|
|
FEE_RETURN_NUT,
|
|
HTLC_NUT,
|
|
MELT_NUT,
|
|
MINT_NUT,
|
|
MPP_NUT,
|
|
P2PK_NUT,
|
|
RESTORE_NUT,
|
|
SCRIPT_NUT,
|
|
STATE_NUT,
|
|
WEBSOCKETS_NUT,
|
|
)
|
|
from ..core.settings import settings
|
|
from ..mint.protocols import SupportsBackends
|
|
|
|
|
|
class LedgerFeatures(SupportsBackends):
|
|
def mint_features(self) -> Dict[int, Union[List[Any], Dict[str, Any]]]:
|
|
mint_method_settings: List[MintMethodSetting] = []
|
|
for method, unit_dict in self.backends.items():
|
|
for unit in unit_dict.keys():
|
|
mint_setting = MintMethodSetting(method=method.name, unit=unit.name)
|
|
if settings.mint_max_peg_in:
|
|
mint_setting.max_amount = settings.mint_max_peg_in
|
|
mint_setting.min_amount = 0
|
|
mint_method_settings.append(mint_setting)
|
|
mint_setting.description = unit_dict[unit].supports_description
|
|
melt_method_settings: List[MeltMethodSetting] = []
|
|
for method, unit_dict in self.backends.items():
|
|
for unit in unit_dict.keys():
|
|
melt_setting = MeltMethodSetting(method=method.name, unit=unit.name)
|
|
if settings.mint_max_peg_out:
|
|
melt_setting.max_amount = settings.mint_max_peg_out
|
|
melt_setting.min_amount = 0
|
|
melt_method_settings.append(melt_setting)
|
|
|
|
supported_dict = dict(supported=True)
|
|
|
|
mint_features: Dict[int, Union[List[Any], Dict[str, Any]]] = {
|
|
MINT_NUT: dict(
|
|
methods=mint_method_settings,
|
|
disabled=settings.mint_peg_out_only,
|
|
),
|
|
MELT_NUT: dict(
|
|
methods=melt_method_settings,
|
|
disabled=False,
|
|
),
|
|
STATE_NUT: supported_dict,
|
|
FEE_RETURN_NUT: supported_dict,
|
|
RESTORE_NUT: supported_dict,
|
|
SCRIPT_NUT: supported_dict,
|
|
P2PK_NUT: supported_dict,
|
|
DLEQ_NUT: supported_dict,
|
|
HTLC_NUT: supported_dict,
|
|
}
|
|
|
|
# signal which method-unit pairs support MPP
|
|
mpp_features = []
|
|
for method, unit_dict in self.backends.items():
|
|
for unit in unit_dict.keys():
|
|
if unit_dict[unit].supports_mpp:
|
|
mpp_features.append(
|
|
{
|
|
"method": method.name,
|
|
"unit": unit.name,
|
|
"mpp": True,
|
|
}
|
|
)
|
|
|
|
if mpp_features:
|
|
mint_features[MPP_NUT] = dict(methods=mpp_features)
|
|
|
|
# specify which websocket features are supported
|
|
# these two are supported by default
|
|
websocket_features: Dict[str, List[Dict[str, Union[str, List[str]]]]] = {
|
|
"supported": []
|
|
}
|
|
# we check the backend to see if "bolt11_mint_quote" is supported as well
|
|
for method, unit_dict in self.backends.items():
|
|
if method == Method["bolt11"]:
|
|
for unit in unit_dict.keys():
|
|
websocket_features["supported"].append(
|
|
{
|
|
"method": method.name,
|
|
"unit": unit.name,
|
|
"commands": ["bolt11_melt_quote", "proof_state"],
|
|
}
|
|
)
|
|
if unit_dict[unit].supports_incoming_payment_stream:
|
|
supported_features: List[str] = list(
|
|
websocket_features["supported"][-1]["commands"]
|
|
)
|
|
websocket_features["supported"][-1]["commands"] = (
|
|
supported_features + ["bolt11_mint_quote"]
|
|
)
|
|
|
|
if websocket_features:
|
|
mint_features[WEBSOCKETS_NUT] = websocket_features
|
|
|
|
return mint_features
|