Merge pull request #135 from LalitDeore/distribution

Suborg distribution for datastore

Author: frikky

db-connector.go

@@ -12600,6 +12600,22 @@ func GetAllCacheKeys(ctx context.Context, orgId string, max int, inputcursor str
 		}
 	}
 
+	foundOrg, err := GetOrg(ctx, orgId)
+	if err == nil && len(foundOrg.ChildOrgs) == 0 && len(foundOrg.CreatorOrg) > 0 && foundOrg.CreatorOrg != orgId {
+		parentOrg, err := GetOrg(ctx, foundOrg.CreatorOrg)
+		if err == nil {
+			parentOrgCache, _, err := GetAllCacheKeys(ctx, parentOrg.Id, max, inputcursor)
+			if err == nil {
+				for _, parentCache := range parentOrgCache {
+					if !ArrayContains(parentCache.SuborgDistribution, orgId) {
+						continue
+					}
+					cacheKeys = append(cacheKeys, parentCache)
+				}
+			}
+		}
+	}
+
 	return cacheKeys, cursor, nil
 }
 

files.go

@@ -2117,7 +2117,7 @@ func HandleSetFileConfig(resp http.ResponseWriter, request *http.Request) {
 		return
 	}
 
-	if user.Role != "admin" {
+	if user.ActiveOrg.Role != "admin" {
 		log.Printf("User (%s) isn't admin during file edit config", user.Username)
 		resp.WriteHeader(401)
 		resp.Write([]byte(`{"success": false, "reason": "only admin can edit file config"}`))

shared.go

@@ -17548,6 +17548,105 @@ func HandleListCacheKeys(resp http.ResponseWriter, request *http.Request) {
 	resp.Write(b)
 }
 
+func HandleCacheConfig(resp http.ResponseWriter, request *http.Request) {
+	cors := HandleCors(resp, request)
+	if cors {
+		return
+	}
+
+	user, err := HandleApiAuthentication(resp, request)
+	if err != nil {
+		log.Printf("[DEBUG] Api authentication failed in cache config: %s", err)
+		resp.WriteHeader(401)
+		resp.Write([]byte(`{"success": false, "reason": "Failed authentication"}`))
+		return
+	}
+
+	if user.ActiveOrg.Role != "admin" {
+		log.Printf("[AUDIT] User %s (%s) tried to list cache keys without admin role", user.Username, user.Id)
+		resp.WriteHeader(401)
+		resp.Write([]byte(`{"success": false, "reason": "Only admins can distribute cache to sub-orgs"}`))
+		return
+	}
+
+	var orgId string
+	location := strings.Split(request.URL.String(), "/")
+	if location[1] == "api" {
+		if len(location) <= 4 {
+			log.Printf("Path too short: %d", len(location))
+			resp.WriteHeader(401)
+			resp.Write([]byte(`{"success": false}`))
+			return
+		}
+
+		orgId = location[4]
+	}
+
+	if len(orgId) == 0 {
+		log.Printf("[ERROR] Missing org id in cache config")
+		resp.WriteHeader(401)
+		resp.Write([]byte(`{"success": false, "reason": "Missing org id"}`))
+		return
+	}
+
+	type cacheConfig struct {
+		Key            string   `json:"key"`
+		Action         string   `json:"action"`
+		SelectedSuborg []string `json:"selected_suborgs"`
+	}
+
+	var config cacheConfig
+
+	body, err := ioutil.ReadAll(request.Body)
+	if err != nil {
+		log.Printf("Error with body read: %s", err)
+		resp.WriteHeader(401)
+		resp.Write([]byte(`{"success": false}`))
+		return
+	}
+
+	err = json.Unmarshal(body, &config)
+	if err != nil {
+		log.Printf("[WARNING] Failed unmarshalling in cache config: %s", err)
+		resp.WriteHeader(401)
+		resp.Write([]byte(`{"success": false}`))
+		return
+	}
+
+	ctx := GetContext(request)
+
+	cacheId := fmt.Sprintf("%s_%s", orgId, config.Key)
+	cache, err := GetCacheKey(ctx, cacheId)
+	if err != nil {
+		log.Printf("[WARNING] Failed getting cache key '%s' for org %s (config)", config.Key, orgId)
+		resp.WriteHeader(400)
+		resp.Write([]byte(fmt.Sprintf(`{"success": false, "reason": "Failed to get key. Does it exist?", "extra": "%s"}`, cache.Key)))
+		return
+	}
+
+	if config.Action == "suborg_distribute" {
+
+		if len(config.SelectedSuborg) == 0 {
+			cache.SuborgDistribution = []string{}
+		} else {
+			cache.SuborgDistribution = config.SelectedSuborg
+		}
+
+		err = SetCacheKey(ctx, *cache)
+		if err != nil {
+			log.Printf("[WARNING] Failed setting cache key '%s' for org %s (config)", config.Key, orgId)
+			resp.WriteHeader(400)
+			resp.Write([]byte(fmt.Sprintf(`{"success": false, "reason": "Failed to set key. Does it exist?", "extra": "%s"}`, cache.Key)))
+			return
+		}
+	}
+
+	log.Printf("[INFO] Successfully updated cache key '%s' for org %s", config.Key, orgId)
+
+	resp.WriteHeader(200)
+	resp.Write([]byte(`{"success": true, "reason" : "Cache updated successfully!"}`))
+}
+
 func HandleDeleteCacheKey(resp http.ResponseWriter, request *http.Request) {
 	cors := HandleCors(resp, request)
 	if cors {

structs.go

@@ -986,10 +986,10 @@ type CacheKeyData struct {
 	Created int64 `json:"created" datastore:"Created"`
 	Edited  int64 `json:"edited" datastore:"Edited"`
 
-	FormattedKey        string `json:"formatted_key,omitempty" datastore:"FormattedKey"`
-	PublicAuthorization string `json:"public_authorization,omitempty" datastore:"PublicAuthorization"` // Used for public authorization
+	FormattedKey        string   `json:"formatted_key,omitempty" datastore:"FormattedKey"`
+	PublicAuthorization string   `json:"public_authorization,omitempty" datastore:"PublicAuthorization"` // Used for public authorization
+	SuborgDistribution  []string `json:"suborg_distribution" datastore:"suborg_distribution"`
 }
-
 type SyncConfig struct {
 	Interval int64  `json:"interval" datastore:"interval"`
 	Apikey   string `json:"api_key" datastore:"api_key"`
@@ -1183,7 +1183,7 @@ type Action struct {
 
 	ParentControlled bool `json:"parent_controlled" datastore:"parent_controlled"` // If the parent workflow node exists, and shouldn't be editable by child workflow
 
-	ParameterLocks   []ParameterLock `json:"parameter_locks" datastore:"parameter_locks"`
+	ParameterLocks []ParameterLock `json:"parameter_locks" datastore:"parameter_locks"`
 }
 
 // Added environment for location to execute
@@ -1209,12 +1209,12 @@ type Trigger struct {
 		X float64 `json:"x" datastore:"x"`
 		Y float64 `json:"y" datastore:"y"`
 	} `json:"position"`
-	Priority       int         `json:"priority" datastore:"priority"`
-	SourceWorkflow string      `json:"source_workflow" yaml:"source_workflow" datastore:"source_workflow"`
-	ExecutionDelay int64       `json:"execution_delay" yaml:"execution_delay" datastore:"execution_delay"`
-	AppAssociation WorkflowApp `json:"app_association" yaml:"app_association" datastore:"app_association"`
-	ParentControlled bool `json:"parent_controlled" datastore:"parent_controlled"` // If the parent workflow node exists, and shouldn't be editable by child workflow
-	
+	Priority         int         `json:"priority" datastore:"priority"`
+	SourceWorkflow   string      `json:"source_workflow" yaml:"source_workflow" datastore:"source_workflow"`
+	ExecutionDelay   int64       `json:"execution_delay" yaml:"execution_delay" datastore:"execution_delay"`
+	AppAssociation   WorkflowApp `json:"app_association" yaml:"app_association" datastore:"app_association"`
+	ParentControlled bool        `json:"parent_controlled" datastore:"parent_controlled"` // If the parent workflow node exists, and shouldn't be editable by child workflow
+
 	// TODO: make this a predictable field
 	// generated from current ID + workflow ID + orgid as seed
 	ReplacementForTrigger string `json:"replacement_for_trigger" datastore:"replacement_for_trigger"` // If this trigger is a replacement for another trigger