initialize.go 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. //go:build ee
  2. // +build ee
  3. package pro
  4. import (
  5. "time"
  6. controller "github.com/gravitl/netmaker/controllers"
  7. "github.com/gravitl/netmaker/logger"
  8. "github.com/gravitl/netmaker/logic"
  9. "github.com/gravitl/netmaker/models"
  10. "github.com/gravitl/netmaker/mq"
  11. proControllers "github.com/gravitl/netmaker/pro/controllers"
  12. proLogic "github.com/gravitl/netmaker/pro/logic"
  13. "github.com/gravitl/netmaker/servercfg"
  14. "golang.org/x/exp/slog"
  15. )
  16. // InitPro - Initialize Pro Logic
  17. func InitPro() {
  18. servercfg.IsPro = true
  19. models.SetLogo(retrieveProLogo())
  20. controller.HttpMiddlewares = append(
  21. controller.HttpMiddlewares,
  22. proControllers.OnlyServerAPIWhenUnlicensedMiddleware,
  23. )
  24. controller.HttpHandlers = append(
  25. controller.HttpHandlers,
  26. proControllers.MetricHandlers,
  27. proControllers.RelayHandlers,
  28. proControllers.UserHandlers,
  29. proControllers.FailOverHandlers,
  30. )
  31. logic.EnterpriseCheckFuncs = append(logic.EnterpriseCheckFuncs, func() {
  32. // == License Handling ==
  33. enableLicenseHook := false
  34. licenseKeyValue := servercfg.GetLicenseKey()
  35. netmakerTenantID := servercfg.GetNetmakerTenantID()
  36. if licenseKeyValue != "" && netmakerTenantID != "" {
  37. enableLicenseHook = true
  38. }
  39. if !enableLicenseHook {
  40. err := initTrial()
  41. if err != nil {
  42. logger.FatalLog0("failed to init trail", err.Error())
  43. }
  44. trialEndDate, err := getTrialEndDate()
  45. if err != nil {
  46. slog.Error("failed to get trial end date", "error", err)
  47. enableLicenseHook = true
  48. }
  49. // check if trial ended
  50. if time.Now().After(trialEndDate) {
  51. // trial ended already
  52. enableLicenseHook = true
  53. }
  54. }
  55. if enableLicenseHook {
  56. slog.Info("starting license checker")
  57. ClearLicenseCache()
  58. if err := ValidateLicense(); err != nil {
  59. slog.Error(err.Error())
  60. return
  61. }
  62. slog.Info("proceeding with Paid Tier license")
  63. logic.SetFreeTierForTelemetry(false)
  64. // == End License Handling ==
  65. AddLicenseHooks()
  66. } else {
  67. addTrialLicenseHook()
  68. }
  69. if servercfg.GetServerConfig().RacAutoDisable {
  70. AddRacHooks()
  71. }
  72. })
  73. logic.ResetFailOver = proLogic.ResetFailOver
  74. logic.ResetFailedOverPeer = proLogic.ResetFailedOverPeer
  75. logic.DenyClientNodeAccess = proLogic.DenyClientNode
  76. logic.IsClientNodeAllowed = proLogic.IsClientNodeAllowed
  77. logic.AllowClientNodeAccess = proLogic.RemoveDeniedNodeFromClient
  78. logic.SetClientDefaultACLs = proLogic.SetClientDefaultACLs
  79. logic.SetClientACLs = proLogic.SetClientACLs
  80. logic.UpdateProNodeACLs = proLogic.UpdateProNodeACLs
  81. logic.GetMetrics = proLogic.GetMetrics
  82. logic.UpdateMetrics = proLogic.UpdateMetrics
  83. logic.DeleteMetrics = proLogic.DeleteMetrics
  84. logic.GetRelays = proLogic.GetRelays
  85. logic.GetAllowedIpsForRelayed = proLogic.GetAllowedIpsForRelayed
  86. logic.RelayedAllowedIPs = proLogic.RelayedAllowedIPs
  87. logic.UpdateRelayed = proLogic.UpdateRelayed
  88. logic.SetRelayedNodes = proLogic.SetRelayedNodes
  89. logic.RelayUpdates = proLogic.RelayUpdates
  90. logic.IsInternetGw = proLogic.IsInternetGw
  91. logic.SetInternetGw = proLogic.SetInternetGw
  92. logic.GetTrialEndDate = getTrialEndDate
  93. mq.UpdateMetrics = proLogic.MQUpdateMetrics
  94. mq.UpdateMetricsFallBack = proLogic.MQUpdateMetricsFallBack
  95. }
  96. func retrieveProLogo() string {
  97. return `
  98. __ __ ______ ______ __ __ ______ __ __ ______ ______
  99. /\ "-.\ \ /\ ___\ /\__ _\ /\ "-./ \ /\ __ \ /\ \/ / /\ ___\ /\ == \
  100. \ \ \-. \ \ \ __\ \/_/\ \/ \ \ \-./\ \ \ \ __ \ \ \ _"-. \ \ __\ \ \ __<
  101. \ \_\\"\_\ \ \_____\ \ \_\ \ \_\ \ \_\ \ \_\ \_\ \ \_\ \_\ \ \_____\ \ \_\ \_\
  102. \/_/ \/_/ \/_____/ \/_/ \/_/ \/_/ \/_/\/_/ \/_/\/_/ \/_____/ \/_/ /_/
  103. ___ ___ ____
  104. ____ ____ ____ / _ \ / _ \ / __ \ ____ ____ ____
  105. /___/ /___/ /___/ / ___/ / , _// /_/ / /___/ /___/ /___/
  106. /___/ /___/ /___/ /_/ /_/|_| \____/ /___/ /___/ /___/
  107. `
  108. }