add structure view

Author: xepozz

src/main/gen/com/github/xepozz/crontab/language/psi/CrontabCommand.java

@@ -0,0 +1,18 @@
+package com.github.xepozz.crontab.ide.structureView
+
+import com.intellij.ide.structureView.StructureViewBuilder
+import com.intellij.ide.structureView.StructureViewModel
+import com.intellij.ide.structureView.TreeBasedStructureViewBuilder
+import com.intellij.lang.PsiStructureViewFactory
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiFile
+
+internal class CrontabStructureViewFactory : PsiStructureViewFactory {
+    override fun getStructureViewBuilder(psiFile: PsiFile): StructureViewBuilder {
+        return object : TreeBasedStructureViewBuilder() {
+            override fun createStructureViewModel(editor: Editor?): StructureViewModel {
+                return CrontabStructureViewModel(editor, psiFile)
+            }
+        }
+    }
+}

src/main/gen/com/github/xepozz/crontab/language/psi/CrontabCronExpression.java

@@ -0,0 +1,26 @@
+package com.github.xepozz.crontab.ide.structureView
+
+import com.github.xepozz.crontab.language.psi.CrontabCommand
+import com.github.xepozz.crontab.language.psi.CrontabSchedule
+import com.github.xepozz.crontab.language.psi.CrontabVariableDefinition
+import com.intellij.ide.structureView.StructureViewModel.ElementInfoProvider
+import com.intellij.ide.structureView.StructureViewModelBase
+import com.intellij.ide.structureView.StructureViewTreeElement
+import com.intellij.ide.util.treeView.smartTree.Sorter
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiFile
+
+class CrontabStructureViewModel(editor: Editor?, psiFile: PsiFile) :
+    StructureViewModelBase(psiFile, editor, RecursiveStructureViewElement(psiFile)), ElementInfoProvider {
+
+    override fun getSorters() = arrayOf(Sorter.ALPHA_SORTER)
+
+    override fun isAlwaysShowsPlus(element: StructureViewTreeElement) = false
+
+    override fun isAlwaysLeaf(element: StructureViewTreeElement) = when (element.value) {
+        is CrontabCommand, is CrontabSchedule, is CrontabVariableDefinition -> true
+        else -> false
+    }
+
+    override fun getSuitableClasses() = arrayOf(CrontabCommand::class.java)
+}

src/main/gen/com/github/xepozz/crontab/language/psi/CrontabSchedule.java

@@ -0,0 +1,55 @@
+package com.github.xepozz.crontab.ide.structureView
+
+import com.github.xepozz.crontab.language.CrontabFile
+import com.github.xepozz.crontab.language.psi.CrontabCommand
+import com.github.xepozz.crontab.language.psi.CrontabCronExpression
+import com.github.xepozz.crontab.language.psi.CrontabSchedule
+import com.github.xepozz.crontab.language.psi.CrontabVariableDefinition
+import com.intellij.ide.projectView.PresentationData
+import com.intellij.ide.structureView.StructureViewTreeElement
+import com.intellij.ide.util.treeView.smartTree.SortableTreeElement
+import com.intellij.ide.util.treeView.smartTree.TreeElement
+import com.intellij.psi.NavigatablePsiElement
+import com.intellij.psi.util.PsiTreeUtil
+
+class RecursiveStructureViewElement(val myElement: NavigatablePsiElement) :
+    StructureViewTreeElement,
+    SortableTreeElement {
+    override fun navigate(requestFocus: Boolean) = myElement.navigate(requestFocus)
+
+    override fun canNavigate() = myElement.canNavigate()
+
+    override fun canNavigateToSource() = myElement.canNavigateToSource()
+
+    override fun getValue() = myElement
+
+    override fun getPresentation() = myElement.presentation
+        ?: PresentationData(myElement.text, null, myElement.getIcon(0), null)
+
+    override fun getChildren(): Array<out TreeElement> {
+        return when (myElement) {
+            is CrontabFile -> PsiTreeUtil.getChildrenOfAnyType(
+                myElement,
+                CrontabCronExpression::class.java,
+                CrontabVariableDefinition::class.java
+            )
+                .map { RecursiveStructureViewElement(it) }
+                .toTypedArray()
+
+            is CrontabCronExpression -> PsiTreeUtil.getChildrenOfAnyType(
+                myElement,
+                CrontabSchedule::class.java,
+                CrontabCommand::class.java
+            )
+                .map { RecursiveStructureViewElement(it) }
+                .toTypedArray()
+
+            else -> StructureViewTreeElement.EMPTY_ARRAY
+        }
+    }
+
+    override fun getAlphaSortKey(): String {
+        val name = myElement.name
+        return name ?: ""
+    }
+}