pin.c

Demonstrates pinning messages

#include <stdio.h>
#include <stdlib.h>
#include <inttypes.h> /* PRIu64, SCNu64 */
#include <assert.h>
 
#include "discord.h"
#include "logmod.h"
 
void
print_usage(void)
{
    printf(
        "\n\nThis bot demonstrates how easy it is to have a"
        " message be pinned.\n"
        "1. Reply to a message with '!pin' or type '!pin <message_id> to pin "
        "it\n"
        "2. Reply to a message with '!unpin' or type '!unpin <message_id> to "
        "unpin it\n"
        "3. Type '!get_pins' to get a id list of pinned messages\n"
        "\nTYPE ANY KEY TO START BOT\n");
}
 
void
on_ready(struct discord *client, const struct discord_ready *event)
{
    logmod_log(INFO, NULL,
               "Pin-Bot succesfully connected to Discord as %s#%s!",
               event->user->username, event->user->discriminator);
}
 
void
on_pin(struct discord *client, const struct discord_message *event)
{
    if (event->author->bot) return;
 
    u64snowflake msg_id = 0;
 
    sscanf(event->content, "%" SCNu64, &msg_id);
 
    if (!msg_id) {
        if (!event->referenced_message) return;
 
        msg_id = event->referenced_message->id;
    }
 
    struct discord_pin_message params = { .reason = "Important message" };
    discord_pin_message(client, event->channel_id, msg_id, &params, NULL);
}
 
void
on_unpin(struct discord *client, const struct discord_message *event)
{
    if (event->author->bot) return;
 
    u64snowflake msg_id = 0;
 
    sscanf(event->content, "%" SCNu64, &msg_id);
 
    if (!msg_id) {
        if (!event->referenced_message) return;
 
        msg_id = event->referenced_message->id;
    }
 
    struct discord_unpin_message params = { .reason = "No longer relevant" };
    discord_unpin_message(client, event->channel_id, msg_id, &params, NULL);
}
 
void
done_get_pins(struct discord *client,
              struct discord_response *resp,
              const struct discord_messages *msgs)
{
    const struct discord_message *event = resp->keep;
    char text[2000] = "No pins on channel";
 
    char *cur = text;
    char *end = &text[sizeof(text) - 1];
 
    for (int i = 0; i < msgs->size; ++i) {
        cur += snprintf(cur, end - cur,
                        "https://discord.com/channels/%" PRIu64 "/%" PRIu64
                        "/%" PRIu64 "\n",
                        event->guild_id, event->channel_id, msgs->array[i].id);
 
        if (cur >= end) break;
    }
 
    struct discord_create_message params = { .content = text };
    discord_create_message(client, event->channel_id, &params, NULL);
}
 
void
fail_get_pins(struct discord *client, struct discord_response *resp)
{
    const struct discord_message *event = resp->keep;
    char text[2000] = "";
 
    snprintf(text, sizeof(text),
             "Failed fetching pinned messages at <#%" PRIu64 ">",
             event->channel_id);
 
    struct discord_create_message params = { .content = text };
    discord_create_message(client, event->channel_id, &params, NULL);
}
 
void
on_get_pins(struct discord *client, const struct discord_message *event)
{
    if (event->author->bot) return;
 
    struct discord_ret_messages ret = {
        .done = &done_get_pins,
        .fail = &fail_get_pins,
        .keep = event,
    };
    discord_get_pinned_messages(client, event->channel_id, &ret);
}
 
int
main(int argc, char *argv[])
{
    const char *config_file;
    if (argc > 1)
        config_file = argv[1];
    else
        config_file = "../config.json";
 
    struct discord *client = discord_from_json(config_file);
    assert(NULL != client && "Couldn't initialize client");
 
    discord_set_on_ready(client, &on_ready);
 
    discord_set_prefix(client, "!");
    discord_set_on_command(client, "pin", &on_pin);
    discord_set_on_command(client, "unpin", &on_unpin);
    discord_set_on_command(client, "get_pins", &on_get_pins);
 
    print_usage();
    fgetc(stdin); // wait for input
 
    discord_run(client);
 
    discord_cleanup(client);
}