Поддержка zstd словаря

api/README.md

@@ -98,9 +98,7 @@ VK API способно возвращать ответ в виде [MessagePack
 СЛОМАННУЮ КОДИРОВКУ.
 
 Для сжатия, вместо классического gzip, можно использовать
-[zstd](https://github.com/facebook/zstd). Сейчас vksdk поддерживает zstd без
-словаря. Если кто знает как получать словарь,
-[отпишитесь сюда](https://github.com/SevereCloud/vksdk/issues/180).
+[zstd](https://github.com/facebook/zstd).
 
 ```go
 vk := api.NewVK(os.Getenv("USER_TOKEN"))
@@ -128,6 +126,18 @@ if err != nil {
 log.Println("msgpack:", len(r)) // msgpack: 650775
 ```
 
+Чтобы загрузить и использовать zstd словарь, воспользуйтесь
+методом `vk.UpdateZstdDict()`
+
+```go
+vk.EnableZstd()
+
+err := vk.UpdateZstdDict()
+if err != nil {
+	log.Fatal(err)
+}
+```
+
 ### Запрос любого метода
 
 Пример запроса [users.get](https://vk.com/dev/users.get)

api/account.go

@@ -108,6 +108,21 @@ func (vk *VK) AccountGetPushSettings(params Params) (response AccountGetPushSett
 	return
 }
 
+// AccountGetZSTDDictResponse struct.
+type AccountGetZSTDDictResponse struct {
+	Link    string `json:"link"`
+	Version string `json:"version"`
+	Hash    string `json:"hash"`
+}
+
+// AccountGetZSTDDict method.
+//
+// https://vk.com/dev/account.getZSTDDict
+func (vk *VK) AccountGetZSTDDict(params Params) (response AccountGetZSTDDictResponse, err error) {
+	err = vk.RequestUnmarshal("account.getZSTDDict", &response, params)
+	return
+}
+
 // AccountRegisterDevice subscribes an iOS/Android/Windows/Mac based device to receive push notifications.
 //
 // https://vk.com/dev/account.registerDevice

api/account_test.go

@@ -124,6 +124,16 @@ func TestVK_AccountGetPushSettings(t *testing.T) {
 	noError(t, err)
 }
 
+func TestVK_AccountGetZSTDDict(t *testing.T) {
+	t.Parallel()
+
+	needUserToken(t)
+
+	vkUser.EnableZstd()
+	err := vkUser.UpdateZstdDict()
+	noError(t, err)
+}
+
 // func TestVK_AccountRegisterDevice(t *testing.T) {
 // TODO: Add test cases
 // }

api/api.go

@@ -98,6 +98,9 @@ type VK struct {
 	msgpack bool
 	zstd    bool
 
+	zstdVersion string
+	zstdDict    zstd.DOption
+
 	mux      sync.Mutex
 	lastTime time.Time
 	rps      int
@@ -260,6 +263,10 @@ func (vk *VK) DefaultHandler(method string, sliceParams ...Params) (Response, er
 
 		req.Header.Set("Accept-Encoding", acceptEncoding)
 
+		if vk.zstdVersion != "" {
+			req.Header.Set("x-zstd-dict-version", vk.zstdVersion)
+		}
+
 		var reader io.Reader
 
 		resp, err := vk.Client.Do(req)
@@ -269,7 +276,7 @@ func (vk *VK) DefaultHandler(method string, sliceParams ...Params) (Response, er
 
 		switch resp.Header.Get("Content-Encoding") {
 		case "zstd":
-			reader, _ = zstd.NewReader(resp.Body)
+			reader, _ = zstd.NewReader(resp.Body, vk.zstdDict)
 		case "gzip":
 			reader, _ = gzip.NewReader(resp.Body)
 		default:
@@ -370,6 +377,55 @@ func (vk *VK) EnableZstd() {
 	vk.zstd = true
 }
 
+// SetZstdDict set vk zstd dict.
+func (vk *VK) SetZstdDict(data []byte) error {
+	// zstd dictionary start with Magic_Number:  4 bytes ID, value 0xEC30A437,
+	// little-endian format.
+	//
+	// https://datatracker.ietf.org/doc/html/rfc8878#section-5
+	indexMagicNumber := bytes.Index(data, []byte{0x37, 0xa4, 0x30, 0xec})
+	if indexMagicNumber == -1 {
+		return fmt.Errorf("api: zstd dictionary not found. %w", zstd.ErrMagicMismatch)
+	}
+
+	vk.zstdVersion = string(data[:indexMagicNumber])
+	vk.zstdDict = zstd.WithDecoderDicts(data[indexMagicNumber:])
+
+	return nil
+}
+
+// LoadZstdDict return zstd dict.
+func (vk *VK) LoadZstdDict() (data []byte, err error) {
+	r, err := vk.AccountGetZSTDDict(nil)
+	if err != nil {
+		return
+	}
+
+	resp, err := vk.Client.Get(r.Link)
+	if err != nil {
+		return
+	}
+
+	data, err = io.ReadAll(resp.Body)
+	if err != nil {
+		return
+	}
+
+	err = resp.Body.Close()
+
+	return
+}
+
+// UpdateZstdDict update vk zstd dict.
+func (vk *VK) UpdateZstdDict() error {
+	data, err := vk.LoadZstdDict()
+	if err != nil {
+		return err
+	}
+
+	return vk.SetZstdDict(data)
+}
+
 func fmtReflectValue(value reflect.Value, depth int) string {
 	switch f := value; value.Kind() {
 	case reflect.Invalid:

doc.go

@@ -7,6 +7,6 @@ package vksdk
 
 // Module constants.
 const (
-	Version = "2.13.0"
+	Version = "2.14.0"
 	API     = "5.131"
 )