add chain.Chainable interface

pkg/handler/forward/local/handler.go

@@ -33,7 +33,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &forwardHandler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -43,6 +42,11 @@ func (h *forwardHandler) Init(md md.Metadata) (err error) {
 	return h.parseMetadata(md)
 }
 
+// implements chain.Chainable interface
+func (h *forwardHandler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 // Forward implements handler.Forwarder.
 func (h *forwardHandler) Forward(group *chain.NodeGroup) {
 	h.group = group

pkg/handler/http/handler.go

@@ -41,7 +41,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &httpHandler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -51,6 +50,11 @@ func (h *httpHandler) Init(md md.Metadata) error {
 	return h.parseMetadata(md)
 }
 
+// implements chain.Chainable interface
+func (h *httpHandler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 func (h *httpHandler) Handle(ctx context.Context, conn net.Conn) {
 	defer conn.Close()
 

pkg/handler/option.go

@@ -2,12 +2,10 @@ package handler
 
 import (
 	"github.com/go-gost/gost/pkg/bypass"
-	"github.com/go-gost/gost/pkg/chain"
 	"github.com/go-gost/gost/pkg/logger"
 )
 
 type Options struct {
-	Chain  *chain.Chain
 	Bypass bypass.Bypass
 	Logger logger.Logger
 }
@@ -20,12 +18,6 @@ func LoggerOption(logger logger.Logger) Option {
 	}
 }
 
-func ChainOption(chain *chain.Chain) Option {
-	return func(opts *Options) {
-		opts.Chain = chain
-	}
-}
-
 func BypassOption(bypass bypass.Bypass) Option {
 	return func(opts *Options) {
 		opts.Bypass = bypass

pkg/handler/relay/handler.go

@@ -34,7 +34,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &relayHandler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -44,6 +43,11 @@ func (h *relayHandler) Init(md md.Metadata) (err error) {
 	return h.parseMetadata(md)
 }
 
+// implements chain.Chainable interface
+func (h *relayHandler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 // Forward implements handler.Forwarder.
 func (h *relayHandler) Forward(group *chain.NodeGroup) {
 	h.group = group

pkg/handler/socks/v4/handler.go

@@ -33,7 +33,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &socks4Handler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -43,6 +42,11 @@ func (h *socks4Handler) Init(md md.Metadata) (err error) {
 	return h.parseMetadata(md)
 }
 
+// implements chain.Chainable interface
+func (h *socks4Handler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 func (h *socks4Handler) Handle(ctx context.Context, conn net.Conn) {
 	defer conn.Close()
 

pkg/handler/socks/v5/handler.go

@@ -35,7 +35,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &socks5Handler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -56,6 +55,11 @@ func (h *socks5Handler) Init(md md.Metadata) (err error) {
 	return
 }
 
+// implements chain.Chainable interface
+func (h *socks5Handler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 func (h *socks5Handler) Handle(ctx context.Context, conn net.Conn) {
 	defer conn.Close()
 

pkg/handler/ss/handler.go

@@ -35,7 +35,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &ssHandler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -45,6 +44,11 @@ func (h *ssHandler) Init(md md.Metadata) (err error) {
 	return h.parseMetadata(md)
 }
 
+// implements chain.Chainable interface
+func (h *ssHandler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 func (h *ssHandler) Handle(ctx context.Context, conn net.Conn) {
 	defer conn.Close()
 

pkg/handler/ss/udp/handler.go

@@ -34,7 +34,6 @@ func NewHandler(opts ...handler.Option) handler.Handler {
 	}
 
 	return &ssuHandler{
-		chain:  options.Chain,
 		bypass: options.Bypass,
 		logger: options.Logger,
 	}
@@ -44,6 +43,11 @@ func (h *ssuHandler) Init(md md.Metadata) (err error) {
 	return h.parseMetadata(md)
 }
 
+// implements chain.Chainable interface
+func (h *ssuHandler) WithChain(chain *chain.Chain) {
+	h.chain = chain
+}
+
 func (h *ssuHandler) Handle(ctx context.Context, conn net.Conn) {
 	defer conn.Close()