feat: botkb package + buttons.go + MessageKeyboard type

Author: trakhimenok

botkb/buttons.go

@@ -0,0 +1,33 @@
+package botkb
+
+type ButtonType int
+
+type Button interface {
+	ButtonType() ButtonType
+}
+
+const (
+	ButtonTypeData ButtonType = iota
+	ButtonTypeURL
+)
+
+var _ Button = (*DataButton)(nil)
+var _ Button = (*UrlButton)(nil)
+
+type DataButton struct {
+	Text string
+	Data string
+}
+
+func (DataButton) ButtonType() ButtonType {
+	return ButtonTypeData
+}
+
+type UrlButton struct {
+	Text string
+	URL  string
+}
+
+func (UrlButton) ButtonType() ButtonType {
+	return ButtonTypeURL
+}

botkb/keyboard.go

@@ -0,0 +1,51 @@
+package botkb
+
+// KeyboardType defines MessageKeyboard type
+type KeyboardType int
+
+//goland:noinspection GoUnusedConst
+const (
+	// KeyboardTypeNone for no MessageKeyboard
+	// Used by: Telegram
+	KeyboardTypeNone KeyboardType = iota
+
+	// KeyboardTypeHide commands to hide MessageKeyboard
+	// Used by: Telegram
+	KeyboardTypeHide
+
+	// KeyboardTypeInline for inline MessageKeyboard
+	// Used by: Telegram
+	KeyboardTypeInline
+
+	// KeyboardTypeBottom for bottom MessageKeyboard
+	// Used by: Telegram
+	KeyboardTypeBottom
+
+	// KeyboardTypeForceReply to force reply from a user
+	// Used by: Telegram
+	KeyboardTypeForceReply
+)
+
+// Keyboard defines MessageKeyboard
+type Keyboard interface {
+	// KeyboardType defines MessageKeyboard type
+	KeyboardType() KeyboardType
+}
+
+var _ Keyboard = (*MessageKeyboard)(nil)
+
+type MessageKeyboard struct {
+	kbType  KeyboardType
+	Buttons [][]Button
+}
+
+func (k MessageKeyboard) KeyboardType() KeyboardType {
+	return k.kbType
+}
+
+func NewMessageKeyboard(kbType KeyboardType, buttons ...[]Button) *MessageKeyboard {
+	return &MessageKeyboard{
+		kbType:  kbType,
+		Buttons: buttons,
+	}
+}

botkb/keyboard_test.go

@@ -0,0 +1,92 @@
+package botkb
+
+import (
+	"reflect"
+	"testing"
+)
+
+func TestNewMessageKeyboard(t *testing.T) {
+	tests := []struct {
+		name    string
+		kbType  KeyboardType
+		buttons [][]Button
+		want    *MessageKeyboard
+	}{
+		{
+			name:    "Empty keyboard with KeyboardTypeNone",
+			kbType:  KeyboardTypeNone,
+			buttons: [][]Button{},
+			want: &MessageKeyboard{
+				kbType:  KeyboardTypeNone,
+				Buttons: [][]Button{},
+			},
+		},
+		{
+			name:    "Empty keyboard with KeyboardTypeInline",
+			kbType:  KeyboardTypeInline,
+			buttons: [][]Button{},
+			want: &MessageKeyboard{
+				kbType:  KeyboardTypeInline,
+				Buttons: [][]Button{},
+			},
+		},
+		{
+			name:   "Keyboard with one row of data buttons",
+			kbType: KeyboardTypeBottom,
+			buttons: [][]Button{
+				{
+					&DataButton{Text: "Button 1", Data: "data1"},
+					&DataButton{Text: "Button 2", Data: "data2"},
+				},
+			},
+			want: &MessageKeyboard{
+				kbType: KeyboardTypeBottom,
+				Buttons: [][]Button{
+					{
+						&DataButton{Text: "Button 1", Data: "data1"},
+						&DataButton{Text: "Button 2", Data: "data2"},
+					},
+				},
+			},
+		},
+		{
+			name:   "Keyboard with multiple rows of mixed buttons",
+			kbType: KeyboardTypeInline,
+			buttons: [][]Button{
+				{
+					&DataButton{Text: "Button 1", Data: "data1"},
+					&UrlButton{Text: "URL 1", URL: "https://example.com"},
+				},
+				{
+					&DataButton{Text: "Button 2", Data: "data2"},
+				},
+			},
+			want: &MessageKeyboard{
+				kbType: KeyboardTypeInline,
+				Buttons: [][]Button{
+					{
+						&DataButton{Text: "Button 1", Data: "data1"},
+						&UrlButton{Text: "URL 1", URL: "https://example.com"},
+					},
+					{
+						&DataButton{Text: "Button 2", Data: "data2"},
+					},
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := NewMessageKeyboard(tt.kbType, tt.buttons...)
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("NewMessageKeyboard() = %v, want %v", got, tt.want)
+			}
+
+			// Also test the KeyboardType method
+			if got.KeyboardType() != tt.kbType {
+				t.Errorf("MessageKeyboard.KeyboardType() = %v, want %v", got.KeyboardType(), tt.kbType)
+			}
+		})
+	}
+}