AbstractCommand.cs 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using Quadarax.Foundation.Common.Console;
  5. using Quadarax.Foundation.Common.Value;
  6. using Quadarax.Foundation.QConsole.Argument;
  7. using Quadarax.Foundation.QConsole.Extensions;
  8. namespace Quadarax.Foundation.QConsole.Command.Base
  9. {
  10. public abstract class AbstractCommand
  11. {
  12. #region *** Private Fields ***
  13. private Engine _engine;
  14. private IDictionary<string, AbstractArgument> _arguments;
  15. #endregion
  16. #region *** Properties ***
  17. public abstract string Name { get; }
  18. public abstract string Description { get; }
  19. public AbstractArgument[] Arguments => _arguments.Values.ToArray();
  20. protected Engine Engine => _engine;
  21. #endregion
  22. #region *** Constructor ***
  23. public AbstractCommand(Engine engine)
  24. {
  25. _engine = engine ?? throw new ArgumentNullException(nameof(engine));
  26. _arguments = new Dictionary<string, AbstractArgument>();
  27. var args = SetupArguments();
  28. foreach (var arg in args)
  29. _arguments.Add(arg.Code, arg);
  30. Init();
  31. WriteDebugInfo($"Command '{Name}' has {_arguments.Count} arguments. Initialized.");
  32. }
  33. #endregion
  34. #region *** Public operations ***
  35. public Result Execute(string input)
  36. {
  37. ConsoleWriter writer = null;
  38. try
  39. {
  40. writer = new ConsoleWriter(ConsoleColor.Green);
  41. foreach (var arg in Arguments)
  42. arg.SetValueAsDefault();
  43. ParseInput(input);
  44. var result = OnExecute();
  45. result = EndExecute(result);
  46. return result;
  47. }
  48. catch (Exception e)
  49. {
  50. return new Result(e);
  51. }
  52. finally
  53. {
  54. writer?.Dispose();
  55. }
  56. }
  57. public override string ToString()
  58. {
  59. return "Command: " + Name;
  60. }
  61. #endregion
  62. #region *** Private operations ***
  63. private void Init()
  64. {
  65. OnCommandInit();
  66. }
  67. private void ParseInput(string input)
  68. {
  69. //var args = input.Split(new[] {Engine.Configuration.CharacterNamedArgumentSeparator}, StringSplitOptions.RemoveEmptyEntries);
  70. var args = Split(input, Engine.Configuration.CharacterTextValueBraceSeparator, new[]
  71. {
  72. Engine.Configuration.CharacterNamedArgumentSeparator,
  73. Engine.Configuration.CharacterOrdinalArgumentSeparator
  74. });
  75. for (int i = 1; i < args.Length; i++)
  76. {
  77. var arg = args[i];
  78. var index = i - 1;
  79. WriteDebugInfo($"Parsing argument [index={index}] by value string '{arg}' in progress..");
  80. AbstractArgument argument = null;
  81. if (arg.StartsWith(Engine.Configuration.CharacterNamedArgumentSeparator))
  82. {
  83. //Named
  84. WriteDebugInfo("Detects argument as named.");
  85. var parsed = ParseNamedArgument(arg);
  86. if (!ContainsArgument(parsed.Item1))
  87. throw new Exception(
  88. $"Invalid argument '{Engine.Configuration.CharacterNamedArgumentSeparator}{parsed.Item1}' for this command.");
  89. argument = GetArgument(parsed.Item1);
  90. if (argument.IsPositional())
  91. if (!((OrdinalArgument)argument).IsCodeAllowed)
  92. throw new Exception($"Usage of ordinal argument '{argument.Code}' as named argument is not allowed!");
  93. if (argument.IsFlag())
  94. ((FlagArgument)argument).Set();
  95. else
  96. argument.Value.Set(parsed.Item2);
  97. }
  98. else
  99. {
  100. //Ordinal
  101. WriteDebugInfo("Detects argument as ordinal.");
  102. argument = GetArgument(index);
  103. if (argument == null)
  104. throw new Exception(
  105. $"Invalid ordinal argument '{arg}' at position {index} for this command.");
  106. argument.Value.Set(arg);
  107. WriteDebugInfo($"Ordinal argument '{argument.Code}' is set to value '{argument.Value.AsString()}'.");
  108. }
  109. }
  110. }
  111. private Tuple<string, string> ParseNamedArgument(string argument)
  112. {
  113. try
  114. {
  115. var argParts = argument.Split(new[] {Engine.Configuration.CharacterNamedArgumentValueSeparator}, StringSplitOptions.RemoveEmptyEntries);
  116. WriteDebugInfo($"Splitting arg-parts '{argument}' to parts (count={argParts.Length}): {string.Join(",",argParts)}");
  117. var argName = argParts[0].Replace(Engine.Configuration.CharacterNamedArgumentSeparator,string.Empty);
  118. var argValue = argParts.Length > 1 ? argParts[1] : string.Empty;
  119. WriteDebugInfo($"Named argument '{argument}' parsed as name='{argName}'; value='{argValue}'.");
  120. return new Tuple<string, string>(argName, argValue);
  121. }
  122. catch (Exception e)
  123. {
  124. WriteDebugInfo($"Error during parsing named argument '{argument}'.");
  125. throw new Exception($"Argument '{argument}' parsing error.", e);
  126. }
  127. }
  128. private string[] Split(string input, string valueBracket, string[] separators)
  129. {
  130. var result = new List<string>();
  131. var trimedInput = input.Trim();
  132. if (string.IsNullOrEmpty(trimedInput))
  133. return result.ToArray();
  134. var cValueBracked = valueBracket.ToCharArray().First();
  135. var cSeparators = separators.Select(x => x.ToCharArray().First());
  136. var nIndexStart = 0;
  137. var nIndexEnd = 0;
  138. var isInsideBracked = false;
  139. int offset = 0;
  140. for (offset = 0; offset < trimedInput.Length; offset++)
  141. {
  142. var current = trimedInput[offset];
  143. // current | isInside | Result
  144. // 1 1 0
  145. // 1 0 1
  146. // 0 1 1
  147. // 0 0 0
  148. isInsideBracked = current == cValueBracked ^ isInsideBracked;
  149. if (isInsideBracked)
  150. continue;
  151. if (cSeparators.Contains(current))
  152. {
  153. nIndexEnd = offset;
  154. result.Add(trimedInput.Substring(nIndexStart,nIndexEnd - nIndexStart));
  155. nIndexStart = nIndexEnd;
  156. isInsideBracked = false;
  157. }
  158. }
  159. if (offset>nIndexStart)
  160. result.Add(trimedInput.Substring(nIndexStart,offset - nIndexStart));
  161. return result.Where(x => !string.IsNullOrEmpty(x.Trim())).Select(x=>x.Trim()).ToArray();
  162. }
  163. private IEnumerable<AbstractArgument> SetupArguments()
  164. {
  165. return OnSetupArguments();
  166. }
  167. protected void WriteDebugInfo(string text)
  168. {
  169. using (var debug = new DebugConsoleWriter())
  170. {
  171. debug.WriteDebugLine(text);
  172. }
  173. }
  174. #endregion
  175. #region *** Protected virtuals ***
  176. protected virtual void OnCommandInit()
  177. {
  178. WriteDebugInfo("Command called OnCommandInit.");
  179. }
  180. protected virtual IEnumerable<AbstractArgument> OnSetupArguments()
  181. {
  182. return new AbstractArgument[0];
  183. //{
  184. // new FlagArgument(Constants.Arguments.General.DebugMode.Code, Constants.Arguments.General.DebugMode.Description, Constants.Arguments.General.DebugMode.Hint, true),
  185. //};
  186. }
  187. protected abstract Result OnExecute();
  188. protected virtual Result EndExecute(Result incommingResult)
  189. {
  190. return incommingResult;
  191. }
  192. #endregion
  193. #region *** Protected operations ***
  194. protected AbstractArgument GetArgument(string code)
  195. {
  196. if (!_arguments.ContainsKey(code))
  197. throw new Exception($"Argument '{code}' is not defined.");
  198. return _arguments[code];
  199. }
  200. protected AbstractArgument GetArgument(int ordinalPosition)
  201. {
  202. return _arguments.Values.Where(x => x.IsPositional()).Skip(ordinalPosition).FirstOrDefault();
  203. }
  204. protected bool ContainsArgument(string code)
  205. {
  206. return _arguments.ContainsKey(code);
  207. }
  208. #endregion
  209. }
  210. }