/*
 * Treebird - Lightweight frontend for Pleroma
 * Copyright (C) 2022 Nekobit
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include "timeline.h"
#include <stdlib.h>
#include "base_page.h"
#include "../config.h"
#include "index.h"
#include "status.h"
#include "easprintf.h"
#include "reply.h"
#include "navigation.h"
#include "query.h"
#include "error.h"
#include "string_helpers.h"

#include "../static/navigation.chtml"
#include "../static/directs_page.chtml"
#include "../static/hashtag_page.chtml"

/* TODO Clean these up and make a meta function, i'm just lazy */

void tl_home(struct session* ssn, mastodont_t* api, int local)
{
    size_t status_count = 0, statuses_html_count = 0;
    struct mstdnt_status* statuses = NULL;
    struct mstdnt_storage storage = { 0 };
    char* status_format = NULL,
        *post_box,
        *navigation_box = NULL,
        *output = NULL;
    char* start_id;

    struct mstdnt_timeline_args args = {
        .local = local,
        .max_id = ssn->post.max_id,
        .since_id = NULL,
        .min_id = ssn->post.min_id,
        .limit = 20
    };

    try_post_status(ssn, api);
    
    if (mastodont_timeline_home(api, &args, &storage, &statuses, &status_count))
    {
        status_format = construct_error(storage.error, E_ERROR, 1, NULL);
    }
    else {
        // Construct statuses into HTML
        status_format = construct_statuses(api, statuses, status_count, NULL, &statuses_html_count);
        if (!status_format)
            status_format = construct_error("Couldn't load posts", E_ERROR, 1, NULL);
    }

    // Create post box
    post_box = construct_post_box(NULL, "", NULL);
    if (statuses)
    {
        // If not set, set it
        start_id = ssn->post.start_id ? ssn->post.start_id : statuses[0].id;
        navigation_box = construct_navigation_box(start_id,
                                                  statuses[0].id,
                                                  statuses[status_count-1].id,
                                                  NULL);
    }
    easprintf(&output, "%s%s%s",
              post_box,
              STR_NULL_EMPTY(status_format),
              STR_NULL_EMPTY(navigation_box));

    struct base_page b = {
        .category = BASE_CAT_HOME,
        .locale = L10N_EN_US,
        .content = output,
        .sidebar_left = NULL
    };

    // Output
    render_base_page(&b, ssn, api);

    // Cleanup
    mastodont_storage_cleanup(&storage);
    mstdnt_cleanup_statuses(statuses, status_count);
    if (status_format) free(status_format);
    if (post_box) free(post_box);
    if (navigation_box) free(navigation_box);
    if (output) free(output);
}

void tl_direct(struct session* ssn, mastodont_t* api)
{
    size_t status_count = 0, statuses_html_count = 0;
    struct mstdnt_status* statuses = NULL;
    struct mstdnt_storage storage = { 0 };
    char* status_format = NULL,
        *navigation_box = NULL,
        *output = NULL,
        *page = NULL;
    char* start_id;

    struct mstdnt_timeline_args args = {
        .with_muted = 0,
        .max_id = ssn->post.max_id,
        .since_id = NULL,
        .min_id = ssn->post.min_id,
        .limit = 20,
    };

    if (mastodont_timeline_direct(api, &args, &storage, &statuses, &status_count))
    {
        status_format = construct_error(storage.error, E_ERROR, 1, NULL);
    }
    else {
        // Construct statuses into HTML
        status_format = construct_statuses(api, statuses, status_count, NULL, &statuses_html_count);
        if (!status_format)
            status_format = construct_error("Couldn't load posts", E_ERROR, 1, NULL);
    }

    // Create post box
    if (statuses)
    {
        // If not set, set it
        start_id = ssn->post.start_id ? ssn->post.start_id : statuses[0].id;
        navigation_box = construct_navigation_box(start_id,
                                                  statuses[0].id,
                                                  statuses[status_count-1].id,
                                                  NULL);
    }
    
    easprintf(&page, "%s%s",
              STR_NULL_EMPTY(status_format),
              STR_NULL_EMPTY(navigation_box));
    
    easprintf(&output, data_directs_page_html, page);

    struct base_page b = {
        .category = BASE_CAT_DIRECT,
        .locale = L10N_EN_US,
        .content = output,
        .sidebar_left = NULL
    };

    // Output
    render_base_page(&b, ssn, api);

    // Cleanup
    mastodont_storage_cleanup(&storage);
    mstdnt_cleanup_statuses(statuses, status_count);
    if (status_format) free(status_format);
    if (navigation_box) free(navigation_box);
    if (output) free(output);
    if (page) free(page);
}

