feat: make txsubmission timeout configurable

Fixes #171

Author: agaffney

protocol/txsubmission/client.go

@@ -11,9 +11,20 @@ type Client struct {
 }
 
 func NewClient(protoOptions protocol.ProtocolOptions, cfg *Config) *Client {
+	if cfg == nil {
+		tmpCfg := NewConfig()
+		cfg = &tmpCfg
+	}
 	c := &Client{
 		config: cfg,
 	}
+	// Update state map with timeout
+	stateMap := StateMap.Copy()
+	if entry, ok := stateMap[STATE_IDLE]; ok {
+		entry.Timeout = c.config.IdleTimeout
+		stateMap[STATE_IDLE] = entry
+	}
+	// Configure underlying Protocol
 	protoConfig := protocol.ProtocolConfig{
 		Name:                PROTOCOL_NAME,
 		ProtocolId:          PROTOCOL_ID,
@@ -23,7 +34,7 @@ func NewClient(protoOptions protocol.ProtocolOptions, cfg *Config) *Client {
 		Role:                protocol.ProtocolRoleClient,
 		MessageHandlerFunc:  c.messageHandler,
 		MessageFromCborFunc: NewMsgFromCbor,
-		StateMap:            StateMap,
+		StateMap:            stateMap,
 		InitialState:        STATE_INIT,
 	}
 	c.Protocol = protocol.New(protoConfig)

protocol/txsubmission/txsubmission.go

@@ -1,6 +1,8 @@
 package txsubmission
 
 import (
+	"time"
+
 	"github.com/cloudstruct/go-ouroboros-network/protocol"
 )
 
@@ -103,6 +105,7 @@ type Config struct {
 	ReplyTxsFunc     ReplyTxsFunc
 	DoneFunc         DoneFunc
 	InitFunc         InitFunc
+	IdleTimeout      time.Duration
 }
 
 // Callback function types
@@ -124,7 +127,9 @@ func New(protoOptions protocol.ProtocolOptions, cfg *Config) *TxSubmission {
 type TxSubmissionOptionFunc func(*Config)
 
 func NewConfig(options ...TxSubmissionOptionFunc) Config {
-	c := Config{}
+	c := Config{
+		IdleTimeout: 300 * time.Second,
+	}
 	// Apply provided options functions
 	for _, option := range options {
 		option(&c)
@@ -167,3 +172,9 @@ func WithInitFunc(initFunc InitFunc) TxSubmissionOptionFunc {
 		c.InitFunc = initFunc
 	}
 }
+
+func WithIdleTimeout(timeout time.Duration) TxSubmissionOptionFunc {
+	return func(c *Config) {
+		c.IdleTimeout = timeout
+	}
+}