source: trunk/gli/src/org/greenstone/gatherer/file/FileManager.java@ 10248

/**
 *#########################################################################
 *
 * A component of the Gatherer application, part of the Greenstone digital
 * library suite from the New Zealand Digital Library Project at the
 * University of Waikato, New Zealand.
 *
 * <BR><BR>
 *
 * Author: John Thompson, NZDL Project, University of Waikato
 *
 * <BR><BR>
 *
 * Copyright (C) 2005 New Zealand Digital Library Project
 *
 * <BR><BR>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * <BR><BR>
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * <BR><BR>
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *########################################################################
 */
package org.greenstone.gatherer.file;

import java.io.File;
import javax.swing.*;
import org.greenstone.gatherer.Dictionary;
import org.greenstone.gatherer.Gatherer;
import org.greenstone.gatherer.collection.CollectionTreeNode;
import org.greenstone.gatherer.gui.LongProgressBar;
import org.greenstone.gatherer.gui.NewFolderOrFilePrompt;
import org.greenstone.gatherer.gui.tree.DragTree;
import org.greenstone.gatherer.util.DragComponent;
import org.greenstone.gatherer.util.Utility;


/** Manages the moving of files within a separate thread.
 * @author John Thompson, NZDL Project, University of Waikato
 */
public class FileManager
{
    /** Not only the queue of files to be moved, but also the object that moves them. */
    static private FileQueue file_queue = null;
    static private RemoteFileQueue remote_file_queue = null;

    public static int FILE_TYPE = 0;
    public static int FOLDER_TYPE = 1;


    /** Constructor.
     * @see org.greenstone.gatherer.file.FileQueue
     */
    public FileManager()
    {
    file_queue = new FileQueue();
    file_queue.start();

    if (Gatherer.isGsdlRemote) {
        remote_file_queue = new RemoteFileQueue();
        remote_file_queue.start();
    }
    }


    /** Determine what action should be carried out by the file queue, and add all of the necessary file jobs. */
    public void action(DragComponent source, FileNode[] source_nodes, DragComponent target, FileNode target_node)
    {
    // Check there is something to do
    if (source_nodes == null) {
        return;
    }

    // We need a unique ID for each file task
    long id = System.currentTimeMillis();

    // If source and target are the same we're moving
    if (source == target) {
        // Start a new move FileTask and we're done
        (new FileTask(id, source, source_nodes, target, target_node, FileJob.MOVE)).start();
        return;
    }

    // If target isn't the RecycleBin, we're copying
    if (!(target instanceof RecycleBin)) {
        // Start a new copy FileTask and we're done
        (new FileTask(id, source, source_nodes, target, target_node, FileJob.COPY)).start();
        return;
    }

    // We're deleting... but first make sure source isn't read-only
    boolean read_only_source = false;

    // The workspace tree is read-only...
    if (source.toString().equals("Workspace")) {
        read_only_source = true;

        // ...except for files from the "Downloaded Files" folder
        String downloaded_files_folder_path = Utility.getCacheDir().getAbsolutePath();
        for (int i = 0; i < source_nodes.length; i++) {
        // Is this the "Downloaded Files" folder?
        if (source_nodes[i].getFile().getAbsolutePath().startsWith(downloaded_files_folder_path)) {
            // Can only delete from the "Downloaded Files" folder if a collection is open
            if (Gatherer.c_man.ready() == false) {
            return;
            }
            read_only_source = false;
        }
        }
    }

    // The source is read-only, so tell the user and abort
    if (read_only_source) {
        JOptionPane.showMessageDialog(Gatherer.g_man, Dictionary.get("FileActions.Read_Only"), Dictionary.get("General.Error"), JOptionPane.ERROR_MESSAGE);
        return;
    }

    // Start a new delete FileTask and we're done
    (new FileTask(id, source, source_nodes, target, target_node, FileJob.DELETE)).start();
    }


    /** Retrieves the file queue object. */
    public FileQueue getQueue()
    {
    return file_queue;
    }


    private class FileTask
    extends Thread
    {
    private long id;
    private DragComponent source;
    private FileNode[] source_nodes;
    private DragComponent target;
    private FileNode target_node;
    private byte type;


    public FileTask(long id, DragComponent source, FileNode[] source_nodes, DragComponent target, FileNode target_node, byte type)
    {
        this.id = id;
        this.source = source;
        this.source_nodes = source_nodes;
        this.target = target;
        this.target_node = target_node;
        this.type = type;
    }


    public void run()
    {
        // Reset the progress bar and set it to indeterminate while calculating its size
        LongProgressBar progress_bar = file_queue.getProgressBar();
        progress_bar.reset();
        progress_bar.setIndeterminate(true);

        // Calculate the progress bar size
        boolean cancelled = file_queue.calculateSize(source_nodes);
        if (!cancelled) {
        file_queue.addJob(id, source, source_nodes, target, target_node, type, true);
        if (Gatherer.isGsdlRemote) {
            remote_file_queue.addJob(id, source, source_nodes, target, target_node, type);
        }
        }

        progress_bar.setIndeterminate(false);
        progress_bar.clear();
    }
    }


    public void newDummyDoc(DragTree tree, CollectionTreeNode parent_node){
    newFolderOrDummyDoc(tree, parent_node, FILE_TYPE);
    }


    public void newFolder(DragTree tree, CollectionTreeNode parent_node) {
    newFolderOrDummyDoc(tree, parent_node, FOLDER_TYPE);
    }


    protected void newFolderOrDummyDoc(DragTree tree, CollectionTreeNode parent_node, int type) {
    // Ask the user for the directories name.
    String extension = "";
    if (type == FILE_TYPE) {
        extension = ".nul";
    }
    NewFolderOrFilePrompt new_folder_prompt = new NewFolderOrFilePrompt(parent_node, type, extension);
    String name = new_folder_prompt.display();
    new_folder_prompt.dispose();
    new_folder_prompt = null;
    // And if the name is non-null...
    if(name != null) {
        FileSystemModel model = (FileSystemModel) tree.getModel();
        File folder_file = new File(parent_node.getFile(), name);
        //... check if it already exists.
        if(folder_file.exists()) { 
        if (type == FILE_TYPE) {
            JOptionPane.showMessageDialog(Gatherer.g_man, Dictionary.get("FileActions.File_Already_Exists_No_Create", name), Dictionary.get("General.Error"), JOptionPane.ERROR_MESSAGE);
        } else {
            JOptionPane.showMessageDialog(Gatherer.g_man, Dictionary.get("FileActions.Folder_Already_Exists", name), Dictionary.get("General.Error"), JOptionPane.ERROR_MESSAGE);
        }
        }
        // Otherwise create it.
        else {
        try {
            if (type == FILE_TYPE) {
            folder_file.createNewFile();
            } else {
            folder_file.mkdirs();
            }
            // Update the parent node to show the new folder
            parent_node.refresh();
            
            // Refresh workspace tree (collection tree is done automatically)
            Gatherer.g_man.refreshWorkspaceTree(DragTree.COLLECTION_CONTENTS_CHANGED);
        } catch (Exception e) {
            if (type == FILE_TYPE) {
            JOptionPane.showMessageDialog(Gatherer.g_man, Dictionary.get("FileActions.File_Create_Error", name), Dictionary.get("General.Error"), JOptionPane.ERROR_MESSAGE);
            } else {
            JOptionPane.showMessageDialog(Gatherer.g_man, Dictionary.get("FileActions.Folder_Create_Error", name), Dictionary.get("General.Error"), JOptionPane.ERROR_MESSAGE);
            }
        }
        }
        
        folder_file = null;
        model = null;
    }
    name = null;
    }
}