AssemblyNameProvider.cs 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  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 VSCodeEditor
  8. {
  9. public interface IAssemblyNameProvider
  10. {
  11. string[] ProjectSupportedExtensions { get; }
  12. ProjectGenerationFlag ProjectGenerationFlag { get; }
  13. string GetAssemblyNameFromScriptPath(string path);
  14. IEnumerable<Assembly> GetAssemblies(Func<string, bool> shouldFileBePartOfSolution);
  15. IEnumerable<string> GetAllAssetPaths();
  16. IEnumerable<string> GetRoslynAnalyzerPaths();
  17. UnityEditor.PackageManager.PackageInfo FindForAssetPath(string assetPath);
  18. ResponseFileData ParseResponseFile(string responseFilePath, string projectDirectory, string[] systemReferenceDirectories);
  19. bool IsInternalizedPackagePath(string path);
  20. void ToggleProjectGeneration(ProjectGenerationFlag preference);
  21. }
  22. internal class AssemblyNameProvider : IAssemblyNameProvider
  23. {
  24. ProjectGenerationFlag m_ProjectGenerationFlag = (ProjectGenerationFlag)EditorPrefs.GetInt("unity_project_generation_flag", 0);
  25. public string[] ProjectSupportedExtensions => EditorSettings.projectGenerationUserExtensions;
  26. public ProjectGenerationFlag ProjectGenerationFlag
  27. {
  28. get => m_ProjectGenerationFlag;
  29. private set
  30. {
  31. EditorPrefs.SetInt("unity_project_generation_flag", (int)value);
  32. m_ProjectGenerationFlag = value;
  33. }
  34. }
  35. public string GetAssemblyNameFromScriptPath(string path)
  36. {
  37. return CompilationPipeline.GetAssemblyNameFromScriptPath(path);
  38. }
  39. public IEnumerable<Assembly> GetAssemblies(Func<string, bool> shouldFileBePartOfSolution)
  40. {
  41. return CompilationPipeline.GetAssemblies()
  42. .Where(i => 0 < i.sourceFiles.Length && i.sourceFiles.Any(shouldFileBePartOfSolution));
  43. }
  44. public IEnumerable<string> GetAllAssetPaths()
  45. {
  46. return AssetDatabase.GetAllAssetPaths();
  47. }
  48. public UnityEditor.PackageManager.PackageInfo FindForAssetPath(string assetPath)
  49. {
  50. return UnityEditor.PackageManager.PackageInfo.FindForAssetPath(assetPath);
  51. }
  52. public ResponseFileData ParseResponseFile(string responseFilePath, string projectDirectory, string[] systemReferenceDirectories)
  53. {
  54. return CompilationPipeline.ParseResponseFile(
  55. responseFilePath,
  56. projectDirectory,
  57. systemReferenceDirectories
  58. );
  59. }
  60. public bool IsInternalizedPackagePath(string path)
  61. {
  62. if (string.IsNullOrWhiteSpace(path))
  63. {
  64. return false;
  65. }
  66. var packageInfo = FindForAssetPath(path);
  67. if (packageInfo == null)
  68. {
  69. return false;
  70. }
  71. var packageSource = packageInfo.source;
  72. switch (packageSource)
  73. {
  74. case PackageSource.Embedded:
  75. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Embedded);
  76. case PackageSource.Registry:
  77. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Registry);
  78. case PackageSource.BuiltIn:
  79. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.BuiltIn);
  80. case PackageSource.Unknown:
  81. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Unknown);
  82. case PackageSource.Local:
  83. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Local);
  84. case PackageSource.Git:
  85. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.Git);
  86. #if UNITY_2019_3_OR_NEWER
  87. case PackageSource.LocalTarball:
  88. return !ProjectGenerationFlag.HasFlag(ProjectGenerationFlag.LocalTarBall);
  89. #endif
  90. }
  91. return false;
  92. }
  93. public void ToggleProjectGeneration(ProjectGenerationFlag preference)
  94. {
  95. if (ProjectGenerationFlag.HasFlag(preference))
  96. {
  97. ProjectGenerationFlag ^= preference;
  98. }
  99. else
  100. {
  101. ProjectGenerationFlag |= preference;
  102. }
  103. }
  104. public IEnumerable<string> GetRoslynAnalyzerPaths()
  105. {
  106. return PluginImporter.GetAllImporters()
  107. .Where(i => !i.isNativePlugin && AssetDatabase.GetLabels(i).SingleOrDefault(l => l == "RoslynAnalyzer") != null)
  108. .Select(i => i.assetPath);
  109. }
  110. }
  111. }