package jalview.gui;

import jalview.bin.Cache;
import jalview.ws.seqfetcher.DbSourceProxy;

import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.Container;
import java.awt.Dimension;
import java.awt.FlowLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.ArrayList;
import java.util.Hashtable;
import java.util.List;
import java.util.Vector;

import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.JViewport;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.MutableTreeNode;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;
import javax.swing.tree.TreeSelectionModel;

import com.sun.java.help.search.DBPartParameters;

public class JDatabaseTree extends JalviewDialog
{
  JButton getDatabaseSelectorButton()
  {
    final JButton viewdbs = new JButton("--- Select Database ---");
    viewdbs.addActionListener(new ActionListener()
    {

      @Override
      public void actionPerformed(ActionEvent arg0)
      {
        showDialog(null);
      }
    });
    return viewdbs;
  }

  JScrollPane svp;

  JTree dbviews;

  private jalview.ws.SequenceFetcher sfetcher;

  public JDatabaseTree(jalview.ws.SequenceFetcher sfetch)
  {
    super();
    /*
     * Dynamically generated database list will need a translation function from
     * internal source to externally distinct names. UNIPROT and UP_NAME are
     * identical DB sources, and should be collapsed.
     */
    DefaultMutableTreeNode tn = null, root = new DefaultMutableTreeNode();
    Hashtable<DbSourceProxy, DefaultMutableTreeNode> tns = new Hashtable<DbSourceProxy, DefaultMutableTreeNode>();
    Hashtable<String, DefaultMutableTreeNode> source = new Hashtable<String, DefaultMutableTreeNode>();
    sfetcher = sfetch;
    String dbs[] = sfetch.getSupportedDb();
    Hashtable ht = new Hashtable();
    for (int i = 0; i < dbs.length; i++)
    {
      tn = source.get(dbs[i]);
      List<DbSourceProxy> srcs = sfetch.getSourceProxy(dbs[i]);
      if (tn == null)
      {
        source.put(dbs[i], tn = new DefaultMutableTreeNode(dbs[i], true));
      }
      for (DbSourceProxy dbp : srcs)
      {
        if (ht.get(dbp.getDbName()) == null)
        {
          tn.add(new DefaultMutableTreeNode(dbp, false));
          ht.put(dbp.getDbName(), dbp.getDbName());
        } else {
          System.err.println("dupe ig for : "+dbs[i]+" \t"+dbp.getDbName());
        }
      }
    }
    for (int i = 0; i < dbs.length; i++)
    {
      tn = source.get(dbs[i]);
      if (tn == null)
      {
        continue;
      }
      if (tn.getChildCount() == 1)
      {
        // remove nodes with only one child
        source.put(dbs[i], tn = (DefaultMutableTreeNode) tn.getChildAt(0));
      }
      root.add(tn);
    }
    initDialogFrame(this, true, false, "Select Database Retrieval Source",
            400, 300);
    svp = new JScrollPane();
    //svp.setAutoscrolls(true);
    dbviews = new JTree(new DefaultTreeModel(root, false));
    dbviews.setCellRenderer(new DbTreeRenderer(this));

    dbviews.getSelectionModel().setSelectionMode(
            TreeSelectionModel.SINGLE_TREE_SELECTION);
    svp.getViewport().setView(dbviews);
    svp.getViewport().setMinimumSize(new Dimension(300,250));
    //svp.setSize(300,250);
//    JPanel panel=new JPanel();
//    panel.setSize(new Dimension(350,220));
//    panel.add(svp);
    
    add(svp, BorderLayout.NORTH);
    JPanel j = new JPanel(new FlowLayout());
    j.setPreferredSize(new Dimension(300,50));
    j.add(ok);
    j.add(cancel);
    add(j, BorderLayout.CENTER);
    validate();
  }

