using System;
using System.Collections.Generic;
using Roslyn.Utilities;

namespace Microsoft.CodeAnalysis.CSharp.Syntax.InternalSyntax;

internal sealed class WithExpressionSyntax : ExpressionSyntax
{
	internal readonly ExpressionSyntax expression;

	internal readonly SyntaxToken withKeyword;

	internal readonly InitializerExpressionSyntax initializer;

	public ExpressionSyntax Expression => expression;

	public SyntaxToken WithKeyword => withKeyword;

	public InitializerExpressionSyntax Initializer => initializer;

	internal WithExpressionSyntax(SyntaxKind kind, ExpressionSyntax expression, SyntaxToken withKeyword, InitializerExpressionSyntax initializer, DiagnosticInfo[]? diagnostics, SyntaxAnnotation[]? annotations)
		: base(kind, diagnostics, annotations)
	{
		((GreenNode)this).SlotCount = 3;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)expression);
		this.expression = expression;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)withKeyword);
		this.withKeyword = withKeyword;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)initializer);
		this.initializer = initializer;
	}

	internal WithExpressionSyntax(SyntaxKind kind, ExpressionSyntax expression, SyntaxToken withKeyword, InitializerExpressionSyntax initializer, SyntaxFactoryContext context)
		: base(kind)
	{
		SetFactoryContext(context);
		((GreenNode)this).SlotCount = 3;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)expression);
		this.expression = expression;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)withKeyword);
		this.withKeyword = withKeyword;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)initializer);
		this.initializer = initializer;
	}

	internal WithExpressionSyntax(SyntaxKind kind, ExpressionSyntax expression, SyntaxToken withKeyword, InitializerExpressionSyntax initializer)
		: base(kind)
	{
		((GreenNode)this).SlotCount = 3;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)expression);
		this.expression = expression;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)withKeyword);
		this.withKeyword = withKeyword;
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)initializer);
		this.initializer = initializer;
	}

	internal override GreenNode? GetSlot(int index)
	{
		return (GreenNode?)(index switch
		{
			0 => expression, 
			1 => withKeyword, 
			2 => initializer, 
			_ => null, 
		});
	}

	internal override SyntaxNode CreateRed(SyntaxNode? parent, int position)
	{
		return (SyntaxNode)(object)new Microsoft.CodeAnalysis.CSharp.Syntax.WithExpressionSyntax(this, parent, position);
	}

	public override void Accept(CSharpSyntaxVisitor visitor)
	{
		visitor.VisitWithExpression(this);
	}

	public override TResult Accept<TResult>(CSharpSyntaxVisitor<TResult> visitor)
	{
		return visitor.VisitWithExpression(this);
	}

	public WithExpressionSyntax Update(ExpressionSyntax expression, SyntaxToken withKeyword, InitializerExpressionSyntax initializer)
	{
		if (expression != Expression || withKeyword != WithKeyword || initializer != Initializer)
		{
			WithExpressionSyntax withExpressionSyntax = SyntaxFactory.WithExpression(expression, withKeyword, initializer);
			DiagnosticInfo[] diagnostics = ((GreenNode)this).GetDiagnostics();
			if (diagnostics != null && diagnostics.Length != 0)
			{
				withExpressionSyntax = GreenNodeExtensions.WithDiagnosticsGreen<WithExpressionSyntax>(withExpressionSyntax, diagnostics);
			}
			SyntaxAnnotation[] annotations = ((GreenNode)this).GetAnnotations();
			if (annotations != null && annotations.Length != 0)
			{
				withExpressionSyntax = GreenNodeExtensions.WithAnnotationsGreen<WithExpressionSyntax>(withExpressionSyntax, (IEnumerable<SyntaxAnnotation>)annotations);
			}
			return withExpressionSyntax;
		}
		return this;
	}

	internal override GreenNode SetDiagnostics(DiagnosticInfo[]? diagnostics)
	{
		return (GreenNode)(object)new WithExpressionSyntax(base.Kind, expression, withKeyword, initializer, diagnostics, ((GreenNode)this).GetAnnotations());
	}

	internal override GreenNode SetAnnotations(SyntaxAnnotation[]? annotations)
	{
		return (GreenNode)(object)new WithExpressionSyntax(base.Kind, expression, withKeyword, initializer, ((GreenNode)this).GetDiagnostics(), annotations);
	}

	internal WithExpressionSyntax(ObjectReader reader)
		: base(reader)
	{
		((GreenNode)this).SlotCount = 3;
		ExpressionSyntax expressionSyntax = (ExpressionSyntax)reader.ReadValue();
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)expressionSyntax);
		expression = expressionSyntax;
		SyntaxToken syntaxToken = (SyntaxToken)reader.ReadValue();
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)syntaxToken);
		withKeyword = syntaxToken;
		InitializerExpressionSyntax initializerExpressionSyntax = (InitializerExpressionSyntax)reader.ReadValue();
		((GreenNode)this).AdjustFlagsAndWidth((GreenNode)(object)initializerExpressionSyntax);
		initializer = initializerExpressionSyntax;
	}

	internal override void WriteTo(ObjectWriter writer)
	{
		((GreenNode)this).WriteTo(writer);
		writer.WriteValue((IObjectWritable)(object)expression);
		writer.WriteValue((IObjectWritable)(object)withKeyword);
		writer.WriteValue((IObjectWritable)(object)initializer);
	}

	static WithExpressionSyntax()
	{
		ObjectBinder.RegisterTypeReader(typeof(WithExpressionSyntax), (Func<ObjectReader, IObjectWritable>)((ObjectReader r) => (IObjectWritable)(object)new WithExpressionSyntax(r)));
	}
}