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.commons.lang.StringUtils;
  23 import org.apache.jackrabbit.webdav.DavResource;
  24 import org.apache.jackrabbit.webdav.io.OutputContext;
  25
  26 import java.util.ArrayList;
  27 import java.util.Arrays;
  28 import java.util.Collections;
  29 import java.util.Date;
  30 import java.util.HashMap;
  31 import java.util.List;
  32 import java.util.Map;
  33 import java.io.PrintWriter;
  34 import java.io.File;
  35
  36 /**
  37  */
  38 public class IndexWriter
  39 {
  40     private final String logicalResource;
  41
  42     private final List<File> localResources;
  43
  44     private final boolean isVirtual;
  45
  46     public IndexWriter(DavResource resource, File localResource, String logicalResource)
  47     {
  48         this.localResources = new ArrayList<File>();
  49         this.localResources.add( localResource );
  50         this.logicalResource = logicalResource;
  51         this.isVirtual = false;
  52     }
  53
  54     public IndexWriter( DavResource resource, List<File> localResources, String logicalResource )
  55     {
  56         this.logicalResource = logicalResource;
  57         this.localResources = localResources;
  58         this.isVirtual = true;
  59     }
  60
  61     public void write(OutputContext outputContext)
  62     {
  63         outputContext.setModificationTime(new Date().getTime());
  64         outputContext.setContentType("text/html");
  65         outputContext.setETag("");
  66         if (outputContext.hasStream())
  67         {
  68             PrintWriter writer = new PrintWriter(outputContext.getOutputStream());
  69             writeDocumentStart(writer);
  70             writeHyperlinks(writer);
  71             writeDocumentEnd(writer);
  72             writer.flush();
  73             writer.close();
  74         }
  75     }
  76
  77     private void writeDocumentStart(PrintWriter writer)
  78     {
  79         writer.println("<html>");
  80         writer.println("<head>");
  81         writer.println("<title>Collection: " + logicalResource + "</title>");
  82         writer.println("</head>");
  83         writer.println("<body>");
  84         writer.println("<h3>Collection: " + logicalResource + "</h3>");
  85
  86         //Check if not root
  87         if (!"/".equals(logicalResource))
  88         {
  89             File file = new File(logicalResource);
  90             String parentName = file.getParent().equals("") ? "/" : file.getParent();
  91
  92             //convert to unix path in case archiva is hosted on windows
  93             parentName = StringUtils.replace(parentName, "\\", "/" );
  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() );
 122                 }
 123             }
 124         }
 125         else
 126         {
 127             // virtual repository - filter unique directories
 128             Map<String, List<String>> uniqueChildFiles = new HashMap<String, List<String>>();
 129             List<String> sortedList = new ArrayList<String>();
 130             for( File resource : localResources )
 131             {
 132                 List<File> files = new ArrayList<File>( Arrays.asList( resource.listFiles() ) );
 133                 for ( File file : files )
 134                 {
 135                     List<String> mergedChildFiles = new ArrayList<String>();
 136                     if( uniqueChildFiles.get( file.getName() ) == null )
 137                     {
 138                         mergedChildFiles.add( file.getAbsolutePath() );
 139                     }
 140                     else
 141                     {
 142                         mergedChildFiles = uniqueChildFiles.get( file.getName() );
 143                         if( !mergedChildFiles.contains( file.getAbsolutePath() ) )
 144                         {
 145                             mergedChildFiles.add( file.getAbsolutePath() );
 146                         }
 147                     }
 148                     uniqueChildFiles.put( file.getName(), mergedChildFiles );
 149                     sortedList.add( file.getName() );
 150                 }
 151             }
 152
 153             Collections.sort( sortedList );
 154             List<String> written = new ArrayList<String>();
 155             for ( String fileName : sortedList )
 156             {
 157                 List<String> childFilesFromMap = uniqueChildFiles.get( fileName );
 158                 for( String childFilePath : childFilesFromMap )
 159                 {
 160                     File childFile = new File( childFilePath );
 161                     if( !written.contains( childFile.getName() ) )
 162                     {
 163                         written.add( childFile.getName() );
 164                         writeHyperlink( writer, fileName, childFile.isDirectory() );
 165                     }
 166                 }
 167             }
 168         }
 169     }
 170
 171     private void writeHyperlink(PrintWriter writer, String resourceName, boolean directory )
 172     {
 173         if (directory)
 174         {
 175             writer.println("<li><a href=\"" + resourceName + "/\">" + resourceName + "</a></li>");
 176         }
 177         else
 178         {
 179             writer.println("<li><a href=\"" + resourceName + "\">" + resourceName + "</a></li>");
 180         }
 181     }
 182 }