如何在增量源代码生成初始化后步骤获取消费项目的命名空间？

你遇到的核心问题是：RegisterPostInitializationOutput属于生成器初始化阶段的逻辑，此时还未加载消费项目的语法/语义模型，无法直接获取项目的命名空间。要实现动态命名空间，需要调整生成逻辑的时机，以下是两种可行方案：

方案一：通过编译选项获取项目默认命名空间 #

消费项目的默认命名空间可以从编译配置中直接获取，我们可以把属性文件的生成逻辑绑定到编译信息上，替代硬编码的命名空间。

修改初始化方法 #

public void Initialize(IncrementalGeneratorInitializationContext context)
{
    // 获取项目的默认命名空间
    var defaultNamespace = context.CompilationProvider
        .Select((compilation, _) => compilation.Options.DefaultNamespace ?? string.Empty);

    // 基于默认命名空间动态生成属性文件
    context.RegisterSourceOutput(defaultNamespace, (spc, ns) =>
    {
        spc.AddSource("SetNTimesAttribute.g.cs", GetSetNTimesAttributeCode(ns));
    });

    // 保留原有pipeline逻辑
    var pipeline = context.SyntaxProvider    
        .CreateSyntaxProvider(SourcesAsString.SyntacticPredicate, SourcesAsString.SemanticTransform)
        .Where(found => found is not null && found.HasValue)
        .Collect()
        .Select(SourcesAsString.TransformType);

    context.RegisterSourceOutput(pipeline, SourcesAsString.Execute);
}

修改属性代码生成方法 #

让方法接收命名空间参数，动态替换硬编码内容：

public static string GetSetNTimesAttributeCode(string targetNamespace)
{
    // 若项目无默认命名空间，回退到原硬编码值
    var ns = string.IsNullOrEmpty(targetNamespace) 
        ? "SetOnceProperties.Sources.Utilities.Attributes" 
        : targetNamespace;
    
    return $@"
// <auto-generated/>

namespace {ns}
{{
    [AttributeUsage(AttributeTargets.Property)]
    internal class SetNTimesAttribute : Attribute
    {{
        public int MaximumSettable {{ get; }}

        public SetNTimesAttribute(int maximumSettable = 1)
        {{
            MaximumSettable = maximumSettable;
        }}
    }}

    [AttributeUsage(AttributeTargets.Property, Inherited = false)]
    internal sealed class SetOnceAttribute : SetNTimesAttribute
    {{ }}
}}";
}

方案二：根据属性使用位置动态生成命名空间 #

如果希望属性的命名空间和消费项目中实际使用它的类的命名空间一致，可以在语义分析阶段收集类的命名空间，再动态生成对应命名空间下的属性。

修改语义分析与生成逻辑 #

// 假设SemanticTransform返回包含类命名空间的自定义对象
public static YourCustomType SemanticTransform(GeneratorSyntaxContext context, CancellationToken ct)
{
    var classDecl = (ClassDeclarationSyntax)context.Node;
    var symbol = context.SemanticModel.GetDeclaredSymbol(classDecl, ct);
    if (symbol == null) return null;
    
    return new YourCustomType 
    { 
        TargetNamespace = symbol.ContainingNamespace.ToDisplayString(),
        // 其他需要的信息
    };
}

public void Initialize(IncrementalGeneratorInitializationContext context)
{
    // 原有pipeline逻辑
    var pipeline = context.SyntaxProvider    
        .CreateSyntaxProvider(SourcesAsString.SyntacticPredicate, SourcesAsString.SemanticTransform)
        .Where(found => found is not null && found.HasValue)
        .Collect();

    // 提取并去重所有使用属性的类的命名空间
    var uniqueNamespaces = pipeline
        .SelectMany(items => items.Select(item => item.TargetNamespace))
        .Distinct();

    // 为每个命名空间生成对应属性文件
    context.RegisterSourceOutput(uniqueNamespaces, (spc, ns) =>
    {
        // 用命名空间替换文件名中的特殊字符，避免冲突
        var fileName = $"SetNTimesAttribute_{ns.Replace('.', '_')}.g.cs";
        spc.AddSource(fileName, GetSetNTimesAttributeCode(ns));
    });

    // 保留原有SourceOutput逻辑
    context.RegisterSourceOutput(pipeline, SourcesAsString.Execute);
}

注意事项 #

1. 方案一适合绝大多数场景，直接复用项目默认命名空间，逻辑简单且无重复生成问题
2. 方案二适合需要属性与使用类同命名空间的场景，但要注意通过Distinct()去重，避免同一命名空间下重复生成属性文件
3. 不要在RegisterPostInitializationOutput中尝试获取项目命名空间，这个阶段没有可用的项目语法/语义信息

内容的提问来源于stack exchange，提问作者TRex