SetVisibilityIntention

Author: tangzx

src/main/java/com/tang/intellij/lua/codeInsight/intention/NamesMacro.kt

@@ -0,0 +1,42 @@
+/*
+ * 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.codeInsight.intention
+
+import com.intellij.codeInsight.lookup.LookupElement
+import com.intellij.codeInsight.lookup.LookupElementBuilder
+import com.intellij.codeInsight.template.Expression
+import com.intellij.codeInsight.template.ExpressionContext
+import com.intellij.codeInsight.template.Macro
+import com.intellij.codeInsight.template.Result
+
+class NamesMacro(vararg val args: String) : Macro() {
+    override fun getPresentableName() = "LuaNamesMacro"
+
+    override fun getName() = "LuaNamesMacro"
+
+    override fun calculateResult(p0: Array<out Expression>, p1: ExpressionContext?): Result? {
+        return null
+    }
+
+    override fun calculateLookupItems(params: Array<out Expression>, context: ExpressionContext?): Array<LookupElement>? {
+        val list = mutableListOf<LookupElement>()
+        for (name in args) {
+            list.add(LookupElementBuilder.create(name))
+        }
+        return list.toTypedArray()
+    }
+}

src/main/java/com/tang/intellij/lua/codeInsight/intention/SetVisibilityIntention.kt

@@ -0,0 +1,65 @@
+/*
+ * 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.codeInsight.intention
+
+import com.intellij.codeInsight.template.TemplateManager
+import com.intellij.codeInsight.template.impl.MacroCallNode
+import com.intellij.codeInsight.template.impl.TextExpression
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiElement
+import com.tang.intellij.lua.comment.psi.LuaDocAccessModifier
+import com.tang.intellij.lua.psi.LuaClassMethodDef
+import com.tang.intellij.lua.psi.LuaCommentOwner
+import com.tang.intellij.lua.psi.LuaFuncBodyOwner
+
+class SetVisibilityIntention : FunctionIntention() {
+
+    override fun isAvailable(bodyOwner: LuaFuncBodyOwner, editor: Editor): Boolean {
+        if (bodyOwner is LuaClassMethodDef) {
+            return bodyOwner.comment?.findTag(LuaDocAccessModifier::class.java) == null
+        }
+        return false
+    }
+
+    override fun invoke(bodyOwner: LuaFuncBodyOwner, editor: Editor) {
+        if (bodyOwner is LuaCommentOwner) {
+            val comment = bodyOwner.comment
+            val funcBody = bodyOwner.funcBody
+            if (funcBody != null) {
+                val templateManager = TemplateManager.getInstance(editor.project)
+                val template = templateManager.createTemplate("", "")
+                if (comment != null) template.addTextSegment("\n")
+                template.addTextSegment("---@")
+                val typeSuggest = MacroCallNode(NamesMacro("public", "protected", "private"))
+                template.addVariable("visibility", typeSuggest, TextExpression("private"), false)
+                template.addEndVariable()
+                if (comment != null) {
+                    editor.caretModel.moveToOffset(comment.textOffset + comment.textLength)
+                } else {
+                    template.addTextSegment("\n")
+                    val e: PsiElement = bodyOwner
+                    editor.caretModel.moveToOffset(e.node.startOffset)
+                }
+                templateManager.startTemplate(editor, template)
+            }
+        }
+    }
+
+    override fun getFamilyName() = text
+
+    override fun getText() = "Set Visibility (public/protected/private)"
+}

src/main/resources/META-INF/plugin.xml

@@ -271,6 +271,10 @@
             <category>Lua</category>
             <className>com.tang.intellij.lua.codeInsight.intention.InvertBooleanIntention</className>
         </intentionAction>
+        <intentionAction>
+            <category>Lua</category>
+            <className>com.tang.intellij.lua.codeInsight.intention.SetVisibilityIntention</className>
+        </intentionAction>
 
         <localInspection
                 implementationClass="com.tang.intellij.lua.codeInsight.inspection.DuplicateClassDeclaration"

src/main/resources/intentionDescriptions/SetVisibilityIntention/description.html

@@ -0,0 +1,21 @@
+<!--
+  ~ 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.
+  -->
+
+<html>
+<body>
+
+</body>
+</html>