add option to filter unsupported extensions

tonkeeper · Aug 14, 2024 · 057e5aa · 057e5aa
1 parent f7107d2
commit 057e5aa
Show file tree

Hide file tree

Showing 5 changed files with 93 additions and 0 deletions.
diff --git a/api/openapi.json b/api/openapi.json
@@ -323,6 +323,22 @@
  "type": "integer"
  }
  },
+ "supportedExtensions": {
+ "description": "comma separated list supported extensions",
+ "explode": false,
+ "in": "query",
+ "name": "supported_extensions",
+ "required": false,
+ "schema": {
+ "example": [
+ "custom_payload"
+ ],
+ "items": {
+ "type": "string"
+ },
+ "type": "array"
+ }
+ },
  "targetBlockIDExtQuery": {
  "description": "target block: (workchain,shard,seqno,root_hash,file_hash)",
  "in": "query",
@@ -6468,6 +6484,9 @@
  },
  {
  "$ref": "#/components/parameters/currenciesQuery"
+ },
+ {
+ "$ref": "#/components/parameters/supportedExtensions"
  }
  ],
  "responses": {

diff --git a/api/openapi.yml b/api/openapi.yml
@@ -614,6 +614,7 @@ paths:
  parameters:
  - $ref: '#/components/parameters/accountIDParameter'
  - $ref: '#/components/parameters/currenciesQuery'
+ - $ref: '#/components/parameters/supportedExtensions'
  responses:
  '200':
  description: account jettons balances
@@ -3002,6 +3003,17 @@ components:
  items:
  type: string
  example: [ "ton", "usd", "rub" ]
+ supportedExtensions:
+ in: query
+ name: supported_extensions
+ description: "comma separated list supported extensions"
+ explode: false
+ required: false
+ schema:
+ type: array
+ items:
+ type: string
+ example: ["custom_payload"]
  currencyQuery:
  in: query
  name: currency

diff --git a/pkg/api/jetton_handlers.go b/pkg/api/jetton_handlers.go
@@ -5,6 +5,7 @@ import (
  "errors"
  "fmt"
  "net/http"
+ "slices"
  "strings"
 
  "github.com/tonkeeper/opentonapi/pkg/bath"
@@ -30,6 +31,9 @@ func (h *Handler) GetAccountJettonsBalances(ctx context.Context, params oas.GetA
  Balances: make([]oas.JettonBalance, 0, len(wallets)),
  }
  for _, wallet := range wallets {
+ if slices.Contains(wallet.Extensions, "custom_payload") && !slices.Contains(params.SupportedExtensions, "custom_payload") {
+ continue
+ }
  jettonBalance, err := h.convertJettonBalance(ctx, wallet, params.Currencies)
  if err != nil {
  continue

diff --git a/pkg/oas/oas_handlers_gen.go b/pkg/oas/oas_handlers_gen.go
diff --git a/pkg/oas/oas_parameters_gen.go b/pkg/oas/oas_parameters_gen.go