using GalaSoft.MvvmLight.Ioc; 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 Tango.Core; using Tango.BL.Entities; using Tango.Logging; using Tango.MachineStudio.Common; using Tango.MachineStudio.Common.Authentication; using Tango.MachineStudio.Common.Modules; using Tango.MachineStudio.DB; using Tango.MachineStudio.MachineDesigner; using Tango.MachineStudio.Stubs; namespace Tango.MachineStudio.UI.Modules { /// /// Represents the Machine Studio default module loader. /// /// /// public class DefaultStudioModuleLoader : ExtendedObject, IStudioModuleLoader { private IAuthenticationProvider _authenticationProvider; private bool _loaded; public event EventHandler ModulesLoaded; /// /// Initializes a new instance of the class. /// /// The authentication provider. public DefaultStudioModuleLoader(IAuthenticationProvider authenticationProvider) { _authenticationProvider = authenticationProvider; AllModules = new ObservableCollection(); UserModules = new ObservableCollection(); _authenticationProvider.CurrentUserChanged += _authenticationProvider_CurrentUserChanged; } /// /// Handles the authentication provider user changed event. /// /// The sender. /// The e. private void _authenticationProvider_CurrentUserChanged(object sender, User e) { LoadModules(); } private ObservableCollection _allModules; /// /// Gets all loaded modules. /// public ObservableCollection AllModules { get { return _allModules; } private set { _allModules = value; RaisePropertyChangedAuto(); } } private ObservableCollection _userModules; /// /// Gets all the user permitted modules. /// public ObservableCollection UserModules { get { return _userModules; } private set { _userModules = value; RaisePropertyChangedAuto(); } } /// /// Loads all available Machine Studio modules. /// public void LoadModules() { if (!_loaded) { //Preloaded AllModules.Add(new StubsModule()); AllModules.Add(new DBModule()); AllModules.Add(new MachineDesignerModule()); //Preloaded AllModules.Clear(); string assemblyFolder = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location); foreach (var file in Directory.GetFiles(assemblyFolder, "*.dll").Where(x => x.Contains("MachineStudio"))) { try { Assembly moduleAssembly = null; moduleAssembly = Assembly.LoadFrom(file); if (moduleAssembly != null) { foreach (var moduleType in moduleAssembly.GetTypes().Where(x => !x.IsInterface && typeof(IStudioModule).IsAssignableFrom(x) && !x.IsAbstract)) { if (!AllModules.ToList().Exists(x => x.GetType() == moduleType)) { try { var module = Activator.CreateInstance(moduleType) as IStudioModule; AllModules.Add(module); } catch (Exception ex) { LogManager.Log(ex, "Could not load module " + moduleType.Name); } } } } } catch { } } _loaded = true; } UserModules.Clear(); if (_authenticationProvider.CurrentUser != null) { UserModules = AllModules.Where(x => _authenticationProvider.CurrentUser.HasPermission(x.Permission)).ToObservableCollection(); } ModulesLoaded?.Invoke(this, new EventArgs()); } /// /// Gets the studio module of type T if loaded. /// /// /// public T GetStudioModule() where T : IStudioModule { return UserModules.OfType().FirstOrDefault(); } } }