package consul

import (
	"google.golang.org/grpc"

	"github.com/hashicorp/consul/acl"
	"github.com/hashicorp/consul/agent/consul/stream"
	agentgrpc "github.com/hashicorp/consul/agent/grpc"
	"github.com/hashicorp/consul/agent/rpc/subscribe"
)

type subscribeBackend struct {
	srv      *Server
	connPool *agentgrpc.ClientConnPool
}

// TODO: refactor Resolve methods to an ACLBackend that can be used by all
// the endpoints.
func (s subscribeBackend) ResolveToken(token string) (acl.Authorizer, error) {
	return s.srv.ResolveToken(token)
}

var _ subscribe.Backend = (*subscribeBackend)(nil)

// Forward requests to a remote datacenter by calling f if the target dc does not
// match the config. Does nothing but return handled=false if dc is not specified,
// or if it matches the Datacenter in config.
//
// TODO: extract this so that it can be used with other grpc services.
func (s subscribeBackend) Forward(dc string, f func(*grpc.ClientConn) error) (handled bool, err error) {
	if dc == "" || dc == s.srv.config.Datacenter {
		return false, nil
	}
	conn, err := s.connPool.ClientConn(dc)
	if err != nil {
		return false, err
	}
	return true, f(conn)
}

func (s subscribeBackend) Subscribe(req *stream.SubscribeRequest) (*stream.Subscription, error) {
	return s.srv.fsm.State().EventPublisher().Subscribe(req)
}