feat(amp): add msgfmt logic for amp module

Author: 35C4n0r

lib/msgfmt/message_box.go

@@ -79,3 +79,24 @@ func removeOpencodeMessageBox(msg string) string {
 	}
 	return strings.Join(lines, "\n")
 }
+
+func removeAmpMessageBox(msg string) string {
+	lines := strings.Split(msg, "\n")
+	msgBoxEndFound := false
+	msgBoxStartIdx := len(lines)
+	for i := len(lines) - 1; i >= 0; i-- {
+		line := strings.TrimSpace(lines[i])
+		if !msgBoxEndFound && strings.HasPrefix(line, "╰") && strings.HasSuffix(line, "╯") {
+			msgBoxEndFound = true
+		}
+		if msgBoxEndFound && strings.HasPrefix(line, "╭") && strings.HasSuffix(line, "╮") {
+			msgBoxStartIdx = i
+			break
+		}
+	}
+	formattedMsg := strings.Join(lines[:msgBoxStartIdx], "\n")
+	if len(formattedMsg) == 0 {
+		return "Welcome to Amp"
+	}
+	return formattedMsg
+}

lib/msgfmt/msgfmt.go

@@ -268,6 +268,13 @@ func formatOpencodeMessage(message string, userInput string) string {
 	return message
 }
 
+func formatAmpMessage(message string, userInput string) string {
+	message = RemoveUserInput(message, userInput, AgentTypeAmp)
+	message = removeAmpMessageBox(message)
+	message = trimEmptyLines(message)
+	return message
+}
+
 func FormatAgentMessage(agentType AgentType, message string, userInput string) string {
 	switch agentType {
 	case AgentTypeClaude:
@@ -283,7 +290,7 @@ func FormatAgentMessage(agentType AgentType, message string, userInput string) s
 	case AgentTypeCopilot:
 		return formatGenericMessage(message, userInput, agentType)
 	case AgentTypeAmp:
-		return formatGenericMessage(message, userInput, agentType)
+		return formatAmpMessage(message, userInput)
 	case AgentTypeCursor:
 		return formatGenericMessage(message, userInput, agentType)
 	case AgentTypeAuggie: