clientfactory.ts

/*
Copyright 2017 - 2019 matrix-appservice-discord

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

import { DiscordBridgeConfigAuth } from "./config";
import { DiscordStore } from "./store";
import { Client as DiscordClient } from "discord.js";
import { Log } from "./log";
import { Util } from "./util";

const log = new Log("ClientFactory");

const READY_TIMEOUT = 30000;

export class DiscordClientFactory {
    private config: DiscordBridgeConfigAuth;
    private store: DiscordStore;
    private botClient: DiscordClient;
    private clients: Map<string, DiscordClient>;
    constructor(store: DiscordStore, config?: DiscordBridgeConfigAuth) {
        this.config = config!;
        this.clients = new Map();
        this.store = store;
    }

    public async init(): Promise<void> {
        if (this.config === undefined) {
            return Promise.reject("Client config not supplied.");
        }
        // We just need to make sure we have a bearer token.
        // Create a new Bot client.
        this.botClient = new DiscordClient({
            fetchAllMembers: true,
            messageCacheLifetime: 5,
            sync: true,
        });

        try {
            await this.botClient.login(this.config.botToken);
        } catch (err) {
            log.error("Could not login as the bot user. This is bad!", err);
            throw err;
        }

    }

    public async getDiscordId(token: string): Promise<string> {
        const client = new DiscordClient({
            fetchAllMembers: false,
            messageCacheLifetime: 5,
            sync: false,
        });

        await client.login(token);
        const id = client.user.id;

        // This can be done asynchronously, because we don't need to block to return the id.
        client.destroy().catch((err) => {
            log.warn("Failed to destroy client ", id);
        });
        return id;
    }

    public async getClient(userId: string | null = null): Promise<DiscordClient> {
        if (userId === null) {
            return this.botClient;
        }

        if (this.clients.has(userId)) {
            log.verbose("Returning cached user client for", userId);
            return this.clients.get(userId) as DiscordClient;
        }

        const discordIds = await this.store.get_user_discord_ids(userId);
        if (discordIds.length === 0) {
            return this.botClient;
        }
        // TODO: Select a profile based on preference, not the first one.
        const token = await this.store.get_token(discordIds[0]);
        const client = new DiscordClient({
            fetchAllMembers: true,
            messageCacheLifetime: 5,
            sync: true,
        });

        const jsLog = new Log("discord.js-ppt");
        client.on("debug", (msg) => { jsLog.verbose(msg); });
        client.on("error", (msg) => { jsLog.error(msg); });
        client.on("warn", (msg) => { jsLog.warn(msg); });

        try {
            await client.login(token);
            log.verbose("Logged in. Storing ", userId);
            this.clients.set(userId, client);
            return client;
        } catch (err) {
            log.warn(`Could not log ${userId} in. Returning bot user for now.`, err);
            return this.botClient;
        }
    }
}