Merge pull request #35 from chenfeng11111/main

支持拼写检查

Author: CppCXY

src/main/kotlin/com/tang/intellij/lua/spellchecker/LuaSpellBundledDictionaryProvider.kt

@@ -0,0 +1,23 @@
+/*
+ * Copyright (c) 2017. tangzx(love.tangzx@qq.com)
+ *
+ * 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.spellchecker
+
+import com.intellij.spellchecker.BundledDictionaryProvider
+
+class LuaSpellBundledDictionaryProvider : BundledDictionaryProvider {
+    override fun getBundledDictionaries() = arrayOf("/spellchecker/lua.dic")
+}

src/main/kotlin/com/tang/intellij/lua/spellchecker/LuaSpellcheckingStrategy.kt

@@ -0,0 +1,51 @@
+/*
+ * Copyright (c) 2017. tangzx(love.tangzx@qq.com)
+ *
+ * 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.spellchecker
+
+import com.intellij.psi.PsiElement
+import com.intellij.spellchecker.inspections.PlainTextSplitter
+import com.intellij.spellchecker.tokenizer.EscapeSequenceTokenizer
+import com.intellij.spellchecker.tokenizer.SpellcheckingStrategy
+import com.intellij.spellchecker.tokenizer.TokenConsumer
+import com.intellij.spellchecker.tokenizer.Tokenizer
+import com.tang.intellij.lua.comment.psi.LuaDocTypes
+import com.tang.intellij.lua.lang.LuaLanguage
+import com.tang.intellij.lua.psi.LuaTypes
+
+class LuaSpellcheckingStrategy : SpellcheckingStrategy() {
+    override fun isMyContext(element: PsiElement): Boolean {
+        return element.language is LuaLanguage
+    }
+
+    override fun getTokenizer(element: PsiElement): Tokenizer<*> {
+        return when(element.node.elementType){
+            LuaDocTypes.STRING -> StringLiteralTokenizer
+            LuaDocTypes.WORD -> TEXT_TOKENIZER
+            LuaTypes.ID -> TEXT_TOKENIZER
+            LuaTypes.STRING -> StringLiteralTokenizer
+            else -> super.getTokenizer(element)
+        }
+    }
+}
+
+
+private object StringLiteralTokenizer : EscapeSequenceTokenizer<PsiElement>() {
+
+    override fun tokenize(element: PsiElement, consumer: TokenConsumer) {
+        consumer.consumeToken(element, PlainTextSplitter.getInstance())
+    }
+}

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

@@ -73,6 +73,9 @@
         <!-- LSP 颜色配置页面 -->
         <colorSettingsPage implementation="com.tang.intellij.lua.highlighting.LuaLspColorSettingsPage"/>
 
+        <spellchecker.support implementationClass="com.tang.intellij.lua.spellchecker.LuaSpellcheckingStrategy" language="Lua"/>
+        <spellchecker.bundledDictionaryProvider implementation="com.tang.intellij.lua.spellchecker.LuaSpellBundledDictionaryProvider"/>
+
         <!-- <readWriteAccessDetector implementation="com.tang.intellij.lua.codeInsight.LuaReadWriteAccessDetector"/> -->
 
         <!--        debugger-->

src/main/resources/spellchecker/lua.dic

@@ -0,0 +1,53 @@
+acos
+asin
+atan
+collectgarbage
+coroutine
+cpath
+difftime
+dofile
+fmod
+frexp
+getfenv
+gethook
+getinfo
+getmetatable
+getregistry
+getupvalue
+gmatch
+gsub
+ipairs
+ldexp
+loadfile
+loadlib
+loadstring
+maxn
+metatable
+modf
+newindex
+pcall
+popen
+randomseed
+rawequal
+rawget
+rawset
+seeall
+setfenv
+sethook
+setlocal
+setlocale
+setmetatable
+setupvalue
+setuservalue
+setvbuf
+sinh
+tanh
+tmpfile
+tmpname
+tonumber
+tostring
+tostring
+traceback
+upvalueid
+upvaluejoin
+xpcall