"""Compute the first and last WOTS sk wots_sk_0 and wots_sk_34."""

from slh_sha256 import SHA256, BlockPad, F

PKseed = 'FA495FB834DEFEA7CC96A81309479135'
SKseed = 'D5213BA4BB6470F1B9EDA88CBC94E627'
ADRS_c = '007cdcef4b8fdb03b005000000000000000000000000'
# wots_sk[0] = PRF(PK.seed, SK.seed, ADRS_c)
wots_sk_0   = SHA256(BlockPad(PKseed) + ADRS_c + SKseed)[:32]
print(f"wots_sk[0]={wots_sk_0}")
#63e73e813025f3da3eaacd528833f005
assert(wots_sk_0 == "63e73e813025f3da3eaacd528833f005")

# i=34, m_i=6
ADRS_c = '007cdcef4b8fdb03b005000000000000002200000000'
wots_sk_34   = SHA256(BlockPad(PKseed) + ADRS_c + SKseed)[:32]
print(f"wots_sk[34]={wots_sk_34}")
#7ab9ab16844abff424d3c775727b9fe0