/*
 * Copyright 2011-2012, Haiku, Inc. All rights reserved.
 * Copyright 2011, Clemens Zeidler <haiku@clemens-zeidler.de>
 *
 * Distributed under the terms of the MIT License.
 */


#include "DefaultNotifier.h"

#include <Catalog.h>
#include <IconUtils.h>
#include <MailDaemon.h>
#include <Roster.h>

#include <MailPrivate.h>


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "Notifier"


DefaultNotifier::DefaultNotifier(const char* accountName, bool inbound,
        ErrorLogWindow* errorWindow, uint32 showMode)
        :
        fAccountName(accountName),
        fIsInbound(inbound),
        fErrorWindow(errorWindow),
        fNotification(B_PROGRESS_NOTIFICATION),
        fShowMode(showMode),
        fTotalItems(0),
        fItemsDone(0),
        fTotalSize(0),
        fSizeDone(0)
{
        BString desc(fIsInbound ? B_TRANSLATE("Fetching mail for %name")
                : B_TRANSLATE("Sending mail for %name"));
        desc.ReplaceFirst("%name", fAccountName);

        BString identifier;
        identifier << accountName << inbound;
                // Two windows for each acocunt : one for sending and the other for
                // receiving mails
        fNotification.SetMessageID(identifier);
        fNotification.SetGroup(B_TRANSLATE("Mail status"));
        fNotification.SetTitle(desc);

        app_info info;
        be_roster->GetAppInfo(B_MAIL_DAEMON_SIGNATURE, &info);
        BBitmap icon(BRect(0, 0, 32, 32), B_RGBA32);
        BNode node(&info.ref);
        BIconUtils::GetVectorIcon(&node, "BEOS:ICON", &icon);
        fNotification.SetIcon(&icon);
}


DefaultNotifier::~DefaultNotifier()
{
}


BMailNotifier*
DefaultNotifier::Clone()
{
        return new DefaultNotifier(fAccountName, fIsInbound, fErrorWindow,
                fShowMode);
}


void
DefaultNotifier::ShowError(const char* error)
{
        fErrorWindow->AddError(B_WARNING_ALERT, error, fAccountName);
}


void
DefaultNotifier::ShowMessage(const char* message)
{
        fErrorWindow->AddError(B_INFO_ALERT, message, fAccountName);
}


void
DefaultNotifier::SetTotalItems(uint32 items)
{
        fTotalItems = items;
        BString progress;
        progress << fItemsDone << "/" << fTotalItems;
        fNotification.SetContent(progress);
}


void
DefaultNotifier::SetTotalItemsSize(uint64 size)
{
        fTotalSize = size;
        fNotification.SetProgress(fSizeDone / (float)fTotalSize);
}


void
DefaultNotifier::ReportProgress(uint32 messages, uint64 bytes,
        const char* message)
{
        fSizeDone += bytes;
        fItemsDone += messages;

        if (fTotalSize > 0)
                fNotification.SetProgress(fSizeDone / (float)fTotalSize);
        else if (fTotalItems > 0) {
                // No size information available
                // Report progress in terms of message count instead
                fNotification.SetProgress(fItemsDone / (float)fTotalItems);
        } else {
                // No message count information either
                // TODO: we should use a B_INFORMATION_NOTIFICATION here, but it is not
                // possible to change the BNotification type after creating it...
                fNotification.SetProgress(0);
        }

        BString progress;
        progress << message << "\t";

        if (fTotalItems > 0)
                progress << fItemsDone << "/" << fTotalItems;

        fNotification.SetContent(progress);

        int timeout = 0; // Default timeout
        if (fItemsDone >= fTotalItems && fTotalItems != 0)
                timeout = 1; // We're done, make the window go away faster

        _NotifyIfAllowed(timeout);
}


void
DefaultNotifier::ResetProgress(const char* message)
{
        fSizeDone = 0;
        fItemsDone = 0;
        fNotification.SetProgress(0);
        if (message != NULL)
                fNotification.SetTitle(message);
        _NotifyIfAllowed(1); // go away faster
}


void
DefaultNotifier::_NotifyIfAllowed(int timeout)
{
        int32 flag;
        if (fIsInbound)
                flag = B_MAIL_SHOW_STATUS_WINDOW_WHEN_ACTIVE;
        else
                flag = B_MAIL_SHOW_STATUS_WINDOW_WHEN_SENDING;

        if ((fShowMode & flag) != 0)
                fNotification.Send(timeout);
}