// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Threading;
using Microsoft.CodeAnalysis.CSharp.Syntax;
using Microsoft.CodeAnalysis.PooledObjects;
using Microsoft.CodeAnalysis.Structure;
using Microsoft.CodeAnalysis.Text;

namespace Microsoft.CodeAnalysis.CSharp.Structure;

internal sealed class SwitchStatementStructureProvider : AbstractSyntaxNodeStructureProvider<SwitchStatementSyntax>
{
    protected override void CollectBlockSpans(
        SyntaxToken previousToken,
        SwitchStatementSyntax node,
        ArrayBuilder<BlockSpan> spans,
        BlockStructureOptions options,
        CancellationToken cancellationToken)
    {
        spans.Add(new BlockSpan(
            isCollapsible: true,
            textSpan: TextSpan.FromBounds(node.CloseParenToken != default ? node.CloseParenToken.Span.End : node.Expression.Span.End, node.CloseBraceToken.Span.End),
            hintSpan: node.Span,
            type: BlockTypes.Conditional));

        foreach (var section in node.Sections)
        {
            if (section.Labels.Count > 0 && section.Statements.Count > 0)
            {
                var start = section.Labels.Last().ColonToken.Span.End;
                var end = section.Statements.Last().Span.End;

                spans.Add(new BlockSpan(
                    isCollapsible: true,
                    textSpan: TextSpan.FromBounds(start, end),
                    hintSpan: section.Span,
                    type: BlockTypes.Statement));
            }
        }
    }
}