[JENKINS-76317]: Add support to define maintenance windows for Clouds

src/main/java/com/sap/prd/jenkins/plugins/agent_maintenance/MaintenanceActionFactory.java → src/main/java/com/sap/prd/jenkins/plugins/agent_maintenance/AgentMaintenanceActionFactory.java

@@ -12,17 +12,17 @@
 
 /** Inject the action link to agents. */
 @Extension
-public class MaintenanceActionFactory extends TransientActionFactory<SlaveComputer> {
+public class AgentMaintenanceActionFactory extends TransientActionFactory<SlaveComputer> {
 
   @Override
   @NonNull
   public Collection<? extends Action> createFor(@NonNull SlaveComputer target) {
     List<Action> result = new ArrayList<>();
     if (!(target instanceof AbstractCloudComputer)
         && target.getActions().stream().noneMatch(x -> x instanceof MaintenanceAction)) {
-      MaintenanceAction action = new MaintenanceAction(target);
+      MaintenanceTarget mt = new MaintenanceTarget(MaintenanceTarget.TargetType.AGENT, target.getName());
+      MaintenanceAction action = new MaintenanceAction(mt);
       result.add(action);
-      target.addAction(new MaintenanceAction(target));
     }
     return result;
   }

src/main/java/com/sap/prd/jenkins/plugins/agent_maintenance/AgentMaintenanceRetentionStrategy.java

@@ -62,8 +62,9 @@ public boolean isManualLaunchAllowed(final SlaveComputer c) {
   @Override
   @GuardedBy("hudson.model.Queue.lock")
   public synchronized long check(final SlaveComputer c) {
-    MaintenanceWindow maintenance = MaintenanceHelper.getInstance().getMaintenance(c.getName());
-    MaintenanceHelper.getInstance().checkRecurring(c.getName());
+    MaintenanceTarget target = new MaintenanceTarget(MaintenanceTarget.TargetType.AGENT, c.getName());
+    MaintenanceWindow maintenance = MaintenanceHelper.getInstance().getMaintenance(target.toKey());
+    MaintenanceHelper.getInstance().checkRecurring(target.toKey());
     LOGGER.log(Level.FINER, "Checking for Maintenance Window for agent {0}. online = {1}, idle = {2}",
         new Object[] { c.getName(), c.isOnline(), c.isIdle() });
     if (maintenance != null) {
@@ -74,13 +75,10 @@ public synchronized long check(final SlaveComputer c) {
         if (maintenance.isKeepUpWhenActive()) {
           if (!maintenance.isMaxWaitTimeFinished()) {
             if (c.isIdle()) {
-              Queue.withLock(new Runnable() {
-                @Override
-                public void run() {
-                  LOGGER.log(Level.INFO, "Disconnecting agent {0} as it was idle when maintenance window started.",
-                      new Object[] { c.getName() });
-                  c.disconnect(maintenance.getOfflineCause(c.getName()));
-                }
+              Queue.withLock(() -> {
+                LOGGER.log(Level.INFO, "Disconnecting agent {0} as it was idle when maintenance window started.",
+                    new Object[] { c.getName() });
+                c.disconnect(maintenance.getOfflineCause(c.getName()));
               });
             }
           } else {

src/main/java/com/sap/prd/jenkins/plugins/agent_maintenance/CloudMaintenanceActionFactory.java

@@ -0,0 +1,35 @@
+package com.sap.prd.jenkins.plugins.agent_maintenance;
+
+import edu.umd.cs.findbugs.annotations.NonNull;
+import hudson.Extension;
+import hudson.model.Action;
+import hudson.slaves.Cloud;
+import jenkins.model.Jenkins;
+import jenkins.model.TransientActionFactory;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+
+@Extension
+public class CloudMaintenanceActionFactory extends TransientActionFactory<Cloud> {
+
+  @NonNull
+  @Override
+  public Collection<? extends Action> createFor(@NonNull Cloud target) {
+    if (!Jenkins.get().hasPermission(Jenkins.ADMINISTER)) {
+      return Collections.emptyList();
+    }
+    MaintenanceTarget mt = new MaintenanceTarget(MaintenanceTarget.TargetType.CLOUD, target.name);
+    MaintenanceAction action = new MaintenanceAction(mt);
+//    target.addAction(action);
+
+    return Collections.singletonList(action);
+  }
+
+  @Override
+  public Class<Cloud> type() {
+    return Cloud.class;
+  }
+}

src/main/java/com/sap/prd/jenkins/plugins/agent_maintenance/CloudMaintenanceProvisioningListener.java

@@ -0,0 +1,36 @@
+package com.sap.prd.jenkins.plugins.agent_maintenance;
+
+import hudson.Extension;
+import hudson.model.queue.CauseOfBlockage;
+import hudson.slaves.Cloud;
+import hudson.slaves.CloudProvisioningListener;
+
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+@Extension
+public class CloudMaintenanceProvisioningListener extends CloudProvisioningListener {
+  private static final Logger LOGGER = Logger.getLogger(CloudMaintenanceProvisioningListener.class.getName());
+
+  @Override
+  public CauseOfBlockage canProvision(Cloud cloud, Cloud.CloudState state, int numExecutors) {
+    try {
+      MaintenanceTarget target = new MaintenanceTarget(MaintenanceTarget.TargetType.CLOUD, cloud.name);
+      LOGGER.log(Level.FINER, "Checking for Maintenance Window for cloud {0}", cloud.name);
+      if (MaintenanceHelper.getInstance()
+              .hasActiveMaintenanceWindows(target.toKey())) {
+
+        return new CauseOfBlockage() {
+          @Override
+          public String getShortDescription() {
+            return "Cloud is in maintenance: " + cloud.name;
+          }
+        };
+      }
+    } catch (Exception e) {
+      LOGGER.log(Level.WARNING, "Failed to check maintenance for cloud " +
+              cloud.name, e);
+    }
+    return null; // Allow provisioning when no active maintenance window
+  }
+}