Merge pull request #85 from sagarPujariLr/develop

It has all the updated code - IP Access Restriction, Permissions, Default Theme

Author: mohammed786

api/account.go

@@ -18,8 +18,14 @@ type SitesToken struct {
 	XToken        string `json:"xtoken"`
 }
 
-func SetSites(appid int64) (*SitesToken, error) {
-	switchapp := conf.AdminConsoleAPIDomain + "/account/switchapp?appid=" + strconv.FormatInt(appid, 10)
+func SetSites(appid int64, owned bool) (*SitesToken, error) {
+	var api string
+	if (owned){
+		api = "/account/switchapp?appid="
+	} else {
+		api = "/account/switchsharedsite?appid="
+	}
+	switchapp := conf.AdminConsoleAPIDomain + api + strconv.FormatInt(appid, 10)
 	switchResp, err := request.Rest(http.MethodGet, switchapp, nil, "")
 	var switchRespObj SitesToken
 	err = json.Unmarshal(switchResp, &switchRespObj)

api/auth.go

@@ -6,6 +6,7 @@ import (
 	"io/ioutil"
 	"net/http"
 	"time"
+	"reflect"
 
 	"github.com/loginradius/lr-cli/cmdutil"
 	"github.com/loginradius/lr-cli/config"
@@ -20,7 +21,7 @@ type LoginResponse struct {
 	AppName       string      `json:"app_name"`
 	Authenticated bool        `json:"authenticated"`
 	NoOfLogins    int32       `json:"no_of_logins"`
-	PlanDetails   interface{} `json"plan_detail"`
+	PlanDetails   interface{} `json:"plan_detail"`
 	XSign         string      `json:"xsign"`
 	XToken        string      `json:"xtoken"`
 }
@@ -44,6 +45,28 @@ type Feature struct {
 	Status  bool   `json:"status"`
 }
 
+type PermissionResponse struct {
+	Permissions Permission `json:"permissions"`
+}
+
+type Permission struct {
+	API_AdminConfiguration         		bool		`json:"API_AdminConfiguration"`
+	API_EditConfiguration       		bool		`json:"API_EditConfiguration"`
+	API_EditCredentials 				bool        `json:"API_EditCredentials"`
+	API_EditThirdPartyCredentials    	bool		`json:"API_EditThirdPartyCredentials"`
+	API_ViewCredentials   				bool		`json:"API_ViewCredentials"`
+	UserManagement_Admin		       	bool		`json:"UserManagement_Admin"`
+	API_ViewConfiguration        		bool		`json:"API_ViewConfiguration"`
+	API_ViewThirdPartyCredentials  		bool		`json:"API_ViewThirdPartyCredentials"`
+	ThirdPartyIntegration_View		    bool		`json:"ThirdPartyIntegration_View"`
+	UserManagement_View				    bool		`json:"UserManagement_View"`
+	SecurityPolicy_View				    bool		`json:"SecurityPolicy_View"`
+	API_AdminThirdPartyCredentials	    bool		`json:"API_AdminThirdPartyCredentials"`
+	ThirdPartyIntegration_Admin		    bool		`json:"ThirdPartyIntegration_Admin"`
+	SecurityPolicy_Admin			    bool		`json:"SecurityPolicy_Admin"`
+	SecurityPolicy_Edit				    bool		`json:"SecurityPolicy_Edit"`
+}
+
 func AuthLogin(params LoginOpts) (*LoginResponse, error) {
 
 	var resObj LoginResponse
@@ -204,6 +227,15 @@ func IsPasswordLessEnabled(features FeatureSchema) bool {
 	return false
 }
 
+func IsIPAutthorizationEnabled(features FeatureSchema) bool {
+	for _, val := range features.Data {
+		if val.Feature == "ip_authorization_enabled" && val.Status {
+			return true
+		}
+	}
+	return false
+}
+
 func storeSiteInfo(data CoreAppData) (map[int64]SitesReponse, map[int64]SharedSitesReponse) {
 	siteInfo := make(map[int64]SitesReponse, len(data.Apps.Data))
 	sharedsiteInfo := make(map[int64]SharedSitesReponse, len(data.Apps.Data))
@@ -219,9 +251,15 @@ func storeSiteInfo(data CoreAppData) (map[int64]SitesReponse, map[int64]SharedSi
 	currentId, err := CurrentID()
 	if err == nil {
 		site, ok := siteInfo[currentId]
+		sharedsite, sharedok := sharedsiteInfo[currentId]
 		if ok {
-			obj, _ := json.Marshal(site)
-			cmdutil.WriteFile("currentSite.json", obj)
+
+				obj, _ := json.Marshal(site)
+				cmdutil.WriteFile("currentSite.json", obj)
+			
+		} else if sharedok {
+			obj, _ := json.Marshal(sharedsite)
+				cmdutil.WriteFile("currentSite.json", obj)
 		}
 	}
 	return siteInfo,sharedsiteInfo
@@ -250,3 +288,58 @@ func UpdatePhoneLogin(feature string, status bool) (*FeatureSchema, error) {
 	}
 	return &resultResp, err
 }
+
+
+func GetPermissionsfromAPI() ( error) {
+	
+	coreAppData := conf.AdminConsoleAPIDomain + "/auth/permissions?"
+	data, err := request.Rest(http.MethodGet, coreAppData, nil, "")
+	var  permissionsResp PermissionResponse
+	var permission Permission
+	if err != nil {
+		return err
+	}
+	err = json.Unmarshal(data, &permissionsResp)
+	permission = permissionsResp.Permissions
+	err = storePermissionData(permission)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+
+func storePermissionData(data Permission) (error) {
+	var permissionobj = cmdutil.PermissionCommands
+	permission := make(map[string]bool, len(permissionobj))
+	for key, val := range permissionobj {
+		v := reflect.ValueOf(&data).Elem().FieldByName(val)
+		if v.IsValid() {
+			permission[key] = v.Bool()
+		} 
+	}
+	obj, err := json.Marshal(permission)
+	if err != nil {
+		return nil
+	}
+	cmdutil.WriteFile("permission.json", obj)
+	return nil
+}
+
+func GetPermission(str string) (bool, error) { 
+	data, err := cmdutil.ReadFile("permission.json")
+	if err != nil {
+	return false, err
+	}
+	permission := make(map[string]bool)
+	err = json.Unmarshal(data, &permission)
+	if err != nil {
+		return false, err
+		}
+	if permission[str] == false {
+		fmt.Println("You don't have access to proceed, request access from the site owner. If you've already been granted access, log out and log back in. If the issue persists, contact LoginRadius support at ")
+		fmt.Println( conf.DashboardDomain + "/support/tickets")
+		return false , nil
+	}
+	return true , nil
+}

api/deployment.go

@@ -131,7 +131,6 @@ type VerifySmtpConfigSchema struct {
 type VerifySmtpConfigError struct {
 	Description 		string		`json:"description"`
 	Message 		string		`json:"message"`
-
 }
 
 func GetSites() (*SitesReponse, error) {

api/platformConfiguration.go

@@ -438,3 +438,19 @@ func UpdatePasswordlessLogin(body []byte) (*PasswordlessLogin, error) {
 	}
 	return &resultResp, err
 }
+
+func UpdateSiteFeatures (data FeatureSchema) error {
+	url := conf.AdminConsoleAPIDomain + "/platform-configuration/app-features?"
+	body, _ := json.Marshal(data)
+	resp, err := request.Rest(http.MethodPost, url, nil, string(body))
+	if err != nil {
+		return err
+	}
+	var resObj FeatureSchema
+	err = json.Unmarshal(resp, &resObj)
+	if err != nil {
+		return err
+	}
+	return nil
+
+}

api/securityConfiguration.go

@@ -20,6 +20,11 @@ type EmailResponse struct {
 	RegistrationType string `json:"RegistrationType"`
 }
 
+type IPResponse struct {
+	AllowedIPs []string `json:"AllowedIPs"`
+	DeniedIPs  []string `json:"DeniedIPs"`
+}
+
 type RegistrationRestrictionTypeSchema struct {
 	SelectedRestrictionType string `json:"selectedRestrictionType"`
 }
@@ -109,4 +114,40 @@ func AddEmailWhitelistBlacklist(restrictionType RegistrationRestrictionTypeSchem
 		}
 	}
 	return nil
-}
+}
+
+func GetIPAccessRestrictionList() (*IPResponse, error) {
+	url := conf.AdminConsoleAPIDomain + "/security-configuration/ip-config?"
+	resp, err := request.Rest(http.MethodGet, url, nil, "")
+	if err != nil {
+		return nil, err
+	}
+	var resObj IPResponse
+	err = json.Unmarshal(resp, &resObj)
+	if err != nil {
+		return nil, err
+	}
+	return &resObj, nil
+
+}
+
+
+func AddIPAccessRestrictionList(disabled bool, data IPResponse) error {
+
+	if disabled  {
+		domainUrl := conf.AdminConsoleAPIDomain + "/security-configuration/ip-config/reset?"
+		_, err := request.Rest(http.MethodPut, domainUrl, nil, "")
+		if err != nil {
+			return err
+		}
+	} else {
+	typeUrl := conf.AdminConsoleAPIDomain + "/security-configuration/ip-config?"
+	body, _ := json.Marshal(data)
+	_, err := request.Rest(http.MethodPut, typeUrl, nil, string(body))
+	if err != nil {
+		return err
+	}
+	} 
+	 
+	return nil
+}