tapestry/tapestry-examples/Workbench/src/java/org/apache/tapestry/workbench/tree/examples/fsmodel/FolderObject.java

// Copyright 2004, 2005 The Apache Software Foundation
//
// Licensed 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.

package org.apache.tapestry.workbench.tree.examples.fsmodel;

import java.io.File;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Vector;

import org.apache.tapestry.contrib.tree.model.ITreeNode;
import org.apache.tapestry.engine.IEngineService;

/**
 * 
 * @author unknown
 */
public class FolderObject extends SFObject
{

    /**
     * serialVersionUID.
     */
    private static final long serialVersionUID = 1889745019885512018L;

    private static final String openImage = "/org/apache/tapestry/workbench/tree/examples/fsmodel/TreeOpen.gif";

    private static final String closedImage = "/org/apache/tapestry/workbench/tree/examples/fsmodel/TreeClosed.gif";
    
    /**
     * @associates <{File}>
     * @supplierCardinality 0..*
     */
    private Vector m_vFiles = null;

    /**
     * @associates <{FolderObject}>
     * @supplierCardinality 0..*
     */
    private Vector m_vFolders = null;

    private boolean m_bShared;

    private final IEngineService _assetService;

    public FolderObject(ITreeNode objParent, File objFile, boolean bInvokeInit,
            IEngineService assetService)
    {
        super(objParent, objFile);

        _assetService = assetService;

        if (bInvokeInit)
            init();
    }

    public void reload()
    {
        m_vFolders = new Vector();
        m_vFiles = new Vector();

        File[] arrFiles = getFile().listFiles();

        if (arrFiles == null)
        {
            return;
        }

        for (int i = 0; i < arrFiles.length; i++)
        {
            if (arrFiles[i].isDirectory())
            {
                m_vFolders.addElement(new FolderObject(this, arrFiles[i], true, _assetService));
            }
            else
            {
                m_vFiles.addElement(new FileObject(this, arrFiles[i], _assetService));
            }
        }
    }

    public boolean isShared()
    {
        return m_bShared;
    }

    public Vector getFolders()
    {
        if (m_vFolders == null)
        {
            reload();
        }
        return m_vFolders;
    }

    public Vector getFiles()
    {
        if (m_vFiles == null)
        {
            reload();
        }
        return m_vFiles;
    }

    public int getChildNumber(Object objChild)
    {
        for (int i = 0; i < m_vFolders.size(); i++)
        {
            Object objChildFolder = m_vFolders.elementAt(i);
            if (objChildFolder.equals(objChild))
            {
                return i;
            }
        }
        return -1;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeNode#containsChild(ITreeNode)
     */
    public boolean containsChild(ITreeNode node)
    {
        return true;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeNode#getAllowsChildren()
     */
    public boolean getAllowsChildren()
    {
        return true;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeNode#getChildCount()
     */
    public int getChildCount()
    {
        return getFolders().size() + getFiles().size();
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeNode#getChildren()
     */
    public Collection getChildren()
    {
        ArrayList arrChildrens = new ArrayList();
        arrChildrens.addAll(getFolders());
        arrChildrens.addAll(getFiles());
        return arrChildrens;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeNode#isLeaf()
     */
    public boolean isLeaf()
    {
        return false;
    }

    public AssetsHolder getAssets()
    {
        if (m_objAssetsHolder == null)
        {
            m_objAssetsHolder = new AssetsHolder(_assetService, openImage, closedImage);
        }
        return m_objAssetsHolder;
    }
}