unrealdragon/nekodl
Watch
1
0
Fork
You've already forked nekodl
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Progress: 50%.

Browse source
This commit is contained in:
unrealdragon 2026-08-25 17:26:04 +05:30
commit fa9ca77bfd
1 changed files with 227 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

187
mainscreen.cpp
View file

@ -33,6 +33,18 @@ MainScreen::MainScreen(std::string search_term, QObject *parent)
}
}
void MainScreen::resumeOldTorrentSession(SearchItem it)
{
qDebug() << "Resuming session: " << it.title;
qDebug() << "Old download path: " << it.download_path.string();
TorrentSession *session = new TorrentSession(it, it.download_path);
auto pid = session->start();
if(pid != 0) {
session->deleteLater();
qDebug() << "Parent done.";
}
}
void MainScreen::renderSearchPage(std::string search_term)
{
if(!manager) {
@ -53,6 +65,15 @@ void MainScreen::renderSearchPage(std::string search_term)
void MainScreen::renderHomePage()
{
qDebug() << "Render Home Page";
if(!socket) {
socket = new Socket();
socket_thread = new QThread(this);
socket->moveToThread(socket_thread);
connect(this, &MainScreen::startSocket, socket, &Socket::start);
connect(socket, &Socket::launchTorrentSession, this, &MainScreen::resumeOldTorrentSession);
socket_thread->start();
emit startSocket();
}
}
std::vector<SearchItem> MainScreen::parseSearchPage(std::string html_page)
@ -198,6 +219,7 @@ int MainScreen::startDownloadIndex(std::vector<SearchItem> res, int index)
return -1;
}
auto download_path = expected_path.value();
res[index].download_path = download_path;
TorrentSession *session = new TorrentSession(res[index], download_path);
auto pid = session->start();
if(pid != 0) {
@ -211,6 +233,7 @@ std::expected<std::string,int> MainScreen::runBasicShell(SearchItem it)
{
qDebug() << "Running Basic Shell";
qDebug() << "Valid commands: ls, cd, select, pwd, rmdir, mkdir, set, lstags and exit.";
qDebug() << "mkdir: Use $title to create a folder with name of title.";
qDebug() << "select: Selects the present working dir"
" as download directory if no arguments are provided."
"Else uses the provided argument as a tag or a path.";
@ -218,8 +241,10 @@ std::expected<std::string,int> MainScreen::runBasicShell(SearchItem it)
qDebug() << "set: used for setting a tag. set [TAG_NAME] [FOLDER PATH]";
qDebug() << "lstags: Lists all tags stored.";
namespace fs = std::filesystem;
std::string extras = "";
while(true) {
std::print("nsh|{}>",fs::current_path());
std::print("nsh|{}>{}",fs::current_path(), extras);
extras = "";
std::string command;
std::getline(std::cin, command);
if(command == "ls") {
@ -319,10 +344,16 @@ std::expected<std::string,int> MainScreen::runBasicShell(SearchItem it)
}
if(command.starts_with("mkdir")) {
auto arg = stripForArgument(command, 1);
fs::path path_to_create = fs::current_path() / fs::path(arg);
bool t = fs::create_directory(path_to_create);
if(!t) {
qDebug() << "Failed to create path: " << QString::fromStdString(path_to_create);
if(!arg.starts_with('$')) {
fs::path path_to_create = fs::current_path() / fs::path(arg);
bool t = fs::create_directory(path_to_create);
if(!t) {
qDebug() << "Failed to create path: " << QString::fromStdString(path_to_create);
}
}
else if(arg == "$title") {
extras = "mkdir " + it.title.toStdString();
continue;
}
}
}
@ -343,6 +374,11 @@ std::string MainScreen::stripForArgument(std::string command, int index)
return "";
}
void MainScreen::statusFromSocket(Status st)
{
}
void MainScreen::renderSearchPage_(QNetworkReply *rep)
{
QByteArray data = rep->readAll();
@ -381,6 +417,7 @@ void SearchItem::printData()
qDebug() << "Downloads: " << downloads;
qDebug() << "Timestamp: " << timestamp;
qDebug() << "Magnet: " << magnet;
qDebug() << "Download path: " << download_path.string();
}
void SearchItem::writeData(std::string file_path)
@ -390,9 +427,14 @@ void SearchItem::writeData(std::string file_path)
qDebug() << "Failed to write to file: " << file_path;
return;
}
qDebug() << "Writing title: " << title;
qDebug() << "Writing url: " << url;
qDebug() << "Writing magnet: " << magnet;
qDebug() << "Writing download_path: " << download_path.string();
out << title.toStdString() << std::endl;
out << url.toDisplayString().toStdString() << std::endl;
out << magnet.toStdString() << std::endl;
out << download_path.string() << std::endl;
out.flush();
out.close();
}
@ -404,14 +446,16 @@ void SearchItem::readFromData(std::string file_path)
qDebug() << "Failed to read from file: " << file_path;
return;
}
std::string stitle, surl, smagnet;
in >> stitle;
in >> surl;
in >> smagnet;
std::string stitle, surl, smagnet, spath;
std::getline(in, stitle);
std::getline(in, surl);
std::getline(in, smagnet);
std::getline(in, spath);
in.close();
title = QString::fromStdString(stitle);
url = QUrl(QString::fromStdString(surl));
magnet = QString::fromStdString(smagnet);
download_path = std::filesystem::canonical(spath);
in.close();
}
@ -470,3 +514,128 @@ std::unordered_map<std::string, std::string> MainScreen::Tags::getAllTags(std::s
}
return t;
}
Socket::Socket(QObject *parent)
{
}
//For each resume file found either start the libtorrent session if it was not running or
//Ask each session their status.
void Socket::start()
{
// Generate a list of all active libtorrent sessions.
namespace fs = std::filesystem;
auto temp_dir = QStandardPaths::writableLocation(QStandardPaths::TempLocation).toStdString();
auto temp_dir_path = fs::path(temp_dir);
fs::directory_iterator itr(temp_dir_path);
for(auto &file_folder: itr) {
if(file_folder.is_regular_file() || file_folder.is_socket()) {
auto file_name = file_folder.path().string();
if(file_name.contains("nekodl")
&& file_name.ends_with(".sock")) {
QLocalSocket *socket = new QLocalSocket(this);
connect(socket, &QLocalSocket::readyRead, this, &Socket::socketReadyRead);
socket->connectToServer(QString::fromStdString(file_name));
qDebug() << "Opening file: " << file_name;
if(socket->isOpen()) {
qDebug() << "Open socket: " << file_name;
QString data = "<STATUS>";
socket->write(data.toUtf8());
socket->flush();
active_sockets[file_name] = socket;
}
else {
qDebug() << "Stale socket: " << file_name;
fs::remove(file_name);
disconnect(socket);
socket->deleteLater();
}
}
}
}
qDebug() << "Checking temp dir.";
if(!recheck) {
if(!timer) {
timer = new QTimer(this);
timer->setInterval(5000);
timer->setSingleShot(true);
connect(timer, &QTimer::timeout, this, [&](){
startNotDownloadingSessions();
});
timer->start();
}
if(checked_resume_files) {
if(timer) {
timer->deleteLater();
}
}
recheck = new QTimer(this);
recheck->setInterval(3000);
recheck->setSingleShot(false);
connect(recheck, &QTimer::timeout, this, [this](){
start();
});
recheck->start();
}
}
// Start all sessions with a .resume_file but
// are not in list of active libtorrent session.
void Socket::startNotDownloadingSessions()
{
namespace fs = std::filesystem;
fs::path resume_file_location = fs::path(
QStandardPaths::writableLocation(QStandardPaths::AppLocalDataLocation).toStdString()
);
resume_file_location = resume_file_location / "resumes";
if(!fs::is_directory(resume_file_location)) {
fs::create_directories(resume_file_location);
}
fs::directory_iterator itr(resume_file_location);
for(auto &file_folder: itr) {
qDebug() << "Checking: " << file_folder.path().string();
if(fs::is_regular_file(file_folder)) {
if(file_folder.path().extension() == ".resume_file") {
//Load resume file.
qDebug() << "Resume file found: " << file_folder.path().string();
auto parent_path = file_folder.path().parent_path();
auto file_name = parent_path / file_folder.path().stem();
auto searchitem_file_path = file_name.string() + ".searchitem";
qDebug() << parent_path.string() << ": " << file_name.string();
qDebug() << "Search item path: " << searchitem_file_path;
if(fs::exists(searchitem_file_path))
{
SearchItem it;
it.readFromData(searchitem_file_path);
qDebug() << "Recovering from: " << it.title;
if(!isTitleInActiveSession(it.title)) {
qDebug() << "Title not in active session list.";
it.printData();
emit launchTorrentSession(it);
}
}
else {
qDebug() << "Failed to recover: " << searchitem_file_path;
}
}
}
}
checked_resume_files = true;
}
bool Socket::isTitleInActiveSession(QString title)
{
return active_sockets.find(title.toStdString()) != active_sockets.end();
}
void Socket::socketReadyRead()
{
}