Browse Source

added ee check

afeiszli 2 years ago
parent
commit
4917c457db
2 changed files with 6 additions and 0 deletions
  1. 1 0
      config/config.go
  2. 5 0
      controllers/server.go

+ 1 - 0
config/config.go

@@ -74,6 +74,7 @@ type ServerConfig struct {
 	BasicAuth             string `yaml:"basic_auth"`
 	LicenseValue          string `yaml:"license_value"`
 	NetmakerAccountID     string `yaml:"netmaker_account_id"`
+	IsEE                  string `yaml:"is_ee"`
 }
 
 // SQLConfig - Generic SQL Config

+ 5 - 0
controllers/server.go

@@ -10,6 +10,7 @@ import (
 	"strings"
 
 	"github.com/gorilla/mux"
+	"github.com/gravitl/netmaker/ee"
 	"github.com/gravitl/netmaker/logger"
 	"github.com/gravitl/netmaker/logic"
 	"github.com/gravitl/netmaker/models"
@@ -136,6 +137,10 @@ func getConfig(w http.ResponseWriter, r *http.Request) {
 	// get params
 
 	scfg := servercfg.GetServerConfig()
+	scfg.IsEE = "no"
+	if ee.IsEnterprise() {
+		scfg.IsEE = "yes"
+	}
 	json.NewEncoder(w).Encode(scfg)
 	//w.WriteHeader(http.StatusOK)
 }