override fun visitElement()

in src/main/kotlin/com/github/mkartashev/hserr/language/HsErrFoldingBuilder.kt [26:58]

31 lines of code
3 McCabe index (conditional complexity)


            override fun visitElement(element: PsiElement) {
                super.visitElement(element)

                if (!element.isValid || element.textRange.length <= 2) {
                    group = null
                    return
                }

                when (element) {
                    is HsErrIntro, is HsErrSection -> {
                        group = FoldingGroup.newGroup("hserr section")
                        descriptors.add(
                            FoldingDescriptor(
                                element.node,
                                element.textRange.shrinkBy(1),
                                group
                            )
                        )
                    }
                    is HsErrSubsection -> {
                        descriptors.add(
                            FoldingDescriptor(
                                element.node,
                                element.textRange.shrinkBy(1),
                                group
                            )
                        )
                    }
                    else -> {
                        group = null
                    }
                }
            }