Rafael Cavazin
2013-07-21 f66e89662c091e082bd1d2feb6ac91513ccff273
src/main/java/com/gitblit/SparkleShareInviteServlet.java
New file
@@ -0,0 +1,113 @@
/*
 * Copyright 2013 gitblit.com.
 *
 * 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 com.gitblit;
import java.io.IOException;
import java.text.MessageFormat;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import com.gitblit.models.RepositoryModel;
import com.gitblit.models.UserModel;
import com.gitblit.utils.StringUtils;
/**
 * Handles requests for Sparkleshare Invites
 *
 * @author James Moger
 *
 */
public class SparkleShareInviteServlet extends HttpServlet {
   private static final long serialVersionUID = 1L;
   public SparkleShareInviteServlet() {
      super();
   }
   @Override
   protected void doPost(HttpServletRequest request, HttpServletResponse response)
         throws ServletException, java.io.IOException {
      processRequest(request, response);
   }
   @Override
   protected void doGet(HttpServletRequest request, HttpServletResponse response)
         throws ServletException, IOException {
      processRequest(request, response);
   }
   protected void processRequest(javax.servlet.http.HttpServletRequest request,
         javax.servlet.http.HttpServletResponse response) throws javax.servlet.ServletException,
         java.io.IOException {
      // extract repo name from request
      String repoUrl = request.getPathInfo().substring(1);
      // trim trailing .xml
      if (repoUrl.endsWith(".xml")) {
         repoUrl = repoUrl.substring(0, repoUrl.length() - 4);
      }
      String servletPath =  Constants.GIT_PATH;
      int schemeIndex = repoUrl.indexOf("://") + 3;
      String host = repoUrl.substring(0, repoUrl.indexOf('/', schemeIndex));
      String path = repoUrl.substring(repoUrl.indexOf(servletPath) + servletPath.length());
      String username = null;
      int fetchIndex = repoUrl.indexOf('@');
      if (fetchIndex > -1) {
         username = repoUrl.substring(schemeIndex, fetchIndex);
      }
      UserModel user;
      if (StringUtils.isEmpty(username)) {
         user = GitBlit.self().authenticate(request);
      } else {
         user = GitBlit.self().getUserModel(username);
      }
      if (user == null) {
         user = UserModel.ANONYMOUS;
         username = "";
      }
      // ensure that the requested repository exists
      RepositoryModel model = GitBlit.self().getRepositoryModel(path);
      if (model == null) {
         response.setStatus(HttpServletResponse.SC_NOT_FOUND);
         response.getWriter().append(MessageFormat.format("Repository \"{0}\" not found!", path));
         return;
      }
      StringBuilder sb = new StringBuilder();
      sb.append("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n");
      sb.append("<sparkleshare><invite>\n");
      sb.append(MessageFormat.format("<address>{0}</address>\n", host));
      sb.append(MessageFormat.format("<remote_path>{0}{1}</remote_path>\n", servletPath, model.name));
      if (GitBlit.getInteger(Keys.fanout.port, 0) > 0) {
         // Gitblit is running it's own fanout service for pubsub notifications
         sb.append(MessageFormat.format("<announcements_url>tcp://{0}:{1}</announcements_url>\n", request.getServerName(), GitBlit.getString(Keys.fanout.port, "")));
      }
      sb.append("</invite></sparkleshare>\n");
      // write invite to client
      response.setContentType("application/xml");
      response.setContentLength(sb.length());
      response.getWriter().append(sb.toString());
   }
}