1 files changed, 16 insertions, 7 deletions
diff --git a/internal/client/api/api.go b/internal/client/api/api.go
index 56cf054..9cfa1d4 100644
--- a/internal/client/api/api.go
+++ b/internal/client/api/api.go
@@ -1,29 +1,38 @@
 package api
 
 import (
+	"errors"
+	"fmt"
 	"log"
 
 	tea "github.com/charmbracelet/bubbletea"
 
 	"github.com/kyren223/eko/internal/client/gateway"
+	"github.com/kyren223/eko/internal/data"
 	"github.com/kyren223/eko/internal/packet"
 	"github.com/kyren223/eko/pkg/assert"
 )
 
+type AppendMessage data.Message
+
 func SendMessage(message string) tea.Cmd {
 	return func() tea.Msg {
 		log.Println("request SendMessage sent")
 		request := packet.SendMessage{Content: message}
-		responsePayload, ok := <-gateway.Send(&request)
+		response, ok := <-gateway.Send(&request)
 		if !ok {
-			log.Println("request timeout")
-			return ""
+			log.Println()
+			return errors.New("request timeout")
 		}
 		log.Println("request SendMessage received response")
 
-		response := responsePayload.(*packet.ErrorMessage)
-		assert.Assert(ok, "server should return an ErrorMessage response for a SendMessage request")
-		assert.Assert(response.IsOk(), "server should always return an OK if we didn't mess up")
-		return ""
+		switch response := response.(type) {
+		case *packet.ErrorMessage:
+			return errors.New(response.Error)
+		case *packet.Messages:
+			assert.Assert(len(response.Messages) == 1, "server must return only the one message that was sent")
+			return AppendMessage(response.Messages[0])
+		}
+		return fmt.Errorf("received invalid response from server: %v", response.Type())
 	}
 }