void tl_public(struct session* ssn, mastodont_t* api, int local, enum base_category cat)
{
    size_t status_count = 0, statuses_html_count = 0;
    struct mstdnt_status* statuses = NULL;
    struct mstdnt_storage storage = { 0 };
    char* status_format = NULL,
        *post_box,
        *navigation_box = NULL,
        *output = NULL;
    char* start_id;

    struct mstdnt_timeline_args args = {
        .local = local,
        .remote = 0,
        .only_media = 0,
        .max_id = ssn->post.max_id,
        .since_id = NULL,
        .min_id = ssn->post.min_id,
        .limit = 20
    };

    try_post_status(ssn, api);
    
    if (mastodont_timeline_public(api, &args, &storage, &statuses, &status_count))
    {
        status_format = construct_error(storage.error, E_ERROR, 1, NULL);
    }
    else {
        // Construct statuses into HTML
        status_format = construct_statuses(api, statuses, status_count, NULL, &statuses_html_count);
        if (!status_format)
            status_format = construct_error("Couldn't load posts", E_ERROR, 1, NULL);
    }

    // Create post box
    post_box = construct_post_box(NULL, "", NULL);
    if (statuses)
    {
        // If not set, set it
        start_id = ssn->post.start_id ? ssn->post.start_id : statuses[0].id;
        navigation_box = construct_navigation_box(start_id,
                                                  statuses[0].id,
                                                  statuses[status_count-1].id,
                                                  NULL);
    }
    easprintf(&output, "%s%s%s",
              post_box,
              STR_NULL_EMPTY(status_format),
              STR_NULL_EMPTY(navigation_box));

    struct base_page b = {
        .category = cat,
        .locale = L10N_EN_US,
        .content = output,
        .sidebar_left = NULL
    };

    // Output
    render_base_page(&b, ssn, api);

    // Cleanup
    mastodont_storage_cleanup(&storage);
    mstdnt_cleanup_statuses(statuses, status_count);
    if (status_format) free(status_format);
    if (post_box) free(post_box);
    if (navigation_box) free(navigation_box);
    if (output) free(output);
}

