org.apache.archiva.webdav.util.IndexWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of archiva-webdav Show documentation
There is a newer version: 2.2.10
package org.apache.archiva.webdav.util;

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

import org.apache.commons.lang.StringUtils;
import org.apache.jackrabbit.webdav.DavResource;
import org.apache.jackrabbit.webdav.io.OutputContext;

import java.io.File;
import java.io.PrintWriter;
import java.text.DateFormat;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;

/**
 */
public class IndexWriter
{
    private final String logicalResource;

    private final List localResources;

    private final boolean isVirtual;

    public IndexWriter( DavResource resource, File localResource, String logicalResource )
    {
        this.localResources = new ArrayList<>();
        this.localResources.add( localResource );
        this.logicalResource = logicalResource;
        this.isVirtual = false;
    }

    public IndexWriter( DavResource resource, List localResources, String logicalResource )
    {
        this.logicalResource = logicalResource;
        this.localResources = localResources;
        this.isVirtual = true;
    }

    public void write( OutputContext outputContext )
    {
        outputContext.setModificationTime( new Date().getTime() );
        outputContext.setContentType( "text/html" );
        outputContext.setETag( "" ); // skygo ETag MRM-1127 seems to be fixed
        if ( outputContext.hasStream() )
        {
            PrintWriter writer = new PrintWriter( outputContext.getOutputStream() );
            writeDocumentStart( writer );
            writeHyperlinks( writer );
            writeDocumentEnd( writer );
            writer.flush();
            writer.close();
        }
    }

    private void writeDocumentStart( PrintWriter writer )
    {
        writer.println("");
        writer.println( "" );
        writer.println( "" );
        writer.println( "Collection: /" + logicalResource + "" );
        writer.println( "" );
        writer.println( "" );
        writer.println( "" );
        writer.println( "" );
        writer.println( "Collection: /" + logicalResource + "" );

        //Check if not root
        if ( logicalResource != null && logicalResource.length() > 0 )
        {
            File file = new File( logicalResource );
            String parentName = file.getParent() == null ? "/" : file.getParent();

            //convert to unix path in case archiva is hosted on windows
            parentName = StringUtils.replace( parentName, "\\", "/" );

            writer.println( "" );
            writer.println( "" + parentName + " (Parent)" );
            writer.println( "" );
        }

        writer.println( "" );
        writer.println( "" );
    }

    private void writeDocumentEnd( PrintWriter writer )
    {
        writer.println( "Name Size (Bytes) Last Modified" );
        writer.println( "" );
        writer.println( "" );
    }

    private void writeHyperlinks( PrintWriter writer )
    {
        if ( !isVirtual )
        {
            for ( File localResource : localResources )
            {
                List files = new ArrayList<>( Arrays.asList( localResource.listFiles() ) );
                Collections.sort( files );

                for ( File file : files )
                {
                    writeHyperlink( writer, file.getName(), file.lastModified(), file.length(), file.isDirectory() );
                }
            }
        }
        else
        {
            // virtual repository - filter unique directories
            Map> uniqueChildFiles = new HashMap<>();
            List sortedList = new ArrayList<>();
            for ( File resource : localResources )
            {
                List files = new ArrayList<>( Arrays.asList( resource.listFiles() ) );
                for ( File file : files )
                {
                    List mergedChildFiles = new ArrayList<>();
                    if ( uniqueChildFiles.get( file.getName() ) == null )
                    {
                        mergedChildFiles.add( file.getAbsolutePath() );
                    }
                    else
                    {
                        mergedChildFiles = uniqueChildFiles.get( file.getName() );
                        if ( !mergedChildFiles.contains( file.getAbsolutePath() ) )
                        {
                            mergedChildFiles.add( file.getAbsolutePath() );
                        }
                    }
                    uniqueChildFiles.put( file.getName(), mergedChildFiles );
                    sortedList.add( file.getName() );
                }
            }

            Collections.sort( sortedList );
            List written = new ArrayList<>();
            for ( String fileName : sortedList )
            {
                List childFilesFromMap = uniqueChildFiles.get( fileName );
                for ( String childFilePath : childFilesFromMap )
                {
                    File childFile = new File( childFilePath );
                    if ( !written.contains( childFile.getName() ) )
                    {
                        written.add( childFile.getName() );
                        writeHyperlink( writer, fileName, childFile.lastModified(), childFile.length(), childFile.isDirectory() );
                    }
                }
            }
        }
    }

    private static String fileDateFormat( long date ) 
    {
        DateFormat dateFormatter = DateFormat.getDateTimeInstance( DateFormat.SHORT, DateFormat.SHORT, Locale.getDefault() );
        Date aDate = new Date( date );
        return dateFormatter.format( aDate );
    }
    
    private void writeHyperlink( PrintWriter writer, String resourceName, long lastModified, long fileSize, boolean directory )
    {
        if ( directory )
        {
            writer.println( "" + resourceName + "  " );
        }
        else
        {
            writer.println( "" + resourceName + "" + fileSize + "  " + fileDateFormat( lastModified ) + "" );
        }
    }
}