old_file_plz_ignore.py

import os
import sys
import asyncio
from pathlib import Path
from typing import List, Tuple
from chia.wallet.sign_coin_spends import sign_coin_spends
from blspy import PrivateKey, AugSchemeMPL
from chia.types.blockchain_format.coin import Coin
from chia.types.blockchain_format.program import Program
from chia.types.blockchain_format.sized_bytes import bytes32
from chia.types.condition_opcodes import ConditionOpcode
from chia.util.hash import std_hash
from chia.util.ints import uint64
from clvm.casts import int_to_bytes
from cdv.cmds.rpc import get_client
from chia.wallet.puzzles.singleton_top_layer_v1_1 import launch_conditions_and_coinsol, pay_to_singleton_puzzle, SINGLETON_MOD_HASH, SINGLETON_MOD, P2_SINGLETON_MOD, SINGLETON_LAUNCHER_HASH, SINGLETON_LAUNCHER, lineage_proof_for_coinsol, puzzle_for_singleton, solution_for_singleton, generate_launcher_coin
from chia.wallet.puzzles.p2_delegated_puzzle_or_hidden_puzzle import puzzle_for_pk, calculate_synthetic_secret_key, DEFAULT_HIDDEN_PUZZLE_HASH, puzzle_for_synthetic_public_key, solution_for_delegated_puzzle
from chia.wallet.trading.offer import OFFER_MOD_HASH, OFFER_MOD
from chia.rpc.full_node_rpc_client import FullNodeRpcClient
from chia.simulator.simulator_full_node_rpc_client import SimulatorFullNodeRpcClient
from chia.util.config import load_config
from chia.util.ints import uint16, uint32
from chia.wallet.derive_keys import master_sk_to_wallet_sk_unhardened
from chia.consensus.default_constants import DEFAULT_CONSTANTS
from chia.types.coin_spend import CoinSpend
from chia.wallet.cat_wallet.cat_utils import construct_cat_puzzle
from chia.wallet.puzzles.tails import GenesisById
from chia.wallet.cat_wallet.cat_utils import CAT_MOD
from chia.wallet.cat_wallet.cat_wallet import CAT_MOD_HASH
from chia_rs import run_chia_program
from chia.types.blockchain_format.program import INFINITE_COST, Program
from chia.util.condition_tools import conditions_dict_for_solution
from chia.wallet.lineage_proof import LineageProof
from chia.wallet.cat_wallet.cat_utils import (
    SpendableCAT,
    construct_cat_puzzle,
    unsigned_spend_bundle_for_spendable_cats,
)
from chia.wallet.trading.offer import Offer
from chia.util.bech32m import decode_puzzle_hash, encode_puzzle_hash
from chia.types.spend_bundle import SpendBundle
from chia.util.bech32m import bech32_decode, bech32_encode, convertbits
from chia.wallet.util.puzzle_compression import (
    compress_object_with_puzzles,
    decompress_object_with_puzzles,
    lowest_best_version,
)
from chia.wallet.puzzles.p2_conditions import puzzle_for_conditions
from chia.util.hash import std_hash


def load_clvm_hex(
    filename
) -> SerializedProgram:
    clvm_hex = open(filename, "r").read().strip()
    assert len(clvm_hex) != 0
    
    clvm_blob = bytes.fromhex(clvm_hex)
    return SerializedProgram.from_bytes(clvm_blob)

ROUTER_MOD: Program = load_clvm_hex("clvm/router.clvm.hex")
PAIR_MOD: Program = load_clvm_hex("clvm/pair.clvm.hex")
LIQUIDITY_TAIL_MOD: Program = load_clvm_hex("clvm/liquidity_tail.clvm.hex")

ROUTER_MOD_HASH = ROUTER_MOD.get_tree_hash()
PAIR_MOD_HASH = PAIR_MOD.get_tree_hash()
LIQUIDITY_TAIL_MOD_HASH = LIQUIDITY_TAIL_MOD.get_tree_hash()

P2_SINGLETON_MOD_HASH = P2_SINGLETON_MOD.get_tree_hash()

