Extension: RepositorySizeGithubAPI to get size of a repo

src/main/java/org/jenkinsci/plugins/github_branch_source/GitHubRepoSizeEstimator.java

@@ -0,0 +1,42 @@
+package org.jenkinsci.plugins.github_branch_source;
+
+import com.cloudbees.plugins.credentials.common.StandardCredentials;
+import hudson.Extension;
+import hudson.model.Item;
+import org.kohsuke.github.GHRepository;
+import org.kohsuke.github.GitHub;
+import jenkins.plugins.git.GitToolChooser;
+
+import java.io.IOException;
+
+public class GitHubRepoSizeEstimator {
+    /**
+     * This extension intends to perform a GET request without any credentials on the provided repository URL
+     * to return the size of repository.
+     */
+    @Extension
+    public static class RepositorySizeGithubAPI extends GitToolChooser.RepositorySizeAPI {
+
+        @Override
+        public boolean isApplicableTo(String repoUrl, Item context, String credentialsId) {
+            StandardCredentials credentials = Connector.lookupScanCredentials(context, repoUrl, credentialsId);
+            GitHubRepositoryInfo info = GitHubRepositoryInfo.forRepositoryUrl(repoUrl);
+            try {
+                GitHub gitHub = Connector.connect(info.getApiUri(), credentials);
+                gitHub.checkApiUrlValidity();
+            } catch (IOException e) {
+                return false;
+            }
+            return true;
+        }
+
+        @Override
+        public Long getSizeOfRepository(String repoUrl, Item context, String credentialsId) throws Exception {
+            StandardCredentials credentials = Connector.lookupScanCredentials(context, repoUrl, credentialsId);
+            GitHubRepositoryInfo info = GitHubRepositoryInfo.forRepositoryUrl(repoUrl);
+            GitHub github = Connector.connect(info.getApiUri(), credentials);
+            GHRepository ghRepository = github.getRepository(info.getRepoOwner() + '/' + info.getRepository());
+            return (long) ghRepository.getSize();
+        }
+    }
+}

src/test/java/org/jenkinsci/plugins/github_branch_source/GitHubRepoSizeEstimatorTest.java

@@ -0,0 +1,98 @@
+package org.jenkinsci.plugins.github_branch_source;
+
+import com.cloudbees.plugins.credentials.*;
+import com.cloudbees.plugins.credentials.common.StandardCredentials;
+import com.cloudbees.plugins.credentials.domains.Domain;
+import com.cloudbees.plugins.credentials.impl.UsernamePasswordCredentialsImpl;
+import hudson.model.FreeStyleBuild;
+import hudson.model.FreeStyleProject;
+import hudson.model.Item;
+import hudson.model.TopLevelItem;
+import hudson.plugins.git.BranchSpec;
+import hudson.plugins.git.GitSCM;
+import hudson.plugins.git.SubmoduleConfig;
+import hudson.plugins.git.UserRemoteConfig;
+import hudson.plugins.git.extensions.GitSCMExtension;
+import hudson.plugins.git.extensions.impl.DisableRemotePoll;
+import hudson.triggers.SCMTrigger;
+import jenkins.model.Jenkins;
+import jenkins.plugins.git.GitToolChooser;
+import org.jenkinsci.plugins.gitclient.JGitTool;
+import org.jenkinsci.plugins.workflow.cps.CpsFlowDefinition;
+import org.jenkinsci.plugins.workflow.job.WorkflowJob;
+import org.jenkinsci.plugins.workflow.job.WorkflowRun;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+import org.jvnet.hudson.test.JenkinsRule;
+import org.kohsuke.github.GitHub;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
+import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
+import static com.github.tomakehurst.wiremock.client.WireMock.*;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.Matchers.notNullValue;
+
+@RunWith(Parameterized.class)
+public class GitHubRepoSizeEstimatorTest extends GitSCMSourceBase {
+
+    public GitHubRepoSizeEstimatorTest(GitHubSCMSource source) {
+        this.source = source;
+    }
+
+    @Parameterized.Parameters(name = "{index}: revision={0}")
+    public static GitHubSCMSource[] revisions() {
+        return new GitHubSCMSource[]{
+                new GitHubSCMSource("cloudbeers", "yolo", null, false),
+                new GitHubSCMSource("", "", "https://github.com/cloudbeers/yolo", true)
+        };
+    }
+
+    @Rule
+    public final JenkinsRule j = new JenkinsRule();
+
+    private CredentialsStore store = null;
+
+    @Before
+    public void enableSystemCredentialsProvider() {
+        SystemCredentialsProvider.getInstance().setDomainCredentialsMap(
+                Collections.singletonMap(Domain.global(), Collections.<Credentials>emptyList()));
+        for (CredentialsStore s : CredentialsProvider.lookupStores(Jenkins.get())) {
+            if (s.getProvider() instanceof SystemCredentialsProvider.ProviderImpl) {
+                store = s;
+                break;
+            }
+        }
+        assertThat("The system credentials provider is enabled", store, notNullValue());
+    }
+
+    @Test
+    public void isApplicableToTest() throws Exception {
+        GitHubRepoSizeEstimator.RepositorySizeGithubAPI api =  j.jenkins.getExtensionList(GitToolChooser.RepositorySizeAPI.class).get(GitHubRepoSizeEstimator.RepositorySizeGithubAPI.class);
+        Item context = Mockito.mock(Item.class);
+        store.addCredentials(Domain.global(), createCredential(CredentialsScope.GLOBAL, "github"));
+        store.save();
+
+        githubApi.stubFor(
+                get(urlEqualTo("/"))
+                        .willReturn(
+                                aResponse()
+                                        .withHeader("Content-Type", "application/json; charset=utf-8")
+                                        .withBodyFile("../__files/body-(root)-XwEI7.json")));
+
+        assertThat(api.isApplicableTo("https://github.com/rishabhBudhouliya/zoom-suspender.git", context, "github"), is(false));
+    }
+
+    private StandardCredentials createCredential(CredentialsScope scope, String id) {
+        return new UsernamePasswordCredentialsImpl(scope, id, "desc: " + id, "username", "password");
+    }
+}