new api

Author: sputn1ck

hyperloop/actions.go

@@ -213,6 +213,8 @@ type FSM struct {
 
 	cfg *Config
 
+	swapAmtFetcher TotalSwapAmountFetcher
+
 	// hyperloop contains all the data required for the hyperloop process.
 	hyperloop *HyperLoop
 
@@ -232,23 +234,25 @@ type FSM struct {
 }
 
 // NewFSM creates a new instant out FSM.
-func NewFSM(cfg *Config) (*FSM, error) {
+func NewFSM(cfg *Config, swapAmtFetcher TotalSwapAmountFetcher) (*FSM, error) {
 
 	hyperloop := &HyperLoop{
 		State: fsm.EmptyState,
 	}
 
-	return NewFSMFromHyperloop(cfg, hyperloop)
+	return NewFSMFromHyperloop(cfg, hyperloop, swapAmtFetcher)
 }
 
 // NewFSMFromHyperloop creates a new instantout FSM from an existing instantout
 // recovered from the database.
-func NewFSMFromHyperloop(cfg *Config, hyperloop *HyperLoop) (*FSM, error) {
+func NewFSMFromHyperloop(cfg *Config, hyperloop *HyperLoop,
+	swapAmtFetcher TotalSwapAmountFetcher) (*FSM, error) {
 
 	instantOutFSM := &FSM{
-		cfg:       cfg,
-		hyperloop: hyperloop,
-		spendChan: make(chan *chainntnfs.SpendDetail),
+		cfg:            cfg,
+		hyperloop:      hyperloop,
+		spendChan:      make(chan *chainntnfs.SpendDetail),
+		swapAmtFetcher: swapAmtFetcher,
 	}
 
 	instantOutFSM.ActionEntryFunc = instantOutFSM.updateHyperloop