using System;
using System.Collections;
using System.Threading;
using System.Text.RegularExpressions;
using System.IO;
using Tamir.SharpSsh;
using Dizzy;
 
namespace Dizzy
{
 
 
	public class ProtocolException : ApplicationException
	{
		public ProtocolException () {}
		public ProtocolException (string Message) {}
	}
 
	public class Protocol
	{
 
		// I dati del nostro povero utente.
		string user, password;
 
		// I transfer attivi
		ArrayList transfers = new ArrayList ();
 
		public delegate void Del (string src, string dest,
		                          int transferredBytes, int totalBytes, string message);
 
 
		ArrayList threads = new ArrayList  ();
 
		Thread finder;
		SFTPConnection sftpUpdater;
 
		public Protocol (ref GlobalConfig config)
		{
			this.user = config.GetValue("user");
			this.password = config.password;
 
			try
			{
				this.sftpUpdater = new SFTPConnection(this.user, this.password);
				this.sftpUpdater.Connect ();
			}
			catch(Exception e)
			{
				throw new ProtocolException("Impossibile connettersi a poisson.phc.unipi.it %% " + e.Message);
			}
 
 
		}
 
		public void Search(string keyword, ref FileTreeView f)
		{
			if(this.finder != null && this.finder.IsAlive)
			{
				Console.WriteLine("Ricerca in corso, tento di interromperla.");
				this.finder.Abort ();
			}
 
			FileSearch fs = new FileSearch (this.user, this.password, keyword, ref f);
			this.finder = new Thread (new ThreadStart (fs.DoSearch));
			this.finder.Start ();
		}
 
 
		public void Disconnect ()
		{
			foreach(FileTransfer t in this.transfers)
			{
				t.Stop ();
			}
			foreach(Thread t in this.threads)
			{
				if(t.IsAlive)
					t.Abort ();
			}
 
			if (this.finder.IsAlive)
			{
				this.finder.Abort ();
			}
		}
 
 
		public void Download(File f, ref TaskTreeView tasks, string downloadFolder)
		{
			// Preparo gli argomento da passare alla funzione.
			ArrayList args = new ArrayList ();
			args.Add (f);
			args.Add (tasks);
			args.Add (downloadFolder);
			Thread t = new Thread (new ParameterizedThreadStart(_Download));
			t.Start (args);
			this.threads.Add (t);
		}
 
		protected void _Download(object args)
		{
			// Questi cast sono piuttosto brutali ma posso passare
			// solo un oggetto a Download e quindi è il meglio che
			// mi riesce di fare.
			File f = (File) ((ArrayList) args)[0];
			TaskTreeView tasks = (TaskTreeView) ((ArrayList) args)[1];
			string downloadFolder = (string) ((ArrayList) args)[2];
			Console.WriteLine (" => Download starting");
			FileTransfer transfer = new FileTransfer (f, ref tasks, this.user, this.password, downloadFolder);
			this.transfers.Add (transfer);
		}
 
	}
 
	public class SFTPConnection
	{
 
		public Sftp sftp;
		public delegate void SFTPEvent (string src, string dest, int transferredBytes,
		                                int totalBytes, string message);
 
		public event SFTPEvent TransferStarted;
		public event SFTPEvent TransferProgress;
		public event SFTPEvent TransferStopped;
 
		public delegate void Del (string src, string dest,
		                          int transferredBytes, int totalBytes, string message);
 
		public SFTPConnection (string user, string password)
		{
 
			// Inizializziamo il protocollo
			this.sftp = new Sftp ("poisson.phc.unipi.it", user);
			this.sftp.Password = password;
 
			this.sftp.OnTransferStart += new FileTransferEvent(this.TransferStartedHandler);
			this.sftp.OnTransferProgress += new FileTransferEvent(this.TransferProgressHandler);
			this.sftp.OnTransferEnd += new FileTransferEvent(this.TransferStoppedHandler);
 
		}
 
		private void TransferStartedHandler(string src, string dest, int transferredBytes,
		                                    int totalBytes, string message)
		{
			Console.WriteLine (" => TransferStarted");
			TransferStarted(src, dest, transferredBytes, totalBytes, message);
		}
 
		private void TransferProgressHandler(string src, string dest, int transferredBytes,
		                                    int totalBytes, string message)
		{
			TransferProgress(src, dest, transferredBytes, totalBytes, message);
		}
 
		private void TransferStoppedHandler(string src, string dest, int transferredBytes,
		                                    int totalBytes, string message)
		{
			TransferStopped(src, dest, transferredBytes, totalBytes, message);
		}
 
		public void Connect()
		{
			sftp.Connect (22);
		}
 
