// 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 Microsoft.CodeAnalysis.Shared.Extensions;
using Microsoft.CodeAnalysis.Text;

namespace Microsoft.CodeAnalysis.Formatting;

internal abstract partial class TreeData
{
    private class NodeAndText : TreeData
    {
        private readonly SourceText _text;

        public NodeAndText(SyntaxNode root, SourceText text)
            : base(root)
        {
            Contract.ThrowIfNull(text);
            _text = text;
        }

        public override int GetOriginalColumn(int tabSize, SyntaxToken token)
        {
            Contract.ThrowIfTrue(token.RawKind == 0);

            var line = _text.Lines.GetLineFromPosition(token.SpanStart);

            return line.GetColumnFromLineOffset(token.SpanStart - line.Start, tabSize);
        }

        public override string GetTextBetween(SyntaxToken token1, SyntaxToken token2)
        {
            if (token1.RawKind == 0)
            {
                // get leading trivia text
                return _text.ToString(TextSpan.FromBounds(token2.FullSpan.Start, token2.SpanStart));
            }

            if (token2.RawKind == 0)
            {
                // get trailing trivia text
                return _text.ToString(TextSpan.FromBounds(token1.Span.End, token1.FullSpan.End));
            }

            return _text.ToString(TextSpan.FromBounds(token1.Span.End, token2.SpanStart));
        }
    }
}