Avoid inserting extraneous quote in Literal and TypedDict complet…

…ions (#39)

src/main/kotlin/com/insyncwithfoo/pyrightls/server/CompletionSupport.kt

@@ -14,6 +14,14 @@ import org.eclipse.lsp4j.InsertTextFormat
 private const val CARET_POSITION = "\$0"
 
 
+private val String.isSingleQuoted: Boolean
+    get() = this.startsWith("'") && this.endsWith("'")
+
+
+private val String.isDoubleQuoted: Boolean
+    get() = this.startsWith("\"") && this.endsWith("\"")
+
+
 private val CompletionItem.isCallable: Boolean
     get() = kind in listOf(
         CompletionItemKind.Method,
@@ -29,6 +37,18 @@ private val CompletionItem.isAutoImportCompletion: Boolean
     }
 
 
+private val CompletionItem.isQuoted: Boolean
+    get() = label.isSingleQuoted || label.isDoubleQuoted
+
+
+private val CompletionItem.quote: Char
+    get() = label[0]
+
+
+private val CompletionParameters.nextCharacter: Char
+    get() = editor.document.charsSequence[offset]
+
+
 private fun CompletionItem.completeWithParentheses() {
     insertText = "$label($CARET_POSITION)"
     insertTextFormat = InsertTextFormat.Snippet
@@ -41,6 +61,11 @@ private fun CompletionItem.useSourceAsDetailIfPossible() {
 }
 
 
+private fun CompletionItem.removeTrailingQuote() {
+    insertText = label.dropLast(1)
+}
+
+
 @Suppress("UnstableApiUsage")
 internal class CompletionSupport(project: Project) : LspCompletionSupport() {
 
@@ -55,6 +80,10 @@ internal class CompletionSupport(project: Project) : LspCompletionSupport() {
             item.useSourceAsDetailIfPossible()
         }
 
+        if (item.isQuoted && parameters.nextCharacter == item.quote) {
+            item.removeTrailingQuote()
+        }
+
         return super.createLookupElement(parameters, item)
     }