  private class DbTreeRenderer extends DefaultTreeCellRenderer implements
          TreeCellRenderer
  {
    JDatabaseTree us;

    public DbTreeRenderer(JDatabaseTree me)
    {
      us = me;
    }

    private Component returnLabel(String txt)
    {
      JLabel jl = new JLabel(txt);
      jl.setFont(JvSwingUtils.getLabelFont());
      return jl;
    }

    @Override
    public Component getTreeCellRendererComponent(JTree tree, Object value,
            boolean selected, boolean expanded, boolean leaf, int row,
            boolean hasFocus)
    {
      String val = "";
      if (value != null && value instanceof DefaultMutableTreeNode)
      {
        DefaultMutableTreeNode vl = (DefaultMutableTreeNode) value;
        value = vl.getUserObject();
        if (value instanceof DbSourceProxy)
        {
          val = (((DbSourceProxy) value).getDbName());
        }
        else
        {
          if (value instanceof String)
          {
            val = ((String) value);
          }
        }
      }
      if (value == null)
      {
        val = ("");
      }
      return super.getTreeCellRendererComponent(tree, val, selected,
              expanded, leaf, row, hasFocus);

    }
  }

  List<DbSourceProxy> selection = null;

  @Override
  protected void raiseClosed()
  {
    for (ActionListener al : lstners)
    {
      al.actionPerformed(null);
    }
  }

  @Override
  protected void okPressed()
  {
    _setSelectionState();
    closeDialog();
  }

  @Override
  protected void cancelPressed()
  {
    closeDialog();
  }

  private void showDialog(Container parent)
  {
    validate();
    waitForInput();
  }

  public boolean hasSelection()
  {
    return selection == null ? false : selection.size() == 0 ? false : true;
  }

  public List<DbSourceProxy> getSelectedSources()
  {
    return selection;
  }

  private void _setSelectionState()
  {
    if (dbviews.getSelectionCount() == 0)
    {
      selection = null;
    }
    List<DbSourceProxy> srcs = new ArrayList<DbSourceProxy>();
    for (TreePath tp : dbviews.getSelectionPaths())
    {
      DefaultMutableTreeNode dmt = (DefaultMutableTreeNode) tp
              .getLastPathComponent();
      if (dmt.getUserObject() != null)
      {
        if (dmt.getUserObject() instanceof DbSourceProxy)
        {
          srcs.add((DbSourceProxy) dmt.getUserObject());
        }
        else
        {
          srcs.addAll(sfetcher.getSourceProxy((String) dmt.getUserObject()));
        }
      }
    }
    selection = srcs;
  }

  public String getSelectedItem()
  {
    if (hasSelection())
    {
      return getSelectedSources().get(0).getDbName();
    }
    return null;
  }

  public String getExampleQueries()
  {
    if (!hasSelection())
    {
      return null;
    }
    StringBuffer sb = new StringBuffer();
    for (DbSourceProxy dbs : getSelectedSources())
    {
      if (sb.length() > 0)
      {
        sb.append(";");
      }
      sb.append(dbs.getTestQuery());
    }
    return sb.toString();
  }

  List<ActionListener> lstners = new Vector<ActionListener>();

  public void addActionListener(ActionListener actionListener)
  {
    lstners.add(actionListener);
  }

  public void removeActionListener(ActionListener actionListener)
  {
    lstners.remove(actionListener);
  }

  public static void main(String args[])
  {
    Cache.getDasSourceRegistry();
    JDatabaseTree jdt = new JDatabaseTree(new jalview.ws.SequenceFetcher());
    JFrame foo = new JFrame();
    foo.setLayout(new BorderLayout());
    foo.add(jdt.getDatabaseSelectorButton(), BorderLayout.CENTER);
    foo.pack();
    foo.setVisible(true);
    int nultimes = 5;
    final Thread us = Thread.currentThread();
    jdt.addActionListener(new ActionListener()
    {

      @Override
      public void actionPerformed(ActionEvent e)
      {
        us.interrupt();
      }
    });
    do
    {
      try
      {
        Thread.sleep(50);
      } catch (InterruptedException x)
      {
        nultimes--;
        if (!jdt.hasSelection())
        {
          System.out.println("No Selection");
        }
        else
        {
          System.out.println("Selection: " + jdt.getSelectedItem());
          int s = 1;
          for (DbSourceProxy pr : jdt.getSelectedSources())
          {
            System.out.println("Source " + s++ + ": " + pr.getDbName()
                    + " (" + pr.getDbSource() + ") Version "
                    + pr.getDbVersion() + ". Test:\t" + pr.getTestQuery());
          }
          System.out.println("Test queries: " + jdt.getExampleQueries());
        }
      }
    } while (nultimes > 0 && foo.isVisible());
    foo.setVisible(false);
  }
}