MERGE

1 files changed, 525 insertions, 0 deletions

diff --git a/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Stubs/ViewModels/MainViewVM.cs b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Stubs/ViewModels/MainViewVM.cs
new file mode 100644
index 000000000..d7e641735
--- /dev/null
+++ b/Software/Visual_Studio/MachineStudio/Modules/Tango.MachineStudio.Stubs/ViewModels/MainViewVM.cs
@@ -0,0 +1,525 @@
+using Microsoft.Win32;
+using System;
+using System.Collections.Generic;
+using System.Collections.ObjectModel;
+using System.IO;
+using System.Linq;
+using System.Reflection;
+using System.Text;
+using System.Threading.Tasks;
+using System.Windows;
+using Tango.Core.Commands;
+using Tango.MachineStudio.Common.Notifications;
+using Tango.MachineStudio.Common.StudioApplication;
+using Tango.Scripting;
+using Tango.Settings;
+using Tango.SharedUI;
+using Tango.Stubs;
+using Tango.Transport;
+using Tango.Transport.Adapters;
+
+namespace Tango.MachineStudio.Stubs.ViewModels
+{
+    /// <summary>
+    /// Represents the script execution utility main view model.
+    /// </summary>
+    /// <seealso cref="Tango.SharedUI.ViewModel" />
+    public class MainViewVM : ViewModel
+    {
+        private UsbTransportAdapter _adapter; //Holds the USB transport adapter.
+        private StubManager _stubManager;
+        private INotificationProvider _notification;
+
+        #region Properties
+
+        public IStudioApplicationManager ApplicationManager { get; set; }
+
+        private bool _useConnectedMachine;
+        /// <summary>
+        /// Gets or sets a value indicating whether [use connected machine].
+        /// </summary>
+        public bool UseConnectedMachine
+        {
+            get { return _useConnectedMachine; }
+            set { _useConnectedMachine = value; RaisePropertyChangedAuto(); InvalidateRelayCommands(); }
+        }
+
+        /// <summary>
+        /// Gets or sets the code tabs.
+        /// </summary>
+        public ObservableCollection<CodeTabVM> CodeTabs { get; set; }
+
+        /// <summary>
+        /// Gets or sets the additional highlight C# types.
+        /// </summary>
+        public ObservableCollection<KeyValuePair<String, Type>> HighlightTypes { get; set; }
+
+        /// <summary>
+        /// Gets or sets the collection of stub snippets.
+        /// </summary>
+        public ObservableCollection<StubSnippetVM> StubSnippets { get; set; }
+
+        private StubSnippetVM _selectedStubSnippet;
+        /// <summary>
+        /// Gets or sets the selected stub snippet.
+        /// </summary>
+        public StubSnippetVM SelectedStubSnippet
+        {
+            get { return _selectedStubSnippet; }
+            set { _selectedStubSnippet = value; RaisePropertyChanged(nameof(SelectedStubSnippet)); }
+        }
+
+        private String _log;
+        /// <summary>
+        /// Gets or sets the current response log.
+        /// </summary>
+        public String Log
+        {
+            get { return _log; }
+            set { _log = value; RaisePropertyChanged(nameof(Log)); }
+        }
+
+        private CodeTabVM _selectedCodeTab;
+        /// <summary>
+        /// Gets or sets the selected code tab.
+        /// </summary>
+        public CodeTabVM SelectedCodeTab
+        {
+            get { return _selectedCodeTab; }
+            set { _selectedCodeTab = value; RaisePropertyChanged(nameof(SelectedCodeTab)); InvalidateRelayCommands(); }
+        }
+
+        private bool _isConnected;
+        /// <summary>
+        /// Gets or sets a value indicating whether the USB adapter is connected.
+        /// </summary>
+        public bool IsConnected
+        {
+            get { return _isConnected; }
+            set { _isConnected = value; RaisePropertyChanged(nameof(IsConnected)); InvalidateRelayCommands(); }
+        }
+
+        private List<String> _ports;
+        /// <summary>
+        /// Gets or sets the available USB ports.
+        /// </summary>
+        public List<String> Ports
+        {
+            get { return _ports; }
+            set { _ports = value; RaisePropertyChanged(nameof(Ports)); }
+        }
+
+        private String _selectedPort;
+        /// <summary>
+        /// Gets or sets the selected USB port.
+        /// </summary>
+        public String SelectedPort
+        {
+            get { return _selectedPort; }
+            set { _selectedPort = value; RaisePropertyChanged(nameof(SelectedPort)); InvalidateRelayCommands(); }
+        }
+
+        private String _status;
+        /// <summary>
+        /// Gets or sets the current status bar text.
+        /// </summary>
+        public String Status
+        {
+            get { return _status; }
+            set { _status = value; RaisePropertyChanged(nameof(Status)); }
+        }
+
+        private bool _isRunning;
+        /// <summary>
+        /// Gets or sets a value indicating whether a stub is currently running.
+        /// </summary>
+        public bool IsRunning
+        {
+            get { return _isRunning; }
+            set { _isRunning = value; RaisePropertyChanged(nameof(IsRunning)); InvalidateRelayCommands(); }
+        }
+
+        #endregion
+
+        #region Commands
+
+        /// <summary>
+        /// Gets or sets the new command.
+        /// </summary>
+        public RelayCommand NewCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the close tab command.
+        /// </summary>
+        public RelayCommand<CodeTabVM> CloseTabCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the run command.
+        /// </summary>
+        public RelayCommand RunCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the stop command.
+        /// </summary>
+        public RelayCommand StopCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the toggle connection command.
+        /// </summary>
+        public RelayCommand ToggleConnectionCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the open command.
+        /// </summary>
+        public RelayCommand OpenCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the save command.
+        /// </summary>
+        public RelayCommand SaveCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the save as command.
+        /// </summary>
+        public RelayCommand SaveAsCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the stub snippet selected command.
+        /// </summary>
+        public RelayCommand StubSnippetSelectedCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the insert snippet command.
+        /// </summary>
+        public RelayCommand<String> InsertSnippetCommand { get; set; }
+
+        /// <summary>
+        /// Gets or sets the exit command.
+        /// </summary>
+        public RelayCommand ExitCommand { get; set; }
+
+        #endregion
+
+        #region Constructors
+
+        /// <summary>
+        /// Initializes a new instance of the <see cref="MainViewVM"/> class.
+        /// </summary>
+        public MainViewVM(IStudioApplicationManager applicationManager, INotificationProvider notification)
+        {
+            ApplicationManager = applicationManager;
+            _notification = notification;
+
+            CodeTabs = new ObservableCollection<CodeTabVM>();
+
+            NewCommand = new RelayCommand(CreateNewTab);
+            CloseTabCommand = new RelayCommand<CodeTabVM>(OnTabClosing);
+            RunCommand = new RelayCommand(RunTab, (x) => (IsConnected || UseConnectedMachine) && !IsRunning && SelectedCodeTab != null);
+            StopCommand = new RelayCommand(StopTab, (x) => (IsConnected || UseConnectedMachine) && IsRunning && SelectedCodeTab != null);
+            InsertSnippetCommand = new RelayCommand<string>((x) => { });
+
+            HighlightTypes = new ObservableCollection<KeyValuePair<string, Type>>();
+            HighlightTypes.Add(new KeyValuePair<string, Type>("stubManager", typeof(StubManager)));
+
+            StubSnippets = new ObservableCollection<StubSnippetVM>();
+
+            foreach (var stubType in StubBase.GetAvailableRequestStubs())
+            {
+                StubSnippetVM snippet = new StubSnippetVM();
+                snippet.Name = stubType.Name.Replace("Stub_", "");
+                snippet.Code = String.Format("stubManager.Run(\"{0}\" ,{1});", stubType.Name, String.Join(", ", stubType.GetProperties(BindingFlags.Public | BindingFlags.Instance).Select(x => x.PropertyType.Name == "string" ? "\"string\"" : x.PropertyType.Name)));
+                StubSnippets.Add(snippet);
+            }
+
+            ToggleConnectionCommand = new RelayCommand(ToggleConnection, (x) => !IsRunning);
+            OpenCommand = new RelayCommand(OpenFile);
+            SaveCommand = new RelayCommand(SaveFile);
+            SaveAsCommand = new RelayCommand(SaveAsFile);
+            StubSnippetSelectedCommand = new RelayCommand(OnStubSnippetSelected);
+            ExitCommand = new RelayCommand(() => Application.Current.Shutdown());
+
+            Ports = new List<string>()
+            {
+                "COM1",
+                "COM2",
+                "COM3",
+                "COM4",
+                "COM5",
+                "COM6",
+                "COM7",
+                "COM8",
+                "COM9",
+            };
+
+            SelectedPort = SettingsManager.Default.StubsUI.SelectedPort != null ? SettingsManager.Default.StubsUI.SelectedPort : Ports.First();
+
+            Status = "Ready";
+
+            if (SettingsManager.Default.StubsUI.LastTabs.Count > 0)
+            {
+                foreach (var file in SettingsManager.Default.StubsUI.LastTabs)
+                {
+                    if (File.Exists(file))
+                    {
+                        OpenFile(file);
+                    }
+                }
+            }
+            else
+            {
+                CreateNewTab();
+            }
+
+            Application.Current.Exit += Current_Exit;
+        }
+
+        #endregion
+
+        #region Virtual Methods
+
+        /// <summary>
+        /// Called when a stub snippet is double clicked.
+        /// </summary>
+        protected virtual void OnStubSnippetSelected()
+        {
+            if (SelectedStubSnippet != null)
+            {
+                if (InsertSnippetCommand != null)
+                {
+                    InsertSnippetCommand.Execute(SelectedStubSnippet.Code);
+                }
+            }
+        }
+
+        /// <summary>
+        /// Called when user closes a script tab.
+        /// </summary>
+        /// <param name="codeTab">The code tab.</param>
+        protected virtual void OnTabClosing(CodeTabVM codeTab)
+        {
+            CodeTabs.Remove(codeTab);
+        }
+
+        #endregion
+
+        #region Private Methods
+
+        /// <summary>
+        /// Saves the selected script file.
+        /// </summary>
+        private void SaveFile()
+        {
+            if (SelectedCodeTab != null)
+            {
+                if (SelectedCodeTab.File == null)
+                {
+                    SaveAsFile();
+                }
+                else
+                {
+                    File.WriteAllText(SelectedCodeTab.File, SelectedCodeTab.Code);
+                }
+            }
+        }
+
+        /// <summary>
+        /// Saves the selected script file.
+        /// </summary>
+        private void SaveAsFile()
+        {
+            if (SelectedCodeTab != null)
+            {
+                SaveFileDialog dlg = new SaveFileDialog();
+                dlg.Filter = "C# Script Files|*.cs";
+                dlg.DefaultExt = ".cs";
+                if (dlg.ShowDialog().Value)
+                {
+                    File.WriteAllText(dlg.FileName, SelectedCodeTab.Code);
+                    SelectedCodeTab.File = dlg.FileName;
+                }
+            }
+        }
+
+        /// <summary>
+        /// Opens a script from HD.
+        /// </summary>
+        private void OpenFile()
+        {
+            OpenFileDialog dlg = new OpenFileDialog();
+            dlg.Filter = "C# Script Files|*.cs";
+            dlg.Multiselect = true;
+            if (dlg.ShowDialog().Value)
+            {
+                foreach (var file in dlg.FileNames)
+                {
+                    OpenFile(file);
+                }
+            }
+        }
+
+        /// <summary>
+        /// Opens the file.
+        /// </summary>
+        /// <param name="file">The file.</param>
+        private void OpenFile(String file)
+        {
+            var newTab = new CodeTabVM();
+            newTab.File = file;
+            newTab.Code = File.ReadAllText(file);
+            CodeTabs.Add(newTab);
+            SelectedCodeTab = newTab;
+        }
+
+        /// <summary>
+        /// Toggles the USB adapter connection.
+        /// </summary>
+        private void ToggleConnection()
+        {
+            try
+            {
+                if (!IsConnected)
+                {
+                    _adapter = new UsbTransportAdapter(SelectedPort);
+                    _adapter.Connect().Wait();
+                    IsConnected = true;
+                }
+                else
+                {
+                    _adapter.Disconnect().Wait();
+                    IsConnected = false;
+                }
+            }
+            catch (Exception ex)
+            {
+                MessageBox.Show(ex.ToString(), "Tango");
+            }
+        }
+
+        /// <summary>
+        /// Creates a new script tab.
+        /// </summary>
+        private void CreateNewTab()
+        {
+            var newTab = new CodeTabVM();
+            CodeTabs.Add(newTab);
+            SelectedCodeTab = newTab;
+        }
+
+        /// <summary>
+        /// Runs the selected script tab.
+        /// </summary>
+        private async void RunTab()
+        {
+            if (UseConnectedMachine && !ApplicationManager.IsMachineConnected)
+            {
+                _notification.ShowError("Cannot execute stub while 'Connected Machine' is set but not machine connected.");
+                return;
+            }
+
+            IsRunning = true;
+            SelectedCodeTab.IsRunning = true;
+            Log = (DateTime.Now.ToTimeString() + ": ") + "Executing script '" + SelectedCodeTab.Title + "'..." + Environment.NewLine;
+
+            await Task.Factory.StartNew(async () =>
+            {
+                try
+                {
+                    ITransportAdapter adapter = _adapter;
+
+                    if (ApplicationManager.IsMachineConnected && UseConnectedMachine)
+                    {
+                        adapter = ApplicationManager.ConnectedMachine.Adapters.First();
+                    }
+
+                    _stubManager = new StubManager(adapter);
+                    var thisStubManager = _stubManager;
+                    _stubManager.Completed += Manager_Completed;
+                    _stubManager.Failed += Manager_Failed;
+                    _stubManager.Executed += Manager_Executed;
+
+                    ScriptEngine engine = new ScriptEngine(new StubOnExecuteParameters(_stubManager));
+
+                    engine.ReferencedAssemblies.Add(this.GetType());
+                    await engine.Run(SelectedCodeTab.Code);
+
+                    if (!thisStubManager.Aborted)
+                    {
+                        IsRunning = false;
+                        SelectedCodeTab.IsRunning = false;
+                    }
+                }
+                catch (Exception ex)
+                {
+                    IsRunning = false;
+                    SelectedCodeTab.IsRunning = false;
+                    MessageBox.Show(ex.Message, "Tango");
+                }
+            });
+        }
+
+        /// <summary>
+        /// Stops the currently current script.
+        /// </summary>
+        private void StopTab()
+        {
+            if (_stubManager != null)
+            {
+                _stubManager.Abort();
+                IsRunning = false;
+                SelectedCodeTab.IsRunning = false;
+                Status = "Stopped!";
+                Log += (DateTime.Now.ToTimeString() + ": ") + "Stopped!" + Environment.NewLine;
+            }
+        }
+
+        #endregion
+
+        #region Event Handlers
+
+        /// <summary>
+        /// Handled the <see cref="StubManager"/> Executed event.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="stubName">Name of the stub.</param>
+        private void Manager_Executed(object sender, string stubName)
+        {
+            Log += (DateTime.Now.ToTimeString() + ": ") + "Executing '" + stubName + "'..." + Environment.NewLine;
+            Status = "Executing " + stubName + "...";
+        }
+
+        /// <summary>
+        /// Handled the <see cref="StubManager"/> Failed event.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="ex">The exception.</param>
+        private void Manager_Failed(object sender, Exception ex)
+        {
+            if (IsRunning)
+            {
+                Log += (DateTime.Now.ToTimeString() + ": ") + ex.Message + Environment.NewLine;
+                Status = "Failed!";
+            }
+        }
+
+        /// <summary>
+        /// Handled the <see cref="StubManager"/> Completed event.
+        /// </summary>
+        /// <param name="sender">The sender.</param>
+        /// <param name="response">The response.</param>
+        private void Manager_Completed(object sender, string response)
+        {
+            Log += (DateTime.Now.ToTimeString() + ": ") + "Response Received:" + Environment.NewLine;
+            Log += (DateTime.Now.ToTimeString() + ": ") + response + Environment.NewLine;
+            Status = "Completed";
+        }
+
+        private void Current_Exit(object sender, ExitEventArgs e)
+        {
+            SettingsManager.Default.MachineStudio.StubsModule.SelectedPort = SelectedPort;
+            SettingsManager.Default.MachineStudio.StubsModule.LastTabs = CodeTabs.Select(x => x.File).ToList();
+            SettingsManager.SaveDefaultSettings();
+        }
+
+        #endregion
+    }
+}