package iss;

import da.*;
import javax.naming.NamingException;
import java.util.Vector;

public class InfoSubSys {
  
  protected DirSearcher ds;
  protected final static String host = "ldap://jemez.extreme.indiana.edu:1849";

  public static void main (String [] args) throws Exception{
    InfoSubSys iss = new InfoSubSys();
    
    Object s = iss.getComponentPath ("ReadVector",
				     "rainier.extreme.indiana.edu");
    if (s == null )
      { 
	System.out.println ("Sorry, none.");
	System.exit(1);
      }
    else {
      //  for (int i=0;i<s.length;i++)
      //System.out.println (s[i]);
      System.out.println (s);
    }
  }

  public InfoSubSys () throws LdapConnectionException {
    ds = new DirSearcher (host);
  }

  public InfoSubSys (String dhost) throws LdapConnectionException {
    ds = new DirSearcher (dhost);
  }

  public String[] getCategories () throws LdapSearchException {
    Query q = new SimpleQuery ("objectclass","softwareType");
    
    try {
      ResultSet r = ds.search (q,DirSearcher.SOFTWARE);
      
      int l = r.getCount();
      if (l==0)
	return null;
      else {
	String [] res = new String [l];
	for (int i=0; i<l; i++) {
	  Entry e = r.getEntry (i);
	  res[i] = e.getString ("cn");
	}
	return res;
      }
    }
    catch (NamingException ne) {
      throw new LdapSearchException (ne.toString());
    }
  }

  public String[] getComponentsByCategory (String cat) 
       throws LdapSearchException {

    Query q = new SimpleQuery ("objectclass","module");
    
    try {
      ResultSet r = ds.search(q,"cn=" + cat + "," + DirSearcher.SOFTWARE);
      int l = r.getCount();
      if (l==0)
	return null;
      else {
	Vector v = new Vector();

	for (int i=0;i<l;i++) {
	  Entry e = r.getEntry(i);
	  String name  = e.getString ("cn");
	  DisName [] machines = e.getDisNameArray ("machine");
	  
	  for (int j=0;j<machines.length;j++)
	    v.addElement (name + " (" + machines[j].getValue ("hn") + ")");

	}

	String [] res = new String [v.size()];
	for (int k=0;k<res.length;k++) 
	  res[k] = (String) v.elementAt (k);
	return res;	
      }
    }
    catch (Exception ne) {
      throw new LdapSearchException (ne.toString());
    }
  }

  public String [] getProxies () throws LdapSearchException{
    Query q = new SimpleQuery ("objectclass","resourceManager");
    try {
      ResultSet r = ds.search(q,DirSearcher.MANAGERS);
      int l = r.getCount();
      if (l==0)
	return null;
      else {
	String [] res = new String [l];
	for (int i=0;i<l;i++) {
	  Entry e = r.getEntry(i);
	  res[i] = e.getString ("hn");
	}
	return res;
      }
    }
    catch (NamingException ne) {
      throw new LdapSearchException (ne.toString());
    }
  }

  public String [] getMachinesByProxy (String proxy)
       throws LdapSearchException{
    Query [] q = new Query [2];
    q[0] = new SimpleQuery ("objectclass","resourceManager");
    q[1] = new SimpleQuery ("hn",proxy);

    Query cq = new CompoundQuery (q,CompoundQuery.AND);

    try {
      ResultSet r = ds.search(cq,DirSearcher.MANAGERS);
      int l = r.getCount();
      if (l==0)
	return null;

      else {
	Entry e = r.getEntry (0);
	DisName [] machines = e.getDisNameArray ("machine");
	String [] s = new String [machines.length];

	for (int i=0;i<s.length;i++) {
	  s[i] = machines[i].getValue ("hn");
	}
	return s;
      }
    }
    catch (Exception ne) {
      throw new LdapSearchException (ne.toString());
    }
  }

  public String [] getComponentsByProxy(String proxy) 
       throws LdapSearchException{
    Query [] q = new Query [2];
    q[0] = new SimpleQuery ("objectclass","resourceManager");
    q[1] = new SimpleQuery ("hn",proxy);
    Vector v = new Vector();
    
    Query cq = new CompoundQuery (q,CompoundQuery.AND);

    try {
      ResultSet r = ds.search(cq,DirSearcher.MANAGERS);
      int l = r.getCount();
      if (l==0)
	return null;
      
      else {
	Entry e = r.getEntry (0);
	DisName [] machines = e.getDisNameArray ("machine");;
	
	for (int i=0;i < machines.length ;i++) {
	  String mach = machines[i].getValue ("hn");
	  Entry machEntry = ds.search (machines[i]);
	  DisName [] modules = machEntry.getDisNameArray ("module");

	  for (int j = 0;j < modules.length;j++) 
	    v.addElement (modules[j].getValue ("cn") + " (" + mach + ")");
	}
	
	String [] res = new String [v.size()];
	for (int i=0;i<v.size();i++)
	  res[i] = (String) v.elementAt(i);
	return res;
      }
    }
    catch (Exception ne) {
      throw new LdapSearchException (ne.toString());
    }
  }

  /** returns either: native, java, perl5, mpi, etc. */

  public String getComponentType (String cn) throws LdapSearchException {   
    Query [] q = new Query[2];
    q[0] = new SimpleQuery ("cn", cn);
    q[1] = new SimpleQuery ("objectclass","module");
    Query cq = new CompoundQuery (q,CompoundQuery.AND);
    try {
      ResultSet r = ds.search (cq,DirSearcher.SOFTWARE);
      if (r==null)
	return null;
      else return r.getEntry(0).getString ("type");
    }
    catch (Exception exp) {
      throw new LdapSearchException (exp.toString());
    }
  }

  public String getComponentPath (String cn,String machine)
       throws LdapSearchException {
    /* Does the machine entry specify a specific path? */
 
    String dnString = "cn=" + cn + ",hn=" + machine +
      ",rst=machines,rt=physical,ou=PSE,o=Extreme! Computing,c=US";
    
    try {
      DisName dn = new DisName (dnString);
      Entry e = ds.search (dn);
      if (e != null) {
	String mpath = e.getString ("path");
	if (mpath.equals ("NULL"))
	  return null;
	else return mpath; 
      }
      else return null; // no entry was found 
    }
    catch (Exception ee) {
      throw new LdapSearchException (ee.toString());
    }       
  }

  public String getComponentBase (String hostname) 
       throws LdapSearchException{
    String disName = "hn=" + hostname + 
      ",rst=machines,rt=physical,ou=PSE,o=Extreme! Computing,c=US";

    try {
      DisName dn = new DisName (disName);
      Entry e = ds.search (dn);
      return e.getString ("base");
    }
    catch (Exception ee) {
      throw new LdapSearchException (ee.toString());
    }
  }
}