@mastra/discord 1.0.0

package/package.json

@@ -0,0 +1,63 @@
+{
+  "name": "@mastra/discord",
+  "version": "1.0.0",
+  "description": "",
+  "main": "dist/index.js",
+  "module": "dist/mylib.esm.js",
+  "typings": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "src"
+  ],
+  "husky": {
+    "hooks": {
+      "pre-commit": "dts lint"
+    }
+  },
+  "engines": {
+    "node": ">=20 <22"
+  },
+  "size-limit": [
+    {
+      "path": "dist/mylib.cjs.production.min.js",
+      "limit": "10 KB"
+    },
+    {
+      "path": "dist/mylib.esm.js",
+      "limit": "10 KB"
+    }
+  ],
+  "devDependencies": {
+    "@jest/globals": "^29.7.0",
+    "@rollup/plugin-image": "^3.0.3",
+    "@size-limit/preset-small-lib": "^11.1.4",
+    "@tsconfig/recommended": "^1.0.7",
+    "@types/jest": "^29.5.12",
+    "@types/lodash": "^4.17.7",
+    "@types/node": "^22.1.0",
+    "dts-cli": "^2.0.5",
+    "husky": "^9.1.4",
+    "jest": "^29.7.0",
+    "size-limit": "^11.1.4",
+    "ts-jest": "^29.2.4",
+    "tslib": "^2.6.3",
+    "typescript": "^5.5.4"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "dependencies": {
+    "fets": "*",
+    "zod": "^3.23.8",
+    "@mastra/core": "0.1.0"
+  },
+  "scripts": {
+    "analyze": "size-limit --why",
+    "build": "dts build",
+    "build:dev": "dts watch",
+    "lint": "dts lint",
+    "size": "size-limit",
+    "start": "dts watch",
+    "test": "jest"
+  }
+}

package/src/Discord.test.ts

@@ -0,0 +1,55 @@
+import {
+  describe,
+  it,
+  beforeAll,
+  afterAll, //expect
+} from '@jest/globals';
+import { Framework } from '@mastra/core';
+
+import { DiscordIntegration } from '.';
+
+// We need to OAuth from admin
+
+const CLIENT_ID = process.env.CLIENT_ID!;
+const CLIENT_SECRET = process.env.CLIENT_SECRET!;
+
+const dbUri = process.env.DB_URL!;
+const connectionId = process.env.CONNECTION_ID!;
+
+const integrationName = 'DISCORD';
+
+const integrationFramework = Framework.init({
+  name: 'TestFramework',
+  integrations: [
+    new DiscordIntegration({
+      config: {
+        CLIENT_ID,
+        CLIENT_SECRET,
+        undefined,
+      },
+    }),
+  ],
+  systemApis: [],
+  systemEvents: {},
+  db: {
+    provider: 'postgres',
+    uri: dbUri,
+  },
+  systemHostURL: 'http://localhost:3000',
+  routeRegistrationPath: '/api/mastra',
+  blueprintDirPath: '',
+});
+
+//const integration = integrationFramework.getIntegration(integrationName) as DiscordIntegration
+
+describe('discord', () => {
+  beforeAll(async () => {});
+
+  it('should 200 on some apis', async () => {
+    //const client = await integration.getApiClient({ connectionId });
+    //const response = await client['/2010-04-01/Accounts.json'].get();
+    //expect(response.status).toBe(200);
+  });
+
+  afterAll(async () => {});
+});

package/src/index.ts

