Carlos Florêncio
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
package client
|
|
|
|
import (
|
|
"github.com/kandev/kandev/internal/agentctl/types"
|
|
)
|
|
|
|
func dispatchWorkspaceShellMessages(msg types.WorkspaceStreamMessage, callbacks WorkspaceStreamCallbacks) bool {
|
|
switch msg.Type {
|
|
case types.WorkspaceMessageTypeShellOutput:
|
|
if callbacks.OnShellOutput != nil {
|
|
callbacks.OnShellOutput(msg.Data)
|
|
}
|
|
case types.WorkspaceMessageTypeShellExit:
|
|
if callbacks.OnShellExit != nil {
|
|
callbacks.OnShellExit(msg.Code)
|
|
}
|
|
case types.WorkspaceMessageTypeConnected:
|
|
if callbacks.OnConnected != nil {
|
|
callbacks.OnConnected()
|
|
}
|
|
case types.WorkspaceMessageTypeError:
|
|
if callbacks.OnError != nil {
|
|
callbacks.OnError(msg.Error)
|
|
}
|
|
default:
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func dispatchWorkspaceGitMessages(msg types.WorkspaceStreamMessage, callbacks WorkspaceStreamCallbacks) bool {
|
|
switch msg.Type {
|
|
case types.WorkspaceMessageTypeGitStatus:
|
|
if callbacks.OnGitStatus != nil && msg.GitStatus != nil {
|
|
callbacks.OnGitStatus(msg.GitStatus)
|
|
}
|
|
case types.WorkspaceMessageTypeGitCommit:
|
|
if callbacks.OnGitCommit != nil && msg.GitCommit != nil {
|
|
callbacks.OnGitCommit(msg.GitCommit)
|
|
}
|
|
case types.WorkspaceMessageTypeGitReset:
|
|
if callbacks.OnGitReset != nil && msg.GitReset != nil {
|
|
callbacks.OnGitReset(msg.GitReset)
|
|
}
|
|
case types.WorkspaceMessageTypeBranchSwitch:
|
|
if callbacks.OnBranchSwitch != nil && msg.BranchSwitch != nil {
|
|
callbacks.OnBranchSwitch(msg.BranchSwitch)
|
|
}
|
|
case types.WorkspaceMessageTypeFileChange:
|
|
if callbacks.OnFileChange != nil && msg.FileChange != nil {
|
|
callbacks.OnFileChange(msg.FileChange)
|
|
}
|
|
default:
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
func dispatchWorkspaceProcessMessages(msg types.WorkspaceStreamMessage, callbacks WorkspaceStreamCallbacks) {
|
|
switch msg.Type {
|
|
case types.WorkspaceMessageTypeProcessOutput:
|
|
if callbacks.OnProcessOutput != nil && msg.ProcessOutput != nil {
|
|
callbacks.OnProcessOutput(msg.ProcessOutput)
|
|
}
|
|
case types.WorkspaceMessageTypeProcessStatus:
|
|
if callbacks.OnProcessStatus != nil && msg.ProcessStatus != nil {
|
|
callbacks.OnProcessStatus(msg.ProcessStatus)
|
|
}
|
|
}
|
|
}
|
|
|
|
// dispatchWorkspaceMessage routes a workspace stream message to the appropriate callback.
|
|
func dispatchWorkspaceMessage(msg types.WorkspaceStreamMessage, callbacks WorkspaceStreamCallbacks) {
|
|
if dispatchWorkspaceShellMessages(msg, callbacks) {
|
|
return
|
|
}
|
|
if dispatchWorkspaceGitMessages(msg, callbacks) {
|
|
return
|
|
}
|
|
dispatchWorkspaceProcessMessages(msg, callbacks)
|
|
}
|