feat(provider): add change count, line count, and complexity count va…

…riables #143 and closed #139 This commit introduces new context variables to calculate and provide the number of changes, lines, and complexity within a file. These variables are added across multiple language providers to ensure consistent support.
phodal · Nov 17, 2024 · e23687c · e23687c
1 parent 0cb29fb
commit e23687c
Show file tree

Hide file tree

Showing 9 changed files with 59 additions and 3 deletions.
diff --git a/core/src/main/kotlin/com/phodal/shirecore/provider/variable/PsiContextVariableProvider.kt b/core/src/main/kotlin/com/phodal/shirecore/provider/variable/PsiContextVariableProvider.kt
@@ -40,6 +40,19 @@ interface PsiContextVariableProvider : VariableProvider<PsiContextVariable> {
         return future.get()
     }
 
+    /// try to load from git
+    fun calculateChangeCount(psiElement: PsiElement?): String {
+        return "0"
+    }
+
+    fun calculateLineCount(psiElement: PsiElement?): String {
+        return psiElement?.containingFile?.text?.lines()?.size.toString()
+    }
+
+    fun calculateComplexityCount(psiElement: PsiElement?): String {
+        return "0"
+    }
+
     companion object {
         private val languageExtension: LanguageExtension<PsiContextVariableProvider> =
             LanguageExtension("com.phodal.shirePsiVariableProvider")

diff --git a/...n/kotlin/com/phodal/shirecore/provider/variable/impl/DefaultPsiContextVariableProvider.kt b/...n/kotlin/com/phodal/shirecore/provider/variable/impl/DefaultPsiContextVariableProvider.kt
@@ -5,7 +5,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 import com.phodal.shirecore.provider.variable.PsiContextVariableProvider
 import com.phodal.shirecore.provider.variable.model.PsiContextVariable
-import com.phodal.shirecore.provider.variable.model.PsiContextVariable.FRAMEWORK_CONTEXT
+import com.phodal.shirecore.provider.variable.model.PsiContextVariable.*
 
 class DefaultPsiContextVariableProvider : PsiContextVariableProvider {
     override fun resolve(variable: PsiContextVariable, project: Project, editor: Editor, psiElement: PsiElement?): String {
@@ -14,6 +14,16 @@ class DefaultPsiContextVariableProvider : PsiContextVariableProvider {
                 return collectFrameworkContext(psiElement, project)
             }
 
+            CHANGE_COUNT -> {
+                return calculateChangeCount(psiElement)
+            }
+            LINE_COUNT -> {
+                return calculateLineCount(psiElement)
+            }
+            COMPLEXITY_COUNT -> {
+                return calculateComplexityCount(psiElement)
+            }
+
             else -> ""
         }
     }

diff --git a/core/src/main/kotlin/com/phodal/shirecore/provider/variable/model/PsiContextVariable.kt b/core/src/main/kotlin/com/phodal/shirecore/provider/variable/model/PsiContextVariable.kt
@@ -76,7 +76,22 @@ enum class PsiContextVariable(
 
     SIMILAR_CODE("similarCode", "Recently 20 files similar code based on the tf-idf search"),
 
-    STRUCTURE("structure", "The structure of the current class, for programming language will be in UML format.")
+    STRUCTURE("structure", "The structure of the current class, for programming language will be in UML format."),
+
+    /**
+     * Represents the number of changes in the current file.
+     */
+    CHANGE_COUNT("changeCount", "The number of changes in the current file"),
+
+    /**
+     * Represents the number of lines in the current file.
+     */
+    LINE_COUNT("lineCount", "The number of lines in the current file"),
+
+    /**
+     * Represents the complexity of the current file.
+     */
+    COMPLEXITY_COUNT("complexityCount", "The complexity of the current file")
     ;
 
     companion object {
@@ -87,7 +102,7 @@ enum class PsiContextVariable(
          * @return the PsiVariable with the given variable name
          */
         fun from(variableName: String): PsiContextVariable? {
-            return values().firstOrNull { it.variableName == variableName }
+            return entries.firstOrNull { it.variableName == variableName }
         }
     }
 }
diff --git a/...shire-go/src/main/kotlin/com/phodal/shirelang/go/variable/GoPsiContextVariableProvider.kt b/...shire-go/src/main/kotlin/com/phodal/shirelang/go/variable/GoPsiContextVariableProvider.kt
@@ -121,6 +121,9 @@ class GoPsiContextVariableProvider : PsiContextVariableProvider {
                     else -> ""
                 }
             }
+            PsiContextVariable.CHANGE_COUNT -> calculateChangeCount(psiElement)
+            PsiContextVariable.LINE_COUNT -> calculateLineCount(psiElement)
+            PsiContextVariable.COMPLEXITY_COUNT -> ""
         } ?: ""
     }
 

