package rpc

import (
	"errors"

	"github.com/spiral/goridge/v2"
	"github.com/spiral/roadrunner/v2/plugins/config"

	"net/rpc"
)

type Plugin interface {
	Name() string
	RpcService() (interface{}, error)
}

// ID contains default service name.
const ID = "rpc"

type services struct {
	service interface{}
	name    string
}

// Service is RPC service.
type Service struct {
	// TODO do we need a pointer here since all receivers are pointers??
	rpc     *rpc.Server
	configProvider config.Provider
	services       []services
	config         Config
	close          chan struct{}
}

// Init rpc service. Must return true if service is enabled.
func (s *Service) Init(cfg config.Provider) error {
	s.configProvider = cfg
	err := s.configProvider.UnmarshalKey(ID, &s.config)
	if err != nil {
		return err
	}

	// TODO Do we need to init defaults
	if s.config.Listen == "" {
		s.config.InitDefaults()
	}

	s.close = make(chan struct{})

	return nil
}

// Serve serves the service.
func (s *Service) Serve() chan error {
	errCh := make(chan error, 1)
	server := rpc.NewServer()
	if server == nil {
		errCh <- errors.New("rpc server is nil")
		return errCh
	}
	s.rpc = server

	if len(s.services) == 0 {
		errCh <- errors.New("no services with RPC")
		return errCh
	}

	// Attach all services
	for i := 0; i < len(s.services); i++ {
		err := s.Register(s.services[i].name, s.services[i].service)
		if err != nil {
			errCh <- err
			return errCh
		}
	}

	ln, err := s.config.Listener()
	if err != nil {
		errCh <- err
		return errCh
	}
	defer func() {
		errCh <- ln.Close()
	}()

	go func() {
		for {
			select {
			case <-s.close:
				return
			default:
				conn, err := ln.Accept()
				if err != nil {
					continue
				}

				go s.rpc.ServeCodec(goridge.NewCodec(conn))
			}
		}
	}()

	return nil
}

// Stop stops the service.
func (s *Service) Stop() error {
	s.close <- struct{}{}
	return nil
}

func (s *Service) Depends() []interface{} {
	return []interface{}{
		s.RpcService,
	}
}

func (s *Service) RpcService(p Plugin) error {
	s.services = append(s.services, services{
		service: p.RpcService(),
		name:    p.Name(),
	})
	return nil
}

// Register publishes in the server the set of methods of the
// receiver value that satisfy the following conditions:
//	- exported method of exported type
//	- two arguments, both of exported type
//	- the second argument is a pointer
//	- one return value, of type error
// It returns an error if the receiver is not an exported type or has
// no suitable methods. It also logs the error using package log.
func (s *Service) Register(name string, svc interface{}) error {
	if s.rpc == nil {
		return errors.New("RPC service is not configured")
	}

	return s.rpc.RegisterName(name, svc)
}

// Client creates new RPC client.
func (s *Service) Client() (*rpc.Client, error) {
	if s.configProvider == nil {
		return nil, errors.New("RPC service is not configured")
	}

	conn, err := s.config.Dialer()
	if err != nil {
		return nil, err
	}

	return rpc.NewClientWithCodec(goridge.NewClientCodec(conn)), nil
}