James Moger
2013-09-30 699e71e76b15081baf746c6ce9c9144f7e5f1ff9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
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());
    }
}