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.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 { public class DefaultStudioModuleLoader : ExtendedObject, IStudioModuleLoader { private IAuthenticationProvider _authenticationProvider; private bool _loaded; public DefaultStudioModuleLoader(IAuthenticationProvider authenticationProvider) { _authenticationProvider = authenticationProvider; AllModules = new ObservableCollection(); UserModules = new ObservableCollection(); _authenticationProvider.CurrentUserChanged += _authenticationProvider_CurrentUserChanged; } private void _authenticationProvider_CurrentUserChanged(object sender, DAL.Observables.User e) { LoadModules(); } private ObservableCollection _allModules; public ObservableCollection AllModules { get { return _allModules; } private set { _allModules = value; RaisePropertyChangedAuto(); } } private ObservableCollection _userModules; public ObservableCollection UserModules { get { return _userModules; } private set { _userModules = value; RaisePropertyChangedAuto(); } } 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))) { if (!AllModules.ToList().Exists(x => x.GetType() == moduleType)) { var module = Activator.CreateInstance(moduleType) as IStudioModule; AllModules.Add(module); } } } } catch { } } _loaded = true; } UserModules.Clear(); if (_authenticationProvider.CurrentUser != null) { UserModules = AllModules.Where(x => _authenticationProvider.CurrentUser.HasPermission(x.Permission)).ToObservableCollection(); } } } }