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( "onclick", "loadRestDocs('" + startPath + "\',\'"+ "rest-docs/" + startPath + "/" + link.attr( "href" ) + "\');" );
  69             link.attr( "href", "#" + startPath + "/" + link.attr( "href" ) );
  70
  71         }
  72
  73         Elements codes = body.select( "code" );
  74
  75         for ( Iterator<Element> elementIterator = codes.iterator(); elementIterator.hasNext(); )
  76         {
  77             Element code = elementIterator.next();
  78             code.attr( "class", code.attr( "class" ) + " nice-code" );
  79         }
  80
  81         //res.appendChild( body.child( 1 ) );
  82
  83         Document res = new Document( "" );
  84         res.appendChild( body.select( "div[id=main]" ).first() );
  85
  86         resp.getOutputStream().write( res.outerHtml().getBytes() );
  87
  88         //IOUtils.copy( is, resp.getOutputStream() );
  89         //super.doGet( req, resp );
  90     }
  91 }