package grpc

import (
	"net"

	"git.solsynth.dev/hydrogen/dealer/pkg/proto"
	"github.com/spf13/viper"
	"google.golang.org/grpc"
	health "google.golang.org/grpc/health/grpc_health_v1"
	"google.golang.org/grpc/reflection"
)

type Server struct {
	proto.UnimplementedStreamControllerServer
	proto.UnimplementedServiceDirectoryServer
}

var S *grpc.Server

func NewGRPC() {
	S = grpc.NewServer()

	health.RegisterHealthServer(S, &Server{})
	proto.RegisterStreamControllerServer(S, &Server{})
	proto.RegisterServiceDirectoryServer(S, &Server{})

	reflection.Register(S)
}

func ListenGRPC() error {
	listener, err := net.Listen("tcp", viper.GetString("grpc_bind"))
	if err != nil {
		return err
	}

	return S.Serve(listener)
}