// 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.Shared.Utilities;
using Roslyn.Utilities;
 
namespace Microsoft.CodeAnalysis.Formatting;
 
internal abstract partial class TreeData
{
    private class Node : TreeData
    {
        public Node(SyntaxNode root)
            : base(root)
        {
            Contract.ThrowIfFalse(root.GetFirstToken(includeZeroWidth: true).RawKind != 0);
        }
 
        public override int GetOriginalColumn(int tabSize, SyntaxToken token)
        {
            Contract.ThrowIfTrue(token.RawKind == 0);
 
            // first find one that has new line text
            var startToken = GetTokenWithLineBreaks(token);
 
            // get last line text from text between them
            var lineText = GetTextBetween(startToken, token).GetLastLineText();
 
            return lineText.GetColumnFromLineOffset(lineText.Length, tabSize);
        }
 
        private static SyntaxToken GetTokenWithLineBreaks(SyntaxToken token)
        {
            var currentToken = token.GetPreviousToken(includeZeroWidth: true);
 
            while (currentToken.RawKind != 0)
            {
                if (currentToken.ToFullString().IndexOf('\n') >= 0)
                {
                    return currentToken;
                }
 
                currentToken = currentToken.GetPreviousToken(includeZeroWidth: true);
            }
 
            return default;
        }
 
        public override string GetTextBetween(SyntaxToken token1, SyntaxToken token2)
        {
            var builder = StringBuilderPool.Allocate();
 
            CommonFormattingHelpers.AppendTextBetween(token1, token2, builder);
 
            return StringBuilderPool.ReturnAndFree(builder);
        }
    }
}