add broadcast command

Author: zhelezkov

package.json

@@ -17,9 +17,11 @@
     "/oclif.manifest.json"
   ],
   "dependencies": {
+    "@dialectlabs/sdk": "^0.2.0",
     "@oclif/core": "^1.9.5",
     "@oclif/plugin-help": "^5",
-    "@oclif/plugin-plugins": "^2.1.0"
+    "@oclif/plugin-plugins": "^2.1.0",
+    "@solana/web3.js": "^1.47.1"
   },
   "devDependencies": {
     "@oclif/test": "^2",

src/commands/broadcast/index.ts

@@ -0,0 +1,93 @@
+import { Backend, Dialect } from '@dialectlabs/sdk';
+import { CliUx, Command, Flags } from '@oclif/core';
+import {
+  envFlag,
+  keypairFlag,
+  modeFlag,
+  sdkEnvFromEnvFlag,
+} from '../../shared/flags';
+import { createWalletFromFile } from '../../shared/wallet';
+
+export default class Broadcast extends Command {
+  static description = 'Command sends messages to multiples accounts';
+
+  static examples = ['<%= config.bin %> <%= command.id %>'];
+
+  static flags = {
+    title: Flags.string({
+      char: 't',
+      description: 'Message title',
+      required: true,
+    }),
+    message: Flags.string({
+      char: 'm',
+      description: 'Message to send',
+      required: true,
+    }),
+    mode: modeFlag,
+    keypair: keypairFlag,
+    env: envFlag,
+  };
+
+  static args = [];
+
+  public async run(): Promise<void> {
+    const { flags } = await this.parse(Broadcast);
+
+    const backends = [];
+    if (flags.mode === 'cloud') {
+      backends.push(Backend.DialectCloud);
+    }
+
+    if (flags.mode === 'solana') {
+      backends.push(Backend.Solana);
+    }
+
+    const wallet = await createWalletFromFile(flags.keypair!);
+
+    const sdk = Dialect.sdk({
+      wallet,
+      environment: sdkEnvFromEnvFlag(flags.env),
+      backends: backends,
+    });
+
+    const dapp = await sdk.dapps.find();
+    if (!dapp) {
+      throw new Error(
+        `You don't have dapp associated with your public key: ${sdk.wallet.publicKey}`,
+      );
+    }
+
+    this.log('Dapp public address:', dapp.publicKey.toString());
+
+    CliUx.ux.action.start('fetching addresses');
+
+    const addresses = await dapp.dappAddresses.findAll();
+
+    CliUx.ux.action.stop(`found ${addresses.length} addresses`);
+
+    if (addresses.length === 0) {
+      return;
+    }
+
+    const confirmed = await CliUx.ux.confirm(
+      `Message preview: 
+${flags.title}
+${flags.message}
+to ${addresses.length} addresses? [y/n]`,
+    );
+
+    if (!confirmed) {
+      return;
+    }
+
+    CliUx.ux.action.start('sending message');
+
+    await dapp.messages.send({
+      title: flags.title,
+      message: flags.message,
+    });
+
+    CliUx.ux.action.stop('sending message');
+  }
+}

src/shared/flags/env.ts

@@ -0,0 +1,31 @@
+import { Environment } from '@dialectlabs/sdk';
+import { Flags } from '@oclif/core';
+
+export const EnvFlagOptions = {
+  LocalDev: 'local-dev',
+  Dev: 'dev',
+  Production: 'prod',
+} as const;
+
+export type EnvFlagOptionValues =
+  typeof EnvFlagOptions[keyof typeof EnvFlagOptions];
+
+export const envFlag = Flags.enum({
+  options: [
+    EnvFlagOptions.LocalDev,
+    EnvFlagOptions.Dev,
+    EnvFlagOptions.Production,
+  ],
+  default: EnvFlagOptions.Production,
+});
+
+export function sdkEnvFromEnvFlag(flagValue: EnvFlagOptionValues): Environment {
+  switch (flagValue) {
+    case EnvFlagOptions.LocalDev:
+      return 'local-development';
+    case EnvFlagOptions.Dev:
+      return 'development';
+    case EnvFlagOptions.Production:
+      return 'production';
+  }
+}

src/shared/flags/index.ts

@@ -0,0 +1,3 @@
+export * from './env';
+export * from './keypair';
+export * from './mode';

src/shared/flags/keypair.ts

@@ -0,0 +1,11 @@
+import { Flags } from '@oclif/core';
+import path from 'node:path';
+
+export const keypairFlag = Flags.file({
+  char: 'k',
+  description: 'Filepath to a keypair',
+  default: process.env.HOME
+    ? path.resolve(process.env.HOME, '.config/solana/id.json')
+    : 'id.json',
+  required: true,
+});

src/shared/flags/mode.ts

@@ -0,0 +1,6 @@
+import { Flags } from '@oclif/core';
+
+export const modeFlag = Flags.enum({
+  options: ['cloud', 'solana'],
+  default: 'cloud',
+});

src/shared/wallet.ts

@@ -0,0 +1,25 @@
+import {
+  DialectWalletAdapter,
+  NodeDialectWalletAdapter,
+} from '@dialectlabs/sdk';
+import { Keypair } from '@solana/web3.js';
+import { readFile } from 'node:fs/promises';
+
+export async function keypairFromFile(keypairPath: string): Promise<Keypair> {
+  const privateKey = await readFile(keypairPath);
+  return Keypair.fromSecretKey(
+    new Uint8Array(JSON.parse(privateKey.toString())),
+  );
+}
+
+export function createWalletFromKeypair(
+  keypair: Keypair,
+): DialectWalletAdapter {
+  return new NodeDialectWalletAdapter(keypair);
+}
+
+export async function createWalletFromFile(
+  keypairPath: string,
+): Promise<DialectWalletAdapter> {
+  return createWalletFromKeypair(await keypairFromFile(keypairPath));
+}

test/commands/broadcast.test.ts

@@ -0,0 +1,17 @@
+import {expect, test} from '@oclif/test'
+
+describe('broadcast', () => {
+  test
+  .stdout()
+  .command(['broadcast'])
+  .it('runs hello', ctx => {
+    expect(ctx.stdout).to.contain('hello world')
+  })
+
+  test
+  .stdout()
+  .command(['broadcast', '--name', 'jeff'])
+  .it('runs hello --name jeff', ctx => {
+    expect(ctx.stdout).to.contain('hello jeff')
+  })
+})

tsconfig.json

@@ -6,9 +6,8 @@
     "outDir": "dist",
     "rootDir": "src",
     "strict": true,
-    "target": "es2019"
+    "target": "es2019",
+    "esModuleInterop": true
   },
-  "include": [
-    "src/**/*"
-  ]
+  "include": ["src/**/*"]
 }