/**
* Calculates indent, based on code style, between parent block and child node
*
* @param parent parent block
* @param child child node
* @return indent
*/
@NotNull
public static Indent getChildIndent(
@NotNull final GroovyBlock parent, @NotNull final ASTNode child) {
ASTNode astNode = parent.getNode();
final PsiElement psiParent = astNode.getPsi();
// For Groovy file
if (psiParent instanceof GroovyFileBase) {
return Indent.getNoneIndent();
}
if (psiParent instanceof GrMethod && child.getPsi() instanceof GrParameterList) {
return Indent.getContinuationIndent();
}
if (GSTRING_TOKENS_INNER.contains(child.getElementType())
&& mGSTRING_BEGIN != child.getElementType()) {
return Indent.getAbsoluteNoneIndent();
}
if (psiParent instanceof GrListOrMap) {
if (mLBRACK.equals(child.getElementType()) || mRBRACK.equals(child.getElementType())) {
return Indent.getNoneIndent();
} else {
return Indent.getContinuationWithoutFirstIndent();
}
}
// For common code block
if (BLOCK_SET.contains(astNode.getElementType())
&& !BLOCK_STATEMENT.equals(astNode.getElementType())) {
return indentForBlock(psiParent, child);
}
if (CASE_SECTION.equals(astNode.getElementType())) {
return indentForCaseSection(child);
}
if (SWITCH_STATEMENT.equals(astNode.getElementType())) {
return indentForSwitchStatement(psiParent, child);
}
if (psiParent instanceof GrLabeledStatement) {
if (child.getPsi() instanceof GrLabel) {
CommonCodeStyleSettings.IndentOptions indentOptions =
parent.getSettings().getIndentOptions();
if (indentOptions.LABEL_INDENT_ABSOLUTE) {
return Indent.getAbsoluteLabelIndent();
}
return Indent.getLabelIndent();
}
}
// for control structures
if (psiParent instanceof GrControlStatement) {
return getControlIndent(psiParent, child);
}
if (psiParent instanceof GrExpression) {
return getExpressionIndent(psiParent, child);
}
if (psiParent instanceof GrVariable
&& child.getPsi() == ((GrVariable) psiParent).getInitializerGroovy()) {
return Indent.getNormalIndent();
}
// For parameter lists
if (psiParent instanceof GrParameterList
|| psiParent instanceof GrExtendsClause
|| psiParent instanceof GrThrowsClause) {
if (parent.getIndent() != null) {
return Indent.getContinuationWithoutFirstIndent();
}
return Indent.getNoneIndent();
}
// For arguments
if (psiParent instanceof GrArgumentList) {
if (child.getElementType() != mLPAREN
&& child.getElementType() != mRPAREN /*&& child.getElementType() != mCOMMA*/) {
return Indent.getContinuationWithoutFirstIndent();
}
}
if ((psiParent instanceof GrDocComment && child.getElementType() != mGDOC_COMMENT_START)
|| psiParent instanceof GrDocTag && child.getElementType() != mGDOC_TAG_NAME) {
return Indent.getSpaceIndent(GDOC_COMMENT_INDENT);
}
if (psiParent instanceof GrNamedArgument
&& child.getPsi() == ((GrNamedArgument) psiParent).getExpression()) {
return Indent.getContinuationIndent();
}
if (child.getPsi() instanceof GrVariable && psiParent instanceof GrVariableDeclaration) {
return Indent.getContinuationWithoutFirstIndent();
}
return Indent.getNoneIndent();
}
private static Indent indentForSwitchStatement(PsiElement psiParent, ASTNode child) {
if (CASE_SECTION.equals(child.getElementType())) {
return getSwitchCaseIndent(psiParent);
}
return Indent.getNoneIndent();
}
