From e15ba6591f79b8f9da4869b7335cd7039df29d30 Mon Sep 17 00:00:00 2001
From: georgebuciuman <georgebuciuman@gmail.com>
Date: Fri, 17 Jan 2025 11:17:28 +0200
Subject: [PATCH] Raise limit for tokens sets when syncing to TS provider
 (#3247)

---
 .changeset/heavy-sheep-float.md                             | 5 +++++
 .../src/storage/TokensStudioTokenStorage.ts                 | 6 ++++++
 .../src/storage/tokensStudio/graphql/getProjectDataQuery.ts | 3 ++-
 3 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100644 .changeset/heavy-sheep-float.md

diff --git a/.changeset/heavy-sheep-float.md b/.changeset/heavy-sheep-float.md
new file mode 100644
index 000000000..ae15ff609
--- /dev/null
+++ b/.changeset/heavy-sheep-float.md
@@ -0,0 +1,5 @@
+---
+"@tokens-studio/figma-plugin": patch
+---
+
+Raise limit for token sets to 100 when syncing with Tokens Studio provider
diff --git a/packages/tokens-studio-for-figma/src/storage/TokensStudioTokenStorage.ts b/packages/tokens-studio-for-figma/src/storage/TokensStudioTokenStorage.ts
index 0e3e2ebb6..7ec8cc3cc 100644
--- a/packages/tokens-studio-for-figma/src/storage/TokensStudioTokenStorage.ts
+++ b/packages/tokens-studio-for-figma/src/storage/TokensStudioTokenStorage.ts
@@ -62,6 +62,12 @@ async function getProjectData(id: string, orgId: string, client: any): Promise<P
     }
 
     const tokenSets = data.data.project.branch.tokenSets.data as TokensSet[];
+    const { totalPages } = data.data.project.branch.tokenSets;
+
+    // TODO: This is a temporary solution until we implement pagination
+    if (totalPages > 1) {
+      notifyToUI('We are currently supporting up to 100 sets, if you encounter this issue and need even more sets please reach out to us on slack or featurebase.', { error: true });
+    }
 
     const returnData = tokenSets.reduce(
       (acc, tokenSet) => {
diff --git a/packages/tokens-studio-for-figma/src/storage/tokensStudio/graphql/getProjectDataQuery.ts b/packages/tokens-studio-for-figma/src/storage/tokensStudio/graphql/getProjectDataQuery.ts
index 6ff19f451..69ae161e1 100644
--- a/packages/tokens-studio-for-figma/src/storage/tokensStudio/graphql/getProjectDataQuery.ts
+++ b/packages/tokens-studio-for-figma/src/storage/tokensStudio/graphql/getProjectDataQuery.ts
@@ -4,7 +4,7 @@ export const GET_PROJECT_DATA_QUERY = gql`
   query Branch($projectId: String!, $organization: String!, $name: String) {
     project(id: $projectId, organization: $organization) {
       branch(name: $name) {
-        tokenSets {
+        tokenSets(limit: 100) {
           data {
             name
             orderIndex
@@ -17,6 +17,7 @@ export const GET_PROJECT_DATA_QUERY = gql`
               value
             }
           }
+          totalPages
         }
         themeGroups {
           data {