transmission/beos/TRWindow.cpp

589 lines
18 KiB
C++
Raw Normal View History

/**
* Copyright (C) 2007 Bryan Varner
*
* 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 substantial 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.
*
* $Id:$
*/
2006-07-16 19:39:23 +00:00
#include "TRWindow.h"
#include <stdio.h>
#include <Alert.h>
#include <Application.h>
#include <File.h>
#include <MenuBar.h>
#include <MenuItem.h>
#include <NodeMonitor.h>
#include <ScrollView.h>
#include <String.h>
#include <malloc.h>
#include <time.h> /* for time() */
#include <unistd.h> /* for sleep() */
2006-07-16 19:39:23 +00:00
#include "Prefs.h"
#include "TRApplication.h"
#include "TRTransfer.h"
#include "TRInfoWindow.h"
BListView *TRWindow::transfers = NULL;
2006-07-16 19:39:23 +00:00
/**
* The Transmission Window! Yay!
*/
TRWindow::TRWindow() : BWindow(BRect(10, 40, 350, 110), "Transmission", B_TITLED_WINDOW,
B_ASYNCHRONOUS_CONTROLS , B_CURRENT_WORKSPACE)
{
engine = NULL;
stopping = false;
quitter = NULL;
2006-07-16 19:39:23 +00:00
Prefs prefs(TRANSMISSION_SETTINGS);
BRect *rectFrame = new BRect();
if (prefs.FindRect("window.frame", rectFrame) == B_OK) {
MoveTo(rectFrame->LeftTop());
ResizeTo(rectFrame->Width(), rectFrame->Height());
} else {
rectFrame->Set(10, 40, 350, 110);
}
Lock();
BRect viewRect(0, 0, rectFrame->Width(), rectFrame->Height());
BMenuBar *menubar = new BMenuBar(viewRect, "MenuBar");
BMenu *menu = new BMenu("File");
menu->AddItem(new BMenuItem("Open", new BMessage(TR_OPEN), 'O', B_COMMAND_KEY));
menu->FindItem(TR_OPEN)->SetTarget(be_app_messenger); // send OPEN to the be_app.
menu->AddSeparatorItem();
menu->AddItem(new BMenuItem("Quit", new BMessage(B_QUIT_REQUESTED), 'Q', B_COMMAND_KEY));
menubar->AddItem(menu);
menu = new BMenu("Torrent");
menu->AddItem(new BMenuItem("Get Info", new BMessage(TR_INFO), 'I', B_COMMAND_KEY));
menu->FindItem(TR_INFO)->SetEnabled(false);
menu->AddSeparatorItem();
menu->AddItem(new BMenuItem("Resume", new BMessage(TR_RESUME)));
menu->AddItem(new BMenuItem("Pause", new BMessage(TR_PAUSE)));
menu->AddItem(new BMenuItem("Remove", new BMessage(TR_REMOVE)));
menubar->AddItem(menu);
menu = new BMenu("Tools");
menu->AddItem(new BMenuItem("Settings", new BMessage(TR_SETTINGS)));
menu->FindItem(TR_SETTINGS)->SetTarget(be_app_messenger);
menu->AddSeparatorItem();
menu->AddItem(new BMenuItem("About Transmission", new BMessage(B_ABOUT_REQUESTED)));
menu->FindItem(B_ABOUT_REQUESTED)->SetTarget(be_app_messenger);
menubar->AddItem(menu);
AddChild(menubar);
SetKeyMenuBar(menubar);
// TODO: Tool Bar? (Well after everything is working based on Menus)
// Setup the transfers ListView
viewRect.Set(2, menubar->Frame().bottom + 3, rectFrame->Width() - 2 - B_V_SCROLL_BAR_WIDTH, rectFrame->Height() - 2);
TRWindow::transfers = new BListView(viewRect, "TorrentList", B_SINGLE_SELECTION_LIST, B_FOLLOW_ALL);
TRWindow::transfers->SetSelectionMessage(new BMessage(TR_SELECT));
2006-07-16 19:39:23 +00:00
AddChild(new BScrollView("TransferScroller", transfers, B_FOLLOW_ALL, 0, false, true));
Unlock();
delete rectFrame;
// Bring up the Transmission Engine
engine = tr_init( "beos" );
2006-07-16 19:39:23 +00:00
LoadSettings();
UpdateList(-1, true);
// Start the message loop without showing the window.
Hide();
Show();
}
static void torrentclose(tr_torrent_t *torrent, void *)
{
tr_torrentClose(torrent);
}
2006-07-16 19:39:23 +00:00
TRWindow::~TRWindow() {
tr_torrentIterate(engine, torrentclose, NULL);
const int MAX_EXIT_WAIT_SECS = 10;
const time_t deadline = time(0) + MAX_EXIT_WAIT_SECS;
2007-08-05 04:04:04 +00:00
while (tr_torrentCount(engine) && time(NULL) < deadline) {
snooze(100000);
}
/* XXX there's no way to make sure the torrent threads are running so this might crash */
2006-07-16 19:39:23 +00:00
tr_close(engine);
stop_watching(this);
delete quitter;
2006-07-16 19:39:23 +00:00
}
void TRWindow::LoadSettings() {
if (engine != NULL) {
Prefs prefs(TRANSMISSION_SETTINGS);
int32 bindPort;
if (prefs.FindInt32("transmission.bindPort", &bindPort) != B_OK) {
bindPort = 9000;
prefs.SetInt32("transmission.bindPort", bindPort);
}
tr_setBindPort(engine, (int)bindPort);
int32 uploadLimit;
if (prefs.FindInt32("transmission.uploadLimit", &uploadLimit) != B_OK) {
uploadLimit = 20;
prefs.SetInt32("transmission.uploadLimit", uploadLimit);
}
tr_setGlobalSpeedLimit(engine, TR_UP, (int)uploadLimit);
2006-07-16 19:39:23 +00:00
}
}
/**
* Rescans the active Torrents folder, and will add all the torrents there to the
* engine. Called during initial Application Start & Stop.
*/
void TRWindow::RescanTorrents() {
if (Lock()) {
TRApplication *app = dynamic_cast<TRApplication*>(be_app);
BEntry *torrentEntry = new BEntry();
status_t err;
if (app->TorrentDir()->InitCheck() == B_OK) {
err = app->TorrentDir()->Rewind();
while (err == B_OK) {
err = app->TorrentDir()->GetNextEntry(torrentEntry, true);
if (err != B_ENTRY_NOT_FOUND) {
AddEntry(torrentEntry);
}
}
}
delete torrentEntry;
Unlock();
}
}
/**
* Adds the file specified by *torrent to the Transmission engine.
* Then adds a new TRTransfer item in the transfers list.
* This item holds cached information about the torrent entry and node.
* These TRTransmission items are _NOT_ guaranteed to render the entry
* they were created from.
*/
void TRWindow::AddEntry(BEntry *torrent) {
node_ref node;
if (torrent->GetNodeRef(&node) == B_OK) {
if (watch_node(&node, B_WATCH_NAME, this) == B_OK) {
BPath path;
torrent->GetPath(&path);
// Try adding the torrent to the engine.
int error;
tr_torrent_t *nTorrent;
nTorrent = tr_torrentInit(engine, path.Path(), GetFolder().String(),
TR_FLAG_PAUSED, &error);
if (nTorrent != NULL && Lock()) { // Success. Add the TRTorrent item.
transfers->AddItem(new TRTransfer(path.Path(), node, nTorrent));
2006-07-16 19:39:23 +00:00
bool autoStart = true;
// Decide if we should auto-start this torrent or not.
BString prefName("download.");
prefName << path.Path() << ".running";
Prefs *prefs = new Prefs(TRANSMISSION_SETTINGS);
if (prefs->FindBool(prefName.String(), &autoStart) != B_OK) {
autoStart = true;
}
delete prefs;
if (autoStart) {
// Start the newly added torrent.
worker_info *startData = (worker_info*)calloc(1, sizeof(worker_info));
startData->window = this;
startData->torrent = nTorrent;
2006-07-16 19:39:23 +00:00
thread_id start_thread = spawn_thread(TRWindow::AsynchStartTorrent, "BirthCanal",
B_NORMAL_PRIORITY, (void *)startData);
if (!((start_thread) < B_OK)) {
resume_thread(start_thread);
} else { // Fallback and start the old way.
StartTorrent(startData->torrent);
2006-07-16 19:39:23 +00:00
free(startData);
}
}
Unlock();
} else {
bool duplicate = false;
TRTransfer* tr;
for (int32 i = 0; i < transfers->CountItems(); i++) {
tr = (TRTransfer*)transfers->ItemAt(i);
if (tr->GetCachedNodeRef() == node) {
duplicate = true;
}
}
if (!duplicate) {
BString errmsg("An error occurred trying to read ");
char namebuf[B_FILE_NAME_LENGTH];
torrent->GetName(namebuf);
errmsg << namebuf;
errmsg << ".";
BAlert *error = new BAlert("Error Opening Torrent",
errmsg.String(),
"Ok", NULL, NULL,
B_WIDTH_AS_USUAL, B_WARNING_ALERT);
error->Go();
torrent->Remove();
}
}
}
}
}
void TRWindow::MessageReceived(BMessage *msg) {
/*
* The only messages we receive from the node_monitor are if we need to
* stop watching the node. Basically, if it's been moved or removed we stop.
*/
if (msg->what == B_NODE_MONITOR) {
node_ref node;
ino_t fromDir;
ino_t toDir;
int32 opcode;
if ((msg->FindInt32("opcode", &opcode) == B_OK) &&
(msg->FindInt64("node", &node.node) == B_OK) &&
(msg->FindInt32("device", &node.device) == B_OK))
{
bool stop = (opcode == B_ENTRY_REMOVED);
if (stop) {
msg->FindInt64("directory", &toDir);
} else { // It must have moved.
stop = ((msg->FindInt64("from directory", &fromDir) == B_OK) &&
(msg->FindInt64("to directory", &toDir) == B_OK) &&
(toDir != fromDir));
}
if (stop) {
watch_node(&node, B_STOP_WATCHING, this);
/* Find the full path from the TRTorrents.
* The index of the TRTorrent that is caching the information
* IS NOT the index of the torrent in the engine. These are
* Totally decoupled, due to the way transmission is written.
*/
remove_info *removeData = (remove_info*)calloc(1, sizeof(remove_info));
removeData->window = this;
2006-07-16 19:39:23 +00:00
TRTransfer* item;
for (int32 i = 0; i < transfers->CountItems(); i++) {
item = (TRTransfer*)transfers->ItemAt(i);
if (item->GetCachedNodeRef() == node) {
strcpy(removeData->path, item->GetCachedPath());
2006-07-16 19:39:23 +00:00
}
}
transfers->DoForEach(TRWindow::RemovePath, (void *)removeData);
free(removeData);
2006-07-16 19:39:23 +00:00
}
}
} else if (msg->what == TR_INFO) {
// Display an Info Window.
TRTransfer *transfer = dynamic_cast<TRTransfer*>(transfers->ItemAt(transfers->CurrentSelection()));
const tr_stat_t *s = tr_torrentStat(transfer->GetTorrent());
const tr_info_t *i = tr_torrentInfo(transfer->GetTorrent());
2006-07-16 19:39:23 +00:00
TRInfoWindow *info = new TRInfoWindow(s, i, tr_torrentGetFolder(transfer->GetTorrent()));
2006-07-16 19:39:23 +00:00
info->MoveTo(Frame().LeftTop() + BPoint(20, 25));
info->Show();
} else if (msg->what == TR_SELECT) {
// Setup the Torrent Menu enabled / disabled state.
int32 selection;
msg->FindInt32("index", &selection);
UpdateList(selection, true);
} else if (msg->what == TR_RESUME) {
worker_info *startData = (worker_info*)calloc(1, sizeof(worker_info));
startData->window = this;
startData->torrent = (dynamic_cast<TRTransfer*>(transfers->ItemAt(transfers->CurrentSelection())))->GetTorrent();
2006-07-16 19:39:23 +00:00
thread_id start_thread = spawn_thread(TRWindow::AsynchStartTorrent, "BirthCanal",
B_NORMAL_PRIORITY, (void *)startData);
if (!((start_thread) < B_OK)) {
resume_thread(start_thread);
} else { // Fallback and start the old way.
StartTorrent(startData->torrent);
2006-07-16 19:39:23 +00:00
free(startData);
}
} else if (msg->what == TR_PAUSE) {
worker_info *stopData = (worker_info*)calloc(1, sizeof(worker_info));
stopData->window = this;
stopData->torrent = (dynamic_cast<TRTransfer*>(transfers->ItemAt(transfers->CurrentSelection())))->GetTorrent();
2006-07-16 19:39:23 +00:00
thread_id stop_thread = spawn_thread(TRWindow::AsynchStopTorrent, "InUtero",
B_NORMAL_PRIORITY, (void *)stopData);
if (!((stop_thread) < B_OK)) {
resume_thread(stop_thread);
} else { // Fallback and stop it the old way.
StopTorrent(stopData->torrent);
2006-07-16 19:39:23 +00:00
free(stopData);
}
} else if (msg->what == TR_REMOVE) {
int32 index = transfers->CurrentSelection();
// Remove the file from the filesystem.
TRTransfer *item = (TRTransfer*)transfers->RemoveItem(index);
tr_torrentClose(item->GetTorrent());
2006-07-16 19:39:23 +00:00
BEntry *entry = new BEntry(item->GetCachedPath(), true);
entry->Remove();
delete entry;
delete item;
UpdateList(transfers->CurrentSelection(), true);
} else if (msg->what == B_SIMPLE_DATA) {
be_app->RefsReceived(msg);
}
BWindow::MessageReceived(msg);
}
/**
* Handles QuitRequests.
* Displays a BAlert asking if the user really wants to quit if torrents are running.
* If affimative, then we'll stop all the running torrents.
*/
bool TRWindow::QuitRequested() {
if (stopping)
return true;
2006-07-16 19:39:23 +00:00
bool quit = false;
int running;
2006-07-16 19:39:23 +00:00
quit_info *quitData = (quit_info*)calloc(1, sizeof(quit_info));
quitData->running = 0;
transfers->DoForEach(TRWindow::CheckQuitStatus, (void *)quitData);
running = quitData->running;
free(quitData);
2006-07-16 19:39:23 +00:00
if (running > 0) {
2006-07-16 19:39:23 +00:00
BString quitMsg("");
quitMsg << "There's " << running << " torrent";
if (running > 1) {
2006-07-16 19:39:23 +00:00
quitMsg << "s";
}
quitMsg << " currently running.\n"
<< "What would you like to do?";
BAlert *confirmQuit = new BAlert("Confirm Quit", quitMsg.String(),
"Cancel", "Quit", NULL,
B_WIDTH_AS_USUAL, B_WARNING_ALERT);
quit = (confirmQuit->Go() == 1);
} else {
quit = true;
}
if (quit) {
Prefs *prefs = new Prefs(TRANSMISSION_SETTINGS);
prefs->SetRect("window.frame", Frame());
BString strItem("");
for (int i = 0; i < transfers->CountItems(); i++) {
2006-07-16 19:39:23 +00:00
strItem = "download.";
tr_torrent_t *torrent = (dynamic_cast<TRTransfer*>(transfers->ItemAt(i)))->GetTorrent();
const tr_info_t *info = tr_torrentInfo(torrent);
const tr_stat_t *stat = tr_torrentStat(torrent);
strItem << info->torrent << ".running";
if (stat->status & (TR_STATUS_CHECK_WAIT | TR_STATUS_CHECK | TR_STATUS_DOWNLOAD | TR_STATUS_SEED)) {
2006-07-16 19:39:23 +00:00
prefs->SetBool(strItem.String(), true);
tr_torrentStop(torrent);
2006-07-16 19:39:23 +00:00
} else {
prefs->SetBool(strItem.String(), false);
}
}
delete prefs;
if (running > 0) {
stopping = true;
BAlert *waiting = new BAlert("Stopping Torrents", "Waiting for torrents to stop...", "Quit");
quitter = new BInvoker(new BMessage(B_QUIT_REQUESTED), BMessenger(be_app));
waiting->Go(quitter);
quit = false;
} else {
be_app->PostMessage(new BMessage(B_QUIT_REQUESTED));
}
2006-07-16 19:39:23 +00:00
}
return quit;
}
void TRWindow::FrameResized(float, float) {
2006-07-16 19:39:23 +00:00
transfers->Invalidate();
}
/**
* Called from the StopTorrent thread.
*/
void TRWindow::StopTorrent(tr_torrent_t *torrent) {
tr_torrentStop(torrent);
2006-07-16 19:39:23 +00:00
UpdateList(transfers->CurrentSelection(), true);
2006-07-16 19:39:23 +00:00
}
BString TRWindow::GetFolder(void) {
2006-07-16 19:39:23 +00:00
// Read the settings.
BString folder("");
Prefs *prefs = new Prefs(TRANSMISSION_SETTINGS);
if (prefs->FindString("download.folder", &folder) != B_OK) {
prefs->SetString("download.folder", "/boot/home/Downloads");
folder << "/boot/home/Downloads";
}
delete prefs;
return folder;
}
/**
* Called from StartTorrent thread.
*/
void TRWindow::StartTorrent(tr_torrent_t *torrent) {
tr_torrentSetFolder(torrent, GetFolder().String());
tr_torrentStart(torrent);
2006-07-16 19:39:23 +00:00
if (transfers->CurrentSelection() >= 0) {
UpdateList(transfers->CurrentSelection(), true);
2006-07-16 19:39:23 +00:00
}
}
/**
* Called from the be_app Pulse();
* This will update the data structures that the TRTorrents use to render,
* and invalidate the view.
*/
void TRWindow::UpdateList(int32 selection = -1, bool menus = true) {
if (stopping)
{
quit_info *quitData = (quit_info*)calloc(1, sizeof(quit_info));
quitData->running = 0;
transfers->DoForEach(TRWindow::CheckQuitStatus, (void *)quitData);
if (quitData->running == 0)
be_app->PostMessage(new BMessage(B_QUIT_REQUESTED));
free(quitData);
}
update_info *upData = (update_info*)calloc(1, sizeof(update_info));
upData->running = false;
upData->selected = selection;
upData->invalid = 0;
2006-07-16 19:39:23 +00:00
transfers->DoForEach(TRWindow::UpdateStats, (void *)upData);
2006-07-16 19:39:23 +00:00
if (menus) {
KeyMenuBar()->FindItem(TR_INFO)->SetEnabled(selection >= 0);
KeyMenuBar()->FindItem(TR_RESUME)->SetEnabled(selection >= 0 && !upData->running);
KeyMenuBar()->FindItem(TR_PAUSE)->SetEnabled(selection >= 0 && upData->running);
KeyMenuBar()->FindItem(TR_REMOVE)->SetEnabled(selection >= 0 && !upData->running);
2006-07-16 19:39:23 +00:00
}
if (upData->invalid > 0 && transfers->LockLooper()) {
transfers->Invalidate();
transfers->UnlockLooper();
2006-07-16 19:39:23 +00:00
}
free(upData);
2006-07-16 19:39:23 +00:00
}
2006-07-16 19:39:23 +00:00
/**
* Thread Function to stop Torrents. This can be expensive and causes the event loop to
* choke.
*/
int32 TRWindow::AsynchStopTorrent(void *data) {
worker_info* stopData = (worker_info*)data;
stopData->window->StopTorrent(stopData->torrent);
2006-07-16 19:39:23 +00:00
free(stopData);
return B_OK;
}
/**
* Thread Function to start Torrents. This can be expensive and causes the event loop to
* choke.
*/
int32 TRWindow::AsynchStartTorrent(void *data) {
worker_info* startData = (worker_info*)data;
startData->window->StartTorrent(startData->torrent);
2006-07-16 19:39:23 +00:00
free(startData);
return B_OK;
}
/**
* Invoked by DoForEach upon the transfers list. This will
* remove the item that is caching the path specified by
* path.
*/
bool TRWindow::RemovePath(BListItem *item, void *data) {
remove_info* removeData = (remove_info*)data;
TRTransfer *transfer = dynamic_cast<TRTransfer*>(item);
if (strcmp(transfer->GetCachedPath(), removeData->path) == 0) {
removeData->window->transfers->RemoveItem(transfer);
return true;
}
return false;
}
/**
* Invoked during QuitRequested, iterates all Transfers and
* checks to see how many are running.
*/
bool TRWindow::CheckQuitStatus(BListItem *item, void *data) {
quit_info* quitData = (quit_info*)data;
TRTransfer *transfer = dynamic_cast<TRTransfer*>(item);
if (transfer->IsRunning()) {
quitData->running++;
}
return false;
}
/**
* Invoked during UpdateList()
*/
bool TRWindow::UpdateStats(BListItem *item, void *data) {
update_info* upData = (update_info*)data;
TRTransfer *transfer = dynamic_cast<TRTransfer*>(item);
int32 index = transfers->IndexOf(transfer);
if (transfer->UpdateStatus(tr_torrentStat(transfer->GetTorrent()), index % 2 == 0)) {
upData->invalid++;
}
if (index == upData->selected) {
upData->running = transfer->IsRunning();
}
return false;
}