package aibridged

import (
	"context"

	"storj.io/drpc"

	"github.com/coder/coder/v2/coderd/aibridged/proto"
)

type Dialer func(ctx context.Context) (DRPCClient, error)

// ClientFunc acquires a DRPCClient, honoring the passed context so a blocking
// acquisition (e.g. waiting for the daemon to connect to coderd) unblocks when
// the context is canceled. Server.Client satisfies it.
type ClientFunc func(context.Context) (DRPCClient, error)

// DRPCClient is the union of various service interfaces the client must support.
type DRPCClient interface {
	proto.DRPCRecorderClient
	proto.DRPCMCPConfiguratorClient
	proto.DRPCAuthorizerClient
	proto.DRPCProviderConfiguratorClient
}

var _ DRPCClient = &Client{}

type Client struct {
	proto.DRPCRecorderClient
	proto.DRPCMCPConfiguratorClient
	proto.DRPCAuthorizerClient
	proto.DRPCProviderConfiguratorClient

	Conn drpc.Conn
}

func (c *Client) DRPCConn() drpc.Conn {
	return c.Conn
}