		public ArrayList SearchFiles()
		{
			ArrayList matches = new ArrayList ();
			Console.WriteLine(" => Scan della nobackup");
			// ArrayList users = sftp.GetFileList ("/nobackup/");
			ArrayList users = new ArrayList();
			users.Add("robol");
			users.Add("bianchi");
			foreach(string folder in users)
			{
				if (folder == ".")
					continue;
				else if (folder == "..")
					continue;
				Console.WriteLine(" => Scan di {0}", folder);
				try {  matches.AddRange (GetFolderContent(folder)); }
				catch(Exception e) { Console.WriteLine (e.Message); }
			}
			Console.WriteLine(" => Scan terminato");
			return matches;
		}
 
		protected ArrayList GetFolderContent(string folder)
		{
			ArrayList matches, files;
			File f;
			string user;
			files = sftp.GetFileList("/nobackup/" + folder + "/");
			matches = new ArrayList ();
			foreach(string entry in files)
			{
				// Controllo che non sia una cartella. Evidentemente
				// questo metodo non è affidabile, ma per ora è il meglio
				// che SFTP ci permetta di fare.
				if(entry.IndexOf(".") == -1)
					matches.AddRange(GetFolderContent(folder + "/" + entry));
				else if(!entry.StartsWith("."))
				{
					user = folder;
					if(user.Contains("/"))
						user = user.Substring(0, user.IndexOf("/"));
					f = new File("/nobackup/" + folder + "/" + entry, user );
					matches.Add(f);
				}
			}
			return matches;
		}
 
		public void Download(File f, string downloadFolder)
		{
			Console.WriteLine (" => Getting {0}", f.name);
			// try
			// {
				Console.WriteLine (f.path);
				this.sftp.Get (f.path, downloadFolder + Path.DirectorySeparatorChar +  f.name);
			// }
			/* catch()
			{
				Console.WriteLine("Error in Get... ");
				Console.WriteLine(e.Message);
			} */
		}
 
		public void Disconnect ()
		{
			sftp.Close ();
		}
	}
 
	public class FileTransfer
	{
 
		TaskTreeView tasks;
		File file;
		Gtk.TreeIter iter;
 
		SFTPConnection sftp;
 
		public delegate void Handler(string src, string dest, int transferredBytes,
		                             int totalBytes, string message);
 
		public FileTransfer (File f, ref TaskTreeView tasks, string user, string password, string downloadFolder)
		{
			// Connessione al server
			this.sftp = new SFTPConnection (user, password);
			this.sftp.Connect ();
 
			this.tasks = tasks;
			this.file = f;
 
			sftp.TransferStarted += new SFTPConnection.SFTPEvent(OnTransferStarted);
			sftp.TransferProgress += new SFTPConnection.SFTPEvent(OnTransferProgress);
			sftp.TransferStopped += new SFTPConnection.SFTPEvent(OnTransferStopped);
 
			Console.WriteLine(" => Starting sftp command");
			sftp.Download (this.file, downloadFolder);
			sftp.Disconnect ();
		}
 
		public void Stop ()
		{
			this.sftp.Disconnect ();
		}
 
		public void OnTransferStarted (string src, string dest, int transferredBytes,
		                          int totalBytes, string message)
		{
			Console.WriteLine(" => TransferStarted received");
			this.iter = this.tasks.AddTask(file.name, 0);
		}
 
		public void OnTransferProgress(string src, string dest, int transferredBytes,
		                               int totalBytes, string message)
		{
			// if(message != "")
				// Console.WriteLine(message);
			int perc = transferredBytes / (totalBytes / 100);
			this.tasks.SetProgress(this.iter, perc);
		}
 
		public void OnTransferStopped (string src, string dest, int transferredBytes,
		                               int totalBytes, string message)
		{
			this.tasks.DeleteRow (this.iter);
		}
	}
 
 
	class FileSearch
{
	SshExec sshexec;
	string username, password, keywords;
	FileTreeView f;
 
	public FileSearch (string username, string password, string keywords, ref FileTreeView f)
	{
		this.username = username;
		this.password = password;
		this.keywords = keywords;
		this.f = f;
	}
 
	public void DoSearch ()
	{
		sshexec = new SshExec("poisson.phc.unipi.it", this.username, this.password);
		sshexec.Connect ();
 
		string command = "locate -i -e --regex /nobackup/.+" + this.keywords;
		string output = sshexec.RunCommand (command);
		f.Clear ();
		foreach(string line in Regex.Split(output, "\n"))
		{
			if (line.StartsWith("/nobackup/") && Regex.Match(line,".*[.].*").Success)
				this.f.AddFile (new File(line, ""));
		}
		sshexec.Close ();
	}
 
}
 
}