Merge pull request #3 from larhauga/add-routing-rules

add routing rule support

Author: larhauga

src/main/java/com/weareadaptive/nexus/casc/plugin/internal/NexusCascPlugin.java

@@ -3,6 +3,7 @@
 import com.weareadaptive.nexus.casc.plugin.internal.config.*;
 import org.apache.shiro.util.ThreadContext;
 import org.eclipse.sisu.Description;
+import org.elasticsearch.cluster.metadata.MappingMetaData;
 import org.sonatype.nexus.CoreApi;
 import org.sonatype.nexus.blobstore.api.BlobStore;
 import org.sonatype.nexus.blobstore.api.BlobStoreConfiguration;
@@ -14,10 +15,13 @@
 import org.sonatype.nexus.common.app.BaseUrlManager;
 import org.sonatype.nexus.common.app.ManagedLifecycle;
 import org.sonatype.nexus.common.app.NotWritableException;
+import org.sonatype.nexus.common.entity.EntityId;
 import org.sonatype.nexus.common.stateguard.StateGuardLifecycleSupport;
 import org.sonatype.nexus.repository.Repository;
+import org.sonatype.nexus.repository.routing.*;
 import org.sonatype.nexus.repository.config.Configuration;
 import org.sonatype.nexus.repository.manager.RepositoryManager;
+import org.sonatype.nexus.repository.security.RepositoryPermissionChecker;
 import org.sonatype.nexus.security.SecurityApi;
 import org.sonatype.nexus.security.SecuritySystem;
 import org.sonatype.nexus.security.authz.AuthorizationManager;
