From ad511546ca81df35772d807772513d7414bfe3e5 Mon Sep 17 00:00:00 2001
From: teidesu <me@tei.su>
Date: Wed, 14 Apr 2021 23:11:37 +0300
Subject: [PATCH] fix(tl-ref): usage count for arrays in return types

---
 packages/tl-reference/scripts/prepare-data.js     | 3 ++-
 packages/tl-reference/src/templates/tl-object.tsx | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/tl-reference/scripts/prepare-data.js b/packages/tl-reference/scripts/prepare-data.js
index 33bd9478..3bc0a5fc 100644
--- a/packages/tl-reference/scripts/prepare-data.js
+++ b/packages/tl-reference/scripts/prepare-data.js
@@ -84,10 +84,11 @@ function prepareData(data) {
             // add hex constructor id
             if (item.id) item.tlId = item.id.toString(16).padStart(8, '0')
 
+            // raw non-array type for usages count
             if (item.arguments) item.arguments.forEach((arg) => {
-                // raw non-array type for usages count
                 arg.rawType = arg.type.replace(/\[]$/, '')
             })
+            if (item.returns) item.rawReturns = item.returns.replace(/\[]$/, '')
 
             // add typescript types for the item and arguments
             // basically copy-pasted from generate-types.js
diff --git a/packages/tl-reference/src/templates/tl-object.tsx b/packages/tl-reference/src/templates/tl-object.tsx
index 901fb08c..2a659964 100644
--- a/packages/tl-reference/src/templates/tl-object.tsx
+++ b/packages/tl-reference/src/templates/tl-object.tsx
@@ -471,7 +471,7 @@ export const query = graphql`
             filter: {
                 prefix: { eq: $prefix }
                 type: { eq: "method" }
-                returns: { eq: $name }
+                rawReturns: { eq: $name }
             }
         ) @include(if: $hasSubtypes) {
             nodes {