mirror of
https://github.com/aljazceru/ark.git
synced 2025-12-18 04:34:19 +01:00
* SDK - GetTransactionsStream add to rest/grpc no session, pattern or user messages provided * SDK - AppDataRepository added, store refactor This patch introduces a significant refactor of the Ark SDK codebase. The primary changes involve: 1. **Domain Package Introduction**: A new `domain` package has been created under `store`, encapsulating core data structures and interfaces such as `ConfigData`, `Transaction`, `SdkRepository`, and various repository interfaces. This enhances modularity and separation of concerns. 2. **Replacement of Store with Domain Objects**: All references to the `store` package's `StoreData` have been replaced with `domain.ConfigData`. This change reflects a shift towards using domain-driven design, where the core business logic is represented by domain objects. 3. **Repository Pattern Implementation**: The code has been refactored to use repositories for accessing and managing data. `ConfigRepository` and `TransactionRepository` interfaces are introduced, providing methods for managing configuration data and transactions, respectively. 4. **New Storage Implementations**: New storage implementations using BadgerDB are introduced for persisting transactions (`transaction_repository.go`). This change indicates a move towards more robust and scalable data storage solutions. 5. **Service Layer Introduction**: A service layer (`service.go`) is introduced, which acts as a facade for the underlying data repositories. This layer abstracts the data access logic and provides a unified interface for interacting with configuration and transaction data. 6. **Code Simplification and Cleanup**: The patch removes redundant code and simplifies the existing codebase by consolidating store-related logic and enhancing the overall structure of the code. 7. **Testing Enhancements**: Test cases are updated to reflect the changes in the data access layer, ensuring that the new repository pattern and domain objects are properly tested. Overall, this refactor aims to improve the maintainability, scalability, and testability of the codebase by adopting best practices such as domain-driven design, repository pattern, and separation of concerns. * 'SDK listen for tx stream This diff represents a substantial refactor and enhancement of the `ark-sdk` client, which involves modifications to multiple files, primarily in the `pkg/client-sdk` and `server/internal/core/application` directories. Here's a summary of the key changes: 1. **Configuration and Initialization**: - Refactor to replace file-based configuration store with a domain-based approach. - Introduced `ListenTransactionStream` configuration to allow real-time transaction event listening. 2. **Client SDK Enhancements**: - The client SDK now supports asynchronous transaction streaming and handling. - Introduced methods to get transaction event channels and stop the client gracefully. - Expanded the `ArkClient` interface to include these new functionalities. 3. **Transaction Handling**: - Added support for listening to and processing transaction events in real-time. - Implemented methods to handle different types of transactions, including boarding, redeem, and round transactions. - Expanded the `Transaction` struct to include more metadata, such as `BoardingVOut`. 4. **Repository Changes**: - Introduced `VtxoRepository` for managing VTXO-related operations. - Modified `AppDataRepository` to include `VtxoRepository`. 5. **Example and Test Adjustments**: - Updated example code to utilize the new transaction streaming capabilities. - Adjusted tests to reflect changes in client initialization and configuration handling. 6. **Dependency Updates**: - Updated `go.mod` and `go.sum` to include new dependencies such as `badger` for data storage and potentially updated versions of existing packages. These changes collectively enhance the functionality of the `ark-sdk` client by enabling real-time transaction handling, improving configuration management, and providing a more robust and scalable client architecture. This update seems to be aimed at improving the efficiency and responsiveness of the client application, likely in response to increased demands for real-time data processing and analysis.' * merge * go work sync * test fix * Tiny refactor (#2) * tiny refactor * renaming * Renamings * Fixes (#3) --------- Co-authored-by: Pietralberto Mazza <18440657+altafan@users.noreply.github.com>
2609 lines
61 KiB
Go
2609 lines
61 KiB
Go
package arksdk
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"math"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/ark-network/ark/common"
|
|
"github.com/ark-network/ark/common/bitcointree"
|
|
"github.com/ark-network/ark/common/tree"
|
|
"github.com/ark-network/ark/pkg/client-sdk/client"
|
|
"github.com/ark-network/ark/pkg/client-sdk/explorer"
|
|
"github.com/ark-network/ark/pkg/client-sdk/internal/utils"
|
|
"github.com/ark-network/ark/pkg/client-sdk/redemption"
|
|
"github.com/ark-network/ark/pkg/client-sdk/types"
|
|
"github.com/ark-network/ark/pkg/client-sdk/wallet"
|
|
"github.com/btcsuite/btcd/btcec/v2/schnorr"
|
|
"github.com/btcsuite/btcd/btcutil"
|
|
"github.com/btcsuite/btcd/btcutil/psbt"
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
"github.com/btcsuite/btcd/txscript"
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/decred/dcrd/dcrec/secp256k1/v4"
|
|
"github.com/lightningnetwork/lnd/lnwallet/chainfee"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type bitcoinReceiver struct {
|
|
to string
|
|
amount uint64
|
|
}
|
|
|
|
func NewBitcoinReceiver(to string, amount uint64) Receiver {
|
|
return bitcoinReceiver{to, amount}
|
|
}
|
|
|
|
func (r bitcoinReceiver) To() string {
|
|
return r.to
|
|
}
|
|
|
|
func (r bitcoinReceiver) Amount() uint64 {
|
|
return r.amount
|
|
}
|
|
|
|
func (r bitcoinReceiver) IsOnchain() bool {
|
|
_, err := btcutil.DecodeAddress(r.to, nil)
|
|
return err == nil
|
|
}
|
|
|
|
type covenantlessArkClient struct {
|
|
*arkClient
|
|
}
|
|
|
|
func NewCovenantlessClient(sdkStore types.Store) (ArkClient, error) {
|
|
cfgData, err := sdkStore.ConfigStore().GetData(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if cfgData != nil {
|
|
return nil, ErrAlreadyInitialized
|
|
}
|
|
|
|
return &covenantlessArkClient{
|
|
&arkClient{
|
|
store: sdkStore,
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func LoadCovenantlessClient(sdkStore types.Store) (ArkClient, error) {
|
|
if sdkStore == nil {
|
|
return nil, fmt.Errorf("missin sdk repository")
|
|
}
|
|
|
|
cfgData, err := sdkStore.ConfigStore().GetData(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if cfgData == nil {
|
|
return nil, ErrNotInitialized
|
|
}
|
|
|
|
clientSvc, err := getClient(
|
|
supportedClients, cfgData.ClientType, cfgData.AspUrl,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to setup transport client: %s", err)
|
|
}
|
|
|
|
explorerSvc, err := getExplorer(cfgData.ExplorerURL, cfgData.Network.Name)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to setup explorer: %s", err)
|
|
}
|
|
|
|
walletSvc, err := getWallet(
|
|
sdkStore.ConfigStore(),
|
|
cfgData,
|
|
supportedWallets,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("faile to setup wallet: %s", err)
|
|
}
|
|
|
|
covenantlessClient := covenantlessArkClient{
|
|
&arkClient{
|
|
Config: cfgData,
|
|
wallet: walletSvc,
|
|
store: sdkStore,
|
|
explorer: explorerSvc,
|
|
client: clientSvc,
|
|
},
|
|
}
|
|
|
|
if cfgData.WithTransactionFeed {
|
|
txStreamCtx, txStreamCtxCancel := context.WithCancel(context.Background())
|
|
covenantlessClient.txStreamCtxCancel = txStreamCtxCancel
|
|
go covenantlessClient.listenForTransactions(txStreamCtx)
|
|
go covenantlessClient.listenForBoardingUtxos(txStreamCtx)
|
|
}
|
|
|
|
return &covenantlessClient, nil
|
|
}
|
|
|
|
func LoadCovenantlessClientWithWallet(
|
|
sdkStore types.Store, walletSvc wallet.WalletService,
|
|
) (ArkClient, error) {
|
|
if sdkStore == nil {
|
|
return nil, fmt.Errorf("missin sdk repository")
|
|
}
|
|
|
|
if walletSvc == nil {
|
|
return nil, fmt.Errorf("missin wallet service")
|
|
}
|
|
|
|
cfgData, err := sdkStore.ConfigStore().GetData(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if cfgData == nil {
|
|
return nil, ErrNotInitialized
|
|
}
|
|
|
|
clientSvc, err := getClient(
|
|
supportedClients, cfgData.ClientType, cfgData.AspUrl,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to setup transport client: %s", err)
|
|
}
|
|
|
|
explorerSvc, err := getExplorer(cfgData.ExplorerURL, cfgData.Network.Name)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to setup explorer: %s", err)
|
|
}
|
|
|
|
covenantlessClient := covenantlessArkClient{
|
|
&arkClient{
|
|
Config: cfgData,
|
|
wallet: walletSvc,
|
|
store: sdkStore,
|
|
explorer: explorerSvc,
|
|
client: clientSvc,
|
|
},
|
|
}
|
|
|
|
if cfgData.WithTransactionFeed {
|
|
txStreamCtx, txStreamCtxCancel := context.WithCancel(context.Background())
|
|
covenantlessClient.txStreamCtxCancel = txStreamCtxCancel
|
|
go covenantlessClient.listenForTransactions(txStreamCtx)
|
|
go covenantlessClient.listenForBoardingUtxos(txStreamCtx)
|
|
}
|
|
|
|
return &covenantlessClient, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) Init(ctx context.Context, args InitArgs) error {
|
|
if err := a.arkClient.init(ctx, args); err != nil {
|
|
return err
|
|
}
|
|
|
|
if args.ListenTransactionStream {
|
|
txStreamCtx, txStreamCtxCancel := context.WithCancel(context.Background())
|
|
a.txStreamCtxCancel = txStreamCtxCancel
|
|
go a.listenForTransactions(txStreamCtx)
|
|
go a.listenForBoardingUtxos(txStreamCtx)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) InitWithWallet(ctx context.Context, args InitWithWalletArgs) error {
|
|
if err := a.arkClient.initWithWallet(ctx, args); err != nil {
|
|
return err
|
|
}
|
|
|
|
if a.WithTransactionFeed {
|
|
txStreamCtx, txStreamCtxCancel := context.WithCancel(context.Background())
|
|
a.txStreamCtxCancel = txStreamCtxCancel
|
|
go a.listenForTransactions(txStreamCtx)
|
|
go a.listenForBoardingUtxos(txStreamCtx)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) listenForTransactions(ctx context.Context) {
|
|
eventChan, closeFunc, err := a.client.GetTransactionsStream(ctx)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get transaction stream")
|
|
return
|
|
}
|
|
defer closeFunc()
|
|
|
|
offchainAddr, _, err := a.wallet.NewAddress(ctx, true)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get new address")
|
|
return
|
|
}
|
|
|
|
desc, err := a.offchainAddressToDefaultVtxoDescriptor(offchainAddr)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get descriptor for new address")
|
|
return
|
|
}
|
|
|
|
for {
|
|
select {
|
|
case event, ok := <-eventChan:
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
if event.Err != nil {
|
|
log.WithError(event.Err).Error("Error in transaction stream")
|
|
continue
|
|
}
|
|
|
|
pendingBoardingTxsMap, newPendingBoardingTxs, err := a.getBoardingPendingTransactions(ctx)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get pending transactions")
|
|
continue
|
|
}
|
|
|
|
if err := a.store.TransactionStore().
|
|
AddTransactions(ctx, newPendingBoardingTxs); err != nil {
|
|
log.WithError(err).Error("Failed to insert new boarding transactions")
|
|
continue
|
|
}
|
|
|
|
a.processTransactionEvent(desc, event, pendingBoardingTxsMap)
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (a *covenantlessArkClient) listenForBoardingUtxos(ctx context.Context) {
|
|
ticker := time.NewTicker(2 * time.Second)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
_, newPendingBoardingTxs, err := a.getBoardingPendingTransactions(ctx)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get pending transactions")
|
|
continue
|
|
}
|
|
|
|
if err := a.store.TransactionStore().
|
|
AddTransactions(ctx, newPendingBoardingTxs); err != nil {
|
|
log.WithError(err).Error("Failed to insert new boarding transactions")
|
|
continue
|
|
}
|
|
case <-ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (a *covenantlessArkClient) getBoardingPendingTransactions(
|
|
ctx context.Context,
|
|
) (map[string]types.Transaction, []types.Transaction, error) {
|
|
oldTxs, err := a.store.TransactionStore().GetAllTransactions(ctx)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
pendingBoardingTxsMap := make(map[string]types.Transaction)
|
|
for _, tx := range oldTxs {
|
|
if tx.IsBoarding() {
|
|
if tx.IsPending {
|
|
pendingBoardingTxsMap[tx.BoardingTxid] = tx
|
|
}
|
|
}
|
|
}
|
|
|
|
boardingTxs, _, err := a.getBoardingTxs(ctx)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
newPendingBoardingTxs := make([]types.Transaction, 0)
|
|
for _, tx := range boardingTxs {
|
|
if tx.IsBoarding() && tx.IsPending {
|
|
if _, ok := pendingBoardingTxsMap[tx.BoardingTxid]; !ok {
|
|
newPendingBoardingTxs = append(newPendingBoardingTxs, tx)
|
|
pendingBoardingTxsMap[tx.BoardingTxid] = tx
|
|
}
|
|
}
|
|
}
|
|
|
|
return pendingBoardingTxsMap, newPendingBoardingTxs, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) processTransactionEvent(
|
|
descriptor string,
|
|
event client.TransactionEvent,
|
|
pendingBoardingTxsMap map[string]types.Transaction,
|
|
) {
|
|
if event.Round != nil {
|
|
boardingAmount := 0
|
|
boardingTxsToUpdate := make([]types.Transaction, 0)
|
|
for _, v := range event.Round.ClaimedBoardingUtxos {
|
|
boardingKey := fmt.Sprintf("%v-%v", v.Txid, v.VOut)
|
|
if tx, ok := pendingBoardingTxsMap[boardingKey]; ok {
|
|
boardingAmount += int(tx.Amount)
|
|
tx.IsPending = false
|
|
boardingTxsToUpdate = append(boardingTxsToUpdate, tx)
|
|
}
|
|
}
|
|
|
|
if err := a.store.TransactionStore().
|
|
UpdateTransactions(context.Background(), boardingTxsToUpdate); err != nil {
|
|
log.WithError(err).Error("Failed to update boarding transactions")
|
|
}
|
|
|
|
spentKeys := make([]types.VtxoKey, 0, len(event.Round.SpentVtxos))
|
|
for _, v := range event.Round.SpentVtxos {
|
|
spentKeys = append(spentKeys, types.VtxoKey{
|
|
Txid: v.Txid,
|
|
VOut: v.VOut,
|
|
})
|
|
}
|
|
|
|
vtxos, err := a.store.VtxoStore().
|
|
GetVtxos(context.Background(), spentKeys)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get spent vtxos")
|
|
return
|
|
}
|
|
|
|
if len(vtxos) > 0 {
|
|
vtxosToUpdate := make([]types.Vtxo, 0)
|
|
for _, v := range vtxos {
|
|
v.Spent = true
|
|
v.Pending = false
|
|
vtxosToUpdate = append(vtxosToUpdate, v)
|
|
}
|
|
|
|
if err := a.store.VtxoStore().
|
|
UpdateVtxos(context.Background(), vtxosToUpdate); err != nil {
|
|
log.WithError(err).Error("Failed to update spent vtxos")
|
|
}
|
|
}
|
|
|
|
vtxosToInsert := make([]types.Vtxo, 0)
|
|
txsToInsert := make([]types.Transaction, 0)
|
|
for _, v := range event.Round.SpendableVtxos {
|
|
if v.Descriptor == descriptor {
|
|
vtxosToInsert = append(vtxosToInsert, types.Vtxo{
|
|
VtxoKey: types.VtxoKey{
|
|
Txid: v.Txid,
|
|
VOut: v.VOut,
|
|
},
|
|
Amount: v.Amount,
|
|
ExpiresAt: v.ExpiresAt,
|
|
RedeemTx: event.Round.Txid,
|
|
Pending: false,
|
|
SpentBy: v.SpentBy,
|
|
Spent: false,
|
|
})
|
|
|
|
if boardingAmount == int(v.Amount) {
|
|
continue
|
|
}
|
|
|
|
txsToInsert = append(txsToInsert, types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
RoundTxid: event.Round.Txid,
|
|
},
|
|
Amount: v.Amount,
|
|
Type: types.TxReceived,
|
|
CreatedAt: time.Now(), //TODO is this ok?
|
|
})
|
|
}
|
|
}
|
|
|
|
if err := a.store.VtxoStore().
|
|
AddVtxos(context.Background(), vtxosToInsert); err != nil {
|
|
log.WithError(err).Error("Failed to insert new vtxos")
|
|
return
|
|
}
|
|
|
|
if err := a.store.TransactionStore().
|
|
AddTransactions(context.Background(), txsToInsert); err != nil {
|
|
log.WithError(err).Error("Failed to insert received transaction")
|
|
return
|
|
}
|
|
|
|
}
|
|
|
|
if event.Redeem != nil {
|
|
vtxosToInsert := make([]types.Vtxo, 0)
|
|
spentKeys := make([]types.VtxoKey, 0, len(event.Redeem.SpentVtxos))
|
|
for _, v := range event.Redeem.SpentVtxos {
|
|
spentKeys = append(spentKeys, types.VtxoKey{
|
|
Txid: v.Txid,
|
|
VOut: v.VOut,
|
|
})
|
|
}
|
|
|
|
vtxos, err := a.store.VtxoStore().
|
|
GetVtxos(context.Background(), spentKeys)
|
|
if err != nil {
|
|
log.WithError(err).Error("Failed to get spent vtxos")
|
|
return
|
|
}
|
|
|
|
if len(vtxos) > 0 {
|
|
inputAmount := uint64(0)
|
|
vtxosToUpdate := make([]types.Vtxo, 0)
|
|
for _, v := range vtxos {
|
|
v.Spent = true
|
|
vtxosToUpdate = append(vtxosToUpdate, v)
|
|
inputAmount += v.Amount
|
|
}
|
|
|
|
if err := a.store.VtxoStore().
|
|
UpdateVtxos(context.Background(), vtxosToUpdate); err != nil {
|
|
log.WithError(err).Error("Failed to update spent vtxos")
|
|
return
|
|
}
|
|
|
|
outputAmount := uint64(0)
|
|
for _, v := range event.Redeem.SpendableVtxos {
|
|
if v.Descriptor == descriptor {
|
|
vtxosToInsert = append(vtxosToInsert, types.Vtxo{
|
|
VtxoKey: types.VtxoKey{
|
|
Txid: v.Txid,
|
|
VOut: v.VOut,
|
|
},
|
|
Amount: v.Amount,
|
|
ExpiresAt: v.ExpiresAt,
|
|
RedeemTx: event.Redeem.Txid,
|
|
Pending: true,
|
|
SpentBy: v.SpentBy,
|
|
Spent: false,
|
|
})
|
|
outputAmount += v.Amount
|
|
}
|
|
}
|
|
|
|
tx := types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
RedeemTxid: event.Redeem.Txid,
|
|
},
|
|
Amount: inputAmount - outputAmount,
|
|
Type: types.TxSent,
|
|
IsPending: true,
|
|
CreatedAt: time.Now(), //TODO is this ok?
|
|
}
|
|
|
|
if err := a.store.TransactionStore().
|
|
AddTransactions(context.Background(), []types.Transaction{tx}); err != nil {
|
|
log.WithError(err).Error("Failed to insert received transaction")
|
|
}
|
|
} else {
|
|
for _, v := range event.Redeem.SpendableVtxos {
|
|
if v.Descriptor == descriptor {
|
|
vtxosToInsert = append(vtxosToInsert, types.Vtxo{
|
|
VtxoKey: types.VtxoKey{
|
|
Txid: v.Txid,
|
|
VOut: v.VOut,
|
|
},
|
|
Amount: v.Amount,
|
|
ExpiresAt: v.ExpiresAt,
|
|
RedeemTx: event.Redeem.Txid,
|
|
Pending: true,
|
|
SpentBy: v.SpentBy,
|
|
Spent: false,
|
|
})
|
|
|
|
tx := types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
RedeemTxid: event.Redeem.Txid,
|
|
},
|
|
Amount: v.Amount,
|
|
Type: types.TxReceived,
|
|
IsPending: true,
|
|
CreatedAt: time.Now(), //TODO is this ok?
|
|
}
|
|
if err := a.store.TransactionStore().
|
|
AddTransactions(context.Background(), []types.Transaction{tx}); err != nil {
|
|
log.WithError(err).Error("Failed to insert received transaction")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if err := a.store.VtxoStore().
|
|
AddVtxos(context.Background(), vtxosToInsert); err != nil {
|
|
log.WithError(err).Error("Failed to insert new vtxos")
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (a *covenantlessArkClient) ListVtxos(
|
|
ctx context.Context,
|
|
) (spendableVtxos, spentVtxos []client.Vtxo, err error) {
|
|
offchainAddrs, _, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
_, pubkey, _, err := common.DecodeAddress(offchainAddrs[0])
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
myPubkey := schnorr.SerializePubKey(pubkey)
|
|
|
|
// The ASP returns the vtxos sent to others via async payments as spendable
|
|
// because they are actually revertable. Since we do not provide any revert
|
|
// feature, we want to ignore them.
|
|
// To understand if the output of a redeem tx is sent or received we look at
|
|
// the inputs and check if they are owned by us.
|
|
// The auxiliary variables below are used to make these checks in an
|
|
// efficient way.
|
|
for _, addr := range offchainAddrs {
|
|
spendable, spent, err := a.client.ListVtxos(ctx, addr)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
for _, v := range spendable {
|
|
if !v.Pending {
|
|
spendableVtxos = append(spendableVtxos, v)
|
|
continue
|
|
}
|
|
script, err := bitcointree.ParseVtxoScript(v.Descriptor)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
reversibleVtxo, ok := script.(*bitcointree.ReversibleVtxoScript)
|
|
if !ok {
|
|
spendableVtxos = append(spendableVtxos, v)
|
|
continue
|
|
}
|
|
|
|
if !bytes.Equal(schnorr.SerializePubKey(reversibleVtxo.Sender), myPubkey) {
|
|
spendableVtxos = append(spendableVtxos, v)
|
|
}
|
|
}
|
|
for _, v := range spent {
|
|
script, err := bitcointree.ParseVtxoScript(v.Descriptor)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
reversibleVtxo, ok := script.(*bitcointree.ReversibleVtxoScript)
|
|
if !ok {
|
|
spentVtxos = append(spentVtxos, v)
|
|
continue
|
|
}
|
|
if !bytes.Equal(schnorr.SerializePubKey(reversibleVtxo.Sender), myPubkey) {
|
|
spentVtxos = append(spentVtxos, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (a *covenantlessArkClient) Balance(
|
|
ctx context.Context, computeVtxoExpiration bool,
|
|
) (*Balance, error) {
|
|
offchainAddrs, boardingAddrs, redeemAddrs, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
const nbWorkers = 3
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(nbWorkers * len(offchainAddrs))
|
|
|
|
chRes := make(chan balanceRes, nbWorkers*len(offchainAddrs))
|
|
for i := range offchainAddrs {
|
|
offchainAddr := offchainAddrs[i]
|
|
boardingAddr := boardingAddrs[i]
|
|
redeemAddr := redeemAddrs[i]
|
|
|
|
go func(addr string) {
|
|
defer wg.Done()
|
|
balance, amountByExpiration, err := a.getOffchainBalance(
|
|
ctx, addr, computeVtxoExpiration,
|
|
)
|
|
if err != nil {
|
|
chRes <- balanceRes{err: err}
|
|
return
|
|
}
|
|
|
|
chRes <- balanceRes{
|
|
offchainBalance: balance,
|
|
offchainBalanceByExpiration: amountByExpiration,
|
|
}
|
|
}(offchainAddr)
|
|
|
|
getDelayedBalance := func(addr string) {
|
|
defer wg.Done()
|
|
|
|
spendableBalance, lockedBalance, err := a.explorer.GetRedeemedVtxosBalance(
|
|
addr, a.UnilateralExitDelay,
|
|
)
|
|
if err != nil {
|
|
chRes <- balanceRes{err: err}
|
|
return
|
|
}
|
|
|
|
chRes <- balanceRes{
|
|
onchainSpendableBalance: spendableBalance,
|
|
onchainLockedBalance: lockedBalance,
|
|
err: err,
|
|
}
|
|
}
|
|
|
|
go getDelayedBalance(boardingAddr)
|
|
go getDelayedBalance(redeemAddr)
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
lockedOnchainBalance := []LockedOnchainBalance{}
|
|
details := make([]VtxoDetails, 0)
|
|
offchainBalance, onchainBalance := uint64(0), uint64(0)
|
|
nextExpiration := int64(0)
|
|
count := 0
|
|
for res := range chRes {
|
|
if res.err != nil {
|
|
return nil, res.err
|
|
}
|
|
if res.offchainBalance > 0 {
|
|
offchainBalance = res.offchainBalance
|
|
}
|
|
if res.onchainSpendableBalance > 0 {
|
|
onchainBalance += res.onchainSpendableBalance
|
|
}
|
|
if res.offchainBalanceByExpiration != nil {
|
|
for timestamp, amount := range res.offchainBalanceByExpiration {
|
|
if nextExpiration == 0 || timestamp < nextExpiration {
|
|
nextExpiration = timestamp
|
|
}
|
|
|
|
fancyTime := time.Unix(timestamp, 0).Format(time.RFC3339)
|
|
details = append(
|
|
details,
|
|
VtxoDetails{
|
|
ExpiryTime: fancyTime,
|
|
Amount: amount,
|
|
},
|
|
)
|
|
}
|
|
}
|
|
if res.onchainLockedBalance != nil {
|
|
for timestamp, amount := range res.onchainLockedBalance {
|
|
fancyTime := time.Unix(timestamp, 0).Format(time.RFC3339)
|
|
lockedOnchainBalance = append(
|
|
lockedOnchainBalance,
|
|
LockedOnchainBalance{
|
|
SpendableAt: fancyTime,
|
|
Amount: amount,
|
|
},
|
|
)
|
|
}
|
|
}
|
|
|
|
count++
|
|
if count == nbWorkers {
|
|
break
|
|
}
|
|
}
|
|
|
|
fancyTimeExpiration := ""
|
|
if nextExpiration != 0 {
|
|
t := time.Unix(nextExpiration, 0)
|
|
if t.Before(time.Now().Add(48 * time.Hour)) {
|
|
// print the duration instead of the absolute time
|
|
until := time.Until(t)
|
|
seconds := math.Abs(until.Seconds())
|
|
minutes := math.Abs(until.Minutes())
|
|
hours := math.Abs(until.Hours())
|
|
|
|
if hours < 1 {
|
|
if minutes < 1 {
|
|
fancyTimeExpiration = fmt.Sprintf("%d seconds", int(seconds))
|
|
} else {
|
|
fancyTimeExpiration = fmt.Sprintf("%d minutes", int(minutes))
|
|
}
|
|
} else {
|
|
fancyTimeExpiration = fmt.Sprintf("%d hours", int(hours))
|
|
}
|
|
} else {
|
|
fancyTimeExpiration = t.Format(time.RFC3339)
|
|
}
|
|
}
|
|
|
|
response := &Balance{
|
|
OnchainBalance: OnchainBalance{
|
|
SpendableAmount: onchainBalance,
|
|
LockedAmount: lockedOnchainBalance,
|
|
},
|
|
OffchainBalance: OffchainBalance{
|
|
Total: offchainBalance,
|
|
NextExpiration: fancyTimeExpiration,
|
|
Details: details,
|
|
},
|
|
}
|
|
|
|
return response, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) SendOnChain(
|
|
ctx context.Context, receivers []Receiver,
|
|
) (string, error) {
|
|
for _, receiver := range receivers {
|
|
if !receiver.IsOnchain() {
|
|
return "", fmt.Errorf("invalid receiver address '%s': must be onchain", receiver.To())
|
|
}
|
|
}
|
|
|
|
return a.sendOnchain(ctx, receivers)
|
|
}
|
|
|
|
func (a *covenantlessArkClient) SendOffChain(
|
|
ctx context.Context,
|
|
withExpiryCoinselect bool, receivers []Receiver,
|
|
) (string, error) {
|
|
for _, receiver := range receivers {
|
|
if receiver.IsOnchain() {
|
|
return "", fmt.Errorf("invalid receiver address '%s': must be offchain", receiver.To())
|
|
}
|
|
}
|
|
|
|
return a.sendOffchain(ctx, withExpiryCoinselect, receivers)
|
|
}
|
|
|
|
func (a *covenantlessArkClient) UnilateralRedeem(ctx context.Context) error {
|
|
if a.wallet.IsLocked() {
|
|
return fmt.Errorf("wallet is locked")
|
|
}
|
|
|
|
offchainAddrs, _, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
vtxos := make([]client.Vtxo, 0)
|
|
for _, offchainAddr := range offchainAddrs {
|
|
spendableVtxos, _, err := a.getVtxos(ctx, offchainAddr, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
vtxos = append(vtxos, spendableVtxos...)
|
|
}
|
|
|
|
totalVtxosAmount := uint64(0)
|
|
for _, vtxo := range vtxos {
|
|
totalVtxosAmount += vtxo.Amount
|
|
}
|
|
|
|
// transactionsMap avoid duplicates
|
|
transactionsMap := make(map[string]struct{}, 0)
|
|
transactions := make([]string, 0)
|
|
|
|
redeemBranches, err := a.getRedeemBranches(ctx, vtxos)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, branch := range redeemBranches {
|
|
branchTxs, err := branch.RedeemPath()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, txHex := range branchTxs {
|
|
if _, ok := transactionsMap[txHex]; !ok {
|
|
transactions = append(transactions, txHex)
|
|
transactionsMap[txHex] = struct{}{}
|
|
}
|
|
}
|
|
}
|
|
|
|
for i, txHex := range transactions {
|
|
for {
|
|
txid, err := a.explorer.Broadcast(txHex)
|
|
if err != nil {
|
|
if strings.Contains(strings.ToLower(err.Error()), "bad-txns-inputs-missingorspent") {
|
|
time.Sleep(1 * time.Second)
|
|
} else {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if len(txid) > 0 {
|
|
log.Infof("(%d/%d) broadcasted tx %s", i+1, len(transactions), txid)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) CollaborativeRedeem(
|
|
ctx context.Context,
|
|
addr string, amount uint64, withExpiryCoinselect bool,
|
|
) (string, error) {
|
|
if a.wallet.IsLocked() {
|
|
return "", fmt.Errorf("wallet is locked")
|
|
}
|
|
|
|
netParams := utils.ToBitcoinNetwork(a.Network)
|
|
if _, err := btcutil.DecodeAddress(addr, &netParams); err != nil {
|
|
return "", fmt.Errorf("invalid onchain address")
|
|
}
|
|
|
|
offchainAddrs, _, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
receivers := []client.Output{
|
|
{
|
|
Address: addr,
|
|
Amount: amount,
|
|
},
|
|
}
|
|
|
|
vtxos := make([]client.Vtxo, 0)
|
|
for _, offchainAddr := range offchainAddrs {
|
|
spendableVtxos, _, err := a.getVtxos(ctx, offchainAddr, withExpiryCoinselect)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
vtxos = append(vtxos, spendableVtxos...)
|
|
}
|
|
|
|
selectedCoins, changeAmount, err := utils.CoinSelect(
|
|
vtxos, amount, a.Dust, withExpiryCoinselect,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if changeAmount > 0 {
|
|
offchainAddr, _, err := a.wallet.NewAddress(ctx, true)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
desc, err := a.offchainAddressToDefaultVtxoDescriptor(offchainAddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
receivers = append(receivers, client.Output{
|
|
Descriptor: desc,
|
|
Amount: changeAmount,
|
|
})
|
|
}
|
|
|
|
inputs := make([]client.Input, 0, len(selectedCoins))
|
|
|
|
for _, coin := range selectedCoins {
|
|
inputs = append(inputs, client.Input{
|
|
Outpoint: client.Outpoint{
|
|
Txid: coin.Txid,
|
|
VOut: coin.VOut,
|
|
},
|
|
Descriptor: coin.Descriptor,
|
|
})
|
|
}
|
|
|
|
roundEphemeralKey, err := secp256k1.GeneratePrivateKey()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
paymentID, err := a.client.RegisterInputsForNextRound(
|
|
ctx,
|
|
inputs,
|
|
hex.EncodeToString(roundEphemeralKey.PubKey().SerializeCompressed()),
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if err := a.client.RegisterOutputsForNextRound(ctx, paymentID, receivers); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
poolTxID, err := a.handleRoundStream(
|
|
ctx, paymentID, selectedCoins, nil, "", receivers, roundEphemeralKey,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return poolTxID, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) SendAsync(
|
|
ctx context.Context,
|
|
withExpiryCoinselect bool, receivers []Receiver,
|
|
) (string, error) {
|
|
if len(receivers) <= 0 {
|
|
return "", fmt.Errorf("missing receivers")
|
|
}
|
|
|
|
netParams := utils.ToBitcoinNetwork(a.Network)
|
|
for _, receiver := range receivers {
|
|
isOnchain, _, err := utils.ParseBitcoinAddress(receiver.To(), netParams)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if isOnchain {
|
|
return "", fmt.Errorf("all receiver addresses must be offchain addresses")
|
|
}
|
|
}
|
|
|
|
offchainAddrs, _, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
_, _, aspPubKey, err := common.DecodeAddress(offchainAddrs[0])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
receiversOutput := make([]client.Output, 0)
|
|
sumOfReceivers := uint64(0)
|
|
|
|
for _, receiver := range receivers {
|
|
_, _, aspKey, err := common.DecodeAddress(receiver.To())
|
|
if err != nil {
|
|
return "", fmt.Errorf("invalid receiver address: %s", err)
|
|
}
|
|
|
|
if !bytes.Equal(
|
|
aspPubKey.SerializeCompressed(), aspKey.SerializeCompressed(),
|
|
) {
|
|
return "", fmt.Errorf("invalid receiver address '%s': must be associated with the connected service provider", receiver)
|
|
}
|
|
|
|
if receiver.Amount() < a.Dust {
|
|
return "", fmt.Errorf("invalid amount (%d), must be greater than dust %d", receiver.Amount(), a.Dust)
|
|
}
|
|
|
|
isSelfTransfer := offchainAddrs[0] == receiver.To()
|
|
|
|
var desc string
|
|
|
|
// reversible vtxo does not make sense for self transfer
|
|
// if the receiver is the same as the sender, handle the output like the change
|
|
if !isSelfTransfer {
|
|
desc, err = a.offchainAddressToReversibleVtxoDescriptor(offchainAddrs[0], receiver.To())
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
} else {
|
|
desc, err = a.offchainAddressToDefaultVtxoDescriptor(receiver.To())
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
receiversOutput = append(receiversOutput, client.Output{
|
|
Descriptor: desc,
|
|
Amount: receiver.Amount(),
|
|
})
|
|
sumOfReceivers += receiver.Amount()
|
|
}
|
|
|
|
vtxos, _, err := a.getVtxos(ctx, offchainAddrs[0], withExpiryCoinselect)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
selectedCoins, changeAmount, err := utils.CoinSelect(
|
|
vtxos, sumOfReceivers, a.Dust, withExpiryCoinselect,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if changeAmount > 0 {
|
|
changeDesc, err := a.offchainAddressToDefaultVtxoDescriptor(offchainAddrs[0])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
changeReceiver := client.Output{
|
|
Descriptor: changeDesc,
|
|
Amount: changeAmount,
|
|
}
|
|
receiversOutput = append(receiversOutput, changeReceiver)
|
|
}
|
|
|
|
inputs := make([]client.Input, 0, len(selectedCoins))
|
|
|
|
for _, coin := range selectedCoins {
|
|
inputs = append(inputs, client.Input{
|
|
Outpoint: client.Outpoint{
|
|
Txid: coin.Txid,
|
|
VOut: coin.VOut,
|
|
},
|
|
Descriptor: coin.Descriptor,
|
|
})
|
|
}
|
|
|
|
redeemTx, err := a.client.CreatePayment(ctx, inputs, receiversOutput)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// TODO verify the redeem tx signature
|
|
|
|
signedRedeemTx, err := a.wallet.SignTransaction(ctx, a.explorer, redeemTx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if err = a.client.CompletePayment(
|
|
ctx, signedRedeemTx,
|
|
); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return signedRedeemTx, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) Claim(ctx context.Context) (string, error) {
|
|
myselfOffchain, _, err := a.wallet.NewAddress(ctx, false)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
_, pendingVtxos, err := a.getVtxos(ctx, myselfOffchain, false)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
_, mypubkey, _, err := common.DecodeAddress(myselfOffchain)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
boardingUtxos, err := a.getClaimableBoardingUtxos(ctx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var pendingBalance uint64
|
|
for _, vtxo := range pendingVtxos {
|
|
pendingBalance += vtxo.Amount
|
|
}
|
|
for _, vtxo := range boardingUtxos {
|
|
pendingBalance += vtxo.Amount
|
|
}
|
|
if pendingBalance == 0 {
|
|
return "", nil
|
|
}
|
|
|
|
desc, err := a.offchainAddressToDefaultVtxoDescriptor(myselfOffchain)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
receiver := client.Output{
|
|
Descriptor: desc,
|
|
Amount: pendingBalance,
|
|
}
|
|
|
|
return a.selfTransferAllPendingPayments(
|
|
ctx,
|
|
pendingVtxos,
|
|
boardingUtxos,
|
|
receiver,
|
|
hex.EncodeToString(mypubkey.SerializeCompressed()),
|
|
)
|
|
}
|
|
|
|
func (a *covenantlessArkClient) GetTransactionHistory(
|
|
ctx context.Context,
|
|
) ([]types.Transaction, error) {
|
|
if a.Config.WithTransactionFeed {
|
|
return a.store.TransactionStore().GetAllTransactions(ctx)
|
|
}
|
|
|
|
if a.Config == nil {
|
|
return nil, fmt.Errorf("client not initialized")
|
|
}
|
|
|
|
spendableVtxos, spentVtxos, err := a.ListVtxos(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
boardingTxs, ignoreVtxos, err := a.getBoardingTxs(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
offchainTxs, err := vtxosToTxsCovenantless(
|
|
a.Config.RoundLifetime, spendableVtxos, spentVtxos, ignoreVtxos,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
txs := append(boardingTxs, offchainTxs...)
|
|
// Sort the slice by age
|
|
sort.Slice(txs, func(i, j int) bool {
|
|
txi := txs[i]
|
|
txj := txs[j]
|
|
if txi.CreatedAt.Equal(txj.CreatedAt) {
|
|
return txi.Type > txj.Type
|
|
}
|
|
return txi.CreatedAt.After(txj.CreatedAt)
|
|
})
|
|
|
|
return txs, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) sendOnchain(
|
|
ctx context.Context, receivers []Receiver,
|
|
) (string, error) {
|
|
if a.wallet.IsLocked() {
|
|
return "", fmt.Errorf("wallet is locked")
|
|
}
|
|
|
|
ptx, err := psbt.New(nil, nil, 2, 0, nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
updater, err := psbt.NewUpdater(ptx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
netParams := utils.ToBitcoinNetwork(a.Network)
|
|
|
|
targetAmount := uint64(0)
|
|
for _, receiver := range receivers {
|
|
targetAmount += receiver.Amount()
|
|
if receiver.Amount() < a.Dust {
|
|
return "", fmt.Errorf("invalid amount (%d), must be greater than dust %d", receiver.Amount(), a.Dust)
|
|
}
|
|
|
|
rcvAddr, err := btcutil.DecodeAddress(receiver.To(), &netParams)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
pkscript, err := txscript.PayToAddrScript(rcvAddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
updater.Upsbt.UnsignedTx.AddTxOut(&wire.TxOut{
|
|
Value: int64(receiver.Amount()),
|
|
PkScript: pkscript,
|
|
})
|
|
updater.Upsbt.Outputs = append(updater.Upsbt.Outputs, psbt.POutput{})
|
|
}
|
|
|
|
utxos, change, err := a.coinSelectOnchain(
|
|
ctx, targetAmount, nil,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if err := a.addInputs(ctx, updater, utxos); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if change > 0 {
|
|
_, changeAddr, err := a.wallet.NewAddress(ctx, true)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
addr, _ := btcutil.DecodeAddress(changeAddr, &netParams)
|
|
|
|
pkscript, err := txscript.PayToAddrScript(addr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
updater.Upsbt.UnsignedTx.AddTxOut(&wire.TxOut{
|
|
Value: int64(change),
|
|
PkScript: pkscript,
|
|
})
|
|
updater.Upsbt.Outputs = append(updater.Upsbt.Outputs, psbt.POutput{})
|
|
}
|
|
|
|
size := updater.Upsbt.UnsignedTx.SerializeSize()
|
|
feeRate, err := a.explorer.GetFeeRate()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
feeAmount := uint64(math.Ceil(float64(size)*feeRate) + 50)
|
|
|
|
if change > feeAmount {
|
|
updater.Upsbt.UnsignedTx.TxOut[len(updater.Upsbt.Outputs)-1].Value = int64(change - feeAmount)
|
|
} else if change == feeAmount {
|
|
updater.Upsbt.UnsignedTx.TxOut = updater.Upsbt.UnsignedTx.TxOut[:len(updater.Upsbt.UnsignedTx.TxOut)-1]
|
|
} else { // change < feeAmount
|
|
if change > 0 {
|
|
updater.Upsbt.UnsignedTx.TxOut = updater.Upsbt.UnsignedTx.TxOut[:len(updater.Upsbt.UnsignedTx.TxOut)-1]
|
|
}
|
|
// reselect the difference
|
|
selected, newChange, err := a.coinSelectOnchain(
|
|
ctx, feeAmount-change, utxos,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if err := a.addInputs(ctx, updater, selected); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if newChange > 0 {
|
|
_, changeAddr, err := a.wallet.NewAddress(ctx, true)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
addr, _ := btcutil.DecodeAddress(changeAddr, &netParams)
|
|
|
|
pkscript, err := txscript.PayToAddrScript(addr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
updater.Upsbt.UnsignedTx.AddTxOut(&wire.TxOut{
|
|
Value: int64(newChange),
|
|
PkScript: pkscript,
|
|
})
|
|
updater.Upsbt.Outputs = append(updater.Upsbt.Outputs, psbt.POutput{})
|
|
}
|
|
}
|
|
|
|
unsignedTx, _ := ptx.B64Encode()
|
|
|
|
signedTx, err := a.wallet.SignTransaction(ctx, a.explorer, unsignedTx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
ptx, err = psbt.NewFromRawBytes(strings.NewReader(signedTx), true)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
for i := range ptx.Inputs {
|
|
if err := psbt.Finalize(ptx, i); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
return ptx.B64Encode()
|
|
}
|
|
|
|
func (a *covenantlessArkClient) sendOffchain(
|
|
ctx context.Context, withExpiryCoinselect bool, receivers []Receiver,
|
|
) (string, error) {
|
|
if a.wallet.IsLocked() {
|
|
return "", fmt.Errorf("wallet is locked")
|
|
}
|
|
|
|
offchainAddrs, _, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(offchainAddrs) <= 0 {
|
|
return "", fmt.Errorf("no funds detected")
|
|
}
|
|
|
|
_, _, aspPubKey, err := common.DecodeAddress(offchainAddrs[0])
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
receiversOutput := make([]client.Output, 0)
|
|
sumOfReceivers := uint64(0)
|
|
|
|
for _, receiver := range receivers {
|
|
_, _, aspKey, err := common.DecodeAddress(receiver.To())
|
|
if err != nil {
|
|
return "", fmt.Errorf("invalid receiver address: %s", err)
|
|
}
|
|
|
|
if !bytes.Equal(
|
|
aspPubKey.SerializeCompressed(), aspKey.SerializeCompressed(),
|
|
) {
|
|
return "", fmt.Errorf("invalid receiver address '%s': must be associated with the connected service provider", receiver.To())
|
|
}
|
|
|
|
if receiver.Amount() < a.Dust {
|
|
return "", fmt.Errorf("invalid amount (%d), must be greater than dust %d", receiver.Amount(), a.Dust)
|
|
}
|
|
|
|
desc, err := a.offchainAddressToDefaultVtxoDescriptor(receiver.To())
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
receiversOutput = append(receiversOutput, client.Output{
|
|
Descriptor: desc,
|
|
Amount: receiver.Amount(),
|
|
})
|
|
sumOfReceivers += receiver.Amount()
|
|
}
|
|
|
|
vtxos := make([]client.Vtxo, 0)
|
|
for _, offchainAddr := range offchainAddrs {
|
|
spendableVtxos, _, err := a.getVtxos(ctx, offchainAddr, withExpiryCoinselect)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
vtxos = append(vtxos, spendableVtxos...)
|
|
}
|
|
|
|
selectedCoins, changeAmount, err := utils.CoinSelect(
|
|
vtxos, sumOfReceivers, a.Dust, withExpiryCoinselect,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if changeAmount > 0 {
|
|
offchainAddr, _, err := a.wallet.NewAddress(ctx, true)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
desc, err := a.offchainAddressToDefaultVtxoDescriptor(offchainAddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
changeReceiver := client.Output{
|
|
Descriptor: desc,
|
|
Amount: changeAmount,
|
|
}
|
|
receiversOutput = append(receiversOutput, changeReceiver)
|
|
}
|
|
|
|
inputs := make([]client.Input, 0, len(selectedCoins))
|
|
for _, coin := range selectedCoins {
|
|
inputs = append(inputs, client.Input{
|
|
Outpoint: client.Outpoint{
|
|
Txid: coin.Txid,
|
|
VOut: coin.VOut,
|
|
},
|
|
Descriptor: coin.Descriptor,
|
|
})
|
|
}
|
|
|
|
roundEphemeralKey, err := secp256k1.GeneratePrivateKey()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
paymentID, err := a.client.RegisterInputsForNextRound(
|
|
ctx, inputs, hex.EncodeToString(roundEphemeralKey.PubKey().SerializeCompressed()),
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if err := a.client.RegisterOutputsForNextRound(
|
|
ctx, paymentID, receiversOutput,
|
|
); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
log.Infof("payment registered with id: %s", paymentID)
|
|
|
|
poolTxID, err := a.handleRoundStream(
|
|
ctx, paymentID, selectedCoins, nil, "", receiversOutput, roundEphemeralKey,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return poolTxID, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) addInputs(
|
|
ctx context.Context,
|
|
updater *psbt.Updater,
|
|
utxos []explorer.Utxo,
|
|
) error {
|
|
// TODO works only with single-key wallet
|
|
offchain, _, err := a.wallet.NewAddress(ctx, false)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, userPubkey, aspPubkey, err := common.DecodeAddress(offchain)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, utxo := range utxos {
|
|
previousHash, err := chainhash.NewHashFromStr(utxo.Txid)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sequence, err := utxo.Sequence()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
updater.Upsbt.UnsignedTx.AddTxIn(&wire.TxIn{
|
|
PreviousOutPoint: wire.OutPoint{
|
|
Hash: *previousHash,
|
|
Index: utxo.Vout,
|
|
},
|
|
Sequence: sequence,
|
|
})
|
|
|
|
vtxoScript := &bitcointree.DefaultVtxoScript{
|
|
Owner: userPubkey,
|
|
Asp: aspPubkey,
|
|
ExitDelay: utxo.Delay,
|
|
}
|
|
|
|
exitClosure := &bitcointree.CSVSigClosure{
|
|
Pubkey: userPubkey,
|
|
Seconds: uint(utxo.Delay),
|
|
}
|
|
|
|
exitLeaf, err := exitClosure.Leaf()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, taprootTree, err := vtxoScript.TapTree()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
leafProof, err := taprootTree.GetTaprootMerkleProof(exitLeaf.TapHash())
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get taproot merkle proof: %s", err)
|
|
}
|
|
|
|
updater.Upsbt.Inputs = append(updater.Upsbt.Inputs, psbt.PInput{
|
|
TaprootLeafScript: []*psbt.TaprootTapLeafScript{
|
|
{
|
|
ControlBlock: leafProof.ControlBlock,
|
|
Script: leafProof.Script,
|
|
LeafVersion: txscript.BaseLeafVersion,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) handleRoundStream(
|
|
ctx context.Context,
|
|
paymentID string,
|
|
vtxosToSign []client.Vtxo,
|
|
boardingUtxos []explorer.Utxo,
|
|
boardingDescriptor string,
|
|
receivers []client.Output,
|
|
roundEphemeralKey *secp256k1.PrivateKey,
|
|
) (string, error) {
|
|
eventsCh, close, err := a.client.GetEventStream(ctx, paymentID)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var pingStop func()
|
|
for pingStop == nil {
|
|
pingStop = a.ping(ctx, paymentID)
|
|
}
|
|
|
|
defer func() {
|
|
pingStop()
|
|
close()
|
|
}()
|
|
|
|
var signerSession bitcointree.SignerSession
|
|
|
|
const (
|
|
start = iota
|
|
roundSigningStarted
|
|
roundSigningNoncesGenerated
|
|
roundFinalization
|
|
)
|
|
|
|
step := start
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return "", fmt.Errorf("context done %s", ctx.Err())
|
|
case notify := <-eventsCh:
|
|
if notify.Err != nil {
|
|
return "", notify.Err
|
|
}
|
|
switch event := notify.Event; event.(type) {
|
|
case client.RoundFinalizedEvent:
|
|
if step != roundFinalization {
|
|
continue
|
|
}
|
|
return event.(client.RoundFinalizedEvent).Txid, nil
|
|
case client.RoundFailedEvent:
|
|
return "", fmt.Errorf("round failed: %s", event.(client.RoundFailedEvent).Reason)
|
|
case client.RoundSigningStartedEvent:
|
|
pingStop()
|
|
if step != start {
|
|
continue
|
|
}
|
|
log.Info("a round signing started")
|
|
signerSession, err = a.handleRoundSigningStarted(
|
|
ctx, roundEphemeralKey, event.(client.RoundSigningStartedEvent),
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
step++
|
|
continue
|
|
case client.RoundSigningNoncesGeneratedEvent:
|
|
if step != roundSigningStarted {
|
|
continue
|
|
}
|
|
pingStop()
|
|
log.Info("round combined nonces generated")
|
|
if err := a.handleRoundSigningNoncesGenerated(
|
|
ctx, event.(client.RoundSigningNoncesGeneratedEvent), roundEphemeralKey, signerSession,
|
|
); err != nil {
|
|
return "", err
|
|
}
|
|
step++
|
|
continue
|
|
case client.RoundFinalizationEvent:
|
|
if step != roundSigningNoncesGenerated {
|
|
continue
|
|
}
|
|
pingStop()
|
|
log.Info("a round finalization started")
|
|
|
|
signedForfeitTxs, signedRoundTx, err := a.handleRoundFinalization(
|
|
ctx, event.(client.RoundFinalizationEvent), vtxosToSign, boardingUtxos, boardingDescriptor, receivers,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if len(signedForfeitTxs) <= 0 && len(vtxosToSign) > 0 {
|
|
log.Info("no forfeit txs to sign, waiting for the next round")
|
|
continue
|
|
}
|
|
|
|
log.Info("finalizing payment... ")
|
|
if err := a.client.SubmitSignedForfeitTxs(ctx, signedForfeitTxs, signedRoundTx); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
log.Info("done.")
|
|
log.Info("waiting for round finalization...")
|
|
step++
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (a *covenantlessArkClient) handleRoundSigningStarted(
|
|
ctx context.Context, ephemeralKey *secp256k1.PrivateKey, event client.RoundSigningStartedEvent,
|
|
) (signerSession bitcointree.SignerSession, err error) {
|
|
sweepClosure := bitcointree.CSVSigClosure{
|
|
Pubkey: a.AspPubkey,
|
|
Seconds: uint(a.RoundLifetime),
|
|
}
|
|
|
|
sweepTapLeaf, err := sweepClosure.Leaf()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
roundTx, err := psbt.NewFromRawBytes(strings.NewReader(event.UnsignedRoundTx), true)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
sharedOutput := roundTx.UnsignedTx.TxOut[0]
|
|
sharedOutputValue := sharedOutput.Value
|
|
|
|
sweepTapTree := txscript.AssembleTaprootScriptTree(*sweepTapLeaf)
|
|
root := sweepTapTree.RootNode.TapHash()
|
|
|
|
signerSession = bitcointree.NewTreeSignerSession(
|
|
ephemeralKey, sharedOutputValue, event.UnsignedTree, root.CloneBytes(),
|
|
)
|
|
|
|
if err = signerSession.SetKeys(event.CosignersPublicKeys); err != nil {
|
|
return
|
|
}
|
|
|
|
nonces, err := signerSession.GetNonces()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
myPubKey := hex.EncodeToString(ephemeralKey.PubKey().SerializeCompressed())
|
|
|
|
err = a.arkClient.client.SubmitTreeNonces(ctx, event.ID, myPubKey, nonces)
|
|
|
|
return
|
|
}
|
|
|
|
func (a *covenantlessArkClient) handleRoundSigningNoncesGenerated(
|
|
ctx context.Context,
|
|
event client.RoundSigningNoncesGeneratedEvent,
|
|
ephemeralKey *secp256k1.PrivateKey,
|
|
signerSession bitcointree.SignerSession,
|
|
) error {
|
|
if signerSession == nil {
|
|
return fmt.Errorf("tree signer session not set")
|
|
}
|
|
|
|
if err := signerSession.SetAggregatedNonces(event.Nonces); err != nil {
|
|
return err
|
|
}
|
|
|
|
sigs, err := signerSession.Sign()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := a.arkClient.client.SubmitTreeSignatures(
|
|
ctx,
|
|
event.ID,
|
|
hex.EncodeToString(ephemeralKey.PubKey().SerializeCompressed()),
|
|
sigs,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) handleRoundFinalization(
|
|
ctx context.Context,
|
|
event client.RoundFinalizationEvent,
|
|
vtxos []client.Vtxo,
|
|
boardingUtxos []explorer.Utxo,
|
|
boardingDescriptor string,
|
|
receivers []client.Output,
|
|
) ([]string, string, error) {
|
|
if err := a.validateCongestionTree(event, receivers); err != nil {
|
|
return nil, "", fmt.Errorf("failed to verify congestion tree: %s", err)
|
|
}
|
|
|
|
offchainAddr, _, err := a.wallet.NewAddress(ctx, false)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
_, myPubkey, _, err := common.DecodeAddress(offchainAddr)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
var forfeits []string
|
|
|
|
if len(vtxos) > 0 {
|
|
signedForfeits, err := a.createAndSignForfeits(
|
|
ctx, vtxos, event.Connectors, event.MinRelayFeeRate, myPubkey,
|
|
)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
forfeits = signedForfeits
|
|
}
|
|
|
|
if len(boardingUtxos) > 0 {
|
|
boardingVtxoScript, err := bitcointree.ParseVtxoScript(boardingDescriptor)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
roundPtx, err := psbt.NewFromRawBytes(strings.NewReader(event.Tx), true)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
// add tapscript leaf
|
|
forfeitClosure := &bitcointree.MultisigClosure{
|
|
Pubkey: myPubkey,
|
|
AspPubkey: a.AspPubkey,
|
|
}
|
|
|
|
forfeitLeaf, err := forfeitClosure.Leaf()
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
_, taprootTree, err := boardingVtxoScript.TapTree()
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
forfeitProof, err := taprootTree.GetTaprootMerkleProof(forfeitLeaf.TapHash())
|
|
if err != nil {
|
|
return nil, "", fmt.Errorf("failed to get taproot merkle proof for boarding utxo: %s", err)
|
|
}
|
|
|
|
tapscript := &psbt.TaprootTapLeafScript{
|
|
ControlBlock: forfeitProof.ControlBlock,
|
|
Script: forfeitProof.Script,
|
|
LeafVersion: txscript.BaseLeafVersion,
|
|
}
|
|
|
|
for i := range roundPtx.Inputs {
|
|
previousOutpoint := roundPtx.UnsignedTx.TxIn[i].PreviousOutPoint
|
|
|
|
for _, boardingUtxo := range boardingUtxos {
|
|
if boardingUtxo.Txid == previousOutpoint.Hash.String() && boardingUtxo.Vout == previousOutpoint.Index {
|
|
roundPtx.Inputs[i].TaprootLeafScript = []*psbt.TaprootTapLeafScript{tapscript}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
b64, err := roundPtx.B64Encode()
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
signedRoundTx, err := a.wallet.SignTransaction(ctx, a.explorer, b64)
|
|
if err != nil {
|
|
return nil, "", err
|
|
}
|
|
|
|
return forfeits, signedRoundTx, nil
|
|
}
|
|
|
|
return forfeits, "", nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) validateCongestionTree(
|
|
event client.RoundFinalizationEvent, receivers []client.Output,
|
|
) error {
|
|
poolTx := event.Tx
|
|
ptx, err := psbt.NewFromRawBytes(strings.NewReader(poolTx), true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !utils.IsOnchainOnly(receivers) {
|
|
if err := bitcointree.ValidateCongestionTree(
|
|
event.Tree, poolTx, a.Config.AspPubkey, a.RoundLifetime,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// if err := common.ValidateConnectors(poolTx, event.Connectors); err != nil {
|
|
// return err
|
|
// }
|
|
|
|
if err := a.validateReceivers(
|
|
ptx, receivers, event.Tree,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Info("congestion tree validated")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) validateReceivers(
|
|
ptx *psbt.Packet,
|
|
receivers []client.Output,
|
|
congestionTree tree.CongestionTree,
|
|
) error {
|
|
netParams := utils.ToBitcoinNetwork(a.Network)
|
|
for _, receiver := range receivers {
|
|
isOnChain, onchainScript, err := utils.ParseBitcoinAddress(
|
|
receiver.Address, netParams,
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid receiver address: %s err = %s", receiver.Address, err)
|
|
}
|
|
|
|
if isOnChain {
|
|
if err := a.validateOnChainReceiver(ptx, receiver, onchainScript); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
if err := a.validateOffChainReceiver(
|
|
congestionTree, receiver,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) validateOnChainReceiver(
|
|
ptx *psbt.Packet,
|
|
receiver client.Output,
|
|
onchainScript []byte,
|
|
) error {
|
|
found := false
|
|
for _, output := range ptx.UnsignedTx.TxOut {
|
|
if bytes.Equal(output.PkScript, onchainScript) {
|
|
if output.Value != int64(receiver.Amount) {
|
|
return fmt.Errorf(
|
|
"invalid collaborative exit output amount: got %d, want %d",
|
|
output.Value, receiver.Amount,
|
|
)
|
|
}
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
if !found {
|
|
return fmt.Errorf("collaborative exit output not found: %s", receiver.Address)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) validateOffChainReceiver(
|
|
congestionTree tree.CongestionTree,
|
|
receiver client.Output,
|
|
) error {
|
|
found := false
|
|
|
|
receiverVtxoScript, err := bitcointree.ParseVtxoScript(receiver.Descriptor)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
outputTapKey, _, err := receiverVtxoScript.TapTree()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
leaves := congestionTree.Leaves()
|
|
for _, leaf := range leaves {
|
|
tx, err := psbt.NewFromRawBytes(strings.NewReader(leaf.Tx), true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, output := range tx.UnsignedTx.TxOut {
|
|
if len(output.PkScript) == 0 {
|
|
continue
|
|
}
|
|
|
|
if bytes.Equal(
|
|
output.PkScript[2:], schnorr.SerializePubKey(outputTapKey),
|
|
) {
|
|
if output.Value != int64(receiver.Amount) {
|
|
continue
|
|
}
|
|
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if found {
|
|
break
|
|
}
|
|
}
|
|
|
|
if !found {
|
|
return fmt.Errorf(
|
|
"off-chain send output not found: %s", receiver.Address,
|
|
)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) createAndSignForfeits(
|
|
ctx context.Context,
|
|
vtxosToSign []client.Vtxo,
|
|
connectors []string,
|
|
feeRate chainfee.SatPerKVByte,
|
|
myPubkey *secp256k1.PublicKey,
|
|
) ([]string, error) {
|
|
parsedForfeitAddr, err := btcutil.DecodeAddress(a.ForfeitAddress, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
forfeitPkScript, err := txscript.PayToAddrScript(parsedForfeitAddr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
parsedScript, err := txscript.ParsePkScript(forfeitPkScript)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
signedForfeits := make([]string, 0)
|
|
connectorsPsets := make([]*psbt.Packet, 0, len(connectors))
|
|
|
|
for _, connector := range connectors {
|
|
p, err := psbt.NewFromRawBytes(strings.NewReader(connector), true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
connectorsPsets = append(connectorsPsets, p)
|
|
}
|
|
|
|
for _, vtxo := range vtxosToSign {
|
|
vtxoScript, err := bitcointree.ParseVtxoScript(vtxo.Descriptor)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vtxoTapKey, vtxoTapTree, err := vtxoScript.TapTree()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
feeAmount, err := common.ComputeForfeitMinRelayFee(feeRate, vtxoTapTree, parsedScript.Class())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vtxoOutputScript, err := common.P2TRScript(vtxoTapKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vtxoTxHash, err := chainhash.NewHashFromStr(vtxo.Txid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
vtxoInput := &wire.OutPoint{
|
|
Hash: *vtxoTxHash,
|
|
Index: vtxo.VOut,
|
|
}
|
|
|
|
forfeitClosure := &bitcointree.MultisigClosure{
|
|
Pubkey: myPubkey,
|
|
AspPubkey: a.AspPubkey,
|
|
}
|
|
|
|
forfeitLeaf, err := forfeitClosure.Leaf()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
leafProof, err := vtxoTapTree.GetTaprootMerkleProof(forfeitLeaf.TapHash())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tapscript := psbt.TaprootTapLeafScript{
|
|
ControlBlock: leafProof.ControlBlock,
|
|
Script: leafProof.Script,
|
|
LeafVersion: txscript.BaseLeafVersion,
|
|
}
|
|
|
|
for _, connectorPset := range connectorsPsets {
|
|
forfeits, err := bitcointree.BuildForfeitTxs(
|
|
connectorPset, vtxoInput, vtxo.Amount, a.Dust, feeAmount, vtxoOutputScript, forfeitPkScript,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(forfeits) <= 0 {
|
|
return nil, fmt.Errorf("no forfeit txs created dust = %d", a.Dust)
|
|
}
|
|
|
|
for _, forfeit := range forfeits {
|
|
forfeit.Inputs[1].TaprootLeafScript = []*psbt.TaprootTapLeafScript{&tapscript}
|
|
|
|
b64, err := forfeit.B64Encode()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
signedForfeit, err := a.wallet.SignTransaction(ctx, a.explorer, b64)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
signedForfeits = append(signedForfeits, signedForfeit)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
return signedForfeits, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) coinSelectOnchain(
|
|
ctx context.Context, targetAmount uint64, exclude []explorer.Utxo,
|
|
) ([]explorer.Utxo, uint64, error) {
|
|
offchainAddrs, boardingAddrs, redemptionAddrs, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
_, myPubkey, _, err := common.DecodeAddress(offchainAddrs[0])
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
myPubkeyStr := hex.EncodeToString(schnorr.SerializePubKey(myPubkey))
|
|
descriptorStr := strings.ReplaceAll(
|
|
a.BoardingDescriptorTemplate, "USER", myPubkeyStr,
|
|
)
|
|
|
|
boardingScript, err := bitcointree.ParseVtxoScript(descriptorStr)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
var boardingTimeout uint
|
|
|
|
if defaultVtxo, ok := boardingScript.(*bitcointree.DefaultVtxoScript); ok {
|
|
boardingTimeout = defaultVtxo.ExitDelay
|
|
} else {
|
|
return nil, 0, fmt.Errorf("unsupported boarding descriptor: %s", descriptorStr)
|
|
}
|
|
|
|
now := time.Now()
|
|
|
|
fetchedUtxos := make([]explorer.Utxo, 0)
|
|
for _, addr := range boardingAddrs {
|
|
utxos, err := a.explorer.GetUtxos(addr)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
for _, utxo := range utxos {
|
|
u := utxo.ToUtxo(boardingTimeout)
|
|
if u.SpendableAt.Before(now) {
|
|
fetchedUtxos = append(fetchedUtxos, u)
|
|
}
|
|
}
|
|
}
|
|
|
|
selected := make([]explorer.Utxo, 0)
|
|
selectedAmount := uint64(0)
|
|
for _, utxo := range fetchedUtxos {
|
|
if selectedAmount >= targetAmount {
|
|
break
|
|
}
|
|
|
|
for _, excluded := range exclude {
|
|
if utxo.Txid == excluded.Txid && utxo.Vout == excluded.Vout {
|
|
continue
|
|
}
|
|
}
|
|
|
|
selected = append(selected, utxo)
|
|
selectedAmount += utxo.Amount
|
|
}
|
|
|
|
if selectedAmount >= targetAmount {
|
|
return selected, selectedAmount - targetAmount, nil
|
|
}
|
|
|
|
fetchedUtxos = make([]explorer.Utxo, 0)
|
|
for _, addr := range redemptionAddrs {
|
|
utxos, err := a.explorer.GetUtxos(addr)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
for _, utxo := range utxos {
|
|
u := utxo.ToUtxo(uint(a.UnilateralExitDelay))
|
|
if u.SpendableAt.Before(now) {
|
|
fetchedUtxos = append(fetchedUtxos, u)
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, utxo := range fetchedUtxos {
|
|
if selectedAmount >= targetAmount {
|
|
break
|
|
}
|
|
|
|
for _, excluded := range exclude {
|
|
if utxo.Txid == excluded.Txid && utxo.Vout == excluded.Vout {
|
|
continue
|
|
}
|
|
}
|
|
|
|
selected = append(selected, utxo)
|
|
selectedAmount += utxo.Amount
|
|
}
|
|
|
|
if selectedAmount < targetAmount {
|
|
return nil, 0, fmt.Errorf(
|
|
"not enough funds to cover amount %d", targetAmount,
|
|
)
|
|
}
|
|
|
|
return selected, selectedAmount - targetAmount, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) getRedeemBranches(
|
|
ctx context.Context, vtxos []client.Vtxo,
|
|
) (map[string]*redemption.CovenantlessRedeemBranch, error) {
|
|
congestionTrees := make(map[string]tree.CongestionTree, 0)
|
|
redeemBranches := make(map[string]*redemption.CovenantlessRedeemBranch, 0)
|
|
|
|
for i := range vtxos {
|
|
vtxo := vtxos[i]
|
|
|
|
// TODO: handle exit for pending changes
|
|
if vtxo.RedeemTx != "" {
|
|
continue
|
|
}
|
|
|
|
if _, ok := congestionTrees[vtxo.RoundTxid]; !ok {
|
|
round, err := a.client.GetRound(ctx, vtxo.RoundTxid)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
congestionTrees[vtxo.RoundTxid] = round.Tree
|
|
}
|
|
|
|
redeemBranch, err := redemption.NewCovenantlessRedeemBranch(
|
|
a.explorer, congestionTrees[vtxo.RoundTxid], vtxo,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
redeemBranches[vtxo.Txid] = redeemBranch
|
|
}
|
|
|
|
return redeemBranches, nil
|
|
}
|
|
|
|
// TODO (@louisinger): return pending balance in dedicated map.
|
|
// Currently, the returned balance is calculated from both spendable and
|
|
// pending vtxos.
|
|
func (a *covenantlessArkClient) getOffchainBalance(
|
|
ctx context.Context, addr string, computeVtxoExpiration bool,
|
|
) (uint64, map[int64]uint64, error) {
|
|
amountByExpiration := make(map[int64]uint64, 0)
|
|
|
|
vtxos, _, err := a.getVtxos(ctx, addr, computeVtxoExpiration)
|
|
if err != nil {
|
|
return 0, nil, err
|
|
}
|
|
var balance uint64
|
|
for _, vtxo := range vtxos {
|
|
balance += vtxo.Amount
|
|
|
|
if vtxo.ExpiresAt != nil {
|
|
expiration := vtxo.ExpiresAt.Unix()
|
|
|
|
if _, ok := amountByExpiration[expiration]; !ok {
|
|
amountByExpiration[expiration] = 0
|
|
}
|
|
|
|
amountByExpiration[expiration] += vtxo.Amount
|
|
}
|
|
}
|
|
|
|
return balance, amountByExpiration, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) getAllBoardingUtxos(
|
|
ctx context.Context,
|
|
) ([]explorer.Utxo, map[string]struct{}, error) {
|
|
_, boardingAddrs, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
utxos := []explorer.Utxo{}
|
|
ignoreVtxos := make(map[string]struct{}, 0)
|
|
for _, addr := range boardingAddrs {
|
|
txs, err := a.explorer.GetTxs(addr)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
for _, tx := range txs {
|
|
for i, vout := range tx.Vout {
|
|
if vout.Address == addr {
|
|
spentStatuses, err := a.explorer.GetTxOutspends(tx.Txid)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
if s := spentStatuses[i]; s.Spent {
|
|
ignoreVtxos[s.SpentBy] = struct{}{}
|
|
}
|
|
createdAt := time.Time{}
|
|
if tx.Status.Confirmed {
|
|
createdAt = time.Unix(tx.Status.Blocktime, 0)
|
|
}
|
|
utxos = append(utxos, explorer.Utxo{
|
|
Txid: tx.Txid,
|
|
Vout: uint32(i),
|
|
Amount: vout.Amount,
|
|
CreatedAt: createdAt,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return utxos, ignoreVtxos, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) getClaimableBoardingUtxos(ctx context.Context) ([]explorer.Utxo, error) {
|
|
offchainAddrs, boardingAddrs, _, err := a.wallet.GetAddresses(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, myPubkey, _, err := common.DecodeAddress(offchainAddrs[0])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
myPubkeyStr := hex.EncodeToString(schnorr.SerializePubKey(myPubkey))
|
|
descriptorStr := strings.ReplaceAll(
|
|
a.BoardingDescriptorTemplate, "USER", myPubkeyStr,
|
|
)
|
|
|
|
boardingScript, err := bitcointree.ParseVtxoScript(descriptorStr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var boardingTimeout uint
|
|
|
|
if defaultVtxo, ok := boardingScript.(*bitcointree.DefaultVtxoScript); ok {
|
|
boardingTimeout = defaultVtxo.ExitDelay
|
|
} else {
|
|
return nil, fmt.Errorf("unsupported boarding descriptor: %s", descriptorStr)
|
|
}
|
|
|
|
claimable := make([]explorer.Utxo, 0)
|
|
now := time.Now()
|
|
|
|
for _, addr := range boardingAddrs {
|
|
boardingUtxos, err := a.explorer.GetUtxos(addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, utxo := range boardingUtxos {
|
|
u := utxo.ToUtxo(boardingTimeout)
|
|
if u.SpendableAt.Before(now) {
|
|
continue
|
|
}
|
|
claimable = append(claimable, u)
|
|
}
|
|
|
|
}
|
|
|
|
return claimable, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) getVtxos(
|
|
ctx context.Context, _ string, computeVtxoExpiration bool,
|
|
) ([]client.Vtxo, []client.Vtxo, error) {
|
|
spendableVtxos, _, err := a.ListVtxos(ctx)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
pendingVtxos := make([]client.Vtxo, 0)
|
|
for _, vtxo := range spendableVtxos {
|
|
if vtxo.RedeemTx != "" {
|
|
pendingVtxos = append(pendingVtxos, vtxo)
|
|
}
|
|
}
|
|
|
|
if !computeVtxoExpiration {
|
|
return spendableVtxos, pendingVtxos, nil
|
|
}
|
|
|
|
redeemBranches, err := a.getRedeemBranches(ctx, spendableVtxos)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
for vtxoTxid, branch := range redeemBranches {
|
|
expiration, err := branch.ExpiresAt()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
for i, vtxo := range spendableVtxos {
|
|
if vtxo.Txid == vtxoTxid {
|
|
spendableVtxos[i].ExpiresAt = expiration
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
return spendableVtxos, pendingVtxos, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) selfTransferAllPendingPayments(
|
|
ctx context.Context, pendingVtxos []client.Vtxo, boardingUtxos []explorer.Utxo, myself client.Output, mypubkey string,
|
|
) (string, error) {
|
|
inputs := make([]client.Input, 0, len(pendingVtxos)+len(boardingUtxos))
|
|
|
|
boardingDescriptor := strings.ReplaceAll(
|
|
a.BoardingDescriptorTemplate, "USER", mypubkey[2:],
|
|
)
|
|
|
|
for _, coin := range pendingVtxos {
|
|
inputs = append(inputs, client.Input{
|
|
Outpoint: client.Outpoint{
|
|
Txid: coin.Txid,
|
|
VOut: coin.VOut,
|
|
},
|
|
Descriptor: coin.Descriptor,
|
|
})
|
|
}
|
|
|
|
for _, utxo := range boardingUtxos {
|
|
inputs = append(inputs, client.Input{
|
|
Outpoint: client.Outpoint{
|
|
Txid: utxo.Txid,
|
|
VOut: utxo.Vout,
|
|
},
|
|
Descriptor: boardingDescriptor,
|
|
})
|
|
}
|
|
outputs := []client.Output{myself}
|
|
|
|
roundEphemeralKey, err := secp256k1.GeneratePrivateKey()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
paymentID, err := a.client.RegisterInputsForNextRound(
|
|
ctx,
|
|
inputs,
|
|
hex.EncodeToString(roundEphemeralKey.PubKey().SerializeCompressed()),
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if err := a.client.RegisterOutputsForNextRound(ctx, paymentID, outputs); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
roundTxid, err := a.handleRoundStream(
|
|
ctx, paymentID, pendingVtxos, boardingUtxos, boardingDescriptor, outputs, roundEphemeralKey,
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return roundTxid, nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) offchainAddressToReversibleVtxoDescriptor(myaddr string, receiveraddr string) (string, error) {
|
|
_, receiverPubkey, aspPubkey, err := common.DecodeAddress(receiveraddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
_, userPubKey, _, err := common.DecodeAddress(myaddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
vtxoScript := bitcointree.ReversibleVtxoScript{
|
|
Owner: receiverPubkey,
|
|
Sender: userPubKey,
|
|
Asp: aspPubkey,
|
|
ExitDelay: uint(a.UnilateralExitDelay),
|
|
}
|
|
|
|
return vtxoScript.ToDescriptor(), nil
|
|
}
|
|
|
|
func (a *covenantlessArkClient) offchainAddressToDefaultVtxoDescriptor(addr string) (string, error) {
|
|
_, userPubKey, aspPubkey, err := common.DecodeAddress(addr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
vtxoScript := bitcointree.DefaultVtxoScript{
|
|
Owner: userPubKey,
|
|
Asp: aspPubkey,
|
|
ExitDelay: uint(a.UnilateralExitDelay),
|
|
}
|
|
|
|
return vtxoScript.ToDescriptor(), nil
|
|
}
|
|
|
|
// getBoardingTxs builds the boarding tx history from onchain utxos:
|
|
// - unspent utxo => pending boarding tx
|
|
// - spent utxo => claimed boarding tx
|
|
//
|
|
// The tx spending an onchain utxo is an ark round, therefore an indexed list
|
|
// of round txids is returned to specify the vtxos to be ignored to build the
|
|
// offchain tx history and prevent duplicates.
|
|
func (a *covenantlessArkClient) getBoardingTxs(
|
|
ctx context.Context,
|
|
) ([]types.Transaction, map[string]struct{}, error) {
|
|
utxos, err := a.getClaimableBoardingUtxos(ctx)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
isPending := make(map[string]bool)
|
|
for _, u := range utxos {
|
|
isPending[u.Txid] = true
|
|
}
|
|
|
|
allUtxos, ignoreVtxos, err := a.getAllBoardingUtxos(ctx)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
unconfirmedTxs := make([]types.Transaction, 0)
|
|
confirmedTxs := make([]types.Transaction, 0)
|
|
for _, u := range allUtxos {
|
|
pending := false
|
|
if isPending[u.Txid] {
|
|
pending = true
|
|
}
|
|
|
|
tx := types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
BoardingTxid: u.Txid,
|
|
},
|
|
Amount: u.Amount,
|
|
Type: types.TxReceived,
|
|
IsPending: pending,
|
|
CreatedAt: u.CreatedAt,
|
|
}
|
|
|
|
emptyTime := time.Time{}
|
|
if u.CreatedAt == emptyTime {
|
|
unconfirmedTxs = append(unconfirmedTxs, tx)
|
|
continue
|
|
}
|
|
confirmedTxs = append(confirmedTxs, tx)
|
|
}
|
|
|
|
txs := append(unconfirmedTxs, confirmedTxs...)
|
|
return txs, ignoreVtxos, nil
|
|
}
|
|
|
|
func findVtxosBySpentBy(allVtxos []client.Vtxo, txid string) (vtxos []client.Vtxo) {
|
|
for _, v := range allVtxos {
|
|
if v.SpentBy == txid {
|
|
vtxos = append(vtxos, v)
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func vtxosToTxsCovenantless(
|
|
roundLifetime int64, spendable, spent []client.Vtxo, ignoreVtxos map[string]struct{},
|
|
) ([]types.Transaction, error) {
|
|
transactions := make([]types.Transaction, 0)
|
|
|
|
indexedTxs := make(map[string]types.Transaction)
|
|
for _, v := range spent {
|
|
// If the vtxo was pending and is spent => it's been claimed.
|
|
if v.Pending {
|
|
transactions = append(transactions, types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
RedeemTxid: v.Txid,
|
|
},
|
|
Amount: v.Amount,
|
|
Type: types.TxReceived,
|
|
IsPending: false,
|
|
CreatedAt: getCreatedAtFromExpiry(roundLifetime, *v.ExpiresAt),
|
|
})
|
|
// Delete any duplicate in the indexed list.
|
|
delete(indexedTxs, v.SpentBy)
|
|
// Ignore the spendable vtxo created by the claim.
|
|
ignoreVtxos[v.SpentBy] = struct{}{}
|
|
continue
|
|
}
|
|
|
|
// If this vtxo spent another one => subtract the amount to find the sent amount.
|
|
if tx, ok := indexedTxs[v.Txid]; ok {
|
|
tx.Amount -= v.Amount
|
|
if v.RedeemTx == "" {
|
|
tx.RedeemTxid = ""
|
|
} else {
|
|
tx.RoundTxid = ""
|
|
}
|
|
indexedTxs[v.Txid] = tx
|
|
}
|
|
|
|
// Add a transaction to the indexed list if not existing, it will be deleted if it's a duplicate.
|
|
tx, ok := indexedTxs[v.SpentBy]
|
|
if !ok {
|
|
indexedTxs[v.SpentBy] = types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
RedeemTxid: v.SpentBy,
|
|
RoundTxid: v.SpentBy,
|
|
},
|
|
Amount: v.Amount,
|
|
Type: types.TxSent,
|
|
IsPending: false,
|
|
CreatedAt: getCreatedAtFromExpiry(roundLifetime, *v.ExpiresAt),
|
|
}
|
|
continue
|
|
}
|
|
|
|
// Otherwise add the amount of this vtxo to the one of the tx in the indexed list.
|
|
tx.Amount += v.Amount
|
|
indexedTxs[v.SpentBy] = tx
|
|
}
|
|
|
|
for _, v := range spendable {
|
|
_, ok1 := ignoreVtxos[v.Txid]
|
|
_, ok2 := ignoreVtxos[v.RoundTxid]
|
|
if ok1 || ok2 {
|
|
continue
|
|
}
|
|
txid := v.RoundTxid
|
|
if txid == "" {
|
|
txid = v.Txid
|
|
}
|
|
|
|
tx, ok := indexedTxs[txid]
|
|
if !ok {
|
|
redeemTxid := ""
|
|
if v.RoundTxid == "" {
|
|
redeemTxid = v.Txid
|
|
}
|
|
transactions = append(transactions, types.Transaction{
|
|
TransactionKey: types.TransactionKey{
|
|
RedeemTxid: redeemTxid,
|
|
RoundTxid: v.RoundTxid,
|
|
},
|
|
Amount: v.Amount,
|
|
Type: types.TxReceived,
|
|
IsPending: v.Pending,
|
|
CreatedAt: getCreatedAtFromExpiry(roundLifetime, *v.ExpiresAt),
|
|
})
|
|
continue
|
|
}
|
|
|
|
tx.Amount -= v.Amount
|
|
if v.RedeemTx == "" {
|
|
tx.RedeemTxid = ""
|
|
} else {
|
|
tx.RoundTxid = ""
|
|
}
|
|
indexedTxs[txid] = tx
|
|
}
|
|
|
|
for _, tx := range indexedTxs {
|
|
transactions = append(transactions, tx)
|
|
}
|
|
|
|
return transactions, nil
|
|
}
|