def get_router_puzzle(pairs):
    return ROUTER_MOD.curry(
        PAIR_MOD_HASH,
        SINGLETON_MOD_HASH,
        P2_SINGLETON_MOD_HASH,
        CAT_MOD_HASH,
        LIQUIDITY_TAIL_MOD_HASH,
        OFFER_MOD_HASH,
        997,
        SINGLETON_LAUNCHER_HASH,
        ROUTER_MOD_HASH,
        Program.to(pairs)
    )

def get_pair_inner_puzzle(singleton_launcher_id, tail_hash, liquidity, xch_reserve, token_reserve):
    return PAIR_MOD.curry(
        PAIR_MOD_HASH,
        (SINGLETON_MOD_HASH, (singleton_launcher_id, SINGLETON_LAUNCHER_HASH)),
        P2_SINGLETON_MOD_HASH,
        CAT_MOD_HASH,
        LIQUIDITY_TAIL_MOD_HASH,
        OFFER_MOD_HASH,
        tail_hash,
        997,
        liquidity,
        xch_reserve,
        token_reserve
    )

def get_pair_puzzle(singleton_launcher_id, tail_hash, liquidity, xch_reserve, token_reserve):
    return puzzle_for_singleton(
        singleton_launcher_id,
        get_pair_inner_puzzle(singleton_launcher_id, tail_hash, liquidity, xch_reserve, token_reserve)
    )

def pair_liquidity_tail_puzzle(pair_launcher_id):
    return LIQUIDITY_TAIL_MOD.curry(
        (SINGLETON_MOD_HASH, (pair_launcher_id, SINGLETON_LAUNCHER_HASH))
    )

def deploy_router_conditions_and_coinspend(parent_coin):
    comment: List[Tuple[str, str]] = [("tibet", "v1")]
    return launch_conditions_and_coinsol(parent_coin, get_router_puzzle(0), comment, 1)


async def get_full_node_client() -> FullNodeRpcClient:
    try:
        client: FullNodeRpcClient = await get_client()
        await client.get_blockchain_state()
        return client
    except:
        client.close()
        await client.await_closed()
        pass
    
    root_path = "/home/yakuhito/.chia/simulator/main"
    config = load_config(root_path, "config.yaml")
    self_hostname = config["self_hostname"]
    rpc_port = config["full_node"]["rpc_port"]
    node_client: SimulatorFullNodeRpcClient = await SimulatorFullNodeRpcClient.create(
        self_hostname, uint16(rpc_port), root_path, config
    )
    await node_client.get_blockchain_state()

    return node_client

async def sign_std_coin_spends(coin_spends, synth_sk):
    async def pk_to_sk(pk):
       return synth_sk

    return await sign_coin_spends(
        coin_spends,
        pk_to_sk,
        DEFAULT_CONSTANTS.AGG_SIG_ME_ADDITIONAL_DATA,
        DEFAULT_CONSTANTS.MAX_BLOCK_COST_CLVM,
    )

    
async def launch_router_with_sk(parent_coin, synth_secret_key):
    conds, launcher_coin_spend = deploy_router_conditions_and_coinspend(parent_coin)
    if parent_coin.amount > 1:
        conds.append(Program.to(
            [
                ConditionOpcode.CREATE_COIN,
                parent_coin.puzzle_hash,
                parent_coin.amount - 1,
            ],
        ))

    p2_coin_spend = CoinSpend(
        parent_coin,
        puzzle_for_synthetic_public_key(synth_secret_key.get_g1()),
        solution_for_delegated_puzzle(Program.to((1, conds)), [])
    )
        
    sb = await sign_std_coin_spends(
        [launcher_coin_spend, p2_coin_spend],
        synth_secret_key
    )

    launcher_id = launcher_coin_spend.coin.name().hex()
    
    return launcher_id, sb

def set_router_launcher_id(launcher_id):
    open("router_launcher_id.txt", "w").write(launcher_id)

def get_router_launcher_id():
    try:
        return open("router_launcher_id.txt", "r").read().strip()
    except:
        return ""

