source: trunk/greenstone3-extensions/vishnu/src/vishnu/builder/CollectionBuilder.java@ 8189

package vishnu.builder;

import java.util.*;
import java.io.*;

public class CollectionBuilder
{
    //String site;
    String collID; // directory name
    String collHome; // full path to collections directory
    String collName;
    String collDesc;
    String indexer;
    long numDocs=0;
    
    private static void usage()
    {
    System.out.println("java vishnu.builder.CollectionBuilder\n -collsHome (directory where collections live) \n -collection (directory name of collection)\n -name (descriptive name for collection) \n -description (collection description)\n -index (\"mg\" or \"luc\")");
    }
    
    
    public static void main(String[] args)
    {
    if( args.length == 0 ){
        usage();
        System.exit(0);
    }
    
    String collID = null;
    String collsHome = null;
    String collName = "";
    String collDesc = "";
    String indexer = null;
    
    int a = 0;
    while( a < args.length ){
        //if( args[a].equals("-gHome") )
        //  gHome = args[++a];
        if (args[a].equals("-collsHome"))
        collsHome = args[++a];
        else if (args[a].equals("-collection"))
        collID = args[++a];
        else if( args[a].equals("-name") )
        collName = args[++a];
        else if( args[a].equals("-description") )
        collDesc = args[++a];
        else if( args[a].equals("-index") )
        indexer = args[++a];
        a++;
    }
    
    if (collsHome==null || collID==null || indexer==null) {
        usage();
        System.exit(0);
    }
    
    if (collName.equals("")) {
        collName = collID;
    }
    if( !indexer.equals("mg") && !indexer.equals("luc") )
        indexer = "luc";
    
    if (!collsHome.endsWith(File.separator)) {
        collsHome += File.separator;
    }
    String coll_home = collsHome + collID + File.separator;
    //CollectionBuilder cb = new CollectionBuilder(site,collName,gHome,indexer);
    CollectionBuilder cb = new CollectionBuilder(coll_home,collID, collName,collDesc, indexer);
    }
    
    
    
    public CollectionBuilder(String coll_home, String coll_id, String coll_name,String coll_desc,  String ind)
    {
    this.collHome = coll_home;
    this.collID = coll_id;
    this.collName = coll_name;
    this.collDesc = coll_desc;
    this.indexer = ind;

    if (!setupCollection()) {
        System.err.println("Couldn't setup the collection. Quitting build");
        return;
    }
    
    
    /**** get text annotation ****/

    FTIndexing();


    /**** extract keywords ****/

    CKIndexing();

    /**** write out the config file ****/
    printConfig();
    }



    private boolean setupCollection()
    {
    // create the index directory
    try {
        File index_dir = new File(collHome + "building");
        if (!index_dir.exists()) {
        index_dir.mkdir();
        } else if (!index_dir.isDirectory()) {
        // index is present but is not a directory
        System.err.println("Trying to create building dir but a file already exists at "+index_dir.toString());
        return false;
        }
    } catch (Exception e) {
        System.err.println("CollectionBuilder.setupCollection Error: "+e);
        return false;
    }

    return true;
    }

    private void printConfig() {

    String index_type="";
    if (indexer.equals("luc")) {
        index_type = "LUCENE";
    } else if (indexer.equals("mg")) {
        index_type = "MG";
    }
    
    try{
        File target = new File(collHome + "config.xml");
        PrintWriter out = new PrintWriter(new FileWriter(target),true);
    
        StringBuffer buffer = new StringBuffer();
        buffer.append("<collection name=\"");
        buffer.append(collName);
        buffer.append("\" description=\"");
        buffer.append(collDesc);
        buffer.append("\" engine=\"");
        buffer.append(index_type);
        buffer.append("\" size=\"");
        buffer.append(String.valueOf(numDocs));
        buffer.append("\" />");
        
        out.print(buffer.toString());
        out.close();

    } catch (IOException e){ System.out.println("Error writing config file: " + e.toString());
    }
    }
    
//     private void printConfig()
//     {

//  String index_type="";
//  if (indexer.equals("luc")) {
//      index_type = "LUCENE";
//  } else if (indexer.equals("mg")) {
//      index_type = "MG";
//  }
//  /**** print buildConfig.xml ****/
//           try{
//      File target = new File(collHome + "building" + File.separator + "buildConfig.xml");
//      FileWriter fw = new FileWriter(target);
//      PrintWriter out = new PrintWriter(fw,true);
        
//      String str = "<buildConfig>\n";
//      str += "<metadataList/>\n";
//      //str += "<metadata name=\"numDocs\">" + String.valueOf(numDocs) + "</metadata>\n";
//      //str += "<metadata name=\"buildType\">" + indexer + "</metadata>\n";
//      //str += "</metadataList>\n";
//      str += "<serviceRackList>\n";
//      str += "<serviceRack name=\"Visualizer\">\n";
//      str += "<engineType name=\""+index_type+"\"/>\n";
//      str += "</serviceRack>\n";
//      str += "</serviceRackList>\n";
//      str += "</buildConfig>\n";
//      out.print(str);
//      out.close();
//  } catch (IOException e){ System.out.println("Error " + e.toString());}
//     }
    

    private void CKIndexing()
    {
    CKWrapper ckw = new CKWrapper();
    
    String in = collHome + "import"+File.separator;
    String out = collHome + "building"+File.separator+"ck_index"+File.separator;
    
    ckw.setCollectionName(collID);
    ckw.setOutputDirectory(out);
    ckw.setInputDirectory(in);
    ckw.startIndexing();
    }
    


    private void FTIndexing()
    {
    String in = collHome + "import"+File.separator;
    String out = collHome + "building"+File.separator + indexer + "_index"+File.separator;
    
    if( indexer.equals("luc") ){
        LuceneWrapper luc = new LuceneWrapper();
        luc.setOutputDirectory(out);
        luc.setInputDirectory(in);
        luc.startIndexing();
    }
    else if( indexer.equals("mg") ){
        MGWrapper mg = new MGWrapper();
        mg.setCollectionName(collID);
        mg.setOutputDirectory(out);
        mg.setInputDirectory(in);
        mg.startIndexing();
    }
    }
}