278 lines
8.1 KiB
C
278 lines
8.1 KiB
C
/*
|
|
* 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 <stdlib.h>
|
|
#include "helpers.h"
|
|
#include "search.h"
|
|
#include "easprintf.h"
|
|
#include "../config.h"
|
|
#include "string_helpers.h"
|
|
#include "base_page.h"
|
|
#include "status.h"
|
|
#include "hashtag.h"
|
|
#include "error.h"
|
|
#include "account.h"
|
|
#include "graphsnbars.h"
|
|
|
|
// Pages
|
|
#include "../static/search.ctmpl"
|
|
#include "../static/search_all.ctmpl"
|
|
|
|
void search_page(FCGX_Request* req,
|
|
struct session* ssn,
|
|
mastodont_t* api,
|
|
enum search_tab tab,
|
|
char* content)
|
|
{
|
|
char* out_data;
|
|
struct search_template tdata = {
|
|
.prefix = config_url_prefix,
|
|
.query = keystr(ssn->query.query),
|
|
.accounts_active = MAKE_FOCUSED_IF(tab, SEARCH_ACCOUNTS),
|
|
.accounts = "Accounts",
|
|
.hashtags_active = MAKE_FOCUSED_IF(tab, SEARCH_HASHTAGS),
|
|
.hashtags = "Hashtags",
|
|
.statuses_active = MAKE_FOCUSED_IF(tab, SEARCH_STATUSES),
|
|
.statuses = "Statuses",
|
|
.results = content
|
|
};
|
|
out_data = tmpl_gen_search(&tdata, NULL);
|
|
|
|
struct base_page b = {
|
|
.category = BASE_CAT_NONE,
|
|
.content = out_data,
|
|
.sidebar_left = NULL
|
|
};
|
|
|
|
// Output
|
|
render_base_page(&b, req, ssn, api);
|
|
free(out_data);
|
|
}
|
|
|
|
void content_search_all(PATH_ARGS)
|
|
{
|
|
struct mstdnt_args m_args;
|
|
set_mstdnt_args(&m_args, ssn);
|
|
struct mstdnt_storage storage = { 0 };
|
|
struct mstdnt_search_args args = {
|
|
.account_id = NULL,
|
|
.type = 0,
|
|
.resolve = 0,
|
|
.following = 0,
|
|
.with_relationships = 0,
|
|
.max_id = NULL,
|
|
.min_id = NULL,
|
|
.since_id = NULL,
|
|
.offset = 0,
|
|
.limit = 20,
|
|
};
|
|
struct mstdnt_search_results results = { 0 };
|
|
|
|
mastodont_search(api, &m_args, keystr(ssn->query.query), &storage, &args, &results);
|
|
|
|
perl_lock();
|
|
dSP;
|
|
ENTER;
|
|
SAVETMPS;
|
|
PUSHMARK(SP);
|
|
HV* session_hv = perlify_session(ssn);
|
|
XPUSHs(newRV_noinc((SV*)session_hv));
|
|
XPUSHs(newRV_noinc((SV*)template_files));
|
|
XPUSHs(newRV_noinc((SV*)perlify_search_results(&results)));
|
|
|
|
// ARGS
|
|
PUTBACK;
|
|
call_pv("search::content_search", G_SCALAR);
|
|
SPAGAIN;
|
|
|
|
// Duplicate so we can free the TMPs
|
|
char* dup = savesharedsvpv(POPs);
|
|
|
|
PUTBACK;
|
|
FREETMPS;
|
|
LEAVE;
|
|
perl_unlock();
|
|
|
|
struct base_page b = {
|
|
.category = BASE_CAT_NONE,
|
|
.content = dup,
|
|
.session = session_hv,
|
|
.sidebar_left = NULL
|
|
};
|
|
|
|
render_base_page(&b, req, ssn, api);
|
|
|
|
mstdnt_cleanup_search_results(&results);
|
|
mastodont_storage_cleanup(&storage);
|
|
Safefree(dup);
|
|
}
|
|
|
|
void content_search_statuses(PATH_ARGS)
|
|
{
|
|
struct mstdnt_args m_args;
|
|
set_mstdnt_args(&m_args, ssn);
|
|
char* statuses_html;
|
|
struct mstdnt_storage storage = { 0 };
|
|
struct mstdnt_search_args args = {
|
|
.account_id = NULL,
|
|
.type = MSTDNT_SEARCH_STATUSES,
|
|
.resolve = 0,
|
|
.following = 0,
|
|
.with_relationships = 0,
|
|
.max_id = NULL,
|
|
.min_id = NULL,
|
|
.since_id = NULL,
|
|
.offset = 0,
|
|
.limit = 20,
|
|
};
|
|
struct mstdnt_search_results results = { 0 };
|
|
|
|
if (mastodont_search(api,
|
|
&m_args,
|
|
keystr(ssn->query.query),
|
|
&storage,
|
|
&args,
|
|
&results) == 0)
|
|
{
|
|
struct construct_statuses_args statuses_args = {
|
|
.highlight_word = keystr(ssn->query.query),
|
|
};
|
|
statuses_html = construct_statuses(ssn, api, results.statuses, results.statuses_len, &statuses_args, NULL);
|
|
if (!statuses_html)
|
|
statuses_html = construct_error("No statuses", E_ERROR, 1, NULL);
|
|
}
|
|
else
|
|
statuses_html = construct_error("An error occured.", E_ERROR, 1, NULL);
|
|
|
|
search_page(req, ssn, api, SEARCH_STATUSES, STR_NULL_EMPTY(statuses_html));
|
|
|
|
if (statuses_html) free(statuses_html);
|
|
mstdnt_cleanup_search_results(&results);
|
|
mastodont_storage_cleanup(&storage);
|
|
}
|
|
|
|
void content_search_accounts(PATH_ARGS)
|
|
{
|
|
struct mstdnt_args m_args;
|
|
set_mstdnt_args(&m_args, ssn);
|
|
char* accounts_html;
|
|
struct mstdnt_storage storage = { 0 };
|
|
struct mstdnt_search_args args = {
|
|
.account_id = NULL,
|
|
.type = MSTDNT_SEARCH_ACCOUNTS,
|
|
.resolve = 0,
|
|
.following = 0,
|
|
.with_relationships = 0,
|
|
.max_id = NULL,
|
|
.min_id = NULL,
|
|
.since_id = NULL,
|
|
.offset = 0,
|
|
.limit = 20,
|
|
};
|
|
struct mstdnt_search_results results = { 0 };
|
|
|
|
if (mastodont_search(api,
|
|
&m_args,
|
|
keystr(ssn->query.query),
|
|
&storage,
|
|
&args,
|
|
&results) == 0)
|
|
{
|
|
accounts_html = construct_accounts(api, results.accts, results.accts_len, 0, NULL);
|
|
if (!accounts_html)
|
|
accounts_html = construct_error("No accounts", E_ERROR, 1, NULL);
|
|
}
|
|
else
|
|
accounts_html = construct_error("An error occured.", E_ERROR, 1, NULL);
|
|
|
|
search_page(req, ssn, api, SEARCH_ACCOUNTS, STR_NULL_EMPTY(accounts_html));
|
|
|
|
if (accounts_html) free(accounts_html);
|
|
mstdnt_cleanup_search_results(&results);
|
|
mastodont_storage_cleanup(&storage);
|
|
}
|
|
|
|
void content_search_hashtags(PATH_ARGS)
|
|
{
|
|
struct mstdnt_args m_args;
|
|
set_mstdnt_args(&m_args, ssn);
|
|
char* tags_html;
|
|
char* tags_graph = NULL;
|
|
char* tags_bars = NULL;
|
|
char* tags_page;
|
|
struct mstdnt_storage storage = { 0 };
|
|
struct mstdnt_search_args args = {
|
|
.account_id = NULL,
|
|
.type = MSTDNT_SEARCH_HASHTAGS,
|
|
.resolve = 0,
|
|
.following = 0,
|
|
.with_relationships = 0,
|
|
.max_id = NULL,
|
|
.min_id = NULL,
|
|
.since_id = NULL,
|
|
.offset = 0,
|
|
.limit = 20,
|
|
};
|
|
struct mstdnt_search_results results = { 0 };
|
|
|
|
if (mastodont_search(api,
|
|
&m_args,
|
|
keystr(ssn->query.query),
|
|
&storage,
|
|
&args,
|
|
&results) == 0)
|
|
{
|
|
tags_html = construct_hashtags(results.tags, results.tags_len, NULL);
|
|
if (!tags_html)
|
|
tags_html = construct_error("No hashtags", E_ERROR, 1, NULL);
|
|
|
|
tags_bars = construct_hashtags_graph(results.tags,
|
|
results.tags_len,
|
|
14,
|
|
NULL);
|
|
if (tags_bars)
|
|
tags_graph = construct_bar_graph_container(tags_bars, NULL);
|
|
|
|
if (tags_bars) free(tags_bars);
|
|
}
|
|
else
|
|
tags_html = construct_error("An error occured.", E_ERROR, 1, NULL);
|
|
|
|
easprintf(&tags_page, "%s%s", STR_NULL_EMPTY(tags_graph), tags_html);
|
|
|
|
search_page(req, ssn, api, SEARCH_HASHTAGS, tags_page);
|
|
|
|
if (tags_html) free(tags_html);
|
|
if (tags_graph) free(tags_graph);
|
|
free(tags_page);
|
|
mstdnt_cleanup_search_results(&results);
|
|
mastodont_storage_cleanup(&storage);
|
|
}
|
|
|
|
HV* perlify_search_results(struct mstdnt_search_results* results)
|
|
{
|
|
if (!results) return NULL;
|
|
|
|
HV* search_hv = newHV();
|
|
hvstores_ref(search_hv, "accounts", perlify_accounts(results->accts, results->accts_len));
|
|
hvstores_ref(search_hv, "statuses", perlify_statuses(results->statuses, results->statuses_len));
|
|
// TODO tags
|
|
|
|
return search_hv;
|
|
}
|