#include "config.h"
#include <bitcoin/address.h>
#include <bitcoin/base58.h>
#include <bitcoin/script.h>
#include <common/addr.h>
#include <common/bech32.h>

char *encode_scriptpubkey_to_addr(const tal_t *ctx,
				  const struct chainparams *chainparams,
				  const u8 *scriptPubkey)
{
	char *out;
	size_t scriptLen = tal_bytelen(scriptPubkey);
	struct bitcoin_address pkh;
	struct ripemd160 sh;
	int witver;

	if (is_p2pkh(scriptPubkey, &pkh))
		return bitcoin_to_base58(ctx, chainparams, &pkh);

	if (is_p2sh(scriptPubkey, &sh))
		return p2sh_to_base58(ctx, chainparams, &sh);

	out = tal_arr(ctx, char, 73 + strlen(chainparams->onchain_hrp));
	if (is_p2tr(scriptPubkey, NULL))
		witver = 1;
	else if (is_p2wpkh(scriptPubkey, NULL) || is_p2wsh(scriptPubkey, NULL))
		witver = 0;
	else {
		return tal_free(out);
	}
	if (!segwit_addr_encode(out, chainparams->onchain_hrp, witver,
				scriptPubkey + 2, scriptLen - 2))
		return tal_free(out);

	return out;
}