ark/pkg/client-sdk/wasm/browser/exports.go

//go:build js && wasm
// +build js,wasm

package browser

import (
	"context"
	"fmt"
	"strings"
	"syscall/js"

	arksdk "github.com/ark-network/ark/pkg/client-sdk"
	"github.com/ark-network/ark/pkg/client-sdk/types"
	"github.com/ark-network/ark/pkg/client-sdk/wallet"
	singlekeywallet "github.com/ark-network/ark/pkg/client-sdk/wallet/singlekey"
	walletstore "github.com/ark-network/ark/pkg/client-sdk/wallet/singlekey/store"
)

var (
	arkSdkClient arksdk.ArkClient
	store        types.Store
)

func init() {
	js.Global().Set("init", InitWrapper())
	js.Global().Set("unlock", UnlockWrapper())
	js.Global().Set("lock", LockWrapper())
	js.Global().Set("locked", IsLockedWrapper())
	js.Global().Set("balance", BalanceWrapper())
	js.Global().Set("receive", ReceiveWrapper())
	js.Global().Set("sendOnChain", SendOnChainWrapper())
	js.Global().Set("sendOffChain", SendOffChainWrapper())
	js.Global().Set("sendAsync", SendAsyncWrapper())
	js.Global().Set("claim", ClaimWrapper())
	js.Global().Set("unilateralRedeem", UnilateralRedeemWrapper())
	js.Global().Set("collaborativeRedeem", CollaborativeRedeemWrapper())
	js.Global().Set("getTransactionHistory", GetTransactionHistoryWrapper())
	js.Global().Set("log", LogWrapper())
	js.Global().Set("dump", DumpWrapper())

	js.Global().Set("getAspUrl", GetAspUrlWrapper())
	js.Global().Set("getAspPubKeyHex", GetAspPubkeyWrapper())
	js.Global().Set("getWalletType", GetWalletTypeWrapper())
	js.Global().Set("getClientType", GetClientTypeWrapper())
	js.Global().Set("getNetwork", GetNetworkWrapper())
	js.Global().Set("getRoundLifetime", GetRoundLifetimeWrapper())
	js.Global().Set("getUnilateralExitDelay", GetUnilateralExitDelayWrapper())
	js.Global().Set("getDust", GetDustWrapper())
}

func NewCovenantClient(
	ctx context.Context, storeSvc types.Store,
) error {
	var err error

	data, err := storeSvc.ConfigStore().GetData(ctx)
	if err != nil {
		return err
	}

	if data == nil {
		arkSdkClient, err = arksdk.NewCovenantClient(storeSvc)
	} else {
		var walletSvc wallet.WalletService
		switch data.WalletType {
		case arksdk.SingleKeyWallet:
			walletSvc, err = getSingleKeyWallet(storeSvc.ConfigStore(), data.Network.Name)
			if err != nil {
				return err
			}
		// TODO: Support HD wallet
		default:
			return fmt.Errorf("unknown wallet type")
		}
		arkSdkClient, err = arksdk.LoadCovenantClientWithWallet(storeSvc, walletSvc)
	}
	if err != nil {
		js.Global().Get("console").Call("error", err.Error())
		return err
	}
	store = storeSvc

	select {}
}

func NewCovenantlessClient(
	ctx context.Context, storeSvc types.Store,
) error {
	var err error

	data, err := storeSvc.ConfigStore().GetData(ctx)
	if err != nil {
		return err
	}

	if data == nil {
		arkSdkClient, err = arksdk.NewCovenantlessClient(storeSvc)
	} else {
		var walletSvc wallet.WalletService
		switch data.WalletType {
		case arksdk.SingleKeyWallet:
			walletSvc, err = getSingleKeyWallet(storeSvc.ConfigStore(), data.Network.Name)
			if err != nil {
				return err
			}
		// TODO: Support HD wallet
		default:
			return fmt.Errorf("unknown wallet type")
		}
		arkSdkClient, err = arksdk.LoadCovenantlessClientWithWallet(storeSvc, walletSvc)
	}
	if err != nil {
		js.Global().Get("console").Call("error", err.Error())
		return err
	}
	store = storeSvc

	select {}
}

func getWalletStore(storeType string) (walletstore.WalletStore, error) {
	if storeType == LocalStorageStore {
		return NewLocalStorageWalletStore()
	}
	// TODO: Support IndexDB store
	return nil, fmt.Errorf("unknown wallet store type")
}

func getSingleKeyWallet(
	configStore types.ConfigStore, network string,
) (wallet.WalletService, error) {
	walletStore, err := getWalletStore(configStore.GetType())
	if err != nil {
		return nil, err
	}
	if strings.Contains(network, "liquid") {
		return singlekeywallet.NewLiquidWallet(configStore, walletStore)
	}
	return singlekeywallet.NewBitcoinWallet(configStore, walletStore)
}