support decorateContext in CloneOptionTrait

Author: long76

src/main/java/jenkins/plugins/git/AbstractGitSCMSource.java

@@ -404,6 +404,7 @@ private <T, C extends GitSCMSourceContext<C, R>, R extends GitSCMSourceRequest>
                 fetch.tags(context.wantTags());
             }
             fetch = fetch.prune(prune);
+            fetch = fetch.shallow(context.wantShallow()).depth(context.depth());
 
             URIish remoteURI = null;
             try {

src/main/java/jenkins/plugins/git/GitSCMSourceContext.java

@@ -94,6 +94,14 @@ public class GitSCMSourceContext<C extends GitSCMSourceContext<C, R>, R extends
      */
     @NonNull
     private String remoteName = AbstractGitSCMSource.DEFAULT_REMOTE_NAME;
+    /**
+     * Perform shallow clone
+     */
+    private boolean shallow;
+    /**
+     * Shallow clone depth
+     */
+    private Integer depth;
 
     /**
      * Constructor.
@@ -192,6 +200,24 @@ public final String remoteName() {
         return remoteName;
     }
 
+    /**
+     * Returns {@code true} if perform shallow clone
+     *
+     * @return {@code true} if perform shallow clone
+     */
+    public final boolean wantShallow() {
+        return shallow;
+    }
+
+    /**
+     * Returns shallow clone depth
+     *
+     * @return shallow clone depth
+     */
+    public final Integer depth() {
+        return depth;
+    }
+
     /**
      * Adds a requirement for branch details to any {@link GitSCMSourceRequest} for this context.
      *
@@ -358,6 +384,32 @@ public final List<RefSpec> asRefSpecs() {
         return result;
     }
 
+    /**
+     * Configures perform shallow clone
+     *
+     * @param shallow {@code true} to perform shallow clone
+     * @return {@code this} for method chaining.
+     */
+    @SuppressWarnings("unchecked")
+    @NonNull
+    public final C shallow(boolean shallow) {
+        this.shallow = shallow;
+        return (C) this;
+    }
+
+    /**
+     * Configures shallow clone depth
+     *
+     * @param depth count atest commits of a repository
+     * @return {@code this} for method chaining.
+     */
+    @SuppressWarnings("unchecked")
+    @NonNull
+    public final C depth(Integer depth) {
+        this.depth = depth;
+        return (C) this;
+    }
+
     /**
      * {@inheritDoc}
      */

src/main/java/jenkins/plugins/git/traits/CloneOptionTrait.java

@@ -27,6 +27,8 @@
 
 import hudson.Extension;
 import hudson.plugins.git.extensions.impl.CloneOption;
+import jenkins.plugins.git.GitSCMSourceContext;
+import jenkins.scm.api.trait.SCMSourceContext;
 import jenkins.scm.api.trait.SCMSourceTrait;
 import org.jenkinsci.Symbol;
 import org.kohsuke.stapler.DataBoundConstructor;
@@ -47,6 +49,12 @@ public CloneOptionTrait(CloneOption extension) {
         super(extension);
     }
 
+    @Override
+    protected void decorateContext(SCMSourceContext<?, ?> context) {
+        CloneOption extension = getExtension();
+        ((GitSCMSourceContext<?, ?>) context).shallow(extension.isShallow()).depth(extension.getDepth());
+    }
+
     /**
      * Our {@link hudson.model.Descriptor}
      */