Готовые хендлеры для User Longpoll V10

longpoll-user/v10/doc.go

@@ -0,0 +1,12 @@
+/*
+Package wrapper implement User Long Poll API wrapper v10.
+
+	w := wrapper.NewWrapper(&lp)
+
+	w.OnNewMessage(func(m wrapper.NewMessage) {
+		fmt.Printf("4 wrapper.NewMessage: %v\n", m)
+	})
+
+Documentation https://vk.com/dev/using_longpoll
+*/
+package wrapper // import "github.com/SevereCloud/vksdk/longpoll-user/v10"

longpoll-user/v10/errors.go

@@ -0,0 +1,3 @@
+package wrapper // import "github.com/SevereCloud/vksdk/longpoll-user/v10"
+
+const errFmtTooShortArray = "cannot parse array into Go struct %s: expected at least %d element(s), got %d"

longpoll-user/v10/extra.go

@@ -0,0 +1,242 @@
+package wrapper // import "github.com/SevereCloud/vksdk/longpoll-user/v10"
+
+import (
+	"fmt"
+	"reflect"
+	"time"
+
+	"github.com/SevereCloud/vksdk/object"
+)
+
+// Action struct for AdditionalData
+type Action struct {
+	SourceAct         string // Service action name with multiple dialogs
+	SourceMid         string // User ID to whom the service action concerns
+	SourceText        string
+	SourceOldText     string
+	SourceMessage     string
+	SourceChatLocalID string
+}
+
+// Possible values for Action.SourceAct
+const (
+	ChatCreate           = "chat_create"              // create chat
+	ChatTitleUpdate      = "chat_title_update"        // change chat name
+	ChatPhotoUpdate      = "chat_photo_update"        // change chat photo
+	ChatInviteUser       = "chat_invite_user"         // invite user to chat
+	ChatInviteUserByLink = "chat_invite_user_by_link" // invite user to chat by link
+	ChatPinMessage       = "chat_pin_message"         // pin message
+	ChatUnpinMessage     = "chat_unpin_message"       // unpin message
+	ChatKickUser         = "chat_kick_user"           // kick out user from chat
+)
+
+func (result *Action) parse(v map[string]interface{}) {
+	if sourceAct, ok := v["source_act"].(string); ok {
+		result.SourceAct = sourceAct
+
+		switch sourceAct {
+		case ChatCreate:
+			if sourceText, ok := v["source_text"].(string); ok {
+				result.SourceText = sourceText
+			}
+
+		case ChatTitleUpdate:
+			if sourceText, ok := v["source_text"].(string); ok {
+				result.SourceText = sourceText
+			}
+
+			if sourceOldText, ok := v["source_old_text"].(string); ok {
+				result.SourceOldText = sourceOldText
+			}
+
+		case ChatInviteUser, ChatKickUser:
+			if sourceMid, ok := v["source_mid"].(string); ok {
+				result.SourceMid = sourceMid
+			}
+
+		case ChatPinMessage, ChatUnpinMessage:
+			if sourceMid, ok := v["source_mid"].(string); ok {
+				result.SourceMid = sourceMid
+			}
+
+			if sourceMessage, ok := v["source_message"].(string); ok {
+				result.SourceMessage = sourceMessage
+			}
+
+			if sourceChatLocalID, ok := v["source_chat_local_id"].(string); ok {
+				result.SourceChatLocalID = sourceChatLocalID
+			}
+		}
+	}
+}
+
+// Keyboard struct.
+// Describes bot keyboard.
+type Keyboard struct {
+	OneTime bool
+	Inline  bool
+	Buttons []object.MessagesKeyboardButton
+}
+
+// AdditionalData struct. If mode contains the flag 2 along with text and the
+// message topic, a JSON-object may be passed. This object contains media
+// attachments or other additional information. Descriptions of the object
+// fields are listed below.
+type AdditionalData struct {
+	Title     string // Message's subject.
+	RefSource string
+	From      string // User ID of who sent the message if the message is from a chat
+	// FromAdmin ID of the administrator who sent the message. It is returned for
+	// messages sent from a community (only for community administrators).
+	FromAdmin   string
+	Emoji       string // Message contains emoji.
+	HasTemplate string
+	Mentions    []int
+	MarkedUsers []int
+	Keyboard    Keyboard // Bot keyboard
+	Action
+}
+
+func (result *AdditionalData) parse(v map[string]interface{}) {
+	if title, ok := v["title"].(string); ok {
+		result.Title = title
+	}
+
+	if refSource, ok := v["ref_source"].(string); ok {
+		result.RefSource = refSource
+	}
+
+	if from, ok := v["from"].(string); ok {
+		result.From = from
+	}
+
+	if fromAdmin, ok := v["from_admin"].(string); ok {
+		result.FromAdmin = fromAdmin
+	}
+
+	if emoji, ok := v["emoji"].(string); ok {
+		result.Emoji = emoji
+	}
+
+	result.Action.parse(v)
+}
+
+// LongPollAttachments type
+type LongPollAttachments map[string]interface{}
+
+// Message object, returning by 3,4,5 and 18
+type Message struct {
+	MessageID             int
+	Flags                 MessageFlag
+	PeerID                int       // destination ID
+	Timestamp             time.Time // message sent time
+	Text                  string    // message text
+	AdditionalData        AdditionalData
+	Attachments           LongPollAttachments // attachments, if mode = 2 was chosen
+	RandomID              int
+	ConversationMessageID int
+	EditTimestamp         time.Time
+}
+
+// Parse func for Message
+func (result *Message) parseMessage(structName string, i []interface{}) error {
+	length := len(i)
+
+	if length < 10 {
+		return fmt.Errorf(errFmtTooShortArray, structName, 10, length)
+	}
+
+	if v, ok := i[1].(float64); ok {
+		result.MessageID = int(v)
+	}
+
+	if v, ok := i[2].(float64); ok {
+		result.Flags = MessageFlag(int(v))
+	}
+
+	if v, ok := i[3].(float64); ok {
+		result.PeerID = int(v)
+	}
+
+	if length > 4 {
+		if v, ok := i[4].(float64); ok {
+			result.Timestamp = time.Unix(int64(v), 0)
+		}
+
+		if v, ok := i[5].(string); ok {
+			result.Text = v
+		}
+
+		if v, ok := i[6].(map[string]interface{}); ok {
+			result.AdditionalData.parse(v)
+		}
+
+		if v, ok := i[7].(map[string]interface{}); ok {
+			result.Attachments = v
+		}
+
+		if length > 8 && length < 11 {
+			if v, ok := i[8].(float64); ok {
+				result.ConversationMessageID = int(v)
+			}
+
+			if v, ok := i[9].(float64); ok {
+				result.EditTimestamp = time.Unix(int64(v), 0)
+			}
+		} else { // if mode = 128
+			if v, ok := i[8].(float64); ok {
+				result.RandomID = int(v)
+			}
+
+			if v, ok := i[9].(float64); ok {
+				result.ConversationMessageID = int(v)
+			}
+
+			if v, ok := i[10].(float64); ok {
+				result.EditTimestamp = time.Unix(int64(v), 0)
+			}
+		}
+	}
+
+	return nil
+}
+
+func interfaceToStringIntMap(m interface{}) (map[string]int, error) {
+	reflectedMap := reflect.ValueOf(m)
+	if reflectedMap.Kind() != reflect.Map {
+		return nil, fmt.Errorf("expected a slice, got %T", m)
+	}
+
+	result := make(map[string]int, reflectedMap.Len())
+
+	for _, key := range reflectedMap.MapKeys() {
+		v, ok := reflectedMap.MapIndex(key).Interface().(float64)
+		if !ok {
+			return nil, fmt.Errorf("cast failed, value type: %T", reflectedMap.MapIndex(key).Interface())
+		}
+
+		result[key.String()] = int(v)
+	}
+
+	return result, nil
+}
+
+func interfaceToIDSlice(slice interface{}) ([]int, error) {
+	reflectedSlice := reflect.ValueOf(slice)
+	if reflectedSlice.Kind() != reflect.Slice {
+		return nil, fmt.Errorf("expected a slice, got %T", slice)
+	}
+
+	result := make([]int, reflectedSlice.Len())
+
+	for i := 0; i < reflectedSlice.Len(); i++ {
+		v, ok := reflectedSlice.Index(i).Interface().(float64)
+		if !ok {
+			return nil, fmt.Errorf("cast failed, value type: %T", reflectedSlice.Index(i).Interface())
+		}
+
+		result[i] = int(v)
+	}
+
+	return result, nil
+}

longpoll-user/v10/extra_test.go

@@ -0,0 +1,46 @@
+package wrapper
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func Test_interfaceToStringIntMap(t *testing.T) {
+	var i interface{}
+
+	err := json.Unmarshal([]byte(`{"v":0}`), &i)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	m, err := interfaceToStringIntMap(i)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	if m["v"] != 0 {
+		t.Error("expected v is 0")
+	}
+}
+
+func Test_interfaceToIDSlice(t *testing.T) {
+	var i interface{}
+
+	err := json.Unmarshal([]byte(`[0, 9]`), &i)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	m, err := interfaceToIDSlice(i)
+	if err != nil {
+		t.Error(err)
+		return
+	}
+
+	if m[1] != 9 {
+		t.Error("expected v is 0")
+	}
+}