aswi2021tri-musketyri-gitlab/Server/ServerApp/User/CommandsAcceptor.cs @ master

//
//	Author: Eliska Mourycova
//

using ServerApp.DataDownload;
using ServerApp.Predictor;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace ServerApp.User
{
	
	/// <summary>
	/// This class tkes care of parsing and executing admin commands
	/// </summary>
	public class CommandsAcceptor
	{
		// the DataDownloader instance
		private DataDownloader dd;

		// the predictor instance
		private IPredictionController model;

		public CommandsAcceptor(DataDownloader dd, IPredictionController controller)
		{
			this.dd = dd;
			this.model = controller;
		}

		/// <summary>
		/// Prints a prompt and waits for input
		/// </summary>
		public void AcceptCommand()
		{
			while (true) {
				Console.Write(">");
				string command = Console.ReadLine();
				if(command != null && command.Length > 0)
				{
					Console.WriteLine("received admin command: " + command);
					Command c = ParseString(command);
					CheckCommand(c);
				}

			}
		}

		/// <summary>
		/// Parses the input string
		/// </summary>
		/// <param name="str">The whole string typed in by the user</param>
		/// <returns>A new instance of the Command class</returns>
		private Command ParseString(string str)
		{

			string[] splits = str.Trim().Split(' ');
			Command command = new Command();

			command.WholeCommand = str.Trim();
			command.MainCommand = splits[0];

			for(int i = 1; i < splits.Length; i++)
			{

				string current = splits[i];
				if (current.StartsWith("-"))
				{
					command.AddFlag(current.Substring(1));
				}
				else
				{
					command.AddData(current);
				}
			}

			return command;
		}

		/// <summary>
		/// Checks the comamnd and decides what to do according to the main command
		/// </summary>
		/// <param name="c">The command</param>
		private void CheckCommand(Command c)
		{

			switch (c.MainCommand)
			{
				case "data":
					HandleDataCommand(c);
					break;
				case "model":
					HandleModelCommand(c);
					break;
				case "help":
					HandleHelpCommand(c);
					break;
				default:
					PrintUnknown(c.MainCommand);
					break;
			}
		}

		/// <summary>
		/// Prints to the console, that the given command is unknown
		/// </summary>
		/// <param name="command">The input command</param>
		private void PrintUnknown(string command)
		{
			Console.WriteLine("Unknown command: " + command);
		}

		/// <summary>
		/// Handles the command linked to data listing and downloading
		/// </summary>
		/// <param name="c">The command</param>
		private void HandleDataCommand(Command c)
		{
			// data -list
			// data -dl 1-2019 12-2020

			if(c.WholeCommand.Equals("data -list"))
			{
				Console.WriteLine("Listing all downloaded data files...");

				var subdirs = dd.DataSubDirectories.Values.ToArray();

				foreach (string sub in subdirs)
				{
					List<string> files = dd.GetData(sub, null, null);
					foreach (string f in files)
					{
						Console.WriteLine(f);
					}
				}
			}
			else if (c.FlagsAndData.ContainsKey("dl"))
			{
				List<string> dates = c.FlagsAndData["dl"];
				if (dates.Count != 2)
				{
					Console.WriteLine("2 date arguments needed!");
					return;
				}

				Date start = Date.ParseDate(dates[0]);
				Date end = Date.ParseDate(dates[1]);

				if (start == null || end == null)
				{
					Console.WriteLine("Date parsing was not successful. Please ensure you enter the dates in the following format: [month]-[year] (e.g. 1-2020). Month must be greater than zero.");
					return;
				}

				Console.WriteLine("Downloading jis, login and weather data from " + start.ToString() + " to " + end.ToString() + "...");

				try
				{
					List<string> files = dd.DownloadData(DataType.JIS, DataFormat.CSV, start, end);
					files.AddRange(dd.DownloadData(DataType.STROJE, DataFormat.CSV, start, end));
					files.AddRange(dd.DownloadData(DataType.POCASI, DataFormat.CSV, start, end));

					Console.WriteLine("Data download complete, saved files: ");
					foreach (string f in files)
						Console.WriteLine(f);

				}
				catch (Exception)
				{
					Console.WriteLine("Data download failed, please ensure the start date is before or same as end date.");
					return;
				}
				
				
			}
			else
			{
				PrintUnknown(c.WholeCommand);
				HandleHelpCommand(c);
			}
		}

		/// <summary>
		/// Handles the command linked to predictor training, rollback and files listing
		/// </summary>
		/// <param name="c">The command</param>
		private void HandleModelCommand(Command c)
		{

			// todo if the command has something else after the flag - abort?
			if (c.FlagsAndData.ContainsKey("files"))
			{
				IEnumerable<string> files = model.GetDataFileNames();
				Console.WriteLine("Listing current model's trianing data...");

				foreach (string f in files)
					Console.WriteLine(f);
			}
			else if (c.FlagsAndData.ContainsKey("retrain"))
			{
				Console.WriteLine("Starting to retrain the predictor...");
				model.Train();
				Console.WriteLine("Predictor retrained.");
			}
			else if (c.FlagsAndData.ContainsKey("rollback"))
			{
				Console.WriteLine("Switching back to the previous version of the predictor...");
				int s = model.Rollback();
				switch (s)
				{
					case 0:
						Console.WriteLine("Rollback succesful.");
						break;
					case 1:
						Console.WriteLine("No previous version of the predictor exists!");
						break;
					case 2:
						Console.WriteLine("Error occured while trying to perform rollback - corrupted or removed Predictor.config file!");
						break;
					default:
						Console.WriteLine("Unknown error occured when performing rollback.");
						break;
				}
			}
			else
			{
				PrintUnknown(c.WholeCommand);
				HandleHelpCommand(c);
			}


			//model.Rollback() // 0 -ok, 1 nejde, 2 - error (kdyz nekdo smaze soubor config)
			//model.GetDataFileNames();
		}

		/// <summary>
		/// Prints help to the console
		/// </summary>
		/// <param name="c"></param>
		private void HandleHelpCommand(Command c)
		{
			Console.WriteLine("List of available commands and their description: ");
			Console.WriteLine("--------------------------------------------------");
			// data
			Console.WriteLine("data [OPTION] [DATE]...");
			Console.WriteLine("   Description: Capable of showing saved data files and downloading new ones.");
			Console.WriteLine("   -list: Lists currently saved data files. This list does not have to match the files which any of the predictors is trained on!");
			Console.WriteLine("      Example: data -list");
			Console.WriteLine("   -dl [startDate] [endDate]: Downloads jis, login and weather files from a specified time span. Both [startDate] and [endDate] are inclusive. Date format must be entered as [month]-[year].");
			Console.WriteLine("      Example: data -dl 1-2019 12-2020");
			Console.WriteLine("");

			// model
			Console.WriteLine("model [OPTION]");
			Console.WriteLine("   Description: Capable of retraining the prediction model, going back to the previous version and listing files on which the predictor was trained. ");
			Console.WriteLine("   -files: Lists the data files on which the currently used model was trained.");
			Console.WriteLine("      Example: model -files");
			Console.WriteLine("   -retrain: Retrains the model using the files currently saved in the data directory.");
			Console.WriteLine("      Example: model -retrain");
			Console.WriteLine("   -rollback: Switches the model back to the previous version.");
			Console.WriteLine("      Example: model -rollback");
		}

	}
}