feat(devins-compiler): add support for custom commands #100

Add support for custom commands in the data provider, allowing the compiler to use custom commands defined in the project. This enables the DevInsCompiler to compile custom commands and integrate them into the output.

Author: phodal

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/DevInsCompiler.kt

@@ -4,6 +4,7 @@ import cc.unitmesh.devti.agent.model.CustomAgentConfig
 import cc.unitmesh.devti.custom.compile.VariableTemplateCompiler
 import cc.unitmesh.devti.language.compiler.exec.*
 import cc.unitmesh.devti.language.dataprovider.BuiltinCommand
+import cc.unitmesh.devti.language.dataprovider.CustomCommand
 import cc.unitmesh.devti.language.parser.CodeBlockElement
 import cc.unitmesh.devti.language.psi.DevInFile
 import cc.unitmesh.devti.language.psi.DevInTypes
@@ -62,6 +63,18 @@ class DevInsCompiler(
             DevInTypes.COMMAND_START -> {
                 val command = BuiltinCommand.fromString(id?.text ?: "")
                 if (command == null) {
+                    CustomCommand.fromString(myProject, id?.text ?: "")?.let { cmd ->
+                        DevInFile.fromString(myProject, cmd.content).let { file ->
+                            DevInsCompiler(myProject, file).compile().let {
+                                output.append(it.output)
+                                result.hasError = it.hasError
+                            }
+                        }
+
+                        return
+                    }
+
+
                     output.append(used.text)
                     logger.warn("Unknown command: ${id?.text}")
                     result.hasError = true

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/dataprovider/CustomCommand.kt

@@ -25,5 +25,9 @@ data class CustomCommand(
             val content = file.inputStream.readBytes().toString(Charsets.UTF_8)
             return CustomCommand(file.nameWithoutExtension, content)
         }
+
+        fun fromString(project: Project, agentName: String): CustomCommand? {
+            return all(project).find { it.commandName == agentName }
+        }
     }
 }