source: other-projects/GlamED/trunk/src/org/honours/agents/CollectionToGreenstone.java@ 26588

package org.honours.agents;

import java.awt.Color;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.util.ArrayList;
import java.util.List;

import org.expeditee.agents.DefaultAgent;
import org.expeditee.gui.Frame;
import org.expeditee.gui.MessageBay;

import org.honours.Main;
import org.honours.collection.*;
import org.honours.greenstone.CollectionContentEditor;
import org.honours.greenstone.CollectionCustomizer;

/**
 * This agent sends a collection (represented by an 
 * Expeditee frameset) back to Greenstone.  
 * @author Korii Scrivener
 */
public class CollectionToGreenstone extends DefaultAgent {

    private static String COLLECTIONS_HOME;
    private static String _collection;
    private static String _collectionPath;
    private static List<String[]> _filesToMove = new ArrayList<String[]>();
    
    
    public static void addFileToMove(String filename, String assocfilepath){
        _filesToMove.add(new String[]{filename,assocfilepath});
    }
    
    public Frame process(Frame frame){
        
        //Check if frame belongs to a collection and then obtain a 
        //collection object.
        Collection collection = Collection.findCollection(frame);

        if(collection == null){
            MessageBay.errorMessage("Error: No collection could be obtained for" + frame.getFramesetName() +
                    ". GSDL exporter is now stopping.");
            return null;
        }
        
        _collection = collection.getName();
        
        
        //Check if GSDL is set.
        if(Main.GSDL_HOME != null){
            COLLECTIONS_HOME = Main.GSDL_HOME + File.separator + "sites" + File.separator + "localsite" + File.separator + "collect";
            _collectionPath = COLLECTIONS_HOME + File.separator + _collection;
        }else{
            MessageBay.errorMessage("ERROR: No Greenstone environment variable set. Now stopping");
            return null;
        }
        
        File collectionFolder = new File(COLLECTIONS_HOME + File.separator + _collection);
        
        if(collectionFolder.isDirectory())
            rebuildCollection(collection);
    
        MessageBay.displayMessage("Success: Collection has been successfully edited & rebuilt", Color.green);
        
        return null;
    }
    
    /**
     * Rebuild the Greenstone collection.
     * @param collection
     */
    private void rebuildCollection(Collection collection){
        try{
            CollectionCustomizer customizer = CollectionCustomizer.getInstance(_collectionPath);

            CollectionContentEditor contentEditor = null;

            //loop through all the items in the collection and read their matching frames.
            for(int i = 0; i < collection.getItems().size(); i++){

                CollectionItem currCI = collection.getItems().get(i);
                Frame currCiFrame = currCI.getFrame();

                if(currCiFrame == null)
                    continue;

                contentEditor = new CollectionContentEditor(currCI,_collectionPath);

                if(i == 0)
                    customizer.customize(currCI);

                contentEditor.editCollectionItems();
            }

            build();
            CollectionCustomizer.reset();

            /*sleep to give time for collection to be rebuilt before 
            *proceeding to move files to index/assoc/assocfilepath directory. Otherwise
            *if you try to move files while rebuilding is still going, they will just be removed.
            *This is done for moving/adding extra files. Maybe it would be better to add them elsewhere
            *i.e. an "images folder...*/
            Thread.sleep(45000);
            
            //move any images to index/assoc folder in GSDL once building is completed.
            moveFiles();

            _filesToMove = new ArrayList<String[]>();
            
        }catch(Exception e){
            e.printStackTrace();
            return;
        }
    }
    
    /**
     * Method to move files currently not part of collection (e.g. images) 
     * to the collection's index/assoc/assocfilepath directory.
     */
    private static void moveFiles()
    {
        try{

            String newPath = _collectionPath + File.separator + "index" + File.separator + "assoc" + File.separator;
            
            for(int i =0; i < _filesToMove.size(); i++){
                String f = _filesToMove.get(i)[0];
                String a = _filesToMove.get(i)[1];
                
                newPath += a + File.separator;
                
                File origFile = new File(f);
                if(origFile.exists()){
                    FileInputStream fis = new FileInputStream(origFile);
                    newPath += origFile.getName();
                    
                    FileOutputStream fos = new FileOutputStream(newPath);
                    
                    byte[] buf = new byte[fis.available()];
                    fis.read(buf);
                    fos.write(buf);
                }
                
                newPath = _collectionPath + File.separator + "index" + File.separator + "assoc" + File.separator;
            }
        }catch(Exception e){
            e.printStackTrace();
        }
    }
    
    
    /**
     * Run batch script to rebuild collection.
     */
    private void build(){
        
        try{
            //"cmd /c start .../build.bat"
            String command = "cmd /c start collectionBuilder.bat";
            createBatchFile();
            Runtime.getRuntime().exec(command);
        }
        catch(Exception e){
            MessageBay.errorMessage("ERROR: Failed to build collection.");
            e.printStackTrace();
            return;
        }
    }
    
    /**
     * Method to create batch file which runs when
     * we want to export a frameset to Greenstone and 
     * rebuild the matching collection.
     */
    private void createBatchFile(){
        
        StringBuffer cmds = new StringBuffer("");
        cmds.append("@echo off\n");
        
        //TODO: Change this so it uses a relative path to GSDL.
        String chdir = "C:\\Research\\Pei-Jones-Taonga\\expeditee-svn\\images\\greenstone3-svn4";
        cmds.append("chdir \"" + ((chdir != null) ? chdir : "") + "\"\n");
        cmds.append("call \"./gs3-setup.bat\"\n");
        
        cmds.append("perl -S buildcol.pl -verbosity 5 -site localsite " + _collection + "\n");
        //cmds.append("perl -S incremental-rebuild.pl -verbosity 5 -site localsite " + _collection + "\n");
        cmds.append("perl -S activate.pl -site localsite " + _collection + "\n");
        cmds.append("pause\n");
        cmds.append("exit\n");
        
        try{
            FileWriter fwriter = new FileWriter("collectionBuilder.bat");
            BufferedWriter out = new BufferedWriter(fwriter);
            out.write(cmds.toString());
            out.close();
        }catch(Exception e){
            e.printStackTrace();
            return;
        }
    }
    
}