IwbZeroModule.cs 6.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. using System.Collections.Generic;
  2. using System.Linq;
  3. using System.Reflection;
  4. using System.Web.Http.Filters;
  5. using System.Web.Mvc;
  6. using Abp;
  7. using Abp.Collections.Extensions;
  8. using Abp.Configuration.Startup;
  9. using Abp.Dependency;
  10. using Abp.Localization;
  11. using Abp.Localization.Dictionaries;
  12. using Abp.Localization.Dictionaries.Xml;
  13. using Abp.Modules;
  14. using Abp.MultiTenancy;
  15. using Abp.Reflection;
  16. using Abp.Reflection.Extensions;
  17. using Abp.Web.Mvc;
  18. using Abp.Web.Mvc.Security.AntiForgery;
  19. using Abp.WebApi;
  20. using Abp.WebApi.Configuration;
  21. using Abp.WebApi.Security.AntiForgery;
  22. using IwbZero.AntiForgery;
  23. using IwbZero.Authorization.Base.Roles;
  24. using IwbZero.Authorization.Base.SystemInfo;
  25. using IwbZero.Authorization.Base.Users;
  26. using IwbZero.MultiTenancy;
  27. using IwbZero.ToolCommon.LogHelpers;
  28. using IwbZero.ToolCommon.StringModel;
  29. using IwbZero.Zero.Configuration;
  30. using NUglify.Helpers;
  31. namespace IwbZero
  32. {
  33. [DependsOn(typeof(AbpKernelModule),typeof(AbpWebMvcModule),typeof(AbpWebApiModule))]
  34. public class IwbZeroModule : AbpModule
  35. {
  36. public override void PreInitialize()
  37. {
  38. Configuration.BackgroundJobs.IsJobExecutionEnabled = true;
  39. IocManager.RegisterIfNot<IIwbZeroEntityTypes, IwbZeroEntityTypes>(); //Registered on services.AddAbpIdentity() for Abp.ZeroCore.
  40. IocManager.Register<IRoleManagementConfig, RoleManagementConfig>();
  41. IocManager.Register<IUserManagementConfig, UserManagementConfig>();
  42. IocManager.Register<ILanguageManagementConfig, LanguageManagementConfig>();
  43. IocManager.Register<IIwbZeroConfig, IwbZeroConfig>();
  44. Configuration.ReplaceService<ITenantStore, TenantStore>(DependencyLifeStyle.Transient);
  45. Configuration.Settings.Providers.Add<IwbZeroSettingProvider>();
  46. foreach (var languageInfo in Configuration.Localization.Languages)
  47. {
  48. Configuration.Localization.Languages.Remove(languageInfo);
  49. }
  50. Configuration.Localization.Languages.Add(new LanguageInfo("zh-Hans", "简体中文", "cn", true));
  51. Configuration.Localization.Languages.Add(new LanguageInfo("en", "English", "us"));
  52. Configuration.Localization.Sources.Add(
  53. new DictionaryBasedLocalizationSource(
  54. IwbZeroConsts.LocalizationSourceName,
  55. new XmlEmbeddedFileLocalizationDictionaryProvider(
  56. typeof(IwbZeroModule).GetAssembly(), "IwbZero.Zero.Localization.Source"
  57. )));
  58. AddIgnoredTypes();
  59. }
  60. public override void Initialize()
  61. {
  62. FillMissingEntityTypes();
  63. IocManager.RegisterAssemblyByConvention(typeof(IwbZeroModule).GetAssembly());
  64. RegisterTenantCache();
  65. }
  66. public override void PostInitialize()
  67. {
  68. var list = new List<Filter>();
  69. GlobalFilters.Filters.ForEach((f) =>
  70. {
  71. if (f.Instance.GetType() != typeof(AbpAntiForgeryMvcFilter))
  72. {
  73. list.Add(f);
  74. }
  75. });
  76. GlobalFilters.Filters.Clear();
  77. foreach (var filter in list)
  78. {
  79. GlobalFilters.Filters.Add(filter.Instance);
  80. }
  81. //GlobalFilters.Filters.Remove(IocManager.Resolve<AbpAntiForgeryMvcFilter>());
  82. GlobalFilters.Filters.Add(IocManager.Resolve<IwbAntiForgeryMvcFilter>());
  83. this.LogInfo("【MVC-Filters】");
  84. this.LogInfo(GlobalFilters.Filters.Obj2String());
  85. var httpConfiguration = IocManager.Resolve<IAbpWebApiConfiguration>().HttpConfiguration;
  86. var list2 = new List<IFilter>();
  87. //httpConfiguration.Filters.Remove(IocManager.Resolve<AbpAntiForgeryApiFilter>());
  88. httpConfiguration.Filters.ForEach((f) =>
  89. {
  90. if (f.Instance.GetType() != typeof(AbpAntiForgeryApiFilter))
  91. {
  92. list2.Add(f.Instance);
  93. }
  94. });
  95. httpConfiguration.Filters.Clear();
  96. foreach (var filter in list2)
  97. {
  98. httpConfiguration.Filters.Add(filter);
  99. }
  100. httpConfiguration.Filters.Add(IocManager.Resolve<IwbAntiForgeryApiFilter>());
  101. this.LogInfo("【API-Filters】");
  102. this.LogInfo(httpConfiguration.Filters.Obj2String());
  103. }
  104. private void AddIgnoredTypes()
  105. {
  106. var ignoredTypes = new[]
  107. {
  108. typeof(SysLog)
  109. };
  110. foreach (var ignoredType in ignoredTypes)
  111. {
  112. Configuration.EntityHistory.IgnoredTypes.AddIfNotContains(ignoredType);
  113. }
  114. }
  115. private void FillMissingEntityTypes()
  116. {
  117. using (var entityTypes = IocManager.ResolveAsDisposable<IIwbZeroEntityTypes>())
  118. {
  119. if (entityTypes.Object.User != null &&
  120. entityTypes.Object.Role != null &&
  121. entityTypes.Object.Tenant != null)
  122. {
  123. return;
  124. }
  125. using (var typeFinder = IocManager.ResolveAsDisposable<ITypeFinder>())
  126. {
  127. var types = typeFinder.Object.FindAll();
  128. entityTypes.Object.Tenant = types.FirstOrDefault(t => typeof(TenantBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  129. entityTypes.Object.Role = types.FirstOrDefault(t => typeof(RoleBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  130. entityTypes.Object.User = types.FirstOrDefault(t => typeof(UserBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  131. entityTypes.Object.Function = types.FirstOrDefault(t => typeof(SysFunctionBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  132. entityTypes.Object.State = types.FirstOrDefault(t => typeof(SysStateBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  133. entityTypes.Object.AttachFile = types.FirstOrDefault(t => typeof(SysAttachFileBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  134. entityTypes.Object.Help = types.FirstOrDefault(t => typeof(SysHelpBase).IsAssignableFrom(t) && !t.GetTypeInfo().IsAbstract);
  135. }
  136. }
  137. }
  138. private void RegisterTenantCache()
  139. {
  140. if (IocManager.IsRegistered<ITenantCache>())
  141. {
  142. return;
  143. }
  144. using (var entityTypes = IocManager.ResolveAsDisposable<IIwbZeroEntityTypes>())
  145. {
  146. var implType = typeof(TenantCache<,>)
  147. .MakeGenericType(entityTypes.Object.Tenant, entityTypes.Object.User);
  148. IocManager.Register(typeof(ITenantCache), implType, DependencyLifeStyle.Transient);
  149. }
  150. }
  151. }
  152. }