package timeflow.app.ui;

import timeflow.data.time.*;
import timeflow.data.db.*;
import timeflow.format.field.*;
import timeflow.format.file.DelimitedFormat;
import timeflow.model.*;

import timeflow.util.*;

import javax.swing.*;

import java.awt.*;
import java.awt.event.*;
import java.text.ParseException;
import java.util.*;
import java.io.*;
import java.net.MalformedURLException;
import java.net.URL;

public class ImportDelimitedPanel extends JFrame 
{
	String fileName;
	SchemaPanel schemaPanel;
	boolean exitOnClose=false; // for testing!
	JScrollPane scroller;
	TFModel model;
	JLabel numLinesLabel=new JLabel();
	
	// for testing:
	public static void main(String[] args) throws Exception
	{
		System.out.println("Starting test of ImportEditor");
		String file="data/probate.tsv";
		String[][] data=DelimitedFormat.readArrayGuessDelim(file, System.out);
		ImportDelimitedPanel editor=new ImportDelimitedPanel(new TFModel());
		editor.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
		editor.setFileName(file);
		editor.setData(data);
		editor.setBounds(50,50,900,800);
		editor.setVisible(true);
		editor.exitOnClose=true;
	}
	
	public ImportDelimitedPanel(final TFModel model)
	{
		super("Import File");
		this.model=model;
		setBackground(Color.white);
		
		setLayout(new BorderLayout());
		JPanel top=new JPanel();
		add(top, BorderLayout.NORTH);
		top.setLayout(new FlowLayout(FlowLayout.LEFT));
		top.setBackground(Color.lightGray);
		top.add(numLinesLabel);
		final JTextField source=new JTextField(12);
		
		JButton done=new JButton("Import This");
		top.add(done);
		done.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				model.setDB(schemaPanel.makeDB(source.getText()), fileName, true, this);
				setVisible(false);
				if (exitOnClose)
					System.exit(0);
			}});
		
		JButton cancel=new JButton("Cancel");
		top.add(cancel);
		cancel.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				setVisible(false);
				if (exitOnClose)
					System.exit(0);
			}});
		
		top.add(new JLabel("        Enter A Source:"));		
		top.add(source);
		schemaPanel=new SchemaPanel();
		schemaPanel.setBackground(Color.white);
		scroller=new JScrollPane(schemaPanel);
		add(scroller, BorderLayout.CENTER);
	}

	public void scrollToTop()
	{
		scroller.getViewport().setViewPosition(new Point(0,0));  
	}
	
	public void setFileName(String fileName)
	{
		this.fileName=fileName;
	}
	
	public void setData(String[][] data)
	{
		numLinesLabel.setText((data.length-1)+" records read.  ");
		schemaPanel.display(data);
	}

	class SchemaPanel extends JPanel
	{
		int numFields, rows;
		String[][] data;
		ArrayList<FieldPanel> panels=new ArrayList<FieldPanel>();
		
		ActDB makeDB(String source)
		{
			// count number of fields that are not ignored.
			int n=0;
			for (FieldPanel fp: panels)
				if (!fp.ignore.isSelected())
					n++;
			
			Class[] types=new Class[n];
			String[] fieldNames=new String[n];
			int[] index=new int[n];
			if (source.trim().length()==0)
				source="[source unspecified]";
			int i=0, j=0;
			for (FieldPanel fp: panels)
			{			
				if (!fp.ignore.isSelected())
				{
					fieldNames[i]=fp.fieldName;
					String typeChoice=(String)fp.typeChoices.getSelectedItem();
					Class type=FieldFormatCatalog.javaClass(typeChoice);
					System.out.println("Type: "+type+" for: "+typeChoice+" from "+fp.fieldName);
					types[i]=type;
					index[i]=j;
					i++;
				}
				j++;
			}
			
			ActDB db= new ArrayDB(fieldNames, types, source);
			HashMap<Integer, StringBuffer> errors=new HashMap<Integer, StringBuffer>();
			for (i=1; i<data.length; i++)
			{
				Act act=db.createAct();				
				for (int k=0; k<n; k++)
				{
					j=index[k];
					String s=data[i][j];
					Field f=db.getField(fieldNames[k]);
					FieldFormat format=FieldFormatCatalog.getFormat(types[k]);
					try
					{
						Object o=format.parse(s);
						act.set(f,o);
					}
					catch (Exception e)
					{
						StringBuffer b=errors.get(i-1);
						if (b==null)
						{
							b=new StringBuffer();
							errors.put(i-1,b);
						}
						else
							b.append("; ");
						b.append(f.getName()+":"+s);
					}
				}
			}
			
			if (errors.size()>0)
			{
				Field error=db.addField("UNPARSED FIELDS", String.class);
				for (int row:errors.keySet())
				{
					db.get(row).set(error, errors.get(row).toString());
				}
			}
			
			for (j=0; j<n; j++)
			{
				System.out.println(db.getField(fieldNames[j]));
			}
			
			return db;
		}
		
		void display(String[][] data)
		{
			removeAll();
			
			this.data=data;
			
			// analyze data.
			Class[] guesses=FieldFormatGuesser.analyze(data, 1, 100);
			
			// go through first row, which is headers.
			String[] headers=data[0];
			
			// if there are duplicate headers, add indicators.
			HashSet<String> h=new HashSet<String>();
			for (int i=0; i<headers.length; i++)
			{
				String base=headers[i];
				int j=2;
				String name=base;
				while (h.contains(name))
				{
					name=base+" "+j;
					j++;
				}
				headers[i]=name;
				h.add(name);
			}
			
			numFields=headers.length;
			int cols=2;
			rows=(int)Math.ceil(numFields/2.0);
			setLayout(new GridLayout(rows, cols));
			for (int i=0; i<numFields; i++)
			{
				Bag<String> vals=new Bag<String>();
				
				for (int j=1; j<data.length; j++)
				{
					vals.add(data[j][i]);
				}
				java.util.List<String> top=vals.listTop(5);
				int n=top.size();
				String[] samples=new String[n];
				for (int j=0; j<n; j++)
				{
					String s=top.get(j);
					samples[j]=(s.length()==0 ? "*MISSING*" : s)+"    ("+vals.num(s)+" times)";
				}
				
				JPanel p=new JPanel();
				add(p);
				p.setLayout(new BorderLayout());
				FieldPanel f=new FieldPanel(headers[i], samples, guesses[i]);
				panels.add(f);
				p.add(f, BorderLayout.CENTER);
				JPanel hr=new JPanel();
				hr.setPreferredSize(new Dimension(20,20));
				p.add(hr, BorderLayout.SOUTH);
			}
		}
		
		public Dimension getPreferredSize()
		{
			return new Dimension(400,150*rows);
		}
	}
	
	class FieldPanel extends JPanel
	{
		JComboBox typeChoices;
		String fieldName;
		JCheckBox ignore;
		JLabel fieldLabel;
		int x1=5, y1=20, y3=150,x2=150, x3=150, x4=375, dh=2;

		
		FieldPanel(String fieldName, String[] sampleValues, Class typeGuess)
		{
			// just going with a null layout here, because it's a lot simpler!
			
			setLayout(null);
			setBackground(Color.white);
			this.fieldName=fieldName;
			
			fieldLabel=new JLabel(" \""+fieldName+'"');	
			fieldLabel.setFont(model.getDisplay().big());
			add(fieldLabel);
			fieldLabel.setBounds(x1,y1,fieldLabel.getPreferredSize().width, fieldLabel.getPreferredSize().height);
			
			typeChoices=new JComboBox();
			for (String choice: FieldFormatCatalog.classNames())
				typeChoices.addItem(choice);
			typeChoices.setSelectedItem(FieldFormatCatalog.humanName(typeGuess));
			add(typeChoices);
			int y2=fieldLabel.getY()+fieldLabel.getHeight()+dh+5;
			typeChoices.setBounds(x1,y2,
					typeChoices.getPreferredSize().width, typeChoices.getPreferredSize().height);
				
			ignore=new JCheckBox("Ignore Field");			
			add(ignore);
			ignore.setBounds(x1,typeChoices.getY()+typeChoices.getHeight()+dh,
					ignore.getPreferredSize().width, ignore.getPreferredSize().height);
			ignore.addActionListener(new ActionListener() {
				@Override
				public void actionPerformed(ActionEvent e) {
					Color c=ignore.isSelected() ? Color.gray : Color.black;
					fieldLabel.setForeground(c);
					typeChoices.setForeground(c);
				}});
			
			JTextArea values=new JTextArea();
			values.setForeground(Color.gray);
			for (int i=0; i<sampleValues.length; i++)
				values.append(sampleValues[i]+"\n");
			add(values);
			values.setBounds(x3,y2,x4-x3,y3-y2);
		}
		
		public Dimension getPreferredSize()
		{
			return new Dimension(500,150);
		}
	}
}