| // Copyright 2015 Google Inc. All Rights Reserved. |
| // |
| // 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.google.gitiles; |
| |
| import static org.eclipse.jgit.http.server.ServletUtils.ATTRIBUTE_REPOSITORY; |
| |
| import com.google.gitiles.GitilesRequestFailureException.FailureReason; |
| import com.google.gitiles.doc.DocServlet; |
| import com.google.gitiles.doc.HtmlSanitizer; |
| import java.io.IOException; |
| import javax.servlet.ServletConfig; |
| import javax.servlet.ServletException; |
| import javax.servlet.http.HttpServlet; |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| import org.eclipse.jgit.lib.ObjectId; |
| import org.eclipse.jgit.lib.Repository; |
| import org.eclipse.jgit.revwalk.RevCommit; |
| import org.eclipse.jgit.revwalk.RevObject; |
| import org.eclipse.jgit.revwalk.RevWalk; |
| import org.eclipse.jgit.transport.resolver.RepositoryResolver; |
| import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException; |
| import org.eclipse.jgit.transport.resolver.ServiceNotEnabledException; |
| |
| /** Serves Markdown at the root of a host. */ |
| public class RootedDocServlet extends HttpServlet { |
| private static final long serialVersionUID = 1L; |
| public static final String BRANCH = "refs/heads/md-pages"; |
| |
| private final RepositoryResolver<HttpServletRequest> resolver; |
| private final DocServlet docServlet; |
| |
| public RootedDocServlet( |
| RepositoryResolver<HttpServletRequest> resolver, |
| GitilesAccess.Factory accessFactory, |
| Renderer renderer) { |
| this(resolver, accessFactory, renderer, HtmlSanitizer.DISABLED_FACTORY); |
| } |
| |
| public RootedDocServlet( |
| RepositoryResolver<HttpServletRequest> resolver, |
| GitilesAccess.Factory accessFactory, |
| Renderer renderer, |
| HtmlSanitizer.Factory htmlSanitizer) { |
| this.resolver = resolver; |
| docServlet = new DocServlet(accessFactory, renderer, htmlSanitizer); |
| } |
| |
| @Override |
| public void init(ServletConfig config) throws ServletException { |
| super.init(config); |
| docServlet.init(config); |
| } |
| |
| @Override |
| public void service(HttpServletRequest req, HttpServletResponse res) |
| throws IOException, ServletException { |
| try (Repository repo = resolver.open(req, null); |
| RevWalk rw = new RevWalk(repo)) { |
| ObjectId id = repo.resolve(BRANCH); |
| if (id == null) { |
| throw new GitilesRequestFailureException(FailureReason.OBJECT_NOT_FOUND); |
| } |
| |
| RevObject obj = rw.peel(rw.parseAny(id)); |
| if (!(obj instanceof RevCommit)) { |
| throw new GitilesRequestFailureException(FailureReason.INCORRECT_OBJECT_TYPE); |
| } |
| |
| req.setAttribute(ATTRIBUTE_REPOSITORY, repo); |
| ViewFilter.setView( |
| req, |
| GitilesView.rootedDoc() |
| .setHostName(req.getServerName()) |
| .setServletPath(req.getContextPath() + req.getServletPath()) |
| .setRevision(BRANCH, obj) |
| .setPathPart(req.getPathInfo()) |
| .build()); |
| |
| docServlet.service(req, res); |
| } catch (ServiceNotAuthorizedException e) { |
| throw new GitilesRequestFailureException(FailureReason.NOT_AUTHORIZED, e); |
| } catch (ServiceNotEnabledException e) { |
| throw new GitilesRequestFailureException(FailureReason.SERVICE_NOT_ENABLED, e); |
| } finally { |
| ViewFilter.removeView(req); |
| req.removeAttribute(ATTRIBUTE_REPOSITORY); |
| } |
| } |
| } |