AssemblyNameProvider.cs 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  1. using System;
  2. using System.Collections.Generic;
  3. using System.Linq;
  4. using UnityEditor;
  5. using UnityEditor.Compilation;
  6. using UnityEditor.PackageManager;
  7. namespace Packages.Rider.Editor.ProjectGeneration
  8. {
  9. internal class AssemblyNameProvider : IAssemblyNameProvider
  10. {
  11. ProjectGenerationFlag m_ProjectGenerationFlag = (ProjectGenerationFlag)EditorPrefs.GetInt("unity_project_generation_flag", 3);
  12. public string[] ProjectSupportedExtensions => EditorSettings.projectGenerationUserExtensions;
  13. public string ProjectGenerationRootNamespace => EditorSettings.projectGenerationRootNamespace;
  14. public ProjectGenerationFlag ProjectGenerationFlag
  15. {
  16. get => m_ProjectGenerationFlag;
  17. private set
  18. {
  19. EditorPrefs.SetInt("unity_project_generation_flag", (int)value);
  20. m_ProjectGenerationFlag = value;
  21. }
  22. }
  23. public string GetAssemblyNameFromScriptPath(string path)
  24. {
  25. return CompilationPipeline.GetAssemblyNameFromScriptPath(path);
  26. }
  27. public IEnumerable<Assembly> GetAssemblies(Func<string, bool> shouldFileBePartOfSolution)
  28. {
  29. foreach (var assembly in CompilationPipeline.GetAssemblies())
  30. {
  31. if (assembly.sourceFiles.Any(shouldFileBePartOfSolution))
  32. {
  33. var options = new ScriptCompilerOptions()
  34. {
  35. ResponseFiles = assembly.compilerOptions.ResponseFiles,
  36. AllowUnsafeCode = assembly.compilerOptions.AllowUnsafeCode,
  37. ApiCompatibilityLevel = assembly.compilerOptions.ApiCompatibilityLevel
  38. };
  39. yield return new Assembly(assembly.name, "Temp\\Bin\\Debug\\",
  40. assembly.sourceFiles,
  41. new[] { "DEBUG", "TRACE" }.Concat(assembly.defines).Concat(EditorUserBuildSettings.activeScriptCompilationDefines).ToArray(),
  42. assembly.assemblyReferences,
  43. assembly.compiledAssemblyReferences,
  44. assembly.flags,
  45. #if UNITY_2020_2_OR_NEWER
  46. options,
  47. assembly.rootNamespace);
  48. #else
  49. options);
  50. #endif
  51. }
  52. }
  53. if (ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.PlayerAssemblies))
  54. {
  55. foreach (var assembly in CompilationPipeline.GetAssemblies(AssembliesType.Player).Where(assembly => assembly.sourceFiles.Any(shouldFileBePartOfSolution)))
  56. {
  57. var options = new ScriptCompilerOptions()
  58. {
  59. ResponseFiles = assembly.compilerOptions.ResponseFiles,
  60. AllowUnsafeCode = assembly.compilerOptions.AllowUnsafeCode,
  61. ApiCompatibilityLevel = assembly.compilerOptions.ApiCompatibilityLevel
  62. };
  63. yield return new Assembly(assembly.name, "Temp\\Bin\\Debug\\Player\\", assembly.sourceFiles,
  64. new[] { "DEBUG", "TRACE" }.Concat(assembly.defines).ToArray(),
  65. assembly.assemblyReferences,
  66. assembly.compiledAssemblyReferences,
  67. assembly.flags,
  68. #if UNITY_2020_2_OR_NEWER
  69. options,
  70. assembly.rootNamespace);
  71. #else
  72. options);
  73. #endif
  74. }
  75. }
  76. }
  77. public string GetProjectName(string assemblyOutputPath, string assemblyName)
  78. {
  79. return assemblyOutputPath.EndsWith("\\Player\\", StringComparison.Ordinal) ? assemblyName + ".Player" : assemblyName;
  80. }
  81. public IEnumerable<string> GetAllAssetPaths()
  82. {
  83. return AssetDatabase.GetAllAssetPaths();
  84. }
  85. public UnityEditor.PackageManager.PackageInfo FindForAssetPath(string assetPath)
  86. {
  87. return UnityEditor.PackageManager.PackageInfo.FindForAssetPath(assetPath);
  88. }
  89. public bool IsInternalizedPackagePath(string path)
  90. {
  91. if (string.IsNullOrEmpty(path.Trim()))
  92. {
  93. return false;
  94. }
  95. var packageInfo = FindForAssetPath(path);
  96. if (packageInfo == null)
  97. {
  98. return false;
  99. }
  100. var packageSource = packageInfo.source;
  101. switch (packageSource)
  102. {
  103. case PackageSource.Embedded:
  104. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Embedded);
  105. case PackageSource.Registry:
  106. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Registry);
  107. case PackageSource.BuiltIn:
  108. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.BuiltIn);
  109. case PackageSource.Unknown:
  110. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Unknown);
  111. case PackageSource.Local:
  112. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Local);
  113. case PackageSource.Git:
  114. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Git);
  115. #if UNITY_2019_3_OR_NEWER
  116. case PackageSource.LocalTarball:
  117. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.LocalTarBall);
  118. #endif
  119. }
  120. return false;
  121. }
  122. public ResponseFileData ParseResponseFile(string responseFilePath, string projectDirectory, string[] systemReferenceDirectories)
  123. {
  124. return CompilationPipeline.ParseResponseFile(
  125. responseFilePath,
  126. projectDirectory,
  127. systemReferenceDirectories
  128. );
  129. }
  130. public IEnumerable<string> GetRoslynAnalyzerPaths()
  131. {
  132. return PluginImporter.GetAllImporters()
  133. .Where(i => !i.isNativePlugin && AssetDatabase.GetLabels(i).SingleOrDefault(l => l == "RoslynAnalyzer") != null)
  134. .Select(i => i.assetPath);
  135. }
  136. public void ToggleProjectGeneration(ProjectGenerationFlag preference)
  137. {
  138. if (ProjectGenerationFlag.HasFlag(preference))
  139. {
  140. ProjectGenerationFlag ^= preference;
  141. }
  142. else
  143. {
  144. ProjectGenerationFlag |= preference;
  145. }
  146. }
  147. public void ResetProjectGenerationFlag()
  148. {
  149. ProjectGenerationFlag = ProjectGenerationFlag.None;
  150. }
  151. }
  152. }