StatefulSwap now checks for unlocked Timelock accounts (#177)

Author: jeffyanta

ocp/rpc/transaction/swap.go

@@ -166,13 +166,16 @@ func (s *transactionServer) StatefulSwap(streamer transactionpb.Transaction_Stat
 		return handleStatefulSwapError(streamer, err)
 	}
 
-	_, err = s.data.GetTimelockByVault(ctx, ownerSourceTimelockVault.PublicKey().ToBase58())
+	sourceTimelockAccountRecord, err := s.data.GetTimelockByVault(ctx, ownerSourceTimelockVault.PublicKey().ToBase58())
 	if err == timelock.ErrTimelockNotFound {
 		return handleStatefulSwapError(streamer, NewSwapValidationError("source timelock vault account not opened"))
 	} else if err != nil {
 		log.With(zap.Error(err)).Warn("failure getting source timelock record")
 		return handleStatefulSwapError(streamer, err)
 	}
+	if !sourceTimelockAccountRecord.IsLocked() {
+		return handleStatefulSwapError(streamer, NewSwapDeniedError("source timelock account isn't locked"))
+	}
 
 	switch initiateReserveSwapReq.FundingSource {
 	case transactionpb.FundingSource_FUNDING_SOURCE_SUBMIT_INTENT:
@@ -273,13 +276,16 @@ func (s *transactionServer) StatefulSwap(streamer transactionpb.Transaction_Stat
 			return handleStatefulSwapError(streamer, err)
 		}
 
-		_, err = s.data.GetTimelockByVault(ctx, ownerDestinationTimelockVault.PublicKey().ToBase58())
+		destinationTimelockAccountRecord, err := s.data.GetTimelockByVault(ctx, ownerDestinationTimelockVault.PublicKey().ToBase58())
 		if err == timelock.ErrTimelockNotFound {
 			return handleStatefulSwapError(streamer, NewSwapValidationError("destination timelock vault account not opened"))
 		} else if err != nil {
 			log.With(zap.Error(err)).Warn("failure getting destination timelock record")
 			return handleStatefulSwapError(streamer, err)
 		}
+		if !destinationTimelockAccountRecord.IsLocked() {
+			return handleStatefulSwapError(streamer, NewSwapDeniedError("destination timelock account isn't locked"))
+		}
 
 		err = vm.EnsureVirtualTimelockAccountIsInitialized(ctx, s.data, ownerDestinationTimelockVault, false)
 		if err != nil {