package grpcs

import (
	"net"

	"github.com/pkg/errors"
	"github.com/rogeecn/atom/container"
	"github.com/rogeecn/atom/providers/log"
	"github.com/rogeecn/atom/utils/opt"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials"
	"google.golang.org/grpc/reflection"
)

func DefaultProvider() container.ProviderContainer {
	return container.ProviderContainer{
		Provider: Provide,
		Options: []opt.Option{
			opt.Prefix(DefaultPrefix),
		},
	}
}

// 给注入启动调用使用
type ServerService interface {
	Name() string
	Register(*grpc.Server)
}

type Grpc struct {
	server *grpc.Server
	config *Config
}

func Provide(opts ...opt.Option) error {
	o := opt.New(opts...)
	var config Config
	if err := o.UnmarshalConfig(&config); err != nil {
		return err
	}

	return container.Container.Provide(func() (*Grpc, error) {
		serverOptions := []grpc.ServerOption{}

		// tls
		if config.Tls != nil {
			tlsConfig, err := credentials.NewServerTLSFromFile(config.Tls.Cert, config.Tls.Key)
			if err != nil {
				return nil, errors.Wrap(err, "Failed to create tls credential")
			}
			serverOptions = append(serverOptions, grpc.Creds(tlsConfig))
		}

		return &Grpc{
			server: grpc.NewServer(serverOptions...),
			config: &config,
		}, nil
	}, o.DiOptions()...)
}

func (g *Grpc) Serve() error {
	ld, err := net.Listen("tcp", g.config.Address())
	if err != nil {
		return errors.Wrapf(err, "bind address failed: %s", g.config.Address())
	}

	log.Infof("grpc server listen on %s", g.config.Address())
	reflection.Register(g.server)

	return g.server.Serve(ld)
}

func (g *Grpc) RegisterService(name string, f func(*grpc.Server)) {
	log.Debug("register service:", name)
	f(g.server)
}