/*
 * Copyright 2014, Ingo Weinhold, ingo_weinhold@gmx.de.
 * Distributed under the terms of the MIT License.
 */


#include "PackageWatchingManager.h"

#include <new>

#include <package/PackageRoster.h>

#include <RegistrarDefs.h>

#include "Debug.h"
#include "EventMaskWatcher.h"


using namespace BPackageKit;
using namespace BPrivate;


PackageWatchingManager::PackageWatchingManager()
{
}


PackageWatchingManager::~PackageWatchingManager()
{
}


void
PackageWatchingManager::HandleStartStopWatching(BMessage* request)
{
        status_t error = request->what == B_REG_PACKAGE_START_WATCHING
                ? _AddWatcher(request) : _RemoveWatcher(request);

        if (error == B_OK) {
                BMessage reply(B_REG_SUCCESS);
                request->SendReply(&reply);
        } else {
                BMessage reply(B_REG_ERROR);
                reply.AddInt32("error", error);
                request->SendReply(&reply);
        }
}


void
PackageWatchingManager::NotifyWatchers(BMessage* message)
{
        int32 event;
        if (message->FindInt32("event", &event) != B_OK) {
                WARNING("No event field in notification message\n");
                return;
        }

        uint32 eventMask;
        switch (event) {
                case B_INSTALLATION_LOCATION_PACKAGES_CHANGED:
                        eventMask = B_WATCH_PACKAGE_INSTALLATION_LOCATIONS;
                        break;
                default:
                        WARNING("Invalid event: %" B_PRId32 "\n", event);
                        return;
        }

        EventMaskWatcherFilter filter(eventMask);
    fWatchingService.NotifyWatchers(message, &filter);
}


status_t
PackageWatchingManager::_AddWatcher(const BMessage* request)
{
        BMessenger target;
        uint32 eventMask;
        status_t error;
        if ((error = request->FindMessenger("target", &target)) != B_OK
                || (error = request->FindUInt32("events", &eventMask)) != B_OK) {
                return error;
        }

        Watcher* watcher = new(std::nothrow) EventMaskWatcher(target, eventMask);
        if (watcher == NULL || !fWatchingService.AddWatcher(watcher)) {
                delete watcher;
                return B_NO_MEMORY;
        }

        return B_OK;
}


status_t
PackageWatchingManager::_RemoveWatcher(const BMessage* request)
{
        BMessenger target;
        status_t error;
        if ((error = request->FindMessenger("target", &target)) != B_OK)
                return error;

        if (!fWatchingService.RemoveWatcher(target))
                return B_BAD_VALUE;

        return B_OK;
}