works

api/api.go

@@ -1,6 +1,9 @@
 package api
 
-import log "github.com/sirupsen/logrus"
+import (
+	"github.com/callebtc/electronwall/config"
+	log "github.com/sirupsen/logrus"
+)
 
 type ApiClient interface {
 	GetNodeInfo(pubkey string) OneML_NodeInfoResponse
@@ -12,23 +15,31 @@ type ApiNodeInfo struct {
 }
 
 func GetApiNodeinfo(pubkey string) (ApiNodeInfo, error) {
-	// get info from 1ml
-	OnemlClient := GetOneMlClient()
-	onemlNodeInfo, err := OnemlClient.GetNodeInfo(pubkey)
-	if err != nil {
-		log.Errorf(err.Error())
-		onemlNodeInfo = OneML_NodeInfoResponse{}
+	response := ApiNodeInfo{
+		OneMl:  OneML_NodeInfoResponse{},
+		Amboss: Amboss_NodeInfoResponse{},
 	}
 
-	// get info from amboss
-	ambossClient := GetAmbossClient()
-	ambossNodeInfo, err := ambossClient.GetNodeInfo(pubkey)
-	if err != nil {
-		log.Errorf(err.Error())
-		ambossNodeInfo = Amboss_NodeInfoResponse{}
+	if config.Configuration.ApiRules.OneMl.Active {
+		// get info from 1ml
+		OnemlClient := GetOneMlClient()
+		onemlNodeInfo, err := OnemlClient.GetNodeInfo(pubkey)
+		if err != nil {
+			log.Errorf(err.Error())
+			onemlNodeInfo = OneML_NodeInfoResponse{}
+		}
+		response.OneMl = onemlNodeInfo
 	}
-	return ApiNodeInfo{
-		OneMl:  onemlNodeInfo,
-		Amboss: ambossNodeInfo,
-	}, err
+
+	if config.Configuration.ApiRules.Amboss.Active {
+		// get info from amboss
+		ambossClient := GetAmbossClient()
+		ambossNodeInfo, err := ambossClient.GetNodeInfo(pubkey)
+		if err != nil {
+			log.Errorf(err.Error())
+			ambossNodeInfo = Amboss_NodeInfoResponse{}
+		}
+		response.Amboss = ambossNodeInfo
+	}
+	return response, nil
 }