probe/auth/service/NoAuthProbeService.go

package service

import (
	"context"
	"path"
	"reflect"
	"time"

	"git.loafle.net/commons/configuration-go"
	cda "git.loafle.net/commons/di-go/annotation"
	cdr "git.loafle.net/commons/di-go/registry"
	logging "git.loafle.net/commons/logging-go"
	crc "git.loafle.net/commons/rpc-go/client"
	ocnc "git.loafle.net/overflow/commons-go/noauthprobe/config"
	ocpc "git.loafle.net/overflow/commons-go/probe/config"
	"git.loafle.net/overflow/probe/client/central"
	"git.loafle.net/overflow/probe/config"

	// For annotation
	_ "git.loafle.net/overflow/commons-go/core/annotation"
)

var NoAuthProbeServiceType = reflect.TypeOf((*NoAuthProbeService)(nil))

func init() {
	cdr.RegisterType(NoAuthProbeServiceType)
}

type NoAuthProbeService struct {
	cda.TypeAnnotation `annotation:"@overflow:RPCService()"`

	ConfigDir    string         `annotation:"@Resource(name='ConfigDir')"`
	Config       *config.Config `annotation:"@Resource(name='Config')"`
	AuthConfig   *ocnc.Auth     `annotation:"@Resource(name='AuthConfig')"`
	AuthDoneChan chan error     `annotation:"@Resource(name='AuthDoneChan')"`

	client *crc.Client
}

func (s *NoAuthProbeService) Init() error {
	client, err := central.NewAuth(s.HandleTempKey, s)
	if nil != err {
		return err
	}
	s.client = client

	return nil
}

func (s *NoAuthProbeService) Start() error {
	if err := s.client.Start(); nil != err {
		return err
	}

	return nil
}

func (s *NoAuthProbeService) Stop() {
	if err := s.client.Stop(context.Background()); nil != err {
		logging.Logger().Error(err)
	}
}

func (s *NoAuthProbeService) Destroy() {
	s.client = nil
}

func (s *NoAuthProbeService) Accept(probeKey string) error {
	logging.Logger().Infof("accepted by central")

	n := time.Now()
	s.AuthConfig.AcceptedDate = &n
	if err := configuration.Save(s.AuthConfig, path.Join(s.ConfigDir, ocnc.ConfigFileName), true); nil != err {
		logging.Logger().Error(err)
		s.AuthDoneChan <- err
		return nil
	}

	s.Config.Probe.Key = &probeKey
	if err := configuration.Save(s.Config, path.Join(s.ConfigDir, ocpc.ConfigFileName), true); nil != err {
		logging.Logger().Error(err)
		s.AuthDoneChan <- err
		return nil
	}

	s.AuthDoneChan <- nil
	return nil
}

func (s *NoAuthProbeService) Deny() error {
	logging.Logger().Infof("denied by central")
	n := time.Now()
	s.AuthConfig.DeniedDate = &n
	if err := configuration.Save(s.AuthConfig, path.Join(s.ConfigDir, ocnc.ConfigFileName), true); nil != err {
		logging.Logger().Error(err)
		s.AuthDoneChan <- err
		return nil
	}

	s.AuthDoneChan <- nil
	return nil
}

func (s *NoAuthProbeService) HandleTempKey(tempKey string) {
	logging.Logger().Infof("registered by central")
	s.AuthConfig.TempKey = &tempKey
	if err := configuration.Save(s.AuthConfig, path.Join(s.ConfigDir, ocnc.ConfigFileName), true); nil != err {
		logging.Logger().Error(err)
		s.AuthDoneChan <- err
		return
	}
}