zulip/web/src/bot_data.js

import _ from "lodash";

import * as people from "./people";

const bots = new Map();

const bot_fields = [
    "api_key",
    "avatar_url",
    "bot_type",
    "default_all_public_streams",
    "default_events_register_stream",
    "default_sending_stream",
    "email",
    "full_name",
    "is_active",
    "owner", // TODO: eliminate
    "owner_id",
    "user_id",
];

const services = new Map();
const services_fields = ["base_url", "interface", "config_data", "service_name", "token"];

export function all_user_ids() {
    return [...bots.keys()];
}

export function add(bot) {
    const clean_bot = _.pick(bot, bot_fields);
    bots.set(bot.user_id, clean_bot);
    const clean_services = bot.services.map((service) => _.pick(service, services_fields));
    services.set(bot.user_id, clean_services);
}

export function deactivate(bot_id) {
    bots.get(bot_id).is_active = false;
}

export function del(bot_id) {
    bots.delete(bot_id);
    services.delete(bot_id);
}

export function update(bot_id, bot_update) {
    const bot = bots.get(bot_id);
    Object.assign(bot, _.pick(bot_update, bot_fields));

    // We currently only support one service per bot.
    const service = services.get(bot_id)[0];
    if (bot_update.services !== undefined && bot_update.services.length > 0) {
        Object.assign(service, _.pick(bot_update.services[0], services_fields));
    }
}

export function get_all_bots_for_current_user() {
    const ret = [];
    for (const bot of bots.values()) {
        if (people.is_my_user_id(bot.owner_id)) {
            ret.push(bot);
        }
    }
    return ret;
}

export function get_editable() {
    const ret = [];
    for (const bot of bots.values()) {
        if (bot.is_active && people.is_my_user_id(bot.owner_id)) {
            ret.push(bot);
        }
    }
    return ret;
}

export function get_all_bots_owned_by_user(user_id) {
    const ret = [];
    for (const bot of bots.values()) {
        if (bot.owner_id === user_id && bot.is_active) {
            ret.push(bot);
        }
    }
    return ret;
}

export function get(bot_id) {
    return bots.get(bot_id);
}

export function get_services(bot_id) {
    return services.get(bot_id);
}

export function initialize(params) {
    bots.clear();
    for (const bot of params.realm_bots) {
        add(bot);
    }
}