async def select_std_coin(client, master_sk, min_amount):
    ph_to_wallet_sk = {}
    for i in range(100):
        wallet_sk = master_sk_to_wallet_sk_unhardened(master_sk, i)
        synth_secret_key = calculate_synthetic_secret_key(wallet_sk, DEFAULT_HIDDEN_PUZZLE_HASH)
        synth_key = synth_secret_key.get_g1()
        puzzle = puzzle_for_synthetic_public_key(synth_key)
        puzzle_hash = puzzle.get_tree_hash()            
        ph_to_wallet_sk[puzzle_hash] = synth_secret_key

    res = await client.get_coin_records_by_puzzle_hashes(ph_to_wallet_sk.keys(), include_spent_coins=False)
    if len(res) == 0:
        if isinstance(client, SimulatorFullNodeRpcClient):
            await client.farm_block([_ for _ in ph_to_wallet_sk.keys()][0])
            return select_std_coin(client, master_sk, min_amount)
        else:
            print("No coins at the given address :(")
            sys.exit(1)
    
    for coin_record in res:
        if coin_record.coin.amount < min_amount:
            continue

        coin = coin_record.coin
        synth_secret_key = ph_to_wallet_sk[coin.puzzle_hash]

        return coin, synth_secret_key

    print("No coins big enough at the given address :(")

async def launch_router():
    client = await get_full_node_client()
    master_sk_hex = ""
    try:
        master_sk_hex = open("master_private_key.txt", "r").read().strip()
    except:
        master_sk_hex = input("Master Private Key: ")

    master_sk = PrivateKey.from_bytes(bytes.fromhex(master_sk_hex))
    parent_coin, synth_secret_key = await select_std_coin(client, master_sk, 3)
        
    launcher_id, sb = await launch_router_with_sk(parent_coin, synth_secret_key)

    print(f"Router launcher coin id: {launcher_id}")
    print("Spend bundle: ", sb)
    check = input("Type 'liftoff' to broadcast tx: ")

    if check.strip() == 'liftoff':
        resp = await client.push_tx(sb)
        print(resp)
        set_router_launcher_id(launcher_id)
        print("Router launcher id saved.")
    else:
        print("that's another word *-*")

    client.close()
    await client.await_closed()

async def create_test_cat(coin, synth_secret_key):
    TOKEN_AMOUNT = 1000000
    coin_id = coin.name()

    tail = GenesisById.construct([coin_id])
    tail_hash = tail.get_tree_hash()
    cat_inner_puzzle = puzzle_for_synthetic_public_key(synth_secret_key.get_g1())

    cat_puzzle = construct_cat_puzzle(CAT_MOD, tail_hash, cat_inner_puzzle)
    cat_puzzle_hash = cat_puzzle.get_tree_hash()

    cat_creation_tx = CoinSpend(
        coin,
        cat_inner_puzzle, # same as this coin's puzzle
        solution_for_delegated_puzzle(Program.to((1, [
            [ConditionOpcode.CREATE_COIN, cat_puzzle_hash, TOKEN_AMOUNT * 10000],
            [ConditionOpcode.CREATE_COIN, coin.puzzle_hash, coin.amount - TOKEN_AMOUNT * 10000],
        ])), [])
    )
    
    cat_coin = Coin(
        coin.name(), # parent
        cat_puzzle_hash,
        TOKEN_AMOUNT * 10000
    )

    cat_inner_solution = solution_for_delegated_puzzle(
        Program.to((1, [
            [ConditionOpcode.CREATE_COIN, 0, -113, tail, []],
            [ConditionOpcode.CREATE_COIN, cat_inner_puzzle.get_tree_hash(), cat_coin.amount]
        ])), []
    )

    cat_eve_spend_bundle = unsigned_spend_bundle_for_spendable_cats(
        CAT_MOD,
        [
            SpendableCAT(
                cat_coin,
                tail_hash,
                cat_inner_puzzle,
                cat_inner_solution,
                limitations_program_reveal=tail,
            )
        ],
    )
    cat_eve_spends = cat_eve_spend_bundle.coin_spends
    
    sb = await sign_std_coin_spends([cat_creation_tx] + cat_eve_spends, synth_secret_key)
    return tail_hash.hex(), sb

