158 lines
3.8 KiB
C++
158 lines
3.8 KiB
C++
/* This file is part of expandurl-mastodon.
|
|
* Copyright © 2018 tastytea <tastytea@tastytea.de>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, version 3.
|
|
*
|
|
* 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <fstream>
|
|
#include <cstdlib> // getenv()
|
|
#include <iostream>
|
|
#include "version.hpp"
|
|
#include "expandurl-mastodon.hpp"
|
|
|
|
using std::cerr;
|
|
using std::string;
|
|
using Mastodon::Easy;
|
|
|
|
Listener::Listener()
|
|
: _instance("")
|
|
, _access_token("")
|
|
, _stream("")
|
|
, _ptr(nullptr)
|
|
{}
|
|
|
|
const bool Listener::start()
|
|
{
|
|
const string filepath = static_cast<const string>(getenv("HOME")) +
|
|
"/.config/expandurl-mastodon.cfg";
|
|
std::ifstream file(filepath);
|
|
|
|
if (file.is_open())
|
|
{
|
|
std::getline(file, _instance);
|
|
_instance = _instance.substr(_instance.find('@') + 1);
|
|
std::getline(file, _access_token);
|
|
file.close();
|
|
}
|
|
else
|
|
{
|
|
cerr << "ERROR: Could not open " << filepath << '\n';
|
|
return false;
|
|
}
|
|
|
|
_thread = std::thread([=]
|
|
{
|
|
Easy masto(_instance, _access_token);
|
|
masto.set_useragent(static_cast<const string>("expandurl-mastodon/") +
|
|
global::version);
|
|
masto.get_stream(Mastodon::API::v1::streaming_user, _stream, _ptr);
|
|
});
|
|
|
|
return true;
|
|
}
|
|
|
|
const void Listener::stop()
|
|
{
|
|
if (_ptr)
|
|
{
|
|
_ptr->abort_stream();
|
|
_thread.join();
|
|
}
|
|
}
|
|
|
|
std::vector<Easy::Notification> Listener::get_new_messages()
|
|
{
|
|
const string buffer = _stream;
|
|
_stream.clear();
|
|
|
|
std::vector<Easy::Notification> v;
|
|
for (const Easy::stream_event &event : Easy::parse_stream(buffer))
|
|
{
|
|
if (event.first == Easy::event_type::Notification)
|
|
{
|
|
v.push_back(Easy::Notification(event.second));
|
|
}
|
|
}
|
|
|
|
return v;
|
|
}
|
|
|
|
Mastodon::Easy::Status Listener::get_status(const std::uint_fast64_t &id)
|
|
{
|
|
Easy masto(_instance, _access_token);
|
|
std::uint_fast16_t ret;
|
|
string answer;
|
|
|
|
ret = masto.get(Easy::v1::statuses_id, {{ "id", { std::to_string(id) }}}, answer);
|
|
if (ret == 0)
|
|
{
|
|
return Easy::Status(answer);
|
|
}
|
|
else
|
|
{
|
|
cerr << "ERROR: " << ret << '\n';
|
|
return Easy::Status();
|
|
}
|
|
}
|
|
|
|
const bool Listener::send_reply(const Easy::Status &status,
|
|
const string &message)
|
|
{
|
|
Easy masto(_instance, _access_token);
|
|
std::uint_fast16_t ret;
|
|
string answer;
|
|
const string id = std::to_string(status.id());
|
|
string strvisibility;
|
|
|
|
switch (status.visibility())
|
|
{
|
|
case Easy::visibility_type::Private:
|
|
strvisibility = "private";
|
|
break;
|
|
case Easy::visibility_type::Direct:
|
|
strvisibility = "direct";
|
|
break;
|
|
default:
|
|
strvisibility = "unlisted";
|
|
break;
|
|
}
|
|
|
|
Easy::parametermap parameters =
|
|
{
|
|
{ "in_reply_to_id", { id } },
|
|
{ "visibility", { strvisibility } },
|
|
{ "status", { '@' + status.account().acct() + ' ' + message } }
|
|
};
|
|
|
|
if (status.sensitive())
|
|
{
|
|
parameters.insert({ "sensitive", { "true" } });
|
|
}
|
|
|
|
if (!status.spoiler_text().empty())
|
|
{
|
|
parameters.insert({ "spoiler_text", { status.spoiler_text() } });
|
|
}
|
|
|
|
ret = masto.post(Easy::v1::statuses, parameters, answer);
|
|
|
|
if (ret == 0)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|