* Copyright 2008 Haiku Inc. All rights reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
* Julun, <host.haiku@gmx.de
*/
#include <PrinterRoster.h>
#include <FindDirectory.h>
#include <Node.h>
#include <NodeMonitor.h>
#include <Path.h>
#include <Printer.h>
#include <new>
namespace BPrivate {
namespace Print {
BPrinterRoster::BPrinterRoster()
: fListener(NULL)
{
BPath path;
find_directory(B_USER_PRINTERS_DIRECTORY, &path, true);
BNode(path.Path()).GetNodeRef(&fUserPrintersNodRef);
fUserPrintersDirectory.SetTo(&fUserPrintersNodRef);
}
BPrinterRoster::~BPrinterRoster()
{
StopWatching();
}
int32
BPrinterRoster::CountPrinters()
{
Rewind();
int32 i = 0;
BPrinter printer;
while (GetNextPrinter(&printer) == B_OK)
i++;
Rewind();
return i;
}
status_t
BPrinterRoster::GetNextPrinter(BPrinter* printer)
{
if (!printer)
return B_BAD_VALUE;
status_t status = fUserPrintersDirectory.InitCheck();
if (!status == B_OK)
return status;
BEntry entry;
bool next = true;
while (status == B_OK && next) {
status = fUserPrintersDirectory.GetNextEntry(&entry);
if (status == B_OK) {
printer->SetTo(entry);
next = !printer->IsValid();
} else {
printer->Unset();
}
}
return status;
}
status_t
BPrinterRoster::GetDefaultPrinter(BPrinter* printer)
{
if (!printer)
return B_BAD_VALUE;
BDirectory dir(&fUserPrintersNodRef);
status_t status = dir.InitCheck();
if (!status == B_OK)
return status;
BEntry entry;
while (dir.GetNextEntry(&entry) == B_OK) {
if (!entry.IsDirectory())
continue;
printer->SetTo(entry);
if (printer->IsValid() && printer->IsDefault())
return B_OK;
}
printer->Unset();
return B_ERROR;
}
status_t
BPrinterRoster::FindPrinter(const BString& name, BPrinter* printer)
{
if (name.Length() <= 0 || !printer)
return B_BAD_VALUE;
BDirectory dir(&fUserPrintersNodRef);
status_t status = dir.InitCheck();
if (!status == B_OK)
return status;
BEntry entry;
while (dir.GetNextEntry(&entry) == B_OK) {
if (!entry.IsDirectory())
continue;
printer->SetTo(entry);
if (printer->IsValid() && printer->Name() == name)
return B_OK;
}
printer->Unset();
return B_ERROR;
}
status_t
BPrinterRoster::Rewind()
{
return fUserPrintersDirectory.Rewind();
}
status_t
BPrinterRoster::StartWatching(const BMessenger& listener)
{
StopWatching();
if (!listener.IsValid())
return B_BAD_VALUE;
fListener = new(std::nothrow) BMessenger(listener);
if (!fListener)
return B_NO_MEMORY;
return watch_node(&fUserPrintersNodRef, B_WATCH_DIRECTORY, *fListener);
}
void
BPrinterRoster::StopWatching()
{
if (fListener) {
stop_watching(*fListener);
delete fListener;
fListener = NULL;
}
}
}
}