@@ -62,6 +66,9 @@ public class NexusCascPlugin extends StateGuardLifecycleSupport {
     private final BlobStoreManager blobStoreManager;
     private final RealmManager realmManager;
     private final CapabilityRegistry capabilityRegistry;
+    private final RoutingRuleStore routingRuleStore;
+    private final RoutingRuleHelper routingRuleHelper;
+    private final RepositoryPermissionChecker repositoryPermissionChecker;
 
     @Inject
     public NexusCascPlugin(
@@ -74,7 +81,11 @@ public NexusCascPlugin(
             final RepositoryManager repositoryManager,
             final BlobStoreManager blobStoreManager,
             final RealmManager realmManager,
-            final CapabilityRegistry capabilityRegistry) throws IllegalAccessException, NoSuchMethodException, InvocationTargetException {
+            final CapabilityRegistry capabilityRegistry,
+            final RoutingRuleStore routingRuleStore,
+            final RoutingRuleHelper routingRuleHelper,
+            final RepositoryPermissionChecker repositoryPermissionChecker
+        ) throws IllegalAccessException, NoSuchMethodException, InvocationTargetException {
         this.baseUrlManager = baseUrlManager;
         this.coreApi = coreApi;
         this.securityApi = securityApi;
@@ -85,6 +96,9 @@ public NexusCascPlugin(
         this.repositoryManager = repositoryManager;
         this.realmManager = realmManager;
         this.capabilityRegistry = capabilityRegistry;
+        this.routingRuleStore = routingRuleStore;
+        this.routingRuleHelper = routingRuleHelper;
+        this.repositoryPermissionChecker = repositoryPermissionChecker;
     }
 
     @Override
@@ -381,6 +395,16 @@ private void applyRepositoryConfig(ConfigRepository repository) {
             log.warn("repository.pruneCleanupPolicies has no effect when no cleanup policies are configured!");
         }
 
+        if (repository.getRoutingRules() != null && !repository.getRoutingRules().isEmpty()) {
+            repository.getRoutingRules().forEach(this::applyRoutingRuleConfig);
+            routingRuleStore.list().forEach(existingRule -> {
+                if (repository.getRoutingRules().stream().noneMatch(rr -> existingRule.name().equals(rr.getName()))) {
+                    log.info("Pruning routing rule {}", existingRule.name());
+                    routingRuleStore.delete(existingRule);
+                }
+            });
+        }
+
         if (repository.getRepositories() != null) {
             repository.getRepositories().forEach(repoConfig -> {
                 Repository existingRepo = repositoryManager.get(repoConfig.getName());
@@ -396,6 +420,16 @@ private void applyRepositoryConfig(ConfigRepository repository) {
 
                     configuration.setAttributes(repoConfig.getAttributes());
 
+                    if ( repoConfig.getRoutingRuleName() != null) {
+                        String ruleName = repoConfig.getRoutingRuleName();
+                        RoutingRule rule = routingRuleStore.getByName(ruleName);
+                        if (rule != null) {
+                            configuration.setRoutingRuleId(rule.id());
+                        } else {
+                            log.error("Routing rule {} does not exist", ruleName);
+                        }
+                    }
+
                     patchRepoAttributes(repoConfig.getAttributes());
 
                     if (repoConfig.getOnline() != null) {
@@ -476,6 +510,22 @@ private void patchRepoAttributes(Map<String, Map<String, Object>> attributes) {
         }
     }
 
+    private void applyRoutingRuleConfig(ConfigRoutingRule routingRuleConfig) {
+        RoutingRule existingRule = routingRuleStore.getByName(routingRuleConfig.getName());
+        if (existingRule != null) {
+            existingRule.matchers(routingRuleConfig.getMatchers());
+            existingRule.description(routingRuleConfig.getDescription());
+            existingRule.mode(RoutingMode.valueOf(routingRuleConfig.getMode()));
+            routingRuleStore.update(existingRule);
+        } else {
+            RoutingRule rule = routingRuleStore.newRoutingRule()
+                .name(routingRuleConfig.getName())
+                .description(routingRuleConfig.getDescription())
+                .mode(RoutingMode.valueOf(routingRuleConfig.getMode()))
+                .matchers(routingRuleConfig.getMatchers());
+            routingRuleStore.create(rule);
+        }
+    }
     /**
      * Apply all configs related to security
      *

src/main/java/com/weareadaptive/nexus/casc/plugin/internal/config/ConfigRepository.java

@@ -10,6 +10,8 @@ public class ConfigRepository {
     private Boolean pruneRepositories;
     private List<ConfigRepositoryEntry> repositories;
 
+    private List<ConfigRoutingRule> routingRules;
+
     public Boolean getPruneBlobStores() {
         return pruneBlobStores;
     }
@@ -57,4 +59,13 @@ public List<ConfigRepositoryEntry> getRepositories() {
     public void setRepositories(List<ConfigRepositoryEntry> repositories) {
         this.repositories = repositories;
     }
+
+    public List<ConfigRoutingRule> getRoutingRules() {
+        return routingRules;
+    }
+
+    public void setRoutingRules(List<ConfigRoutingRule> routingRules) {
+        this.routingRules = routingRules;
+    }
+
 }

src/main/java/com/weareadaptive/nexus/casc/plugin/internal/config/ConfigRepositoryEntry.java

@@ -1,5 +1,7 @@
 package com.weareadaptive.nexus.casc.plugin.internal.config;
 
+import org.sonatype.nexus.common.entity.EntityId;
+
 import java.util.HashMap;
 import java.util.Map;
 
@@ -9,6 +11,8 @@ public class ConfigRepositoryEntry {
     private Boolean online;
     private Map<String, Map<String, Object>> attributes = new HashMap<>();
 
+    private String routingRuleName;
+
     public String getName() {
         return name;
     }
@@ -40,4 +44,12 @@ public Map<String, Map<String, Object>> getAttributes() {
     public void setAttributes(Map<String, Map<String, Object>> attributes) {
         this.attributes = attributes;
     }
+    public String getRoutingRuleName() {
+        return routingRuleName;
+    }
+
+    public void setRoutingRuleName(String routingRuleName) {
+        this.routingRuleName = routingRuleName;
+    }
+
 }

src/main/java/com/weareadaptive/nexus/casc/plugin/internal/config/ConfigRoutingRule.java

@@ -0,0 +1,42 @@
+package com.weareadaptive.nexus.casc.plugin.internal.config;
+
+import java.util.List;
+
+public class ConfigRoutingRule {
+    private String name;
+    private String description;
+    private String mode;
+    private List<String> matchers;
+
+    public String getDescription() {
+        return description;
+    }
+
+    public void setDescription(String description) {
+        this.description = description;
+    }
+
+    public String getMode() {
+        return mode;
+    }
+
+    public void setMode(String mode) {
+        this.mode = mode;
+    }
+
+    public List<String> getMatchers() {
+        return matchers;
+    }
+
+    public void setMatchers(List<String> matchers) {
+        this.matchers = matchers;
+    }
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+}