aboutsummaryrefslogtreecommitdiffstats
path: root/Software/Visual_Studio/MachineStudio/Tango.MachineStudio.UI/Modules/DefaultStudioModuleLoader.cs
blob: 28f56e521b22becec1a162dd738654f70517eec6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
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;

namespace Tango.MachineStudio.UI.Modules
{
    /// <summary>
    /// Represents the Machine Studio default <see cref="IStudioModuleLoader">module loader</see>.
    /// </summary>
    /// <seealso cref="Tango.Core.ExtendedObject" />
    /// <seealso cref="Tango.MachineStudio.Common.Modules.IStudioModuleLoader" />
    public class DefaultStudioModuleLoader : ExtendedObject, IStudioModuleLoader
    {
        private IAuthenticationProvider _authenticationProvider;
        private bool _loaded;

        /// <summary>
        /// Initializes a new instance of the <see cref="DefaultStudioModuleLoader"/> class.
        /// </summary>
        /// <param name="authenticationProvider">The authentication provider.</param>
        public DefaultStudioModuleLoader(IAuthenticationProvider authenticationProvider)
        {
            _authenticationProvider = authenticationProvider;
            AllModules = new SynchronizedObservableCollection<IStudioModule>();
            UserModules = new SynchronizedObservableCollection<IStudioModule>();
            _authenticationProvider.CurrentUserChanged += _authenticationProvider_CurrentUserChanged;
        }

        /// <summary>
        /// Handles the authentication provider user changed event.
        /// </summary>
        /// <param name="sender">The sender.</param>
        /// <param name="e">The e.</param>
        private void _authenticationProvider_CurrentUserChanged(object sender, User e)
        {
            if (e != null)
            {
                LoadModules();
            }
        }

        private SynchronizedObservableCollection<IStudioModule> _allModules;
        /// <summary>
        /// Gets all loaded modules.
        /// </summary>
        public SynchronizedObservableCollection<IStudioModule> AllModules
        {
            get { return _allModules; }
            private set { _allModules = value; RaisePropertyChangedAuto(); }
        }

        private SynchronizedObservableCollection<IStudioModule> _userModules;
        /// <summary>
        /// Gets all the user permitted modules.
        /// </summary>
        public SynchronizedObservableCollection<IStudioModule> UserModules
        {
            get { return _userModules; }
            private set { _userModules = value; RaisePropertyChangedAuto(); }
        }

        /// <summary>
        /// Loads all available Machine Studio modules.
        /// </summary>
        public void LoadModules()
        {
            if (!_loaded)
            {
                ////Preloaded

                //LogManager.Log(String.Format("Loading module '{0}'...", nameof(StubsModule)));
                //AllModules.Add(new StubsModule());
                //LogManager.Log(String.Format("Loading module '{0}'...", nameof(DBModule)));
                //AllModules.Add(new DBModule());
                //LogManager.Log(String.Format("Loading module '{0}'...", nameof(MachineDesignerModule)));
                //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
                                    {
                                        LogManager.Log(String.Format("Loading module '{0}'...", moduleType.Name));
                                        var module = Activator.CreateInstance(moduleType) as IStudioModule;
                                        //SimpleIoc.Default.Register(() => module);
                                        AllModules.Add(module);
                                    }
                                    catch (Exception ex)
                                    {
                                        LogManager.Log(ex, "Could not load module " + moduleType.Name);
                                    }
                                }
                            }
                        }
                    }
                    catch { }
                }

                _loaded = true;
            }

            AllModules = AllModules.OrderBy(x => x.GetType().GetCustomAttribute<StudioModuleAttribute>().Index).ToSynchronizedObservableCollection();

            UserModules.Clear();

            if (_authenticationProvider.CurrentUser != null)
            {
                UserModules = AllModules.Where(x => _authenticationProvider.CurrentUser.HasPermission(x.Permission)).ToSynchronizedObservableCollection();
            }

            ModulesLoaded?.Invoke(this, new EventArgs());
        }

        /// <summary>
        /// Gets the studio module of type T if loaded.
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <returns></returns>
        public T GetStudioModule<T>() where T : IStudioModule
        {
            return UserModules.OfType<T>().FirstOrDefault();
        }

        /// <summary>
        /// Occurs when all modules are initialized.
        /// </summary>
        public event EventHandler ModulesLoaded;
    }
}