@@ -0,0 +1,201 @@
+import { Integration, OpenAPI, IntegrationCredentialType, IntegrationAuth } from '@mastra/core';
+import { createClient, type OASClient, type NormalizeOAS } from 'fets';
+
+// @ts-ignore
+import DiscordLogo from './assets/discord.png';
+import { openapi } from './openapi';
+import { components } from './openapi-components';
+import { paths } from './openapi-paths';
+
+type DiscordConfig = {
+  CLIENT_ID: string;
+  CLIENT_SECRET: string;
+
+  [key: string]: any;
+};
+
+export class DiscordIntegration extends Integration {
+  availableScopes = [
+    {
+      key: `activities.read`,
+      description: `allows your app to fetch data from a user's "Now Playing/Recently Played" list - requires Discord approval`,
+    },
+    {
+      key: `activities.write`,
+      description: `allows your app to update a user's activity - requires Discord approval (NOT REQUIRED FOR GAMESDK ACTIVITY MANAGER)`,
+    },
+    {
+      key: `applications.builds.read`,
+      description: `allows your app to read build data for a user's applications`,
+    },
+    {
+      key: `applications.builds.upload`,
+      description: `allows your app to upload/update builds for a user's applications - requires Discord approval`,
+    },
+    {
+      key: `applications.commands`,
+      description: `allows your app to use commands in a guild`,
+    },
+    {
+      key: `applications.commands.permissions.update`,
+      description: `allows your app to update permissions for its commands in a guild a user has permissions to`,
+    },
+    {
+      key: `applications.entitlements`,
+      description: `allows your app to read entitlements for a user's applications`,
+    },
+    {
+      key: `applications.store.update`,
+      description: `allows your app to read and update store data (SKUs, store listings, achievements, etc.) for a user's applications`,
+    },
+    {
+      key: `bot`,
+      description: `for oauth2 bots, this puts the bot in the user's selected guild by default`,
+    },
+    {
+      key: `connections`,
+      description: `allows /users/@me/connections to return linked third-party accounts`,
+    },
+    {
+      key: `dm_channels.read`,
+      description: `allows your app to see information about the user's DMs and group DMs - requires Discord approval`,
+    },
+    {
+      key: `email`,
+      description: `enables /users/@me to return an email`,
+    },
+    {
+      key: `gdm.join`,
+      description: `allows your app to join users to a group dm`,
+    },
+    {
+      key: `guilds`,
+      description: `allows /users/@me/guilds to return basic information about all of a user's guilds`,
+    },
+    {
+      key: `guilds.join`,
+      description: `allows /guilds/{guild.id}/members/{user.id} to be used for joining users to a guild`,
+    },
+    {
+      key: `guilds.members.read`,
+      description: `allows /users/@me/guilds/{guild.id}/member to return a user's member information in a guild`,
+    },
+    {
+      key: `identify`,
+      description: `allows /users/@me without email`,
+    },
+    {
+      key: `messages.read`,
+      description: `for local rpc server api access, this allows you to read messages from all client channels (otherwise restricted to channels/guilds your app creates)`,
+    },
+    {
+      key: `relationships.read`,
+      description: `allows your app to know a user's friends and implicit relationships - requires Discord approval`,
+    },
+    {
+      key: `rpc`,
+      description: `for local rpc server access, this allows you to control a user's local Discord client - requires Discord approval`,
+    },
+    {
+      key: `rpc.activities.write`,
+      description: `for local rpc server access, this allows you to update a user's activity - requires Discord approval`,
+    },
+    {
+      key: `rpc.notifications.read`,
+      description: `for local rpc server access, this allows you to receive notifications pushed out to the user - requires Discord approval`,
+    },
+    {
+      key: `rpc.screenshare.read`,
+      description: `for local rpc server access, this allows you to read a user's screenshare status- requires Discord approval`,
+    },
+    {
+      key: `rpc.screenshare.write`,
+      description: `for local rpc server access, this allows you to update a user's screenshare settings- requires Discord approval`,
+    },
+    {
+      key: `rpc.video.read`,
+      description: `for local rpc server access, this allows you to read a user's video status - requires Discord approval`,
+    },
+    {
+      key: `rpc.video.write`,
+      description: `for local rpc server access, this allows you to update a user's video settings - requires Discord approval`,
+    },
+    {
+      key: `rpc.voice.read`,
+      description: `for local rpc server access, this allows you to read a user's voice settings and listen for voice events - requires Discord approval`,
+    },
+    {
+      key: `rpc.voice.write`,
+      description: `for local rpc server access, this allows you to update a user's voice settings - requires Discord approval`,
+    },
+    {
+      key: `voice`,
+      description: `allows your app to connect to voice on user's behalf and see all the voice members - requires Discord approval`,
+    },
+    {
+      key: `webhook.incoming`,
+      description: `this generates a webhook that is returned in the oauth token response for authorization code grants`,
+    },
+  ];
+
+  constructor({ config }: { config: DiscordConfig }) {
+    super({
+      ...config,
+      authType: IntegrationCredentialType.OAUTH,
+      name: 'DISCORD',
+      logoUrl: DiscordLogo,
+    });
+  }
+
+  getOpenApiSpec() {
+    return { paths, components } as unknown as OpenAPI;
+  }
+
+  getApiClient = async ({ connectionId }: { connectionId: string }): Promise<OASClient<NormalizeOAS<openapi>>> => {
+    const connection = await this.dataLayer?.getConnection({ name: this.name, connectionId });
+
+    if (!connection) {
+      throw new Error(`Connection not found for connectionId: ${connectionId}`);
+    }
+
+    const authenticator = this.getAuthenticator();
+    const { accessToken } = await authenticator.getAuthToken({ k_id: connection.id });
+
+    const client = createClient<NormalizeOAS<openapi>>({
+      endpoint: `https://discord.com/api/v10`,
+      globalParams: {
+        headers: {
+          Authorization: `Bearer ${accessToken}`,
+        },
+      },
+    });
+
+    return client as any;
+  };
+
+  registerEvents() {
+    this.events = {};
+    return this.events;
+  }
+
+  getAuthenticator() {
+    return new IntegrationAuth({
+      dataAccess: this.dataLayer!,
+      // @ts-ignore
+      onConnectionCreated: () => {
+        // TODO
+      },
+      config: {
+        INTEGRATION_NAME: this.name,
+        AUTH_TYPE: this.config.authType,
+        CLIENT_ID: this.config.CLIENT_ID,
+        CLIENT_SECRET: this.config.CLIENT_SECRET,
+        REDIRECT_URI: this.config.REDIRECT_URI || this.corePresets.redirectURI,
+        SERVER: `https://discord.com/api/v10`,
+        AUTHORIZATION_ENDPOINT: `discord.com/oauth2/authorize`,
+        TOKEN_ENDPOINT: `https://discord.com/api/oauth2/token`,
+        SCOPES: this.config.SCOPES || [],
+      },
+    });
+  }
+}