reservation: add protocol version

Author: sputn1ck

fsm/stateparser/stateparser.go

@@ -45,7 +45,7 @@ func run() error {
 
 	case "reservation":
 		reservationFSM := &reservation.FSM{}
-		err = writeMermaidFile(fp, reservationFSM.GetReservationStates())
+		err = writeMermaidFile(fp, reservationFSM.GetServerInitiatedReservationStates())
 		if err != nil {
 			return err
 		}

instantout/reservation/actions.go

@@ -58,6 +58,7 @@ func (f *FSM) InitAction(ctx context.Context,
 		reservationRequest.expiry,
 		reservationRequest.heightHint,
 		keyRes.KeyLocator,
+		ProtocolVersionServerInitiated,
 	)
 	if err != nil {
 		return f.HandleError(err)

instantout/reservation/fsm.go

@@ -8,6 +8,17 @@ import (
 	"github.com/lightninglabs/loop/swapserverrpc"
 )
 
+type ProtocolVersion uint32
+
+const (
+	// ProtocolVersionUndefined is the default protocol version.
+	ProtocolVersionUndefined ProtocolVersion = 0
+
+	// ProtocolVersionServerInitiated is the protocol version where the
+	// server initiates the reservation.
+	ProtocolVersionServerInitiated ProtocolVersion = 1
+)
+
 const (
 	// defaultObserverSize is the size of the fsm observer channel.
 	defaultObserverSize = 15
@@ -43,9 +54,10 @@ type FSM struct {
 }
 
 // NewFSM creates a new reservation FSM.
-func NewFSM(cfg *Config) *FSM {
+func NewFSM(cfg *Config, protocolVersion ProtocolVersion) *FSM {
 	reservation := &Reservation{
-		State: fsm.EmptyState,
+		State:           fsm.EmptyState,
+		ProtocolVersion: protocolVersion,
 	}
 
 	return NewFSMFromReservation(cfg, reservation)
@@ -54,15 +66,24 @@ func NewFSM(cfg *Config) *FSM {
 // NewFSMFromReservation creates a new reservation FSM from an existing
 // reservation recovered from the database.
 func NewFSMFromReservation(cfg *Config, reservation *Reservation) *FSM {
+
 	reservationFsm := &FSM{
 		cfg:         cfg,
 		reservation: reservation,
 	}
 
+	var states fsm.States
+	switch reservation.ProtocolVersion {
+	case ProtocolVersionServerInitiated:
+		states = reservationFsm.GetServerInitiatedReservationStates()
+	default:
+		states = make(fsm.States)
+	}
+
 	reservationFsm.StateMachine = fsm.NewStateMachineWithState(
-		reservationFsm.GetReservationStates(), reservation.State,
-		defaultObserverSize,
+		states, reservation.State, defaultObserverSize,
 	)
+
 	reservationFsm.ActionEntryFunc = reservationFsm.updateReservation
 
 	return reservationFsm
@@ -133,9 +154,9 @@ var (
 	OnUnlocked = fsm.EventType("OnUnlocked")
 )
 
-// GetReservationStates returns the statemap that defines the reservation
-// state machine.
-func (f *FSM) GetReservationStates() fsm.States {
+// GetServerInitiatedReservationStates returns the statemap that defines the
+// reservation state machine, where the server initiates the reservation.
+func (f *FSM) GetServerInitiatedReservationStates() fsm.States {
 	return fsm.States{
 		fsm.EmptyState: fsm.State{
 			Transitions: fsm.Transitions{

instantout/reservation/manager.go

@@ -113,7 +113,7 @@ func (m *Manager) newReservation(ctx context.Context, currentHeight uint32,
 	// Create the reservation state machine. We need to pass in the runCtx
 	// of the reservation manager so that the state machine will keep on
 	// running even if the grpc conte
-	reservationFSM := NewFSM(m.cfg)
+	reservationFSM := NewFSM(m.cfg, ProtocolVersionServerInitiated)
 
 	// Add the reservation to the active reservations map.
 	m.Lock()

instantout/reservation/reservation.go

@@ -37,6 +37,10 @@ type Reservation struct {
 	// ID is the unique identifier of the reservation.
 	ID ID
 
+	// ProtocolVersion is the version of the protocol used for the
+	// reservation.
+	ProtocolVersion ProtocolVersion
+
 	// State is the current state of the reservation.
 	State fsm.StateType
 
@@ -69,8 +73,8 @@ type Reservation struct {
 
 func NewReservation(id ID, serverPubkey, clientPubkey *btcec.PublicKey,
 	value btcutil.Amount, expiry, heightHint uint32,
-	keyLocator keychain.KeyLocator) (*Reservation,
-	error) {
+	keyLocator keychain.KeyLocator, protocolVersion ProtocolVersion) (
+	*Reservation, error) {
 
 	if id == [32]byte{} {
 		return nil, errors.New("id is empty")
@@ -103,6 +107,7 @@ func NewReservation(id ID, serverPubkey, clientPubkey *btcec.PublicKey,
 		KeyLocator:       keyLocator,
 		Expiry:           expiry,
 		InitiationHeight: int32(heightHint),
+		ProtocolVersion:  protocolVersion,
 	}, nil
 }
 

instantout/reservation/store.go

@@ -83,6 +83,7 @@ func (r *SQLStore) CreateReservation(ctx context.Context,
 		ClientKeyFamily:  int32(reservation.KeyLocator.Family),
 		ClientKeyIndex:   int32(reservation.KeyLocator.Index),
 		InitiationHeight: reservation.InitiationHeight,
+		ProtocolVersion:  int32(reservation.ProtocolVersion),
 	}
 
 	updateArgs := sqlc.InsertReservationUpdateParams{
@@ -287,6 +288,7 @@ func sqlReservationToReservation(row sqlc.Reservation,
 		),
 		InitiationHeight: row.InitiationHeight,
 		State:            fsm.StateType(lastUpdate.UpdateState),
+		ProtocolVersion:  ProtocolVersion(row.ProtocolVersion),
 	}, nil
 }
 

instantout/reservation/store_test.go

@@ -33,6 +33,7 @@ func TestSqlStore(t *testing.T) {
 			Family: 1,
 			Index:  1,
 		},
+		ProtocolVersion: ProtocolVersionServerInitiated,
 	}
 
 	err := store.CreateReservation(ctxb, reservation)

loopdb/sqlc/migrations/000014_reservation_protocol_version.down.sql

@@ -0,0 +1,3 @@
+-- protocol_version is used to determine the version of the reservation protocol
+-- that was used to create the reservation.
+ALTER TABLE reservations DROP COLUMN protocol_Version;

loopdb/sqlc/migrations/000014_reservation_protocol_version.up.sql

@@ -0,0 +1,3 @@
+-- protocol_version is used to determine the version of the reservation protocol
+-- that was used to create the reservation.
+ALTER TABLE reservations ADD COLUMN protocol_Version INTEGER NOT NULL DEFAULT 1;