James Moger
2011-10-06 a652a6886031ff805fcabd852925c8a11b25dc1c
src/com/gitblit/utils/FederationUtils.java
@@ -26,7 +26,10 @@
import java.security.SecureRandom;
import java.security.cert.CertificateException;
import java.security.cert.X509Certificate;
import java.text.MessageFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@@ -39,8 +42,14 @@
import javax.net.ssl.X509TrustManager;
import javax.servlet.http.HttpServletResponse;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.gitblit.Constants.FederationProposalResult;
import com.gitblit.Constants.FederationRequest;
import com.gitblit.FederationServlet;
import com.gitblit.IStoredSettings;
import com.gitblit.Keys;
import com.gitblit.models.FederationModel;
import com.gitblit.models.FederationProposal;
import com.gitblit.models.RepositoryModel;
@@ -75,6 +84,8 @@
   private static final DummyHostnameVerifier HOSTNAME_VERIFIER;
   private static final Logger LOGGER = LoggerFactory.getLogger(FederationUtils.class);
   static {
      SSLContext context = null;
      try {
@@ -89,21 +100,143 @@
   }
   /**
    * Returns the list of federated gitblit instances that this instance will
    * try to pull.
    *
    * @return list of registered gitblit instances
    */
   public static List<FederationModel> getFederationRegistrations(IStoredSettings settings) {
      List<FederationModel> federationRegistrations = new ArrayList<FederationModel>();
      List<String> keys = settings.getAllKeys(Keys.federation._ROOT);
      keys.remove(Keys.federation.name);
      keys.remove(Keys.federation.passphrase);
      keys.remove(Keys.federation.allowProposals);
      keys.remove(Keys.federation.proposalsFolder);
      keys.remove(Keys.federation.defaultFrequency);
      keys.remove(Keys.federation.sets);
      Collections.sort(keys);
      Map<String, FederationModel> federatedModels = new HashMap<String, FederationModel>();
      for (String key : keys) {
         String value = key.substring(Keys.federation._ROOT.length() + 1);
         List<String> values = StringUtils.getStringsFromValue(value, "\\.");
         String server = values.get(0);
         if (!federatedModels.containsKey(server)) {
            federatedModels.put(server, new FederationModel(server));
         }
         String setting = values.get(1);
         if (setting.equals("url")) {
            // url of the origin Gitblit instance
            federatedModels.get(server).url = settings.getString(key, "");
         } else if (setting.equals("token")) {
            // token for the origin Gitblit instance
            federatedModels.get(server).token = settings.getString(key, "");
         } else if (setting.equals("frequency")) {
            // frequency of the pull operation
            federatedModels.get(server).frequency = settings.getString(key, "");
         } else if (setting.equals("folder")) {
            // destination folder of the pull operation
            federatedModels.get(server).folder = settings.getString(key, "");
         } else if (setting.equals("bare")) {
            // whether pulled repositories should be bare
            federatedModels.get(server).bare = settings.getBoolean(key, true);
         } else if (setting.equals("mirror")) {
            // are the repositories to be true mirrors of the origin
            federatedModels.get(server).mirror = settings.getBoolean(key, true);
         } else if (setting.equals("mergeAccounts")) {
            // merge remote accounts into local accounts
            federatedModels.get(server).mergeAccounts = settings.getBoolean(key, false);
         } else if (setting.equals("sendStatus")) {
            // send a status acknowledgment to source Gitblit instance
            // at end of git pull
            federatedModels.get(server).sendStatus = settings.getBoolean(key, false);
         } else if (setting.equals("notifyOnError")) {
            // notify administrators on federation pull failures
            federatedModels.get(server).notifyOnError = settings.getBoolean(key, false);
         } else if (setting.equals("exclude")) {
            // excluded repositories
            federatedModels.get(server).exclusions = settings.getStrings(key);
         } else if (setting.equals("include")) {
            // included repositories
            federatedModels.get(server).inclusions = settings.getStrings(key);
         }
      }
      // verify that registrations have a url and a token
      for (FederationModel model : federatedModels.values()) {
         if (StringUtils.isEmpty(model.url)) {
            LOGGER.warn(MessageFormat.format(
                  "Dropping federation registration {0}. Missing url.", model.name));
            continue;
         }
         if (StringUtils.isEmpty(model.token)) {
            LOGGER.warn(MessageFormat.format(
                  "Dropping federation registration {0}. Missing token.", model.name));
            continue;
         }
         // set default frequency if unspecified
         if (StringUtils.isEmpty(model.frequency)) {
            model.frequency = settings.getString(Keys.federation.defaultFrequency, "60 mins");
         }
         federationRegistrations.add(model);
      }
      return federationRegistrations;
   }
   /**
    * Sends a federation poke to the Gitblit instance at remoteUrl. Pokes are
    * sent by an pulling Gitblit instance to an origin Gitblit instance as part
    * of the proposal process. This is to ensure that the pulling Gitblit
    * instance has an IP route to the origin instance.
    *
    * @param remoteUrl
    *            the remote Gitblit instance to send a federation proposal to
    * @param proposal
    *            a complete federation proposal
    * @return true if there is a route to the remoteUrl
    */
   public static boolean poke(String remoteUrl) throws Exception {
      String url = FederationServlet.asFederationLink(remoteUrl, null, FederationRequest.POKE);
      Gson gson = new Gson();
      String json = gson.toJson("POKE");
      int status = writeJson(url, json);
      return status == HttpServletResponse.SC_OK;
   }
   /**
    * Sends a federation proposal to the Gitblit instance at remoteUrl
    * 
    * @param remoteUrl
    *            the remote Gitblit instance to send a federation proposal to
    * @param proposal
    *            a complete federation proposal
    * @return true if the proposal was received
    * @return the federation proposal result code
    */
   public static boolean propose(String remoteUrl, FederationProposal proposal) throws Exception {
   public static FederationProposalResult propose(String remoteUrl, FederationProposal proposal)
         throws Exception {
      String url = FederationServlet
            .asFederationLink(remoteUrl, null, FederationRequest.PROPOSAL);
      Gson gson = new GsonBuilder().setPrettyPrinting().create();
      String json = gson.toJson(proposal);
      int status = writeJson(url, json);
      return status == HttpServletResponse.SC_OK;
      switch (status) {
      case HttpServletResponse.SC_FORBIDDEN:
         // remote Gitblit Federation disabled
         return FederationProposalResult.FEDERATION_DISABLED;
      case HttpServletResponse.SC_BAD_REQUEST:
         // remote Gitblit did not receive any JSON data
         return FederationProposalResult.MISSING_DATA;
      case HttpServletResponse.SC_METHOD_NOT_ALLOWED:
         // remote Gitblit not accepting proposals
         return FederationProposalResult.NO_PROPOSALS;
      case HttpServletResponse.SC_NOT_ACCEPTABLE:
         // remote Gitblit failed to poke this Gitblit instance
         return FederationProposalResult.NO_POKE;
      case HttpServletResponse.SC_OK:
         // received
         return FederationProposalResult.ACCEPTED;
      default:
         return FederationProposalResult.ERROR;
      }
   }
   /**