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

   1 package org.apache.maven.archiva.webdav.util;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import org.apache.jackrabbit.webdav.DavResource;
  23 import org.apache.jackrabbit.webdav.io.OutputContext;
  24
  25 import java.util.ArrayList;
  26 import java.util.Arrays;
  27 import java.util.Collections;
  28 import java.util.Date;
  29 import java.util.HashMap;
  30 import java.util.List;
  31 import java.util.Map;
  32 import java.io.PrintWriter;
  33 import java.io.File;
  34
  35 /**
  36  * @author <a href="mailto:james@atlassian.com">James William Dumay</a>
  37  */
  38 public class IndexWriter
  39 {
  40     private final DavResource resource;
  41
  42     private final String logicalResource;
  43
  44     private final List<File> localResources;
  45
  46     private final boolean isVirtual;
  47
  48     public IndexWriter(DavResource resource, File localResource, String logicalResource)
  49     {
  50         this.resource = resource;
  51         this.localResources = new ArrayList<File>();
  52         this.localResources.add( localResource );
  53         this.logicalResource = logicalResource;
  54         this.isVirtual = false;
  55     }
  56
  57     public IndexWriter( DavResource resource, List<File> localResources, String logicalResource )
  58     {
  59         this.resource = resource;
  60         this.logicalResource = logicalResource;
  61         this.localResources = localResources;
  62         this.isVirtual = true;
  63     }
  64
  65     public void write(OutputContext outputContext)
  66     {
  67         outputContext.setModificationTime(new Date().getTime());
  68         outputContext.setContentType("text/html");
  69         outputContext.setETag("");
  70         if (outputContext.hasStream())
  71         {
  72             PrintWriter writer = new PrintWriter(outputContext.getOutputStream());
  73             writeDocumentStart(writer);
  74             writeHyperlinks(writer);
  75             writeDocumentEnd(writer);
  76             writer.flush();
  77             writer.close();
  78         }
  79     }
  80
  81     private void writeDocumentStart(PrintWriter writer)
  82     {
  83         writer.println("<html>");
  84         writer.println("<head>");
  85         writer.println("<title>Collection: " + logicalResource + "<title>");
  86         writer.println("</head>");
  87         writer.println("<h3>Collection: " + logicalResource + "</h3>");
  88
  89         //Check if not root
  90         if (!"/".equals(logicalResource))
  91         {
  92             File file = new File(logicalResource);
  93             String parentName = file.getParent().equals("") ? "/" : file.getParent();
  94
  95             writer.println("<ul>");
  96             writer.println("<li><a href=\"../\">" + parentName + "</a> <i><small>(Parent)</small></i></li>");
  97             writer.println("</ul>");
  98         }
  99
 100         writer.println("<ul>");
 101     }
 102
 103     private void writeDocumentEnd(PrintWriter writer)
 104     {
 105         writer.println("</ul>");
 106         writer.println("</body>");
 107         writer.println("</html>");
 108     }
 109
 110     private void writeHyperlinks(PrintWriter writer)
 111     {
 112         if( !isVirtual )
 113         {
 114             for( File localResource : localResources )
 115             {
 116                 List<File> files = new ArrayList<File>( Arrays.asList( localResource.listFiles() ) );
 117                 Collections.sort( files );
 118
 119                 for ( File file : files )
 120                 {
 121                     writeHyperlink( writer, file.getName(), file.isDirectory(), false );
 122                 }
 123             }
 124         }
 125         else
 126         {
 127             // virtual repository - filter unique directories
 128             Map<String, File> uniqueChildFiles = new HashMap<String, File>();
 129             for( File resource : localResources )
 130             {
 131                 List<File> files = new ArrayList<File>( Arrays.asList( resource.listFiles() ) );
 132
 133                 for ( File file : files )
 134                 {
 135                     if( uniqueChildFiles.get( file.getName() ) == null )
 136                     {
 137                         uniqueChildFiles.put( file.getName(), file );
 138                     }
 139                 }
 140             }
 141
 142             List<File> uniqueChildFilesInList = new ArrayList<File>();
 143             uniqueChildFilesInList.addAll( uniqueChildFiles.values() );
 144             Collections.sort( uniqueChildFilesInList );
 145
 146             for ( File file : uniqueChildFilesInList )
 147             {
 148                 writeHyperlink( writer, file.getName(), file.isDirectory(), true );
 149             }
 150         }
 151     }
 152
 153     private void writeHyperlink(PrintWriter writer, String resourceName, boolean directory, boolean isBrowse )
 154     {
 155         if (directory)
 156         {
 157             writer.println("<li><a href=\"./" + resourceName + "/\">" + resourceName + "</a></li>");
 158         }
 159         else
 160         {
 161             writer.println("<li><a href=\"./" + resourceName + "\">" + resourceName + "</a></li>");
 162         }
 163     }
 164 }