async def launch_test_token_cmd():
    client = await get_full_node_client()
    master_sk_hex = ""
    try:
        master_sk_hex = open("master_private_key.txt", "r").read().strip()
    except:
        master_sk_hex = input("Master Private Key: ")

    master_sk = PrivateKey.from_bytes(bytes.fromhex(master_sk_hex))
    coin, synth_secret_key = await select_std_coin(client, master_sk, 1000000 * 10000)
    
    tail_hash, sb = await create_test_cat(coin, synth_secret_key)
    print(f"New TAIL: {tail_hash}")

    print("Spend bundle: ", sb)
    check = input("Type 'liftoff' to broadcast tx: ")
    if check.strip() == 'liftoff':
        resp = await client.push_tx(sb)
        print(resp)
    else:
        print("that's another word *-*")

    client.close()
    await client.await_closed()

unspent_singletons = {} # map launcher_id (hex string) -> [coin, creation_spend]
async def get_unspent_singleton_info(client, launcher_id):
    coin, creation_spend = unspent_singletons.get(launcher_id, [None, None])
    if coin == None:
        res = await client.get_coin_record_by_name(bytes.fromhex(launcher_id))
        coin = res.coin
    res = await client.get_coin_record_by_name(coin.name())

    while res.spent:
        coin_spend = await client.get_puzzle_and_solution(coin.name(), res.spent_block_index)
        conditions_dict = conditions_dict_for_solution(coin_spend.puzzle_reveal, coin_spend.solution, INFINITE_COST)
        for cwa in conditions_dict[ConditionOpcode.CREATE_COIN]: #cwa = condition with args
            if len(cwa.vars) >= 2 and cwa.vars[1] == b"\x01":
                creation_spend = coin_spend
                coin = Coin(
                    coin.name(),
                    cwa.vars[0],
                    1
                )
                break
        res = await client.get_coin_record_by_name(coin.name())

    unspent_singletons[launcher_id] = [res.coin, creation_spend]
    return unspent_singletons[launcher_id]

def get_pairs(last_router_coin_spend):
    pairs = []
    if last_router_coin_spend.coin.puzzle_hash != SINGLETON_LAUNCHER_HASH:
        creation_inner_puzzle_hash = Program.from_bytes(bytes(last_router_coin_spend.puzzle_reveal)).uncurry()[1]
        pair_launcher_id = Coin(last_router_coin_spend.coin.name(), SINGLETON_LAUNCHER_HASH, 2).name()

        arr = Program.from_bytes(bytes(last_router_coin_spend.solution)).as_python()[-1]
        pairs = Program.from_bytes(bytes([_ for _ in creation_inner_puzzle_hash.as_iter()][-1])).uncurry()[-1]
        pairs = [_ for _ in pairs.as_iter()][-1].as_python()
        if len(pairs) == 0:
            pairs = []
        pairs.append((arr[1], pair_launcher_id))
    return pairs

async def create_pair(client, coin, synth_secret_key, router_launcher_id, tail_hash):
    current_router_coin, creation_spend = await get_unspent_singleton_info(client, router_launcher_id)
    
    lineage_proof = lineage_proof_for_coinsol(creation_spend)
    pairs = get_pairs(creation_spend)
        
    router_inner_puzzle = get_router_puzzle(pairs)
    router_inner_solution = Program.to([
        current_router_coin.name(),
        bytes.fromhex(tail_hash)
    ])

    router_singleton_puzzle = puzzle_for_singleton(bytes.fromhex(router_launcher_id), router_inner_puzzle)
    router_singleton_solution = solution_for_singleton(lineage_proof, current_router_coin.amount, router_inner_solution)
    router_singleton_spend = CoinSpend(current_router_coin, router_singleton_puzzle, router_singleton_solution)

    pair_launcher_coin = Coin(current_router_coin.name(), SINGLETON_LAUNCHER_HASH, 2)
    pair_puzzle = get_pair_puzzle(
        pair_launcher_coin.name(),
        bytes.fromhex(tail_hash),
        0, 0, 0
    )

    comment: List[Tuple[str, str]] = []

    # launch_conditions_and_coinsol would not work here since we spend a coin with amount 2
    # and the solution says the amount is 1 *-*
    pair_launcher_solution = Program.to(
        [
            pair_puzzle.get_tree_hash(),
            1,
            comment,
        ]
    )
    assert_launcher_announcement = [
        ConditionOpcode.ASSERT_COIN_ANNOUNCEMENT,
        std_hash(pair_launcher_coin.name() + pair_launcher_solution.get_tree_hash()),
    ]
    pair_launcher_spend = CoinSpend(
        pair_launcher_coin,
        SINGLETON_LAUNCHER,
        pair_launcher_solution,
    )

    # first condition is CREATE_COIN, which we took care of
    # important to note: router also takes care of assert_launcher_announcement, but we need it to link the fund spend to the bundle

    fund_spend = CoinSpend(
        coin,
        puzzle_for_synthetic_public_key(synth_secret_key.get_g1()),
        solution_for_delegated_puzzle(Program.to((1, [
            [ConditionOpcode.CREATE_COIN, coin.puzzle_hash, coin.amount - 2],
            assert_launcher_announcement
        ])), [])
    )
    
    pair_launcher_id = Coin(current_router_coin.name(), SINGLETON_LAUNCHER_HASH, 2).name().hex()
    sb = await sign_std_coin_spends([router_singleton_spend, pair_launcher_spend, fund_spend], synth_secret_key)
    return pair_launcher_id, sb

