add plugin system

ingress/plugin.go

@@ -0,0 +1,56 @@
+package ingress
+
+import (
+	"context"
+
+	ingress_pkg "github.com/go-gost/core/ingress"
+	"github.com/go-gost/plugin/ingress/proto"
+	xlogger "github.com/go-gost/x/logger"
+)
+
+type pluginIngress struct {
+	client  proto.IngressClient
+	options options
+}
+
+// NewPluginIngress creates a plugin ingress.
+func NewPluginIngress(opts ...Option) ingress_pkg.Ingress {
+	var options options
+	for _, opt := range opts {
+		opt(&options)
+	}
+	if options.logger == nil {
+		options.logger = xlogger.Nop()
+	}
+
+	p := &pluginIngress{
+		options: options,
+	}
+	if options.client != nil {
+		p.client = proto.NewIngressClient(options.client)
+	}
+	return p
+}
+
+func (p *pluginIngress) Get(ctx context.Context, host string) string {
+	if p.client == nil {
+		return ""
+	}
+
+	r, err := p.client.Get(ctx,
+		&proto.GetRequest{
+			Host: host,
+		})
+	if err != nil {
+		p.options.logger.Error(err)
+		return ""
+	}
+	return r.GetEndpoint()
+}
+
+func (p *pluginIngress) Close() error {
+	if p.options.client != nil {
+		return p.options.client.Close()
+	}
+	return nil
+}