///////////////////////////////////////////////////////////////////////////////////////////////
// checkstyle: Checks Java source code and other text files for adherence to a set of rules.
// Copyright (C) 2001-2026 the original author or authors.
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
///////////////////////////////////////////////////////////////////////////////////////////////

package com.puppycrawl.tools.checkstyle.checks.indentation;

import com.puppycrawl.tools.checkstyle.api.DetailAST;
import com.puppycrawl.tools.checkstyle.api.TokenTypes;
import com.puppycrawl.tools.checkstyle.utils.TokenUtil;

/**
 * Handler for operator new.
 */
public class NewHandler extends AbstractExpressionHandler {

    /**
     * Token types that require line wrapping indentation for new keyword.
     */
    private static final int[] LINE_WRAP_NEW_PARENT_TYPES = {
        TokenTypes.ASSIGN,
        TokenTypes.LITERAL_RETURN,
        TokenTypes.LITERAL_THROW,
    };

    /** The AST which is handled by this handler. */
    private final DetailAST mainAst;

    /**
     * Construct an instance of this handler with the given indentation context,
     * abstract syntax tree, and parent handler.
     *
     * @param context        the indentation check
     * @param ast           the abstract syntax tree
     * @param parent        the parent handler
     */
    public NewHandler(IndentationContext context,
                      DetailAST ast,
                      AbstractExpressionHandler parent) {
        super(context, "new", ast, parent);
        mainAst = ast;
    }

    @Override
    public void checkIndentation() {
        // if new is on the line start and it is not the part of assignment.
        if (isOnStartOfLine(mainAst)) {
            final int columnNo = expandedTabsColumnNo(mainAst);
            final IndentLevel level = getIndentImpl();

            final boolean forceStrictCondition = getContext().isForceStrictCondition();
            if (forceStrictCondition && !level.isAcceptable(columnNo)
                || !forceStrictCondition && level.isGreaterThan(columnNo)) {
                logError(mainAst, "", columnNo, level);
            }
        }

        final DetailAST firstChild = mainAst.getFirstChild();
        if (firstChild != null) {
            checkExpressionSubtree(firstChild, getIndent(), false, false);
        }

        final DetailAST expression = mainAst.findFirstToken(TokenTypes.ELIST);
        if (checkNestedNew(expression) && isOnStartOfLine(expression)) {
            final IndentLevel indentLevel = new IndentLevel(getIndent(),
                    getLineWrappingIndent());
            checkExpressionSubtree(expression, indentLevel, false, false);
        }

        final DetailAST lparen = mainAst.findFirstToken(TokenTypes.LPAREN);
        checkLeftParen(lparen);
    }

    /**
     * Check if nested {@code new} present.
     *
     * @param expression expression
     *
     * @return true if nested new is present.
     */
    public boolean checkNestedNew(DetailAST expression) {
        boolean result = false;
        if (expression != null && expression.getFirstChild() != null) {
            final boolean isNestedNewPresent = expression.getFirstChild()
                .findFirstToken(TokenTypes.LITERAL_NEW) != null;
            if (!isNestedNewPresent) {
                result = true;
            }
        }
        return result;
    }

    @Override
    public IndentLevel getSuggestedChildIndent(AbstractExpressionHandler child) {
        final int offset;
        if (TokenUtil.isOfType(child.getMainAst(), TokenTypes.OBJBLOCK)) {
            offset = getBasicOffset();
        }
        else {
            offset = getLineWrappingIndent();
        }
        return new IndentLevel(getIndent(), offset);
    }

    @Override
    protected IndentLevel getIndentImpl() {
        IndentLevel result;
        // if our expression isn't first on the line, just use the start
        // of the line
        if (isOnStartOfLine(mainAst)) {
            result = super.getIndentImpl();

            final boolean isLineWrappedNew = TokenUtil.isOfType(mainAst.getParent().getParent(),
                LINE_WRAP_NEW_PARENT_TYPES);

            final int ternaryLevel = getTernaryNestingLevel();

            if (isLineWrappedNew || doesNewNeedLineWrappingIndent()) {
                result = new IndentLevel(result, getLineWrappingIndent());
            }

            if (ternaryLevel >= 2) {
                for (int idx = 1; idx < ternaryLevel; idx++) {
                    result = new IndentLevel(result, getLineWrappingIndent());
                }
            }
        }
        else {
            result = new IndentLevel(getLineStart(mainAst));
        }

        return result;
    }

    /**
     * A shortcut for the corresponding property.
     *
     * @return value of lineWrappingIndentation property from the indentation context
     */
    private int getLineWrappingIndent() {
        return getContext().getLineWrappingIndentation();
    }

    @Override
    protected boolean shouldIncreaseIndent() {
        return false;
    }

    /**
     * Checks if the new keyword needs line wrapping indentation.
     * This applies when new is within an assignment, return, throw, or ternary operator
     * (where the ternary is part of an assignment, return, or throw statement).
     *
     * @return true if the new keyword needs line wrapping indentation
     */
    private boolean doesNewNeedLineWrappingIndent() {
        DetailAST ast = mainAst.getParent();

        while (TokenUtil.isOfType(ast, TokenTypes.DOT, TokenTypes.METHOD_CALL, TokenTypes.EXPR)) {
            ast = ast.getParent();
        }

        return TokenUtil.isOfType(ast, LINE_WRAP_NEW_PARENT_TYPES)
                || ast.getType() == TokenTypes.QUESTION && isParentAssignReturnOrThrow(ast);
    }

    /**
     * Checks if the parent of the given AST is an assignment, return or throw statement.
     *
     * @param ast the AST node to check
     * @return true if the parent is ASSIGN, LITERAL_RETURN or LITERAL_THROW
     */
    private static boolean isParentAssignReturnOrThrow(DetailAST ast) {
        DetailAST parent = ast.getParent();
        while (TokenUtil.isOfType(parent, TokenTypes.EXPR, TokenTypes.QUESTION)) {
            parent = parent.getParent();
        }
        return TokenUtil.isOfType(parent, LINE_WRAP_NEW_PARENT_TYPES);
    }

    /**
     * Counts how many ternary operator levels the new keyword is nested in,
     * where the outermost ternary is part of an assignment, return, or throw.
     *
     * @return the number of ternary nesting levels, or 0 if not in a valid ternary context
     */
    private int getTernaryNestingLevel() {
        DetailAST ast = mainAst.getParent();

        while (TokenUtil.isOfType(ast, TokenTypes.DOT, TokenTypes.METHOD_CALL, TokenTypes.EXPR)) {
            ast = ast.getParent();
        }

        int level = 0;
        while (ast.getType() == TokenTypes.QUESTION) {
            level++;
            do {
                ast = ast.getParent();
            } while (ast.getType() == TokenTypes.EXPR);
        }

        int result = 0;
        if (TokenUtil.isOfType(ast, LINE_WRAP_NEW_PARENT_TYPES)) {
            result = level;
        }

        return result;
    }

}