async def create_pair_cmd(tail_hash):
    client = await get_full_node_client()
    master_sk_hex = ""
    try:
        master_sk_hex = open("master_private_key.txt", "r").read().strip()
    except:
        master_sk_hex = input("Master Private Key: ")
    master_sk = PrivateKey.from_bytes(bytes.fromhex(master_sk_hex))
    router_launcher_id = get_router_launcher_id()
    if router_launcher_id == "":
        print("Please set the router launcher id first.")
        return

    coin, synth_secret_key = await select_std_coin(client, master_sk, 2)
    pair_launcher_id, sb = await create_pair(client, coin, synth_secret_key, router_launcher_id, tail_hash)
    print(f"Pair launcher id: {pair_launcher_id}")

    print("Pair ready to be created.")
    check = input("Type 'magic' to broadcast tx: ")

    if check.strip() == 'magic':
        resp = await client.push_tx(sb)
        print(resp)
    else:
        print("that's another word *-*")

    client.close()
    await client.await_closed()
    
def pair_initial_liquidity_inner_solution(pair_coin_id, token_amount, xch_amount, liquidity_inner_puzzle_hash, liquidity_parent_id):
    return Program.to([
        Program.to([pair_coin_id, b"\x00" * 32, b"\x00" * 32]),
        0,
        Program.to([token_amount, liquidity_inner_puzzle_hash, liquidity_parent_id, xch_amount])
    ])

