PrototypeFile.java

package org.codehaus.mojo.unix.sysvpkg.prototype;

/*
 * The MIT License
 *
 * Copyright 2009 The Codehaus.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of
 * this software and associated documentation files (the "Software"), to deal in
 * the Software without restriction, including without limitation the rights to
 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
 * of the Software, and to permit persons to whom the Software is furnished to do
 * so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

import fj.*;
import fj.data.*;
import static fj.data.List.*;
import static fj.data.Option.*;
import org.apache.commons.vfs.*;
import org.codehaus.mojo.unix.*;
import static org.codehaus.mojo.unix.FileAttributes.*;
import static org.codehaus.mojo.unix.PackageFileSystem.*;
import org.codehaus.mojo.unix.util.*;
import static org.codehaus.mojo.unix.util.RelativePath.*;
import org.codehaus.mojo.unix.util.line.*;
import static org.codehaus.mojo.unix.util.vfs.VfsUtil.*;
import org.joda.time.*;

import java.io.*;

/**
 * @author <a href="mailto:trygvis@codehaus.org">Trygve Laugstøl</a>
 * @version $Id$
 */
public class PrototypeFile
    implements LineProducer
{
    public static final LocalDateTime START_OF_TIME = new LocalDateTime( 0, DateTimeZone.UTC );

    private List<String> iFiles = nil();

    private PackageFileSystem<PrototypeEntry> fileSystem;

    public PrototypeFile( DirectoryEntry defaultDirectory )
    {
        Validate.validateNotNull( defaultDirectory );

        UnixFsObject.Directory root = UnixFsObject.directory( BASE, START_OF_TIME, EMPTY );
        DirectoryEntry rootEntry = new DirectoryEntry( Option.<String>none(), root );

        fileSystem = create( rootEntry, defaultDirectory );
    }

    public void addIFileIf( File file, String name )
    {
        if ( file == null || !file.canRead() )
        {
            return;
        }

        iFiles = iFiles.cons( "i " + name + "=" + file.getAbsolutePath() );
    }

    public void addIFileIf( Option<File> file, String name )
    {
        if ( file.isNone() )
        {
            return;
        }

        iFiles = iFiles.cons( "i " + name + "=" + file.some().getAbsolutePath() );
    }

    public void addIFileIf( File file )
    {
        if ( file == null || !file.canRead() )
        {
            return;
        }

        addIFileIf( file, file.getName() );
    }

    public boolean hasPath( RelativePath path )
    {
        return fileSystem.hasPath( path );
    }

    public void addDirectory( UnixFsObject.Directory directory )
    {
        fileSystem = fileSystem.addDirectory( new DirectoryEntry( directory.attributes.bind( findClassTag ), directory ) );
    }

    public void addFile( FileObject fromFile, UnixFsObject.RegularFile file )
    {
        fileSystem = fileSystem.addFile( new FileEntry( file.attributes.bind( findClassTag ), some( false ), file, some( asFile( fromFile ) ) ) );
    }

    public void addSymlink( UnixFsObject.Symlink symlink )
    {
        fileSystem = fileSystem.addSymlink( new SymlinkEntry( symlink.attributes.bind( findClassTag ), symlink ) );
    }

    public void apply( F2<UnixFsObject, FileAttributes, FileAttributes> f )
    {
        fileSystem = fileSystem.apply( f );
    }

    public void streamTo( final LineStreamWriter stream )
    {
        System.out.println( PackageFileSystemFormatter.<PrototypeEntry>flatFormatter().print( fileSystem ) );

        stream.
            addAllLines( iFiles.reverse() );

        for ( PackageFileSystemObject<PrototypeEntry> object : fileSystem.prettify().toList().filter( filterRoot ) )
        {
            object.getExtension().streamTo( stream );
        }
    }

    public static Option<String> findClassTag( FileAttributes fileAttributes )
    {
        String prefix = "class:";

        for ( String tag : fileAttributes.tags )
        {
            if ( tag.startsWith( prefix ) )
            {
                return some( tag.substring( prefix.length() ) );
            }
        }

        return none();
    }

    public static F<FileAttributes, Option<String>> findClassTag = new F<FileAttributes, Option<String>>()
    {
        public Option<String> f( FileAttributes fileAttributes )
        {
            return findClassTag( fileAttributes );
        }
    };

    private F<PackageFileSystemObject<PrototypeEntry>, Boolean> filterRoot = new F<PackageFileSystemObject<PrototypeEntry>, Boolean>()
    {
        public Boolean f( PackageFileSystemObject<PrototypeEntry> fileSystemObject )
        {
            return !fileSystemObject.getUnixFsObject().path.isBase();
        }
    };
}