staticaddr: deposit timeout fsm

Author: hieblmi

staticaddr/fsm.go

@@ -0,0 +1,173 @@
+package staticaddr
+
+import (
+	"context"
+
+	"github.com/btcsuite/btcd/chaincfg"
+	"github.com/btcsuite/btcd/txscript"
+	"github.com/btcsuite/btcd/wire"
+	"github.com/lightninglabs/lndclient"
+	"github.com/lightninglabs/loop/fsm"
+	"github.com/lightninglabs/loop/staticaddr/script"
+	"github.com/lightningnetwork/lnd/input"
+	"github.com/lightningnetwork/lnd/keychain"
+)
+
+// DepositContext contains the context parameters for a deposit state machine.
+type DepositContext struct {
+	Deposit *Deposit
+}
+
+// States.
+var (
+	Deposited           = fsm.StateType("Deposited")
+	SweepExpiredDeposit = fsm.StateType("SweepExpiredDeposit")
+
+	PublishedExpiredDeposit = fsm.StateType("PublishedExpiredDeposit")
+
+	Failed = fsm.StateType("DepositFailed")
+)
+
+// Events.
+var (
+	OnStart  = fsm.EventType("OnStart")
+	OnExpiry = fsm.EventType("OnExpiry")
+	OnSwept  = fsm.EventType("OnSwept")
+)
+
+// Config contains the services required for the instant out FSM.
+type Config struct {
+	// Store is used to store the instant out.
+	Store SqlStore
+
+	Lnd lndclient.LndServices
+
+	Manager *ManagerConfig
+
+	// Network is the network that is used for the swap.
+	Network *chaincfg.Params
+}
+
+// FSM is the state machine that handles the instant out.
+type FSM struct {
+	*fsm.StateMachine
+
+	addressParameters *AddressParameters
+
+	staticAddress *script.StaticAddress
+
+	deposit *Deposit
+
+	ctx context.Context
+
+	// cfg contains all the services that the reservation manager needs to
+	// operate.
+	cfg *ManagerConfig
+}
+
+// NewFSM creates a new state machine that can action on all static address
+// feature requests.
+func NewFSM(ctx context.Context, addressParameters *AddressParameters,
+	staticAddress *script.StaticAddress, deposit *Deposit,
+	cfg *ManagerConfig) (*FSM, error) {
+
+	depoFsm := &FSM{
+		ctx:               ctx,
+		addressParameters: addressParameters,
+		staticAddress:     staticAddress,
+		deposit:           deposit,
+		cfg:               cfg,
+	}
+
+	depoFsm.StateMachine = fsm.NewStateMachine(depoFsm.DepositStates(), 20)
+	depoFsm.ActionEntryFunc = depoFsm.DepositEntryFunction
+
+	return depoFsm, nil
+}
+
+// DepositStates returns the states a deposit can be in.
+func (f *FSM) DepositStates() fsm.States {
+	return fsm.States{
+		fsm.EmptyState: fsm.State{
+			Transitions: fsm.Transitions{
+				OnStart: Deposited,
+			},
+			Action: fsm.NoOpAction,
+		},
+		Deposited: fsm.State{
+			Transitions: fsm.Transitions{
+				OnExpiry:    SweepExpiredDeposit,
+				fsm.OnError: Failed,
+			},
+			Action: fsm.NoOpAction,
+		},
+		SweepExpiredDeposit: fsm.State{
+			Transitions: fsm.Transitions{
+				OnSwept:     PublishedExpiredDeposit,
+				fsm.OnError: Failed,
+			},
+			Action: f.SweepExpiredDepositAction,
+		},
+		PublishedExpiredDeposit: fsm.State{
+			Action: fsm.NoOpAction,
+		},
+		Failed: fsm.State{
+			Action: fsm.NoOpAction,
+		},
+	}
+}
+
+// DepositEntryFunction is called after every action and updates the reservation
+// in the db.
+func (f *FSM) DepositEntryFunction(notification fsm.Notification) {
+	log.Debugf("DepositEntryFunction: %v", notification)
+}
+
+// Infof logs an info message with the deposit outpoint.
+func (f *FSM) Infof(format string, args ...interface{}) {
+	log.Infof(
+		"Deposit %v: "+format,
+		append(
+			[]interface{}{f.deposit.OutPoint},
+			args...,
+		)...,
+	)
+}
+
+// Debugf logs a debug message with the deposit outpoint.
+func (f *FSM) Debugf(format string, args ...interface{}) {
+	log.Debugf(
+		"Deposit %v: "+format,
+		append(
+			[]interface{}{f.deposit.OutPoint},
+			args...,
+		)...,
+	)
+}
+
+// Errorf logs an error message with the deposit outpoint.
+func (f *FSM) Errorf(format string, args ...interface{}) {
+	log.Errorf(
+		"Deposit %v: "+format,
+		append(
+			[]interface{}{f.deposit.OutPoint},
+			args...,
+		)...,
+	)
+}
+
+// SignDescriptor returns the sign descriptor for the static address output.
+func (f *FSM) SignDescriptor() (lndclient.SignDescriptor, error) {
+	return lndclient.SignDescriptor{
+		WitnessScript: f.staticAddress.TimeoutLeaf.Script,
+		KeyDesc: keychain.KeyDescriptor{
+			PubKey: f.addressParameters.ClientPubkey,
+		},
+		Output: wire.NewTxOut(
+			int64(f.deposit.Value), f.addressParameters.PkScript,
+		),
+		HashType:   txscript.SigHashDefault,
+		InputIndex: 0,
+		SignMethod: input.TaprootScriptSpendSignMethod,
+	}, nil
+}