RESTServiceURLBuilder.java

/**
 * Copyright (c) Istituto Nazionale di Fisica Nucleare, 2006-2014.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.italiangrid.voms.request.impl;

import java.net.MalformedURLException;
import java.net.URL;
import java.util.Iterator;
import java.util.List;

import org.italiangrid.voms.VOMSError;
import org.italiangrid.voms.request.VOMSACRequest;
import org.italiangrid.voms.request.VOMSServerInfo;

/**
 * An helper class which builds a URL suitable for an HTTPS VOMS REST request
 * starting from the VOMS URI as available in a {@link VOMSServerInfo} object.
 * 
 * @author Valerio Venturi
 * 
 */
public class RESTServiceURLBuilder {

  public URL build(VOMSServerInfo endpoint, VOMSACRequest request) {

    URL url = null;

    try {

      url = new URL("https", endpoint.getURL().getHost(), endpoint.getURL()
        .getPort(), buildPath(endpoint, request));

    } catch (MalformedURLException e) {

      throw new VOMSError("Malformed URI: " + e.getMessage());
    }

    return url;
  }

  private String buildPath(VOMSServerInfo endpoint, VOMSACRequest request) {

    StringBuilder stringBuilder = new StringBuilder();

    stringBuilder.append("/generate-ac?fqans=");

    if (request.getRequestedFQANs().isEmpty()) {
      // Take voname from endpoint info
      stringBuilder.append("/" + endpoint.getVoName());

    } else {

      List<String> FQANs = request.getRequestedFQANs();

      Iterator<String> i = FQANs.iterator();

      boolean first = true;

      while (i.hasNext()) {

        if (!first)
          stringBuilder.append(",");

        stringBuilder.append((String) i.next());

        first = false;
      }
    }

    String targetString = targetListAsCommaSeparatedList(request.getTargets());

    if (targetString != null && targetString.trim().length() != 0) {

      stringBuilder.append("&targets=");
      stringBuilder.append(targetString);
    }

    stringBuilder.append("&lifetime=");
    stringBuilder.append(request.getLifetime());

    return stringBuilder.toString();
  }

  private String targetListAsCommaSeparatedList(List<String> targets) {

    StringBuilder targetStringBuilder = new StringBuilder();

    for (String target : targets) {

      targetStringBuilder.append(target);
      targetStringBuilder.append(',');
    }

    return targetStringBuilder.toString();
  }
}