基于JetBrains平台的微信小程序插件

package com.zxy.ijplugin.wechat_miniprogram.lang.wxml.tag; import com.intellij.application.options.CodeStyle; import com.intellij.application.options.editor.WebEditorOptions; import com.intellij.codeInsight.TailType; import com.intellij.codeInsight.completion.CompletionUtil; import com.intellij.codeInsight.completion.InsertHandler; import com.intellij.codeInsight.completion.InsertionContext; import com.intellij.codeInsight.daemon.impl.quickfix.EmptyExpression; import com.intellij.codeInsight.editorActions.XmlEditUtil; import com.intellij.codeInsight.editorActions.XmlTagNameSynchronizer; import com.intellij.codeInsight.lookup.Lookup; import com.intellij.codeInsight.lookup.LookupElement; import com.intellij.codeInsight.lookup.LookupItem; import com.intellij.codeInsight.template.Template; import com.intellij.codeInsight.template.TemplateEditingAdapter; import com.intellij.codeInsight.template.TemplateManager; import com.intellij.codeInsight.template.impl.MacroCallNode; import com.intellij.codeInsight.template.macro.CompleteMacro; import com.intellij.codeInsight.template.macro.CompleteSmartMacro; import com.intellij.codeInspection.InspectionProfile; import com.intellij.codeInspection.htmlInspections.XmlEntitiesInspection; import com.intellij.injected.editor.EditorWindow; import com.intellij.lang.ASTNode; import com.intellij.openapi.command.WriteCommandAction; import com.intellij.openapi.command.undo.UndoManager; import com.intellij.openapi.editor.Document; import com.intellij.openapi.editor.Editor; import com.intellij.openapi.editor.RangeMarker; import com.intellij.openapi.editor.ScrollType; import com.intellij.openapi.project.Project; import com.intellij.openapi.util.Ref; import com.intellij.openapi.util.text.StringUtil; import com.intellij.profile.codeInspection.InspectionProjectProfileManager; import com.intellij.psi.PsiDocumentManager; import com.intellij.psi.PsiElement; import com.intellij.psi.PsiFile; import com.intellij.psi.codeStyle.CodeStyleSettings; import com.intellij.psi.formatter.xml.HtmlCodeStyleSettings; import com.intellij.psi.formatter.xml.XmlCodeStyleSettings; import com.intellij.psi.html.HtmlTag; import com.intellij.psi.util.PsiTreeUtil; import com.intellij.psi.xml.XmlTag; import com.intellij.psi.xml.XmlTokenType; import com.intellij.xml.*; import com.intellij.xml.XmlExtension.AttributeValuePresentation; import com.intellij.xml.actions.GenerateXmlTagAction; import com.intellij.xml.impl.schema.XmlElementDescriptorImpl; import com.intellij.xml.util.HtmlUtil; import com.intellij.xml.util.XmlUtil; import com.zxy.ijplugin.wechat_miniprogram.lang.wxml.utils.WXMLAttributeInsertUtils; import com.zxy.ijplugin.wechat_miniprogram.lang.wxml.utils.WXMLUtils; import kotlin.collections.ArraysKt; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import java.util.*; /** * copy from {@see com.intellij.codeInsight.completion.XmlTagInsertHandler} */ public class WXMLTagInsertHandler implements InsertHandler<LookupElement> { public static final WXMLTagInsertHandler INSTANCE = new WXMLTagInsertHandler(); public static void insertIncompleteTag(char completionChar, final Editor editor, XmlTag tag) { XmlTag originalElement = CompletionUtil.getOriginalElement(tag); XmlElementDescriptor descriptor = originalElement != null ? originalElement.getDescriptor() : tag.getDescriptor(); if (descriptor == null) return; final Project project = editor.getProject(); assert project != null; TemplateManager templateManager = TemplateManager.getInstance(project); Template template = templateManager.createTemplate("", ""); template.setToIndent(true); // temp code PsiFile containingFile = tag.getContainingFile(); boolean htmlCode = HtmlUtil.hasHtml(containingFile) || HtmlUtil.supportsXmlTypedHandlers(containingFile); template.setToReformat(!htmlCode); StringBuilder indirectRequiredAttrs = addRequiredAttributes(tag, template, containingFile); final boolean chooseAttributeName = addTail(completionChar, descriptor, htmlCode, tag, template, indirectRequiredAttrs); templateManager.startTemplate(editor, template, new TemplateEditingAdapter() { private RangeMarker myAttrValueMarker; @Override public void waitingForInput(Template template) { int offset = editor.getCaretModel().getOffset(); myAttrValueMarker = editor.getDocument().createRangeMarker(offset + 1, offset + 4); } @Override public void templateFinished(@NotNull final Template template, boolean brokenOff) { final int offset = editor.getCaretModel().getOffset(); if (chooseAttributeName && offset > 0) { char c = editor.getDocument().getCharsSequence().charAt(offset - 1); if (c == '/' || (c == ' ' && brokenOff)) { WriteCommandAction.writeCommandAction(project).run( () -> editor.getDocument().replaceString(offset, offset + 3, ">")); } } } @Override public void templateCancelled(final Template template) { if (myAttrValueMarker == null) { return; } if (UndoManager.getInstance(project).isUndoOrRedoInProgress()) { return; } if (chooseAttributeName && myAttrValueMarker.isValid()) { final int startOffset = myAttrValueMarker.getStartOffset(); final int endOffset = myAttrValueMarker.getEndOffset(); WriteCommandAction.writeCommandAction(project).run( () -> editor.getDocument().replaceString(startOffset, endOffset, ">")); } } }); } @Nullable private static StringBuilder addRequiredAttributes(@Nullable XmlTag tag, Template template, PsiFile containingFile) { Set<String> notRequiredAttributes = Collections.emptySet(); if (tag instanceof HtmlTag) { final InspectionProfile profile = InspectionProjectProfileManager.getInstance( tag.getProject()).getCurrentProfile(); XmlEntitiesInspection inspection = (XmlEntitiesInspection) profile.getUnwrappedTool( XmlEntitiesInspection.REQUIRED_ATTRIBUTES_SHORT_NAME, tag); if (inspection != null) { StringTokenizer tokenizer = new StringTokenizer(inspection.getAdditionalEntries()); notRequiredAttributes = new HashSet<>(); while (tokenizer.hasMoreElements()) notRequiredAttributes.add(tokenizer.nextToken()); } } XmlAttributeDescriptor[] attributes = WXMLUtils.getWXMLAttributeDescriptors(tag); StringBuilder indirectRequiredAttrs = null; if (WebEditorOptions.getInstance().isAutomaticallyInsertRequiredAttributes()) { final XmlExtension extension = XmlExtension.getExtension(containingFile); for (XmlAttributeDescriptor attributeDecl : attributes) { String attributeName = attributeDecl.getName(tag); boolean shouldBeInserted = extension.shouldBeInserted(attributeDecl); if (!shouldBeInserted) continue; AttributeValuePresentation presenter = extension.getAttributeValuePresentation(tag, attributeName, XmlEditUtil.getAttributeQuote(containingFile)); boolean htmlCode = HtmlUtil.hasHtml(containingFile) || HtmlUtil.supportsXmlTypedHandlers(