ShrlAlgo/RevitGen
1
0
Fork 0
mirror of https://github.com/ShrlAlgo/RevitGen.git synced 2026-03-07 17:28:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

整理代码

Browse Source
This commit is contained in:
GG Z
2025-09-03 22:11:56 +08:00
parent da46257be2
commit b25f8d7ec9
4 changed files with 16 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
RevitGen.Generator/RevitCommandGenerator.cs
View File

@@ -10,7 +10,6 @@ namespace RevitGen.Generator
[Generator(LanguageNames.CSharp)]
public class RevitCommandGenerator : ISourceGenerator
{
// ★★ 1. 修正常量，使用 RevitCommandAttribute ★★
private const string RevitCommandAttributeFullName = "RevitGen.Attributes.RevitCommandAttribute";
private const string CommandHandlerAttributeFullName = "RevitGen.Attributes.CommandHandlerAttribute";
@@ -27,18 +26,15 @@ namespace RevitGen.Generator
if (!(context.SyntaxReceiver is SyntaxReceiver receiver))
{
// ... [日志和返回逻辑不变] ...
return;
}
log.AppendLine($"// Candidate classes found by SyntaxReceiver: {receiver.CandidateClasses.Count}");
if (receiver.CandidateClasses.Count == 0)
{
// ... [日志和返回逻辑不变] ...
return;
}
// ★★ 2. 确保我们查找的是正确的特性符号 ★★
var attributeSymbol = context.Compilation.GetTypeByMetadataName(RevitCommandAttributeFullName);
if (attributeSymbol == null)
{
@@ -80,7 +76,6 @@ namespace RevitGen.Generator
if (commandClasses.Any())
{
// ... [生成 partial 类和 App 类的逻辑不变] ...
foreach (var classSymbol in commandClasses)
{
var partialClassSource = SourceGenerationHelper.GenerateCommandPartialClass(classSymbol);

1
RevitGen.Generator/SourceGenerationHelper.cs
View File

@@ -12,7 +12,6 @@ namespace RevitGen.Generator
/// </summary>
internal static class SourceGenerationHelper
{
// 定义我们属性的完整名称，以使用最终确定的、正确的命名方案。
private const string RevitCommandAttributeFullName = "RevitGen.Attributes.RevitCommandAttribute";
private const string CommandHandlerAttributeFullName = "RevitGen.Attributes.CommandHandlerAttribute";

30
RevitGen.Generator/SyntaxReceiver.cs
View File

@@ -17,24 +17,24 @@ namespace RevitGen.Generator
public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
{
// 检查节点是否是一个类声明，并且它带有属性
//if (syntaxNode is ClassDeclarationSyntax classDeclarationSyntax &&
// classDeclarationSyntax.AttributeLists.Count > 0)
//{
// // 检查类是否被声明为 partial
// foreach (var modifier in classDeclarationSyntax.Modifiers)
// {
// if (modifier.ValueText == "partial")
// {
// CandidateClasses.Add(classDeclarationSyntax);
// break;
// }
// }
//}
if (syntaxNode is ClassDeclarationSyntax classDeclarationSyntax &&
classDeclarationSyntax.AttributeLists.Count > 0)
classDeclarationSyntax.AttributeLists.Count > 0)
{
CandidateClasses.Add(classDeclarationSyntax);
// 检查类是否被声明为 partial
foreach (var modifier in classDeclarationSyntax.Modifiers)
{
if (modifier.ValueText == "partial")
{
CandidateClasses.Add(classDeclarationSyntax);
break;
}
}
}
//if (syntaxNode is ClassDeclarationSyntax classDeclarationSyntax &&
// classDeclarationSyntax.AttributeLists.Count > 0)
//{
// CandidateClasses.Add(classDeclarationSyntax);
//}
}
}
}

7
RevitGenTest/RevitAddin.cs
View File

@@ -1,10 +1,5 @@
using System;
using System.Drawing;
using Autodesk.Revit.Attributes;
using Autodesk.Revit.DB;
using Autodesk.Revit.DB;
using Autodesk.Revit.UI;
using Autodesk.Revit.UI.Selection;
using RevitGen.Attributes;