Add support for TokenMacro in URL of Stash Server [JENKINS-32324] Fix…

…es GH jenkinsci#139

pom.xml

@@ -3,7 +3,7 @@
     <parent>
         <groupId>org.jenkins-ci.plugins</groupId>
         <artifactId>plugin</artifactId>
-        <version>2.23</version>
+        <version>2.26</version>
         <relativePath />
     </parent>
     <packaging>hpi</packaging>

src/main/java/org/jenkinsci/plugins/stashNotifier/StashNotifier.java

@@ -377,19 +377,16 @@ protected Collection<String> lookupCommitSha1s(
 	 *
 	 * @param logger    the logger to log messages to
 	 * @param run
+         * @param stashServer
 	 * @return			the HttpClient
 	 */
-	protected HttpClient getHttpClient(PrintStream logger, Run<?, ?> run) throws Exception {
+	protected HttpClient getHttpClient(PrintStream logger, Run<?, ?> run, String stashServer) throws Exception {
         boolean ignoreUnverifiedSSL = ignoreUnverifiedSSLPeer;
-        String stashServer = stashServerBaseUrl;
+
         DescriptorImpl descriptor = getDescriptor();
 
         CertificateCredentials certificateCredentials = getCredentials(CertificateCredentials.class, run.getParent());
 
-        if ("".equals(stashServer) || stashServer == null) {
-            stashServer = descriptor.getStashRootUrl();
-        }
-
         URL url = new URL(stashServer);
         HttpClientBuilder builder = HttpClientBuilder.create();
         RequestConfig.Builder requestBuilder = RequestConfig.custom();
@@ -691,9 +688,13 @@ protected NotificationResult notifyStash(
 			final StashBuildState state) throws Exception {
 		HttpEntity stashBuildNotificationEntity
 			= newStashBuildNotificationEntity(run, state, listener);
+
+                String stashURL= expandStashURL(run, listener);
+
+                logger.println("Notifying Stash at \""+stashURL+"\"");
 
-		HttpPost req = createRequest(stashBuildNotificationEntity, run.getParent(), commitSha1);
-		HttpClient client = getHttpClient(logger, run);
+		HttpPost req = createRequest(stashBuildNotificationEntity, run.getParent(), commitSha1, stashURL);
+		HttpClient client = getHttpClient(logger, run, stashURL);
 		try {
 			HttpResponse res = client.execute(req);
 			if (res.getStatusLine().getStatusCode() != 204) {
@@ -712,7 +713,7 @@ protected NotificationResult notifyStash(
      *
      * @param clazz The type of {@link com.cloudbees.plugins.credentials.Credentials} to return.
      * @param project The hierarchical project context within which the credentials are searched for.
-     * @return The first credentials of the given type that are found withing the project hierarchy, or null otherwise.
+     * @return The first credentials of the given type that are found within the project hierarchy, or null otherwise.
      */
     private <T extends Credentials> T getCredentials(final Class<T> clazz, final Item project) {
 
@@ -781,19 +782,16 @@ protected  <C extends Credentials> List<C> lookupCredentials(Class<C> type, Item
 	 * @param stashBuildNotificationEntity	a entity containing the parameters
 	 * 										for Stash
 	 * @param commitSha1	the SHA1 of the commit that was built
+     * @param url
 	 * @return				the HTTP POST request to the Stash build API
 	 */
     protected HttpPost createRequest(
 			final HttpEntity stashBuildNotificationEntity,
-            final Item project,
-			final String commitSha1) throws AuthenticationException {
-
-		String url = stashServerBaseUrl;
-        DescriptorImpl descriptor = getDescriptor();
-
-        if ("".equals(url) || url == null)
-            url = descriptor.getStashRootUrl();
-
+                        final Item project,
+			final String commitSha1,
+                        final String url) throws AuthenticationException {
+
+
 		HttpPost req = new HttpPost(
 				url
 				+ "/rest/build-status/1.0/commits/"
@@ -819,6 +817,28 @@ protected HttpPost createRequest(
 		return req;
 	}
 
+    private String expandStashURL(Run<?, ?> run, final TaskListener listener) {
+        String url = stashServerBaseUrl;
+        DescriptorImpl descriptor = getDescriptor();
+        if ("".equals(url) || url == null) {
+            url = descriptor.getStashRootUrl();
+        }
+
+        try {
+            if (!(run instanceof AbstractBuild<?, ?>)) {
+                url = TokenMacro.expandAll(run, new FilePath(run.getRootDir()), listener, url);
+            } else {
+                url = TokenMacro.expandAll((AbstractBuild<?, ?>) run, listener, url);
+            }
+
+        } catch (IOException | InterruptedException | MacroEvaluationException ex) {
+                PrintStream logger = listener.getLogger();
+                logger.println("Unable to expand Stash Server URL");
+		ex.printStackTrace(logger);
+            }
+        return url;
+    }
+
 	/**
 	 * Returns the HTTP POST entity body with the JSON representation of the
 	 * run result to be sent to the Stash build API.

src/test/java/org/jenkinsci/plugins/stashNotifier/StashNotifierTest.java

@@ -198,7 +198,7 @@ public void test_build_http_client_with_proxy() throws Exception {
         PrintStream logger = mock(PrintStream.class);
 
         //when
-        sn.getHttpClient(logger, build);
+        sn.getHttpClient(logger, build, "http://localhost");
 
         //then
         ArgumentCaptor<HttpHost> proxyCaptor = ArgumentCaptor.forClass(HttpHost.class);
@@ -239,7 +239,7 @@ public void test_build_http_client_https() throws Exception {
         PrintStream logger = mock(PrintStream.class);
 
         //when
-        sn.getHttpClient(logger, build);
+        sn.getHttpClient(logger, build,"https://localhost");
 
         //then
         verify(httpClientBuilder).setSSLSocketFactory(any(SSLConnectionSocketFactory.class));
@@ -535,7 +535,7 @@ public void test_createRequest() throws AuthenticationException {
         when(CredentialsMatchers.firstOrNull(anyCollection(), any(CredentialsMatcher.class))).thenReturn(credential);
 
         //when
-        HttpPost request = sn.createRequest(mock(HttpEntity.class), mock(Item.class), sha1);
+        HttpPost request = sn.createRequest(mock(HttpEntity.class), mock(Item.class), sha1, "http://localhost");
 
         //then
         assertThat(request, is(not(nullValue())));
@@ -682,14 +682,14 @@ private NotificationResult notifyStash(int statusCode) throws Exception {
         when(buildListener.getLogger()).thenReturn(logger);
         doReturn("someKey1").when(sn).getBuildKey(eq(build), eq(buildListener));
         HttpPost httpPost = mock(HttpPost.class);
-        doReturn(httpPost).when(sn).createRequest(any(HttpEntity.class), any(Item.class), anyString());
+        doReturn(httpPost).when(sn).createRequest(any(HttpEntity.class), any(Item.class), anyString(), anyString());
         CloseableHttpResponse resp = mock(CloseableHttpResponse.class);
         StatusLine sl = mock(StatusLine.class);
         when(sl.getStatusCode()).thenReturn(statusCode);
         when(resp.getStatusLine()).thenReturn(sl);
         when(resp.getEntity()).thenReturn(new StringEntity(""));
         when(client.execute(eq(httpPost))).thenReturn(resp);
-        doReturn(client).when(sn).getHttpClient(any(PrintStream.class), any(AbstractBuild.class));
+        doReturn(client).when(sn).getHttpClient(any(PrintStream.class), any(AbstractBuild.class), anyString());
         return sn.notifyStash(logger, build, sha1, buildListener, StashBuildState.FAILED);
     }