void tl_list(struct session* ssn, mastodont_t* api, char* list_id)
{
    size_t status_count, statuses_html_count;
    struct mstdnt_status* statuses = NULL;
    struct mstdnt_storage storage = { 0 };
    char* status_format, *post_box, *navigation_box = NULL, *start_id;
    char* output = NULL;

    struct mstdnt_timeline_args args = {
        .max_id = ssn->post.max_id,
        .since_id = NULL,
        .min_id = ssn->post.min_id,
        .limit = 20,
    };

    try_post_status(ssn, api);
    
    if (mastodont_timeline_list(api, list_id, &args, &storage, &statuses, &status_count))
    {
        status_format = construct_error(storage.error, E_ERROR, 1, NULL);
    }
    else {
        // Construct statuses into HTML
        status_format = construct_statuses(api, statuses, status_count, NULL, &statuses_html_count);
        if (!status_format)
            status_format = construct_error("No statuses", E_ERROR, 1, NULL);
    }

    // Create post box
    post_box = construct_post_box(NULL, "", NULL);
    if (statuses)
    {
        // If not set, set it
        start_id = ssn->post.start_id ? ssn->post.start_id : statuses[0].id;
        navigation_box = construct_navigation_box(start_id,
                                                  statuses[0].id,
                                                  statuses[status_count-1].id,
                                                  NULL);
    }
    easprintf(&output, "%s%s%s",
              post_box,
              STR_NULL_EMPTY(status_format),
              STR_NULL_EMPTY(navigation_box));

    struct base_page b = {
        .category = BASE_CAT_LISTS,
        .locale = L10N_EN_US,
        .content = output,
        .sidebar_left = NULL
    };

    // Output
    render_base_page(&b, ssn, api);

    // Cleanup
    mastodont_storage_cleanup(&storage);
    mstdnt_cleanup_statuses(statuses, status_count);
    free(status_format);
    if (post_box) free(post_box);
    if (output) free(output);
    if (navigation_box) free(navigation_box);
}


void tl_tag(struct session* ssn, mastodont_t* api, char* tag_id)
{
    size_t status_count, statuses_html_count;
    struct mstdnt_status* statuses = NULL;
    struct mstdnt_storage storage = { 0 };
    char* status_format, *navigation_box = NULL, *start_id;
    char* output = NULL;

    struct mstdnt_timeline_args args = {
        .max_id = ssn->post.max_id,
        .since_id = NULL,
        .min_id = ssn->post.min_id,
        .limit = 20,
    };

    if (mastodont_timeline_tag(api, tag_id, &args, &storage, &statuses, &status_count))
        status_format = construct_error(storage.error, E_ERROR, 1, NULL);
    else {
        // Construct statuses into HTML
        status_format = construct_statuses(api, statuses, status_count, NULL, &statuses_html_count);
        if (!status_format)
            status_format = construct_error("No statuses", E_ERROR, 1, NULL);
    }

    // Create post box
    if (statuses)
    {
        // If not set, set it
        start_id = ssn->post.start_id ? ssn->post.start_id : statuses[0].id;
        navigation_box = construct_navigation_box(start_id,
                                                  statuses[0].id,
                                                  statuses[status_count-1].id,
                                                  NULL);
    }
    easprintf(&output, data_hashtag_page_html,
              tag_id,
              STR_NULL_EMPTY(status_format),
              STR_NULL_EMPTY(navigation_box));

    struct base_page b = {
        .category = BASE_CAT_NONE,
        .locale = L10N_EN_US,
        .content = output,
        .sidebar_left = NULL
    };

    // Output
    render_base_page(&b, ssn, api);

    // Cleanup
    mastodont_storage_cleanup(&storage);
    mstdnt_cleanup_statuses(statuses, status_count);
    free(status_format);
    if (output) free(output);
    if (navigation_box) free(navigation_box);
}

void content_tl_home(struct session* ssn, mastodont_t* api, char** data)
{
    (void)data;
    if (ssn->cookies.logged_in)
        tl_home(ssn, api, 0);
    else
        content_tl_federated(ssn, api, data);
}

void content_tl_direct(struct session* ssn, mastodont_t* api, char** data)
{
    (void)data;
    tl_direct(ssn, api);
}

void content_tl_federated(struct session* ssn, mastodont_t* api, char** data)
{
    (void)data;
    tl_public(ssn, api, 0, BASE_CAT_FEDERATED);
}

void content_tl_local(struct session* ssn, mastodont_t* api, char** data)
{
    (void)data;
    tl_public(ssn, api, 1, BASE_CAT_LOCAL);
}

void content_tl_list(struct session* ssn, mastodont_t* api, char** data)
{
    tl_list(ssn, api, data[0]);
}

void content_tl_tag(struct session* ssn, mastodont_t* api, char** data)
{
    tl_tag(ssn, api, data[0]);
}