async def add_liquidity_offer_initial(client, pair_id, tail_hash, token_amount, xch_amount, initial_offer_str, current_pair_coin=None, last_singleton_spend=None):
    if current_pair_coin is None or last_singleton_spend == None:
        current_pair_coin, last_singleton_spend = await get_unspent_singleton_info(client, pair_id)
    
    # how does a deposit offer look like? well, I'm glad you asked
    # ephemeral = will appear in the 'offered' section of the offer
    #        ---------------------COIN ANNOUNCEMENT--|
    #        |                                      \/                    
    # user offer -> initial offer coin -> intermediary liquidity cat -> ephemeral liquidity cat
    #                                              /\ \/
    #                                           pair singleton
    #                                           \/          \/
    #                                 new xch reserve       new token reserve
    # thank god I have a whiteboard

    # 1. identify initial offer coin
    initial_offer = Offer.from_bech32(initial_offer_str)
    initial_offer_coin = None
    for addition in initial_offer.additions():
        if addition.puzzle_hash == OFFER_MOD_HASH:
            initial_offer_coin = addition
            break
    if initial_offer_coin is None:
        print("sth is wrong with the offer :(")
        sys.exit(1)

    # 2. identify announcements to assert
    announcement_asserts = []
    intial_spend_bundle = initial_offer.to_spend_bundle()
    for coin_spend in intial_spend_bundle.coin_spends:
        if coin_spend.coin.parent_coin_info == b"\x00" * 32:
            continue

        _, conditions_dict, __ = conditions_dict_for_solution(coin_spend.puzzle_reveal, coin_spend.solution, INFINITE_COST)
        # std coins create coin announcements that need to be asserted
        for cwa in conditions_dict.get(ConditionOpcode.CREATE_COIN_ANNOUNCEMENT, []): #cwa = condition with args
            announcement_asserts.append([
                ConditionOpcode.ASSERT_COIN_ANNOUNCEMENT,
                std_hash(coin_spend.coin.name() + cwa.vars[0])
            ])

    # 3. create intermediary spend
    pair_inner_puzzle = get_pair_inner_puzzle(bytes.fromhex(pair_id), bytes.fromhex(tail_hash), 0, 0, 0)
    
    liquidity_cat_tail = pair_liquidity_tail_puzzle(bytes.fromhex(pair_id))
    intermediary_liquidity_cat_tail_solution = Program.to([pair_inner_puzzle.get_tree_hash()])
    intermediary_liquidity_cat_inner_puzzle = Program.to(
        (1, [
            [ConditionOpcode.CREATE_COIN, 0, -113, liquidity_cat_tail, intermediary_liquidity_cat_tail_solution],
            [ConditionOpcode.CREATE_COIN, OFFER_MOD_HASH, token_amount]
        ] + announcement_asserts)
    )
    
    intermediary_liquidity_cat_inner_solution = Program.to([])
    intermediary_liquidity_cat_puzzle = construct_cat_puzzle(CAT_MOD, liquidity_cat_tail.get_tree_hash(), intermediary_liquidity_cat_inner_puzzle)
    intermediary_liquidity_cat = Coin(initial_offer_coin.name(), intermediary_liquidity_cat_puzzle.get_tree_hash(), token_amount)
    
    intermediary_liquidity_cat_solution = Program.to([
        intermediary_liquidity_cat_inner_solution,
        [],
        intermediary_liquidity_cat.name(),
        [initial_offer_coin.name(), intermediary_liquidity_cat_puzzle.get_tree_hash(), token_amount],
        [initial_offer_coin.name(), intermediary_liquidity_cat_inner_puzzle.get_tree_hash(), token_amount],
        [],
        []
    ])
    intermediary_liquidity_cat_spend = CoinSpend(intermediary_liquidity_cat, intermediary_liquidity_cat_puzzle, intermediary_liquidity_cat_solution)

    # debug
    open("/tmp/p", "w").write(bytes(intermediary_liquidity_cat_spend.puzzle_reveal).hex())
    open("/tmp/s", "w").write(bytes(intermediary_liquidity_cat_spend.solution).hex())
    print("coin name: ", intermediary_liquidity_cat_spend.coin.name().hex())
    os.system("brun -x $(cat /tmp/p) $(cat /tmp/s)")
    a = b'00' # input("announcement hash: ").strip().encode()
    print(a)
    print("final annoucnement assert: ", std_hash(intermediary_liquidity_cat_spend.coin.name() + a).hex())
    print("singleton inner puzzle hash: ", pair_inner_puzzle.get_tree_hash().hex())
    # input("SLEEP")
    # debug

    # 4. spend the initial offer coin
    initial_offer_coin_solution = Program.to([
        Program.to([
            initial_offer_coin.name(),
            [intermediary_liquidity_cat_puzzle.get_tree_hash(), token_amount],
            [OFFER_MOD_HASH, initial_offer_coin.amount - token_amount] # really interesting stuff going on here
            # we're basically re-offering the change (giving it back)
        ])
    ])
    initial_offer_coin_spend = CoinSpend(initial_offer_coin, OFFER_MOD, initial_offer_coin_solution)

    # 5. create ephemereal cat spend
    # take nonce from given offer

    # debug
    # REMOVE
    a = None
    for k in initial_offer.get_requested_payments().keys():
        a = k
    notarized_payment = initial_offer.get_requested_payments()[a][0]

    # notarized_payment = initial_offer.get_requested_payments()[liquidity_cat_tail.get_tree_hash()][0]
    eph_cat_nonce = notarized_payment.nonce
    eph_cat_memos = notarized_payment.memos

    eph_cat_puzzle = construct_cat_puzzle(CAT_MOD, liquidity_cat_tail.get_tree_hash(), OFFER_MOD)
    eph_cat = Coin(intermediary_liquidity_cat.name(), eph_cat_puzzle.get_tree_hash(), token_amount)
    eph_cat_inner_solution = Program.to([
        Program.to([
            eph_cat_nonce,
            [eph_cat_memos[0], token_amount, eph_cat_memos]
        ])
    ])

    eph_cat_spend_bundle = unsigned_spend_bundle_for_spendable_cats(
        CAT_MOD,
        [
            SpendableCAT(
                eph_cat,
                liquidity_cat_tail.get_tree_hash(),
                OFFER_MOD,
                eph_cat_inner_solution,
                lineage_proof=LineageProof(intermediary_liquidity_cat.parent_coin_info, intermediary_liquidity_cat_inner_puzzle.get_tree_hash(), token_amount)
            )
        ],
    )
    eph_cat_spend = eph_cat_spend_bundle.coin_spends[0]

    # 4. Spend pair / singleton
    p2_singleton_puzzle = pay_to_singleton_puzzle(bytes.fromhex(pair_id))
    p2_singleton_puzzle_cat =  construct_cat_puzzle(CAT_MOD, bytes.fromhex(tail_hash), p2_singleton_puzzle)
    
    pair_puzzle = get_pair_puzzle(bytes.fromhex(pair_id), bytes.fromhex(tail_hash), 0, 0, 0)
    
    pair_inner_solution = pair_initial_liquidity_inner_solution(
        current_pair_coin.name(),
        token_amount,
        xch_amount,
        intermediary_liquidity_cat_inner_puzzle.get_tree_hash(),
        intermediary_liquidity_cat.parent_coin_info
    )
    lineage_proof = lineage_proof_for_coinsol(last_singleton_spend)
    pair_solution = solution_for_singleton(lineage_proof, current_pair_coin.amount, pair_inner_solution)
    
    print("pair puzzle hash: ", pair_puzzle.get_tree_hash().hex())
    singleton_spend = CoinSpend(current_pair_coin, pair_puzzle, pair_solution)

    # 5. create ephemeral XCH reserve coin
    eph_xch_reserve_coin = Coin(
        b"\x00" * 32,
        OFFER_MOD_HASH,
        xch_amount
    )
    eph_xch_reserve_solution = Program.to([
        Program.to([
            current_pair_coin.name(),
            [p2_singleton_puzzle.get_tree_hash(), xch_amount]
        ])
    ])
    eph_xch_reserve_spend = CoinSpend(eph_xch_reserve_coin, OFFER_MOD, eph_xch_reserve_solution)

    # 6. create ephemeral token reserve coin
    eph_token_reserve_puzzle = construct_cat_puzzle(CAT_MOD, bytes.fromhex(tail_hash), OFFER_MOD)
    eph_token_reserve_coin = Coin(
        b"\x00" * 32,
        eph_token_reserve_puzzle.get_tree_hash(),
        token_amount
    )
    eph_token_reserve_inner_solution = Program.to([
        Program.to([
            current_pair_coin.name(),
            [p2_singleton_puzzle_cat.get_tree_hash(), token_amount]
        ])
    ])
    eph_token_reserve_spend = CoinSpend(
        eph_token_reserve_coin,
        eph_token_reserve_puzzle,
        eph_token_reserve_inner_solution
    )

    initial_offer_sb = initial_offer.to_spend_bundle()
    cs_from_initial_offer = [cs for cs in initial_offer_sb.coin_spends if cs.coin.parent_coin_info != b"\x00" * 32]

    sb = SpendBundle(
        [
            initial_offer_coin_spend,
            singleton_spend,
            intermediary_liquidity_cat_spend,
            eph_cat_spend,
            eph_xch_reserve_spend,
            eph_token_reserve_spend
        ] + cs_from_initial_offer,
        initial_offer_sb.aggregated_signature # AugSchemeMPL.aggregate([])
    )

    mods: List[bytes] = [bytes(s.puzzle_reveal.to_program().uncurry()[0]) for s in sb.coin_spends]
    
    # debug
    version = max(lowest_best_version(mods), 5)
    # version = max(lowest_best_version(mods), 6)
    offer_bytes = compress_object_with_puzzles(bytes(sb), version)
    offer_str = bech32_encode("offer:tibet:add_initial_liquidity", convertbits(list(offer_bytes), 8, 5))

    # offer = Offer.from_bech32(offer_str)
    # final_offer = Offer.aggregate([offer, initial_offer])
    # debug
    # final_offer = offer

    # debug
    # return final_offer.to_bech32("offer:tibet:add_initial_liquidity")
    return offer_str

