@atomist/automation-client/lib/spi/message/MessageClientSupport.ts

Atomist API for software low-level client

import { SlackMessage } from "@atomist/slack-messages";
import { toStringArray } from "../../internal/util/string";
import {
    GraphClient,
    QueryNoCacheOptions,
} from "../graph/GraphClient";
import {
    addressSlackChannels,
    addressSlackUsers,
    Destination,
    MessageClient,
    MessageOptions,
    RequiredMessageOptions,
    SlackMessageClient,
} from "./MessageClient";

export abstract class MessageClientSupport implements MessageClient {

    public respond(msg: any,
                   options?: MessageOptions): Promise<any> {
        return this.doSend(msg, [], options);
    }

    public send(msg: any,
                destinations: Destination | Destination[],
                options?: MessageOptions): Promise<any> {
        const dests = (Array.isArray(destinations)) ? destinations : [destinations];
        return this.doSend(msg, dests, options);
    }

    public abstract delete(destinations: Destination | Destination[],
                           options: RequiredMessageOptions): Promise<void>;

    protected abstract doSend(msg: any,
                              destinations: Destination[],
                              options?: MessageOptions): Promise<any>;

}

export const Query = `
query ChatTeam {
  ChatTeam {
    id
  }
}`;

export class DefaultSlackMessageClient implements MessageClient, SlackMessageClient {

    constructor(private readonly delegate: MessageClient,
                private readonly graphClient: GraphClient) { }

    public respond(msg: any,
                   options?: MessageOptions): Promise<any> {
        return this.delegate.respond(msg, options);
    }

    public send(msg: any,
                destinations: Destination | Destination[],
                options?: MessageOptions): Promise<any> {
        return this.delegate.send(msg, destinations, options);
    }

    public async delete(destinations: Destination | Destination[],
                        options: RequiredMessageOptions): Promise<void> {
        return this.delegate.delete(destinations, options);
    }

    public addressUsers(msg: string | SlackMessage,
                        users: string | string[],
                        options?: MessageOptions): Promise<any> {
        if (!users || (Array.isArray(users) && users.length === 0)) {
            throw new Error("Please pass at least one user");
        }
        return lookupChatTeam(this.graphClient)
            .then(chatTeamId =>
                this.delegate.send(msg, addressSlackUsers(chatTeamId, ...toStringArray(users)), options));
    }

    public addressChannels(msg: string | SlackMessage,
                           channels: string | string[],
                           options?: MessageOptions): Promise<any> {
        if (!channels || (Array.isArray(channels) && channels.length === 0)) {
            throw new Error("Please pass at least one channel");
        }
        return lookupChatTeam(this.graphClient)
            .then(chatTeamId =>
                this.delegate.send(msg, addressSlackChannels(chatTeamId, ...toStringArray(channels)), options));

    }
}

export function lookupChatTeam(graphClient: GraphClient): Promise<string> {
    if (graphClient) {
        return graphClient.query<any, any>({
            query: Query,
            variables: {},
            options: QueryNoCacheOptions,
        })
            .then(result => {
                if (result.ChatTeam && result.ChatTeam.length !== 1) {
                    return Promise.reject("More then one or no ChatTeam found. Please use fully qualified " +
                        "message addressing available on MessageClient");
                } else {
                    return result.ChatTeam[0].id;
                }
            });
    } else {
        return Promise.reject("No GraphClient to lookup ChatTeam. Please use fully qualified message " +
            "addressing available on MessageClient");
    }
}