Retrieve userAccountId when connecting Confluence (#3387)

dust-tt · Jan 23, 2024 · 23b68e2 · 23b68e2
1 parent bf47395
commit 23b68e2
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 1 deletion.
diff --git a/connectors/src/connectors/confluence/index.ts b/connectors/src/connectors/confluence/index.ts
@@ -8,6 +8,7 @@ import type { ConnectorsAPIErrorResponse } from "@dust-tt/types";
 import { confluenceConfig } from "@connectors/connectors/confluence/lib/config";
 import {
   getConfluenceCloudInformation,
+  getConfluenceUserAccountId,
   listConfluenceSpaces,
 } from "@connectors/connectors/confluence/lib/confluence_api";
 import type { ConfluenceSpaceType } from "@connectors/connectors/confluence/lib/confluence_client";
@@ -52,6 +53,8 @@ export async function createConfluenceConnector(
     return new Err(new Error("Confluence access token is invalid"));
   }
 
+  const userAccountId = await getConfluenceUserAccountId(confluenceAccessToken);
+
   const { id: cloudId, url: cloudUrl } = confluenceCloudInformation;
   try {
     const connector = await sequelize_conn.transaction(async (transaction) => {
@@ -70,6 +73,7 @@ export async function createConfluenceConnector(
           cloudId,
           connectorId: connector.id,
           url: cloudUrl,
+          userAccountId,
         },
         { transaction }
       );

diff --git a/connectors/src/connectors/confluence/lib/confluence_api.ts b/connectors/src/connectors/confluence/lib/confluence_api.ts
@@ -18,6 +18,14 @@ export async function getConfluenceCloudInformation(accessToken: string) {
   }
 }
 
+export async function getConfluenceUserAccountId(accessToken: string) {
+  const client = new ConfluenceClient(accessToken);
+
+  const userAccount = await client.getUserAccount();
+
+  return userAccount.account_id;
+}
+
 async function fetchConfluenceConfiguration(connectorId: ModelId) {
   const confluenceConfig = await ConfluenceConfiguration.findOne({
     where: {

diff --git a/connectors/src/connectors/confluence/lib/confluence_client.ts b/connectors/src/connectors/confluence/lib/confluence_client.ts
@@ -69,6 +69,13 @@ const ConfluenceListPagesCodec = t.type({
   }),
 });
 
+const ConfluenceUserProfileCodec = t.intersection([
+  t.type({
+    account_id: t.string,
+  }),
+  CatchAllCodec,
+]);
+
 function extractCursorFromLinks(links: { next?: string }): string | null {
   if (!links.next) {
     return null;
@@ -182,4 +189,8 @@ export class ConfluenceClient {
       ConfluencePageWithBodyCodec
     );
   }
+
+  async getUserAccount() {
+    return this.request("/me", ConfluenceUserProfileCodec);
+  }
 }
diff --git a/connectors/src/lib/models/confluence.ts b/connectors/src/lib/models/confluence.ts
@@ -18,6 +18,7 @@ export class ConfluenceConfiguration extends Model<
 
   declare cloudId: string;
   declare url: string;
+  declare userAccountId: string;
 
   declare connectorId: ForeignKey<Connector["id"]>;
 }
@@ -46,11 +47,18 @@ ConfluenceConfiguration.init(
       type: DataTypes.STRING,
       allowNull: false,
     },
+    userAccountId: {
+      type: DataTypes.STRING,
+      allowNull: false,
+    },
   },
   {
     sequelize: sequelize_conn,
     modelName: "confluence_configurations",
-    indexes: [{ fields: ["connectorId"], unique: true }],
+    indexes: [
+      { fields: ["connectorId"], unique: true },
+      { fields: ["userAccountId"] },
+    ],
   }
 );
 Connector.hasOne(ConfluenceConfiguration);