async def add_liquidity_offer(client, router_launcher_id, pair_id, token_tail, token_amount, initial_offer, xch_amount=0):
    current_pair_coin, last_singleton_spend = await get_unspent_singleton_info(client, pair_id)
    # todo: get pair tail hash from last_singleton_spend or router if last spend is launcher
    if last_singleton_spend.coin.puzzle_hash == SINGLETON_LAUNCHER_HASH:
        return await add_liquidity_offer_initial(client, pair_id, token_tail, token_amount, xch_amount, initial_offer, current_pair_coin=current_pair_coin, last_singleton_spend=last_singleton_spend)

    # todo: normal offer

    return "offer1"

async def add_liquidity_cmd(offer, pair_launcher_id, token_tail_hash, token_amount, xch_amount):
    client = await get_full_node_client()
    router_launcher_id = get_router_launcher_id()
    # master_sk_hex = ""
    # try:
    #     master_sk_hex = open("master_private_key.txt", "r").read().strip()
    # except:
    #     master_sk_hex = input("Master Private Key: ")
    # master_sk = PrivateKey.from_bytes(bytes.fromhex(master_sk_hex))
    # if router_launcher_id == "":
    #     print("Please set the router launcher id first.")
    #     return

    # coin, synth_secret_key = await select_std_coin(client, master_sk, 2)
    offer = await add_liquidity_offer(client, router_launcher_id, pair_id, "da4c8b525434b5ee171221ae02aa74dd53fba70516d59eb5e9cee44c310e69c1", token_amount, offer, xch_amount=xch_amount)
    open("/tmp/deposit_offer", "w").write(offer) # debug
    print("Deposit offer written to /tmp/deposit_offer.") # debug

    client.close()
    await client.await_closed()

