test pre format processor

Author: tangzx

src/main/java/com/tang/intellij/lua/editor/formatter/LuaPreFormatProcessor.kt

@@ -0,0 +1,78 @@
+/*
+ * Copyright (c) 2017. tangzx([email protected])
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.tang.intellij.lua.editor.formatter
+
+import com.intellij.lang.ASTNode
+import com.intellij.openapi.util.TextRange
+import com.intellij.psi.PsiDocumentManager
+import com.intellij.psi.PsiElement
+import com.intellij.psi.impl.source.codeStyle.PreFormatProcessor
+import com.intellij.util.DocumentUtil
+import com.tang.intellij.lua.psi.LuaFile
+import com.tang.intellij.lua.psi.LuaTypes
+import com.tang.intellij.lua.psi.LuaVisitor
+
+/**
+ * 移除不用的 ;,
+ * Created by tangzx on 2017/7/8.
+ */
+class LuaPreFormatProcessor : PreFormatProcessor {
+    override fun process(astNode: ASTNode, sourceRange: TextRange): TextRange {
+        var range = sourceRange
+        val psi = astNode.psi
+        if (psi.containingFile is LuaFile) {
+            val list = mutableListOf<Pair<Int, Boolean>>()
+            psi.accept(object : LuaVisitor() {
+                override fun visitElement(o: PsiElement) {
+                    if (o.node.elementType == LuaTypes.SEMI) {
+                        val nextSibling = o.nextSibling
+                        var replaceWithSpace = false
+                        if (nextSibling != null) {
+                            if (nextSibling.node.startOffset == o.node.startOffset + 1) {
+                                replaceWithSpace = true
+                            }
+                        }
+                        list.add(Pair(o.node.startOffset, replaceWithSpace))
+                    } else o.acceptChildren(this)
+                }
+            })
+
+            val psiDocumentManager = PsiDocumentManager.getInstance(psi.project)
+            val document = psiDocumentManager.getDocument(psi.containingFile)
+            document ?: return range
+            DocumentUtil.executeInBulk(document, true) {
+                psiDocumentManager.doPostponedOperationsAndUnblockDocument(document)
+
+                var removed = 0
+                list.forEach {
+                    if (it.second) {
+                        document.replaceString(it.first, it.first + 1, " ")
+                    } else {
+                        document.deleteString(it.first, it.first + 1)
+                        removed++
+                    }
+                }
+
+                psiDocumentManager.commitDocument(document)
+
+                range = TextRange(sourceRange.startOffset, sourceRange.length - removed)
+            }
+
+        }
+        return range
+    }
+}