application.go

package potens

import (
	"crypto/rsa"
	"net"
	"regexp"
	"strconv"
	"strings"

	"github.com/kubex/potens-go/definition"
	"github.com/kubex/potens-go/identity"
	"github.com/kubex/potens-go/kapp"
	"github.com/kubex/proto-go/application"
	"github.com/kubex/proto-go/discovery"
	"github.com/kubex/proto-go/eventpipe"
	"go.uber.org/zap"
	"google.golang.org/grpc"
)

//Application Helper struct for your application
type Application struct {
	/** RunTime **/
	Port          int
	IP            net.IP
	instanceID    string
	consoleDomain string
	currentStatus discovery.ServiceStatus
	appServer     *kapp.ApplicationServer

	/** Definition **/
	Name       string
	definition *definition.AppDefinition
	appRelease discovery.AppRelease

	/** Identity **/
	identity *identity.AppIdentity
	pk       *rsa.PrivateKey
	kh       string

	/** Utility **/
	logger *zap.Logger

	/** gRPC **/
	server         *grpc.Server
	services       *serviceCache
	logDiscoveryHB bool
}

type serviceCache struct {
	discoveryClient     discovery.DiscoveryClient
	discoveryHost       string
	discoveryPort       string
	discoveryRegistered bool
	socketHandler       *socketHandler
	eventPipeClient     eventpipe.EventPipeClient
}

//FatalErr If an error is provided, Log().Fatal()
func (app *Application) FatalErr(err error) {
	if err != nil {
		app.Log().Fatal("Oops", zap.Error(err))
	}
}

//GlobalAppID returns the global app ID for the application
func (app *Application) GlobalAppID() string {
	return app.definition.GlobalAppID()
}

//VendorID returns the vendor ID for this application
func (app *Application) VendorID() string {
	return app.definition.VendorID
}

//AppID returns the app ID for this application
func (app *Application) AppID() string {
	return app.definition.AppID
}

//InstanceID returns the instance ID for this session
func (app *Application) InstanceID() string {
	return app.instanceID
}

//PortString returns the port as a string
func (app *Application) PortString() string {
	return strconv.FormatInt(int64(app.Port), 10)
}

//ConsoleDomain returns the kubex console domain being used
func (app *Application) ConsoleDomain() string {
	return app.consoleDomain
}

//IsProduction returns if you are in production environment
func (app *Application) IsProduction() bool {
	return app.consoleDomain == KubexProductionConsoleDomain
}

//Log returns a logger to log against
func (app *Application) Log() *zap.Logger {
	return app.logger
}

//ServiceKey returns a key that can be used to pre-fix environment variables
func (app *Application) ServiceKey() string {
	return strings.ToUpper(strings.Replace(regexp.MustCompile("[^A-Za-z0-9\\-_]").ReplaceAllString(app.Name, ""), "-", "_", -1))
}

//AppServer grpc application server, add routing and functionality for your app on this
func (app *Application) AppServer() *kapp.ApplicationServer {
	if app.appServer == nil {
		app.appServer = kapp.New()
	}
	return app.appServer
}

//RegisterAsAppServer listen to requests as a kubex application
func (app *Application) RegisterAsAppServer() *kapp.ApplicationServer {
	srv := app.AppServer()
	application.RegisterApplicationServer(app.GetGrpcServer(), srv)
	return srv
}

//ExposeAndServe DiscoveryOnline && Serve
func (app *Application) ExposeAndServe() error {
	discoveryErr := app.DiscoveryOnline()
	if discoveryErr != nil {
		return discoveryErr
	}
	err := app.Serve()
	app.DiscoveryOffline()
	return err
}