source.dussan.org Git - archiva.git/blob

   1 package org.apache.archiva.web.docs;
   2 /*
   3  * Licensed to the Apache Software Foundation (ASF) under one
   4  * or more contributor license agreements.  See the NOTICE file
   5  * distributed with this work for additional information
   6  * regarding copyright ownership.  The ASF licenses this file
   7  * to you under the Apache License, Version 2.0 (the
   8  * "License"); you may not use this file except in compliance
   9  * with the License.  You may obtain a copy of the License at
  10  *
  11  *   http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing,
  14  * software distributed under the License is distributed on an
  15  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  16  * KIND, either express or implied.  See the License for the
  17  * specific language governing permissions and limitations
  18  * under the License.
  19  */
  20
  21 import org.apache.commons.lang.StringUtils;
  22 import org.jsoup.Jsoup;
  23 import org.jsoup.nodes.Document;
  24 import org.jsoup.nodes.Element;
  25 import org.jsoup.select.Elements;
  26 import org.slf4j.Logger;
  27 import org.slf4j.LoggerFactory;
  28
  29 import javax.servlet.ServletException;
  30 import javax.servlet.http.HttpServlet;
  31 import javax.servlet.http.HttpServletRequest;
  32 import javax.servlet.http.HttpServletResponse;
  33 import java.io.IOException;
  34 import java.io.InputStream;
  35 import java.util.Iterator;
  36
  37 /**
  38  * @author Olivier Lamy
  39  * @since 1.4-M4
  40  */
  41 public class RestDocsServlet
  42     extends HttpServlet
  43 {
  44     private Logger logger = LoggerFactory.getLogger( getClass() );
  45
  46     @Override
  47     protected void doGet( HttpServletRequest req, HttpServletResponse resp )
  48         throws ServletException, IOException
  49     {
  50
  51         logger.debug( "docs request to path: {}", req.getPathInfo() );
  52
  53         String path = StringUtils.removeStart( req.getPathInfo(), "/" );
  54         InputStream is = Thread.currentThread().getContextClassLoader().getResourceAsStream( path );
  55
  56         String startPath = StringUtils.substringBefore( path, "/" );
  57
  58         // replace all links !!
  59         Document document = Jsoup.parse( is, "UTF-8", "" );
  60
  61         Element body = document.body().child( 0 );
  62
  63         Elements links = body.select( "a[href]" );
  64
  65         for ( Iterator<Element> elementIterator = links.iterator(); elementIterator.hasNext(); )
  66         {
  67             Element link = elementIterator.next();
  68             link.attr( "href", "#" + startPath + "/" + link.attr( "href" ) );
  69         }
  70
  71         Elements codes = body.select( "code" );
  72
  73         for ( Iterator<Element> elementIterator = codes.iterator(); elementIterator.hasNext(); )
  74         {
  75             Element code = elementIterator.next();
  76             code.attr( "class", code.attr( "class" ) + " nice-code" );
  77         }
  78
  79         //default generated enunciate use h1/h2/h3 which is quite big so transform to h3/h4/h5
  80
  81         Elements headers = body.select( "h1" );
  82
  83         for ( Iterator<Element> elementIterator = headers.iterator(); elementIterator.hasNext(); )
  84         {
  85             Element header = elementIterator.next();
  86             header.tagName( "h3" );
  87         }
  88
  89         headers = body.select( "h2" );
  90
  91         for ( Iterator<Element> elementIterator = headers.iterator(); elementIterator.hasNext(); )
  92         {
  93             Element header = elementIterator.next();
  94             header.tagName( "h4" );
  95         }
  96
  97         headers = body.select( "h3" );
  98
  99         for ( Iterator<Element> elementIterator = headers.iterator(); elementIterator.hasNext(); )
 100         {
 101             Element header = elementIterator.next();
 102             header.tagName( "h5" );
 103         }
 104
 105         Document res = new Document( "" );
 106         res.appendChild( body.select( "div[id=main]" ).first() );
 107
 108         resp.getOutputStream().write( res.outerHtml().getBytes() );
 109
 110     }
 111 }