telodendria/src/Routes/RouteMessages.c

/*
 * Copyright (C) 2022-2024 Jordan Bancino <@jordan:bancino.net> with
 * other valuable contributors. See CONTRIBUTORS.txt for the full list.
 *
 * Permission is hereby granted, free of charge, to any person
 * obtaining a copy of this software and associated documentation files
 * (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
#include <Routes.h>


#include <Cytoplasm/HashMap.h>
#include <Cytoplasm/Memory.h>
#include <Cytoplasm/Json.h>
#include <Cytoplasm/Str.h>

#include <Config.h>
#include <User.h>
#include <Room.h>

#include <string.h>
#include <stdlib.h>

#include <Schema/Filter.h>

ROUTE_IMPL(RouteMessages, path, argp)
{
    RouteArgs *args = argp;
    Db *db = args->matrixArgs->db;

    HashMap *params = HttpRequestParams(args->context);
    HashMap *response = NULL;

    User *user = NULL;
    char *token = NULL;

    char *roomId = ArrayGet(path, 0);
    char *from = HashMapGet(params, "from");
    char *limitStr = HashMapGet(params, "limit");
    char *end = NULL;
    char *sender = NULL;
    char *serverName = NULL;
    CommonID *id = NULL;
    int limit = strtol(limitStr, NULL, 10);

    Array *messages = NULL;

    Room *room = NULL;

    char *err;

    if (!limit || limit < 0)
    {
        limit = 10;
    }
    if (!roomId)
    {
        /* Should be impossible */
        HttpResponseStatus(args->context, HTTP_INTERNAL_SERVER_ERROR);
        return MatrixErrorCreate(M_UNKNOWN, NULL);
    }
    if (HttpRequestMethodGet(args->context) != HTTP_GET)
    {
        err = "Unknown request method.";
        HttpResponseStatus(args->context, HTTP_BAD_REQUEST);
        response = MatrixErrorCreate(M_UNRECOGNIZED, err);
        goto finish;
    }

    serverName = ConfigGetServerName(db);
    if (!serverName)
    {
        HttpResponseStatus(args->context, HTTP_INTERNAL_SERVER_ERROR);
        response = MatrixErrorCreate(M_UNKNOWN, NULL);
        goto finish;
    }

    response = MatrixGetAccessToken(args->context, &token);
    if (response)
    {
        goto finish;
    }

    user = UserAuthenticate(db, token);
    if (!user)
    {
        HttpResponseStatus(args->context, HTTP_UNAUTHORIZED);
        response = MatrixErrorCreate(M_UNKNOWN_TOKEN, NULL);
        goto finish;
    }
    id = UserIdParse(UserGetName(user), serverName);
    id->sigil = '@';
    sender = ParserRecomposeCommonID(*id);

    room = RoomLock(db, roomId);
    if (!RoomContainsUser(room, sender))
    {
        err = "User is not in the room.";
        HttpResponseStatus(args->context, HTTP_UNAUTHORIZED);
        response = MatrixErrorCreate(M_FORBIDDEN, err);
        goto finish;
    }
    RoomUnlock(room);

    /* TODO: Filters, to, and friends */
    messages = UserFetchMessages(user, limit, from, &end);
    if (!messages)
    {
        Room *r = RoomLock(db, roomId);
        Array *leaf = RoomPrevEventsGet(r);
        HashMap *start = JsonValueAsObject(ArrayGet(leaf, 0));
        char *startId = StrDuplicate(
            JsonValueAsString(HashMapGet(start, "event_id"))
        );
        char *token = UserNewMessageToken(user, roomId, startId);
        RoomUnlock(r);

        messages = UserFetchMessages(user, limit, token, &end);
        Free(token);
        Free(startId);
    }
    response = HashMapCreate();
    JsonSet(response, JsonValueArray(messages), 1, "chunk");
    if (end)
    {
        JsonSet(response, JsonValueString(end), 1, "end");
        Free(end);
    }

    UserFreeMessageToken(user, from);
finish:
    UserUnlock(user);
    UserIdFree(id);
    if (sender)
    {
        Free(sender);
    }
    if (serverName)
    {
        Free(serverName);
    }
    return response;
}