feat: marketing mutation + query (#331)

NibiruChain · Mar 6, 2024 · 61c7fe8 · 61c7fe8
1 parent 26176e3
commit 61c7fe8
Show file tree

Hide file tree

Showing 6 changed files with 170 additions and 48 deletions.
diff --git a/src/gql/heart-monitor/heart-monitor.test.ts b/src/gql/heart-monitor/heart-monitor.test.ts
@@ -98,8 +98,9 @@ import {
   defaultTask,
   QueryMarketingArgs,
   MarketingFields,
-  GQLMarketingMutationGqlUpdateTwitterUserArgs,
-  GQLTwitterUser,
+  QueryMarketingMutationArgs,
+  GQLMarketingMutationFields,
+  defaultAccountLinksInfo,
 } from ".."
 
 const nibiruUrl = "testnet-1"
@@ -436,43 +437,38 @@ test("markPriceCandlesSubscription", async () => {
 })
 
 const testMarketingMutation = async (
-  args: GQLMarketingMutationGqlUpdateTwitterUserArgs,
-  fields?: GQLTwitterUser
+  args: QueryMarketingMutationArgs,
+  fields?: GQLMarketingMutationFields
 ) => {
   const resp = await heartMonitor.marketingMutation(args, {}, fields)
   expect(resp).toHaveProperty("marketing")
 
   if (resp.marketing) {
     const { marketing } = resp
 
-    checkFields(
-      [marketing],
-      [
-        "completedTasks",
-        "creationTimestamp",
-        "displayName",
-        "followersCount",
-        "followingCount",
-        "id",
-        "likes",
-        "listedCount",
-        "nibiAddress",
-        "tweets",
-        "tweetsCount",
-        "username",
-      ]
-    )
+    checkFields([marketing], ["updateTwitterUser", "linkAccounts"])
   }
 }
 
 // Create JIT JWT for this test
 test.skip("marketingMutation", async () => {
   await testMarketingMutation({
-    input: {
-      userId: "800528778854182912",
-      nibiAddress: "nibi1p6luzkxeufy29reymgjqnl5mv6a6gae07cphed",
-      displayName: "WildFyre",
-      username: "wildfyreapp",
+    updateTwitterUser: {
+      input: {
+        userId: "800528778854182912",
+        nibiAddress: "nibi1p6luzkxeufy29reymgjqnl5mv6a6gae07cphed",
+        displayName: "WildFyre",
+        username: "wildfyreapp",
+      },
+    },
+    linkAccounts: {
+      input: {
+        nibiAddress: "nibi1p6luzkxeufy29reymgjqnl5mv6a6gae07cphed",
+        twitterUser: {
+          displayName: "WildFyre",
+          username: "wildfyreapp",
+        },
+      },
     },
   })
 })
@@ -489,7 +485,15 @@ const testMarketingQuery = async (
 
     checkFields(
       [marketing],
-      ["likes", "retweets", "tasks", "tweets", "twitterUser"]
+      [
+        "likes",
+        "retweets",
+        "tasks",
+        "tweets",
+        "twitterUser",
+        "accountLinksInfo",
+        "lastUpdatedTimestamp",
+      ]
     )
   }
 }
@@ -506,6 +510,12 @@ test.skip("marketingQuery", async () => {
     likes: {
       where: { userId: "1516130689028087815" },
     },
+    accountLinksInfo: {
+      where: {
+        nibiAddress: "nibi1p6luzkxeufy29reymgjqnl5mv6a6gae07cphed",
+        twitterId: "800528778854182912",
+      },
+    },
   })
   await testMarketingQuery(
     {
@@ -518,21 +528,31 @@ test.skip("marketingQuery", async () => {
       likes: {
         where: { userId: "" },
       },
+      accountLinksInfo: {
+        where: {
+          nibiAddress: "",
+          twitterId: "",
+        },
+      },
     },
     {
+      accountLinksInfo: defaultAccountLinksInfo,
       twitterUser: defaultTwitterUser,
       tweets: defaultTweet,
       likes: defaultLike,
       tasks: defaultTask,
+      lastUpdatedTimestamp: "",
     }
   )
   await testMarketingQuery(
     {},
     {
+      accountLinksInfo: defaultAccountLinksInfo,
       twitterUser: defaultTwitterUser,
       tweets: defaultTweet,
       likes: defaultLike,
       tasks: defaultTask,
+      lastUpdatedTimestamp: "",
     }
   )
 })

diff --git a/src/gql/heart-monitor/heart-monitor.ts b/src/gql/heart-monitor/heart-monitor.ts
@@ -112,6 +112,8 @@ import {
   GqlOutMarketingMutation,
   GQLTwitterUser,
   DeepPartial,
+  QueryMarketingMutationArgs,
+  GQLMarketingMutationFields,
 } from ".."
 
 /** IHeartMonitor is an interface for a Heart Monitor GraphQL API.
@@ -154,9 +156,9 @@ export interface IHeartMonitor {
   ) => Promise<GqlOutInflation>
 
   readonly marketingMutation: (
-    args: GQLMarketingMutationGqlUpdateTwitterUserArgs,
+    args: QueryMarketingMutationArgs,
     headers: HeadersInit,
-    fields?: DeepPartial<GQLTwitterUser>
+    fields?: DeepPartial<GQLMarketingMutationFields>
   ) => Promise<GqlOutMarketingMutation>
 
   readonly marketingQuery: (
@@ -338,9 +340,9 @@ export class HeartMonitor implements IHeartMonitor {
   ) => marketingQuery(args, this.gqlEndpt, fields)
 
   marketingMutation = async (
-    args: GQLMarketingMutationGqlUpdateTwitterUserArgs,
+    args: QueryMarketingMutationArgs,
     headers: HeadersInit,
-    fields?: DeepPartial<GQLTwitterUser>
+    fields?: DeepPartial<GQLMarketingMutationFields>
   ) => marketingMutation(args, this.gqlEndpt, headers, fields)
 
   markPriceCandles = async (

diff --git a/src/gql/mutation/marketing.ts b/src/gql/mutation/marketing.ts
@@ -7,18 +7,79 @@ import {
   GQLTwitterUser,
   GQLMarketingMutationGqlUpdateTwitterUserArgs,
   DeepPartial,
+  GQLMarketingMutationGqlLinkAccountsArgs,
+  GQLAccountLinksInfo,
+  defaultAccountLinksInfo,
 } from ".."
 
+export type QueryMarketingMutationArgs = {
+  updateTwitterUser?: GQLMarketingMutationGqlUpdateTwitterUserArgs
+  linkAccounts?: GQLMarketingMutationGqlLinkAccountsArgs
+}
+
 export interface GqlOutMarketingMutation {
   marketing?: GQLMutation["marketing"]
 }
 
+export type GQLMarketingMutationFields = DeepPartial<{
+  updateTwitterUser?: DeepPartial<GQLTwitterUser>
+  linkAccounts?: DeepPartial<GQLAccountLinksInfo>
+}>
+
 export const marketingMutationString = (
-  args: GQLMarketingMutationGqlUpdateTwitterUserArgs,
+  args: QueryMarketingMutationArgs,
   excludeParentObject: boolean,
-  fields?: DeepPartial<GQLTwitterUser>
-) =>
-  `mutation {
+  fields?: DeepPartial<GQLMarketingMutationFields>
+) => {
+  const GQLMarketingMutationQuery: string[] = []
+
+  if (fields?.updateTwitterUser) {
+    GQLMarketingMutationQuery.push(
+      gqlQuery(
+        "updateTwitterUser",
+        args.updateTwitterUser ?? {},
+        convertObjectToPropertiesString(fields.updateTwitterUser),
+        true
+      )
+    )
+  }
+
+  if (fields?.linkAccounts) {
+    GQLMarketingMutationQuery.push(
+      gqlQuery(
+        "linkAccounts",
+        args.linkAccounts ?? {},
+        convertObjectToPropertiesString(fields.linkAccounts),
+        true
+      )
+    )
+  }
+
+  // Default Objects
+
+  if (args.updateTwitterUser && !fields?.updateTwitterUser) {
+    GQLMarketingMutationQuery.push(
+      gqlQuery(
+        "updateTwitterUser",
+        args.updateTwitterUser,
+        convertObjectToPropertiesString(defaultTwitterUser),
+        true
+      )
+    )
+  }
+
+  if (args.linkAccounts && !fields?.linkAccounts) {
+    GQLMarketingMutationQuery.push(
+      gqlQuery(
+        "linkAccounts",
+        args.linkAccounts,
+        convertObjectToPropertiesString(defaultAccountLinksInfo),
+        true
+      )
+    )
+  }
+
+  return `mutation {
     marketing {
       ${gqlQuery(
         "updateTwitterUser",
@@ -30,11 +91,12 @@ export const marketingMutationString = (
       )}
     }
   }`
+}
 
 export const marketingMutation = async (
-  args: GQLMarketingMutationGqlUpdateTwitterUserArgs,
+  args: QueryMarketingMutationArgs,
   endpt: string,
   headers: HeadersInit,
-  fields?: DeepPartial<GQLTwitterUser>
+  fields?: DeepPartial<GQLMarketingMutationFields>
 ): Promise<GqlOutMarketingMutation> =>
   doGqlQuery(marketingMutationString(args, true, fields), endpt, headers)
diff --git a/src/gql/query/marketing.ts b/src/gql/query/marketing.ts
@@ -1,12 +1,15 @@
 import {
   convertObjectToPropertiesString,
   DeepPartial,
+  defaultAccountLinksInfo,
   defaultLike,
   defaultTask,
   defaultTweet,
   defaultTwitterUser,
   doGqlQuery,
+  GQLAccountLinksInfo,
   GQLLike,
+  GQLMarketingQueryGqlAccountLinksInfoArgs,
   GQLMarketingQueryGqlLikesArgs,
   GQLMarketingQueryGqlTasksArgs,
   GQLMarketingQueryGqlTweetsArgs,
@@ -23,6 +26,7 @@ export type QueryMarketingArgs = {
   tweets?: GQLMarketingQueryGqlTweetsArgs
   likes?: GQLMarketingQueryGqlLikesArgs
   tasks?: GQLMarketingQueryGqlTasksArgs
+  accountLinksInfo?: GQLMarketingQueryGqlAccountLinksInfoArgs
 }
 
 export interface GqlOutMarketingQuery {
@@ -34,6 +38,8 @@ export type MarketingFields = DeepPartial<{
   tweets?: DeepPartial<GQLTweet>
   likes?: DeepPartial<GQLLike>
   tasks?: DeepPartial<GQLTask>
+  accountLinksInfo?: DeepPartial<GQLAccountLinksInfo>
+  lastUpdatedTimestamp?: string
 }>
 
 export const marketingQueryString = (
@@ -86,6 +92,17 @@ export const marketingQueryString = (
     )
   }
 
+  if (fields?.accountLinksInfo) {
+    marketingQuery.push(
+      gqlQuery(
+        "accountLinksInfo",
+        args.accountLinksInfo ?? {},
+        convertObjectToPropertiesString(fields.accountLinksInfo),
+        true
+      )
+    )
+  }
+
   // Default Objects
 
   if (args.likes && !fields?.likes) {
@@ -132,6 +149,22 @@ export const marketingQueryString = (
     )
   }
 
+  if (args.accountLinksInfo && !fields?.accountLinksInfo) {
+    marketingQuery.push(
+      gqlQuery(
+        "accountLinksInfo",
+        args.accountLinksInfo,
+        convertObjectToPropertiesString(defaultAccountLinksInfo),
+        true
+      )
+    )
+  }
+
+  // Add lastUpdatedTimestamp if specified
+  if (fields?.lastUpdatedTimestamp) {
+    marketingQuery.push("lastUpdatedTimestamp")
+  }
+
   return `
         marketing {
           ${marketingQuery.join("\n")}