mirror of
https://github.com/gravitl/netmaker.git
synced 2025-09-13 00:24:31 +08:00
230e062c84
* set free tier limits through config * add host limit to config * check for host limit on free tier * fix license validation, replace node limit with hosts * add hosts to telemetry data * debug init * validate license every 1hr * hook manager, api to fetch server usage * hook manager, server usage api * encode json server usage api * update ngork url * update license validation endpoint * avoid setting limits on eer * adding hotfix * correct users limits env var * add comments to exported funcs --------- Co-authored-by: afeiszli <alex.feiszli@gmail.com>
71 lines
2.6 KiB
Go
71 lines
2.6 KiB
Go
//go:build ee
|
|
// +build ee
|
|
|
|
package ee
|
|
|
|
import (
|
|
controller "github.com/gravitl/netmaker/controllers"
|
|
"github.com/gravitl/netmaker/ee/ee_controllers"
|
|
eelogic "github.com/gravitl/netmaker/ee/logic"
|
|
"github.com/gravitl/netmaker/logger"
|
|
"github.com/gravitl/netmaker/logic"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/gravitl/netmaker/servercfg"
|
|
)
|
|
|
|
// InitEE - Initialize EE Logic
|
|
func InitEE() {
|
|
setIsEnterprise()
|
|
servercfg.Is_EE = true
|
|
models.SetLogo(retrieveEELogo())
|
|
controller.HttpHandlers = append(
|
|
controller.HttpHandlers,
|
|
ee_controllers.MetricHandlers,
|
|
ee_controllers.NetworkUsersHandlers,
|
|
ee_controllers.UserGroupsHandlers,
|
|
ee_controllers.RelayHandlers,
|
|
)
|
|
logic.EnterpriseCheckFuncs = append(logic.EnterpriseCheckFuncs, func() {
|
|
// == License Handling ==
|
|
ValidateLicense()
|
|
logger.Log(0, "proceeding with Paid Tier license")
|
|
logic.SetFreeTierForTelemetry(false)
|
|
// == End License Handling ==
|
|
AddLicenseHooks()
|
|
resetFailover()
|
|
})
|
|
logic.EnterpriseFailoverFunc = eelogic.SetFailover
|
|
logic.EnterpriseResetFailoverFunc = eelogic.ResetFailover
|
|
logic.EnterpriseResetAllPeersFailovers = eelogic.WipeAffectedFailoversOnly
|
|
logic.DenyClientNodeAccess = eelogic.DenyClientNode
|
|
logic.IsClientNodeAllowed = eelogic.IsClientNodeAllowed
|
|
logic.AllowClientNodeAccess = eelogic.RemoveDeniedNodeFromClient
|
|
}
|
|
|
|
func resetFailover() {
|
|
nets, err := logic.GetNetworks()
|
|
if err == nil {
|
|
for _, net := range nets {
|
|
err = eelogic.ResetFailover(net.NetID)
|
|
if err != nil {
|
|
logger.Log(0, "failed to reset failover on network", net.NetID, ":", err.Error())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func retrieveEELogo() string {
|
|
return `
|
|
__ __ ______ ______ __ __ ______ __ __ ______ ______
|
|
/\ "-.\ \ /\ ___\ /\__ _\ /\ "-./ \ /\ __ \ /\ \/ / /\ ___\ /\ == \
|
|
\ \ \-. \ \ \ __\ \/_/\ \/ \ \ \-./\ \ \ \ __ \ \ \ _"-. \ \ __\ \ \ __<
|
|
\ \_\\"\_\ \ \_____\ \ \_\ \ \_\ \ \_\ \ \_\ \_\ \ \_\ \_\ \ \_____\ \ \_\ \_\
|
|
\/_/ \/_/ \/_____/ \/_/ \/_/ \/_/ \/_/\/_/ \/_/\/_/ \/_____/ \/_/ /_/
|
|
|
|
___ ___ ____
|
|
____ ____ ____ / _ \ / _ \ / __ \ ____ ____ ____
|
|
/___/ /___/ /___/ / ___/ / , _// /_/ / /___/ /___/ /___/
|
|
/___/ /___/ /___/ /_/ /_/|_| \____/ /___/ /___/ /___/
|
|
|
|
`
|
|
}
|