async def main():
    if len(sys.argv) < 2:
        print("Possible commands: launch_router, set_router, launch_test_token, create_pair, add_liquidity")
    elif sys.argv[1] == "launch_router":
        await launch_router()
    elif sys.argv[1] == "set_router":
        if len(sys.argv) == 3:
            set_router_launcher_id(sys.argv[2])
        else:
            print("Usage: set_router [launcher_id]")
    elif sys.argv[1] == "launch_test_token":
        await launch_test_token_cmd()
    elif sys.argv[1] == "create_pair":
        if len(sys.argv) == 3:
            await create_pair_cmd(sys.argv[2])
        else:
            print("Usage: create_pair [tail_hash_of_asset]")
    elif sys.argv[1] == "add_liquidity":
        if len(sys.argv) == 5 or len(sys.argv) == 6:
            xch_amount = 0
            if len(sys.argv) == 6:
                xch_amount = int(sys.argv[5])

            pair_launcher_id = bytes.fromhex(sys.argv[2])
            token_tail_hash = bytes.fromhex(sys.argv[3])
            liquidity_tail_hash = pair_liquidity_tail_puzzle(pair_launcher_id).get_tree_hash()

            print(f"Liquidity asset id: {liquidity_tail_hash.hex()}")
            print(f"Token asset id: {token_tail_hash.hex()}")
            print(f"Generating offer...")
            # chia wallet make_offer -o 10:2 -o 1:0.003 -r 11:1 -p /tmp/deposit_offer
            # debug
            offer = open("/tmp/deposit_offer", "r").read()
            await add_liquidity_cmd(offer, pair_launcher_id, token_tail_hash, int(sys.argv[4]), xch_amount)
        else:
            print("Usage: add_liquidity [pair_launcher_id] [token_asset_id] [amount_token] [amount_xch_if_first_deposit]")
    else:
        print("Unknown command.")

if __name__ == "__main__":
    asyncio.run(main())

# TAIL ID for simulator: d08398e38f8e20f0304a13f38d4f4cf4e488993e149149e91ac258d933af9274
# pair ID: c5deaf6f7acb36d0c661e1a91b08c9ac4b1f840d73d2954f69e1f035357774c5