102 lines
3.5 KiB
C#
102 lines
3.5 KiB
C#
using System.Linq;
|
|
using Microsoft.CodeAnalysis;
|
|
|
|
namespace Fantasy.SourceGenerator.Common
|
|
{
|
|
/// <summary>
|
|
/// 编译环境检测帮助类
|
|
/// 提供检测 Fantasy 框架、Unity 环境等公共方法
|
|
/// </summary>
|
|
public static class CompilationHelper
|
|
{
|
|
/// <summary>
|
|
/// 检查是否定义了 Fantasy 框架的预编译符号
|
|
/// 只有定义了 FANTASY_NET的项目才会生成代码
|
|
/// </summary>
|
|
public static bool HasFantasyNETDefine(Compilation compilation)
|
|
{
|
|
// 遍历所有语法树的预处理符号
|
|
foreach (var tree in compilation.SyntaxTrees)
|
|
{
|
|
var defines = tree.Options.PreprocessorSymbolNames;
|
|
if (defines.Contains("FANTASY_NET"))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 检查是否定义了 Fantasy 框架的预编译符号
|
|
/// 只有定义了 FANTASY_UNITY的项目才会生成代码
|
|
/// </summary>
|
|
public static bool HasFantasyUNITYDefine(Compilation compilation)
|
|
{
|
|
// 遍历所有语法树的预处理符号
|
|
foreach (var tree in compilation.SyntaxTrees)
|
|
{
|
|
var defines = tree.Options.PreprocessorSymbolNames;
|
|
if (defines.Contains("FANTASY_UNITY"))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 检查是否定义了 Fantasy 框架的预编译符号
|
|
/// 只有定义了 FANTASY_NET 或 FANTASY_UNITY 的项目才会生成代码
|
|
/// </summary>
|
|
public static bool HasFantasyDefine(Compilation compilation)
|
|
{
|
|
// 遍历所有语法树的预处理符号
|
|
foreach (var tree in compilation.SyntaxTrees)
|
|
{
|
|
var defines = tree.Options.PreprocessorSymbolNames;
|
|
if (defines.Contains("FANTASY_NET") || defines.Contains("FANTASY_UNITY"))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/// <summary>
|
|
/// 检测是否是 Unity 编译环境
|
|
/// 优先检查是否引用了 UnityEngine 核心程序集或类型，这是最可靠的方式
|
|
/// </summary>
|
|
public static bool IsUnityCompilation(Compilation compilation)
|
|
{
|
|
// 方法1: 检查是否引用了 UnityEngine.CoreModule 或 UnityEngine 程序集（最可靠）
|
|
foreach (var reference in compilation.References)
|
|
{
|
|
if (reference is PortableExecutableReference peRef)
|
|
{
|
|
var display = peRef.Display ?? "";
|
|
// Unity 2017+ 使用模块化程序集
|
|
if (display.Contains("UnityEngine.CoreModule.dll") ||
|
|
display.Contains("UnityEngine.dll"))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
// 方法2: 检查是否能够找到 UnityEngine 命名空间的核心类型
|
|
var unityMonoBehaviour = compilation.GetTypeByMetadataName("UnityEngine.MonoBehaviour");
|
|
var unityGameObject = compilation.GetTypeByMetadataName("UnityEngine.GameObject");
|
|
if (unityMonoBehaviour != null || unityGameObject != null)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
}
|