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

//
//	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
{
	

	public class CommandsAcceptor
	{
		private DataDownloader dd;
		private IPredictionController model;

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

		public void AcceptCommand()
		{
			while (true) {
				string command = Console.ReadLine();
				Console.WriteLine("received admin command: " + command);
				Command c = ParseString(command);
				CheckCommand(c);
				//Program.NotifyUserMessage(command);
			}
		}

		private static 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;
		}

		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;
			}
		}

		private void PrintUnknown(string command)
		{
			Console.WriteLine("Unknown command: " + command);
		}

		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 ex)
				{
					Console.WriteLine("Data download failed, please ensure the start date is before or same as end date.");
					return;
				}
				
				
			}
		}

		private void HandleModelCommand(Command c)
		{
			// TODO:
			// model -files
			// model -retrain
			// model -rollback
		}

		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");
		}

		private void PrintUsage(Command c)
		{
			
		}
	}
}