mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-09-20 08:05:33 +03:00
e5b94d34a0
* Rename chialisp file extensions * Add pre-commit check for .clvm files * Delete sha256tree files * remove hash files in favor of central dictionary * Add check for missing files like hex and hash entries * Enhance clsp pre-commit check * Actually check hash matches * Update tools/manage_clvm.py Co-authored-by: Kyle Altendorf <sda@fstab.net> * Update tools/manage_clvm.py Co-authored-by: Kyle Altendorf <sda@fstab.net> * Fix Windows file writing * Fix setup.py package_data fields * Load hash dict at runtime * Move away from exception pattern * Bad equality check * Minor fixes * remove trailing whitespace fix --------- Co-authored-by: Kyle Altendorf <sda@fstab.net>
65 lines
2.4 KiB
Python
65 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
|
|
from chia.types.blockchain_format.program import Program
|
|
from chia.types.blockchain_format.sized_bytes import bytes32
|
|
from chia.wallet.puzzles.load_clvm import load_clvm_maybe_recompile
|
|
from chia.wallet.util.curry_and_treehash import calculate_hash_of_quoted_mod_hash, curry_and_treehash
|
|
|
|
SINGLETON_TOP_LAYER_MOD = load_clvm_maybe_recompile("singleton_top_layer_v1_1.clsp")
|
|
SINGLETON_TOP_LAYER_MOD_HASH = SINGLETON_TOP_LAYER_MOD.get_tree_hash()
|
|
SINGLETON_TOP_LAYER_MOD_HASH_QUOTED = calculate_hash_of_quoted_mod_hash(SINGLETON_TOP_LAYER_MOD_HASH)
|
|
SINGLETON_LAUNCHER_PUZZLE = load_clvm_maybe_recompile("singleton_launcher.clsp")
|
|
SINGLETON_LAUNCHER_PUZZLE_HASH = SINGLETON_LAUNCHER_PUZZLE.get_tree_hash()
|
|
|
|
|
|
def get_inner_puzzle_from_singleton(puzzle: Program) -> Optional[Program]:
|
|
"""
|
|
Extract the inner puzzle of a singleton
|
|
:param puzzle: Singleton puzzle
|
|
:return: Inner puzzle
|
|
"""
|
|
r = puzzle.uncurry()
|
|
if r is None:
|
|
return None
|
|
inner_f, args = r
|
|
if not is_singleton(inner_f):
|
|
return None
|
|
SINGLETON_STRUCT, INNER_PUZZLE = list(args.as_iter())
|
|
return Program(INNER_PUZZLE)
|
|
|
|
|
|
def is_singleton(inner_f: Program) -> bool:
|
|
"""
|
|
Check if a puzzle is a singleton mod
|
|
:param inner_f: puzzle
|
|
:return: Boolean
|
|
"""
|
|
return inner_f == SINGLETON_TOP_LAYER_MOD
|
|
|
|
|
|
def create_singleton_puzzle_hash(innerpuz_hash: bytes32, launcher_id: bytes32) -> bytes32:
|
|
"""
|
|
Return Hash ID of the whole Singleton Puzzle
|
|
:param innerpuz_hash: Singleton inner puzzle tree hash
|
|
:param launcher_id: launcher coin name
|
|
:return: Singleton full puzzle hash
|
|
"""
|
|
# singleton_struct = (MOD_HASH . (LAUNCHER_ID . LAUNCHER_PUZZLE_HASH))
|
|
singleton_struct = Program.to((SINGLETON_TOP_LAYER_MOD_HASH, (launcher_id, SINGLETON_LAUNCHER_PUZZLE_HASH)))
|
|
|
|
return curry_and_treehash(SINGLETON_TOP_LAYER_MOD_HASH_QUOTED, singleton_struct.get_tree_hash(), innerpuz_hash)
|
|
|
|
|
|
def create_singleton_puzzle(innerpuz: Program, launcher_id: bytes32) -> Program:
|
|
"""
|
|
Create a full Singleton puzzle
|
|
:param innerpuz: Singleton inner puzzle
|
|
:param launcher_id:
|
|
:return: Singleton full puzzle
|
|
"""
|
|
# singleton_struct = (MOD_HASH . (LAUNCHER_ID . LAUNCHER_PUZZLE_HASH))
|
|
singleton_struct = Program.to((SINGLETON_TOP_LAYER_MOD_HASH, (launcher_id, SINGLETON_LAUNCHER_PUZZLE_HASH)))
|
|
return SINGLETON_TOP_LAYER_MOD.curry(singleton_struct, innerpuz)
|