From f04f1026836363c4eb7372b79146bfe6722dffc0 Mon Sep 17 00:00:00 2001
From: Andrey Vlasovskikh <andrey.vlasovskikh@jetbrains.com>
Date: Fri, 9 Apr 2021 00:56:39 +0300
Subject: [PATCH] PY-46321 Suggest "pycon" as the main alias for Python console
 code on GitHub

GitOrigin-RevId: 892043c4a320ebb6618cc3b526f2fdb72ae5057f
---
 .../python/markdown/PyCodeFenceLanguageProvider.kt     | 10 +++++++++-
 .../com/jetbrains/python/markdown/PyCodeFenceTest.kt   |  5 ++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/python/python-markdown/src/com/jetbrains/python/markdown/PyCodeFenceLanguageProvider.kt b/python/python-markdown/src/com/jetbrains/python/markdown/PyCodeFenceLanguageProvider.kt
index e277149afcad..4a16311959b3 100644
--- a/python/python-markdown/src/com/jetbrains/python/markdown/PyCodeFenceLanguageProvider.kt
+++ b/python/python-markdown/src/com/jetbrains/python/markdown/PyCodeFenceLanguageProvider.kt
@@ -2,6 +2,7 @@ package com.jetbrains.python.markdown
 
 import com.intellij.codeInsight.completion.CompletionParameters
 import com.intellij.codeInsight.lookup.LookupElement
+import com.intellij.codeInsight.lookup.LookupElementBuilder
 import com.intellij.lang.Language
 import com.jetbrains.python.PythonLanguage
 import com.jetbrains.python.documentation.doctest.PyDocstringLanguageDialect
@@ -23,5 +24,12 @@ class PyCodeFenceLanguageProvider : CodeFenceLanguageProvider {
       else -> null
     }
 
-  override fun getCompletionVariantsForInfoString(parameters: CompletionParameters): List<LookupElement> = emptyList()
+  override fun getCompletionVariantsForInfoString(parameters: CompletionParameters): List<LookupElement> {
+    return listOf(
+      LookupElementBuilder
+        .create("pycon")
+        .withIcon(PyDocstringLanguageDialect.getInstance().associatedFileType?.icon)
+        .withTypeText("Python console"),
+    )
+  }
 }
\ No newline at end of file
diff --git a/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt b/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt
index b2009aa52a65..c0e15633bee0 100644
--- a/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt
+++ b/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt
@@ -2,6 +2,9 @@ package com.jetbrains.python.markdown
 
 import com.jetbrains.python.fixtures.PyTestCase
 
+/**
+ * Tests for [PyCodeFenceLanguageProvider].
+ */
 class PyCodeFenceTest : PyTestCase() {
   fun testPythonInjected() {
     myFixture.configureByText("a.md", """
@@ -121,7 +124,7 @@ class PyCodeFenceTest : PyTestCase() {
       ```
     """.trimIndent())
     myFixture.completeBasic()
-    assertContainsElements(lookupStrings, "doctest", "python")
+    assertContainsElements(lookupStrings, "doctest", "python", "pycon")
   }
 
   private val lookupStrings: List<String>