diff --git a/...java/src/main/kotlin/com/phodal/shirelang/java/variable/JavaPsiContextVariableProvider.kt b/...java/src/main/kotlin/com/phodal/shirelang/java/variable/JavaPsiContextVariableProvider.kt
@@ -49,6 +49,9 @@ class JavaPsiContextVariableProvider : PsiContextVariableProvider {
             } ?: ""
 
             FRAMEWORK_CONTEXT -> return collectFrameworkContext(psiElement, project)
+            PsiContextVariable.CHANGE_COUNT -> calculateChangeCount(psiElement)
+            PsiContextVariable.LINE_COUNT -> calculateLineCount(psiElement)
+            PsiContextVariable.COMPLEXITY_COUNT -> ""
         }
     }
 }

diff --git a/.../src/main/kotlin/com/phodal/shirelang/javascript/variable/JSPsiContextVariableProvider.kt b/.../src/main/kotlin/com/phodal/shirelang/javascript/variable/JSPsiContextVariableProvider.kt
@@ -102,6 +102,9 @@ class JSPsiContextVariableProvider : PsiContextVariableProvider {
                     else -> null
                 } ?: ""
             }
+            PsiContextVariable.CHANGE_COUNT -> calculateChangeCount(psiElement)
+            PsiContextVariable.LINE_COUNT -> calculateLineCount(psiElement)
+            PsiContextVariable.COMPLEXITY_COUNT -> ""
         }
     }
 

diff --git a/...roto/src/main/kotlin/com/phodal/shirelang/proto/provider/ShireProtoPsiVariableProvider.kt b/...roto/src/main/kotlin/com/phodal/shirelang/proto/provider/ShireProtoPsiVariableProvider.kt
@@ -73,6 +73,9 @@ class ShireProtoPsiVariableProvider : PsiContextVariableProvider {
             PsiContextVariable.IS_NEED_CREATE_FILE -> ""
             PsiContextVariable.TARGET_TEST_FILE_NAME -> ""
             PsiContextVariable.UNDER_TEST_METHOD_CODE -> ""
+            PsiContextVariable.CHANGE_COUNT -> calculateChangeCount(psiElement)
+            PsiContextVariable.LINE_COUNT -> calculateLineCount(psiElement)
+            PsiContextVariable.COMPLEXITY_COUNT -> ""
         }
     }
 }
diff --git a/...on/src/main/kotlin/com/phodal/shirelang/python/provider/ShirePythonPsiVariableProvider.kt b/...on/src/main/kotlin/com/phodal/shirelang/python/provider/ShirePythonPsiVariableProvider.kt
@@ -108,6 +108,9 @@ class ShirePythonPsiVariableProvider : PsiContextVariableProvider {
 
             PsiContextVariable.SIMILAR_CODE -> TODO()
             PsiContextVariable.STRUCTURE -> TODO()
+            PsiContextVariable.CHANGE_COUNT -> calculateChangeCount(psiElement)
+            PsiContextVariable.LINE_COUNT -> calculateLineCount(psiElement)
+            PsiContextVariable.COMPLEXITY_COUNT -> ""
         }
     }
 }
diff --git a/shirelang/src/main/kotlin/com/phodal/shirelang/provider/ShirePsiVariableProvider.kt b/shirelang/src/main/kotlin/com/phodal/shirelang/provider/ShirePsiVariableProvider.kt
@@ -31,6 +31,9 @@ class ShirePsiVariableProvider : PsiContextVariableProvider {
             PsiContextVariable.CALLED_METHOD -> ""
             PsiContextVariable.SIMILAR_CODE -> ""
             PsiContextVariable.STRUCTURE -> ""
+            PsiContextVariable.CHANGE_COUNT -> calculateChangeCount(psiElement)
+            PsiContextVariable.LINE_COUNT -> calculateLineCount(psiElement)
+            PsiContextVariable.COMPLEXITY_COUNT -> ""
         }
     }
 }