Class/ function code getting APIs; Git related APIs

ide-former-plugin/gradle.properties

@@ -18,7 +18,7 @@ platformVersion = 2023.2.4
 # Plugin Dependencies -> https://plugins.jetbrains.com/docs/intellij/plugin-dependencies.html
 # To find actual PythonCore version go to https://plugins.jetbrains.com/plugin/631-python/versions and check, which version is matched with platformVersion
 # Example: platformPlugins = com.intellij.java, com.jetbrains.php:203.4449.22
-platformPlugins = com.intellij.java, org.jetbrains.kotlin, PythonCore:232.10203.2
+platformPlugins = com.intellij.java, org.jetbrains.kotlin, PythonCore:232.10203.2, Git4Idea
 
 # Gradle Releases -> https://github.com/gradle/gradle/releases
 gradleVersion = 8.3

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileClasses/FileClasses.kt

@@ -3,6 +3,6 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileClasse
 import org.jetbrains.research.ideFormerPlugin.api.models.IdeApiMethod
 
 interface FileClasses : IdeApiMethod {
-    fun getClassesNames(): List<String>?
+    fun getClassesNames(): List<String>
     fun getClassCode(className: String): String?
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileClasses/JavaFileClasses.kt

@@ -2,7 +2,7 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileClasse
 
 import com.intellij.psi.PsiClass
 import com.intellij.psi.PsiDirectory
-import org.jetbrains.research.ideFormerPlugin.api.models.utils.getFilePsiElementsOfType
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.*
 
 class JavaFileClasses(
     private val projectDirectory: PsiDirectory,
@@ -14,10 +14,15 @@ class JavaFileClasses(
         javaClasses = getFilePsiElementsOfType<PsiClass>(projectDirectory, javaFileName)
     }
 
-    override fun getClassesNames(): List<String>? =
-        javaClasses?.mapNotNull { it.name }
+    override fun getClassesNames(): List<String> {
+        if (javaClasses == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return javaClasses!!.mapNotNull { it.name }
+    }
 
-    override fun getClassCode(className: String): String? = javaClasses
-        ?.firstOrNull { it.name == className }
-        ?.text
+    override fun getClassCode(className: String): String? {
+        if (javaClasses == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return javaClasses!!
+            .firstOrNull { it.name == className }
+            ?.text
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileClasses/KtFileClasses.kt

@@ -2,6 +2,7 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileClasse
 
 import com.intellij.psi.PsiDirectory
 import org.jetbrains.kotlin.psi.KtClass
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.UNCALLED_EXECUTE_BEFORE_RESULT_GETTING
 import org.jetbrains.research.ideFormerPlugin.api.models.utils.getFilePsiElementsOfType
 
 class KtFileClasses(
@@ -14,9 +15,15 @@ class KtFileClasses(
         ktClasses = getFilePsiElementsOfType<KtClass>(projectDirectory, ktFileName)
     }
 
-    override fun getClassesNames(): List<String>? = ktClasses?.mapNotNull { it.name }
+    override fun getClassesNames(): List<String> {
+        if (ktClasses == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return ktClasses!!.mapNotNull { it.name }
+    }
 
-    override fun getClassCode(className: String): String? = ktClasses
-        ?.firstOrNull { it.name == className }
-        ?.text
+    override fun getClassCode(className: String): String? {
+        if (ktClasses == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return ktClasses!!
+            .firstOrNull { it.name == className }
+            ?.text
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileClasses/PyFileClasses.kt

@@ -2,6 +2,7 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileClasse
 
 import com.intellij.psi.PsiDirectory
 import com.jetbrains.python.psi.PyClass
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.UNCALLED_EXECUTE_BEFORE_RESULT_GETTING
 import org.jetbrains.research.ideFormerPlugin.api.models.utils.getFilePsiElementsOfType
 
 class PyFileClasses(
@@ -14,9 +15,15 @@ class PyFileClasses(
         pyClasses = getFilePsiElementsOfType<PyClass>(projectDirectory, pyFileName)
     }
 
-    override fun getClassesNames(): List<String>? = pyClasses?.mapNotNull { it.name }
+    override fun getClassesNames(): List<String> {
+        if (pyClasses == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return pyClasses!!.mapNotNull { it.name }
+    }
 
-    override fun getClassCode(className: String): String? = pyClasses
-        ?.firstOrNull { it.name == className }
-        ?.text
+    override fun getClassCode(className: String): String? {
+        if (pyClasses == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return pyClasses!!
+            .firstOrNull { it.name == className }
+            ?.text
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileFunctions/FileFunctions.kt

@@ -3,6 +3,6 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileFuncti
 import org.jetbrains.research.ideFormerPlugin.api.models.IdeApiMethod
 
 interface FileFunctions : IdeApiMethod {
-    fun getFunctionsNames(): List<String>?
+    fun getFunctionsNames(): List<String>
     fun getFunctionCode(functionName: String): String?
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileFunctions/JavaFileFunctions.kt

@@ -2,6 +2,7 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileFuncti
 
 import com.intellij.psi.PsiDirectory
 import com.intellij.psi.PsiMethod
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.UNCALLED_EXECUTE_BEFORE_RESULT_GETTING
 import org.jetbrains.research.ideFormerPlugin.api.models.utils.getFilePsiElementsOfType
 
 class JavaFileFunctions(
@@ -14,8 +15,15 @@ class JavaFileFunctions(
         javaFunctions = getFilePsiElementsOfType<PsiMethod>(projectDirectory, javaFileName)
     }
 
-    override fun getFunctionsNames(): List<String>? = javaFunctions?.map { it.name }
-    override fun getFunctionCode(functionName: String): String? = javaFunctions
-        ?.firstOrNull { it.name == functionName }
-        ?.text
+    override fun getFunctionsNames(): List<String> {
+        if (javaFunctions == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return javaFunctions!!.map { it.name }
+    }
+
+    override fun getFunctionCode(functionName: String): String? {
+        if (javaFunctions == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return javaFunctions!!
+            .firstOrNull { it.name == functionName }
+            ?.text
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileFunctions/KtFileFunctions.kt

@@ -2,6 +2,7 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileFuncti
 
 import com.intellij.psi.PsiDirectory
 import org.jetbrains.kotlin.psi.KtFunction
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.UNCALLED_EXECUTE_BEFORE_RESULT_GETTING
 import org.jetbrains.research.ideFormerPlugin.api.models.utils.getFilePsiElementsOfType
 
 class KtFileFunctions(
@@ -14,8 +15,15 @@ class KtFileFunctions(
         ktFunctions = getFilePsiElementsOfType<KtFunction>(projectDirectory, ktFileName)
     }
 
-    override fun getFunctionsNames(): List<String>? = ktFunctions?.mapNotNull { it.name }
-    override fun getFunctionCode(functionName: String): String? = ktFunctions
-        ?.firstOrNull { it.name == functionName }
-        ?.text
+    override fun getFunctionsNames(): List<String> {
+        if (ktFunctions == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return ktFunctions!!.mapNotNull { it.name }
+    }
+
+    override fun getFunctionCode(functionName: String): String? {
+        if (ktFunctions == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return ktFunctions!!
+            .firstOrNull { it.name == functionName }
+            ?.text
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileFunctions/PyFileFunctions.kt

@@ -2,6 +2,7 @@ package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileFuncti
 
 import com.intellij.psi.PsiDirectory
 import com.jetbrains.python.psi.PyFunction
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.UNCALLED_EXECUTE_BEFORE_RESULT_GETTING
 import org.jetbrains.research.ideFormerPlugin.api.models.utils.getFilePsiElementsOfType
 
 class PyFileFunctions(
@@ -14,8 +15,15 @@ class PyFileFunctions(
         pyFunctions = getFilePsiElementsOfType<PyFunction>(projectDirectory, pyFileName)
     }
 
-    override fun getFunctionsNames(): List<String>? = pyFunctions?.mapNotNull { it.name }
-    override fun getFunctionCode(functionName: String): String? = pyFunctions
-        ?.firstOrNull { it.name == functionName }
-        ?.text
+    override fun getFunctionsNames(): List<String> {
+        if (pyFunctions == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return pyFunctions!!.mapNotNull { it.name }
+    }
+
+    override fun getFunctionCode(functionName: String): String? {
+        if (pyFunctions == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return pyFunctions!!
+            .firstOrNull { it.name == functionName }
+            ?.text
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/FileText.kt → ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileText/GetFileText.kt

@@ -1,19 +1,23 @@
-package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated
+package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileText
 
 import com.intellij.psi.PsiDirectory
 import org.jetbrains.research.ideFormerPlugin.api.models.IdeApiMethod
-import org.jetbrains.research.ideFormerPlugin.api.models.utils.findFileRecursively
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.*
 
-class FileText(
+class GetFileText(
     private val projectDirectory: PsiDirectory,
     private val fileName: String
 ) : IdeApiMethod {
     private var fileText: String? = null
 
     override fun execute() {
+        projectDirectory.refresh()
         val psiFile = projectDirectory.findFileRecursively(fileName)
         fileText = psiFile.text
     }
 
-    fun getFileText(): String? = fileText
+    fun getFileText(): String {
+        if (fileText == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+        return fileText!!
+    }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileRelated/fileText/SetFileText.kt

@@ -0,0 +1,30 @@
+package org.jetbrains.research.ideFormerPlugin.api.models.fileRelated.fileText
+
+import com.intellij.psi.PsiDirectory
+import org.jetbrains.research.ideFormerPlugin.api.models.ReversibleApiMethod
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.*
+
+class SetFileText(
+    private val projectDirectory: PsiDirectory,
+    private val fileName: String,
+    private val newText: String
+) : ReversibleApiMethod {
+    private var previousText: String? = null
+
+    override fun execute() {
+        projectDirectory.refresh()
+        val psiFile = projectDirectory.findFileRecursively(fileName)
+        previousText = psiFile.text
+        psiFile.setText(newText)
+    }
+
+    override fun reverse() {
+        if (previousText == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+
+        previousText!!.let {
+            val psiFile = projectDirectory.findFileRecursively(fileName)
+            psiFile.setText(it)
+            previousText = null
+        }
+    }
+}

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileSystemRelated/ChangeDirectory.kt

@@ -1,28 +1,32 @@
 package org.jetbrains.research.ideFormerPlugin.api.models.fileSystemRelated
 
 import com.intellij.psi.PsiDirectory
-import org.jetbrains.research.ideFormerPlugin.api.models.utils.DEFAULT_DIRECTORY_NAME
-import org.jetbrains.research.ideFormerPlugin.api.models.utils.findSubdirectoryRecursively
 import org.jetbrains.research.ideFormerPlugin.api.models.ReversibleApiMethod
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.*
 import org.jetbrains.research.ideFormerPlugin.stateKeeper.IdeStateKeeper
 
 class ChangeDirectory(
     private val ideStateKeeper: IdeStateKeeper,
     private val targetDirectoryName: String = DEFAULT_DIRECTORY_NAME
 ) : ReversibleApiMethod {
-    private var prevDir: PsiDirectory? = null
+    private var previousDirectory: PsiDirectory? = null
 
     override fun execute() {
-        val targetDir = ideStateKeeper.currentProjectDirectory.findSubdirectoryRecursively(targetDirectoryName)
+        val currentProjectDirectory = ideStateKeeper.currentProjectDirectory
+        currentProjectDirectory.refresh()
 
-        prevDir = ideStateKeeper.currentProjectDirectory
+        val targetDir = currentProjectDirectory.findDirectoryRecursively(targetDirectoryName)
+
+        previousDirectory = currentProjectDirectory
         ideStateKeeper.currentProjectDirectory = targetDir
     }
 
     override fun reverse() {
-        prevDir?.let {
+        if (previousDirectory == null) error(UNCALLED_EXECUTE_BEFORE_RESULT_GETTING)
+
+        previousDirectory!!.let {
             ideStateKeeper.currentProjectDirectory = it
-            prevDir = null
+            previousDirectory = null
         }
     }
 }

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileSystemRelated/CreateDirectory.kt

@@ -0,0 +1,20 @@
+package org.jetbrains.research.ideFormerPlugin.api.models.fileSystemRelated
+
+import com.intellij.psi.PsiDirectory
+import org.jetbrains.research.ideFormerPlugin.api.models.ReversibleApiMethod
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.createSubdirectoryByName
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.deleteSubdirectoryByName
+
+class CreateDirectory(
+    private val projectDirectory: PsiDirectory,
+    private val directoryName: String
+) : ReversibleApiMethod {
+
+    override fun execute() {
+        projectDirectory.createSubdirectoryByName(directoryName)
+    }
+
+    override fun reverse() {
+        projectDirectory.deleteSubdirectoryByName(directoryName)
+    }
+}

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileSystemRelated/CreateFile.kt

@@ -0,0 +1,21 @@
+package org.jetbrains.research.ideFormerPlugin.api.models.fileSystemRelated
+
+import com.intellij.psi.PsiDirectory
+import org.jetbrains.research.ideFormerPlugin.api.models.ReversibleApiMethod
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.*
+
+class CreateFile(
+    private val projectDirectory: PsiDirectory,
+    private val fileName: String
+) : ReversibleApiMethod {
+
+    override fun execute() {
+        projectDirectory.refresh()
+        projectDirectory.createFileByName(fileName)
+    }
+
+    override fun reverse() {
+        projectDirectory.refresh()
+        projectDirectory.deleteFileByName(fileName)
+    }
+}

ide-former-plugin/ide-core/src/main/kotlin/org/jetbrains/research/ideFormerPlugin/api/models/fileSystemRelated/DeleteFile.kt

@@ -0,0 +1,28 @@
+package org.jetbrains.research.ideFormerPlugin.api.models.fileSystemRelated
+
+import com.intellij.psi.PsiDirectory
+import org.jetbrains.research.ideFormerPlugin.api.models.ReversibleApiMethod
+import org.jetbrains.research.ideFormerPlugin.api.models.utils.*
+
+class DeleteFile(
+    private val projectDirectory: PsiDirectory,
+    private val fileName: String
+) : ReversibleApiMethod {
+    private var deletedFileText: String? = projectDirectory.findFileRecursively(fileName).text
+
+    override fun execute() {
+        projectDirectory.refresh()
+        // TODO: getting deleting file text should be here. when execute() is called it may be different
+        projectDirectory.deleteFileByName(fileName)
+    }
+
+    override fun reverse() {
+        projectDirectory.refresh()
+        val psiFile = projectDirectory.createFileByName(fileName)
+
+        deletedFileText!!.let {
+            psiFile.setText(it)
+            deletedFileText = null
+        }
+    }
+}