// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
using System.Collections.Immutable;
 
namespace Microsoft.AspNetCore.Razor.Language.CodeGeneration;
 
public abstract class CodeTarget(RazorCodeDocument codeDocument, ImmutableArray<ICodeTargetExtension> targetExtensions)
{
    public RazorCodeDocument CodeDocument => codeDocument;
    public RazorCodeGenerationOptions Options => codeDocument.CodeGenerationOptions;
    public ImmutableArray<ICodeTargetExtension> Extensions => targetExtensions;
 
    public static CodeTarget CreateDefault(
        RazorCodeDocument codeDocument,
        Action<CodeTargetBuilder>? configure = null)
    {
        ArgHelper.ThrowIfNull(codeDocument);
 
        var builder = new DefaultCodeTargetBuilder(codeDocument);
        configure?.Invoke(builder);
        return builder.Build();
    }
 
    public abstract IntermediateNodeWriter CreateNodeWriter();
 
    public TExtension? GetExtension<TExtension>()
        where TExtension : class, ICodeTargetExtension
    {
        foreach (var extension in Extensions)
        {
            if (extension is TExtension match)
            {
                return match;
            }
        }
 
        return null;
    }
 
    public bool HasExtension<TExtension>()
        where TExtension : class, ICodeTargetExtension
    {
        foreach (var extension in Extensions)
        {
            if (extension is TExtension)
            {
                return true;
            }
        }
 
        return false;
    }
}