/*
 * Copyright 2017 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Brian Hill
 */


#include "RepositoriesWindow.h"

#include <Alert.h>
#include <Application.h>
#include <Catalog.h>
#include <FindDirectory.h>
#include <LayoutBuilder.h>
#include <NodeMonitor.h>
#include <Region.h>
#include <Screen.h>

#include "AddRepoWindow.h"
#include "constants.h"


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "RepositoriesWindow"


RepositoriesWindow::RepositoriesWindow()
        :
        BWindow(BRect(50, 50, 500, 400), B_TRANSLATE_SYSTEM_NAME("Repositories"),
                B_TITLED_WINDOW, B_NOT_ZOOMABLE | B_ASYNCHRONOUS_CONTROLS
                        | B_AUTO_UPDATE_SIZE_LIMITS),
        fAddWindow(NULL),
        fPackageNodeStatus(B_ERROR)
{
        fView = new RepositoriesView();
        BLayoutBuilder::Group<>(this, B_VERTICAL).Add(fView).End();

        // Size and location on screen
        BRect frame = fSettings.GetFrame();
        ResizeTo(frame.Width(), frame.Height());
        BScreen screen;
        BRect screenFrame = screen.Frame();
        if (screenFrame.right < frame.right || screenFrame.left > frame.left
                || screenFrame.top > frame.top || screenFrame.bottom < frame.bottom) {
                CenterOnScreen();
        }
        else
                MoveTo(frame.left, frame.top);
        Show();

        fMessenger.SetTo(this);

        // Find the pkgman settings or cache directory
        BPath packagePath;
        // /boot/system/settings/package-repositories
        status_t status = find_directory(B_SYSTEM_SETTINGS_DIRECTORY,
                &packagePath);
        if (status == B_OK)
                status = packagePath.Append("package-repositories");
        else {
                // /boot/system/cache/package-repositories
                status = find_directory(B_SYSTEM_CACHE_DIRECTORY, &packagePath);
                if (status == B_OK)
                        status = packagePath.Append("package-repositories");
        }
        if (status == B_OK) {
                BNode packageNode(packagePath.Path());
                if (packageNode.InitCheck()==B_OK && packageNode.IsDirectory())
                        fPackageNodeStatus = packageNode.GetNodeRef(&fPackageNodeRef);
        }

        // watch the pkgman settings or cache directory for changes
        _StartWatching();
}


RepositoriesWindow::~RepositoriesWindow()
{
        _StopWatching();
}


void
RepositoriesWindow::_StartWatching()
{
        if (fPackageNodeStatus == B_OK) {
                status_t result = watch_node(&fPackageNodeRef, B_WATCH_DIRECTORY, this);
                fWatchingPackageNode = (result == B_OK);
        }
}


void
RepositoriesWindow::_StopWatching()
{
        if (fPackageNodeStatus == B_OK && fWatchingPackageNode) {
                watch_node(&fPackageNodeRef, B_STOP_WATCHING, this);
                fWatchingPackageNode = false;
        }
}


bool
RepositoriesWindow::QuitRequested()
{
        if (fView->IsTaskRunning()) {
                BAlert *alert = new BAlert("tasks",
                        B_TRANSLATE_COMMENT("Some tasks are still running. Stop these "
                                "tasks and quit?", "Application quit alert message"),
                        B_TRANSLATE_COMMENT("Stop and quit", "Button label"),
                        kCancelLabel, NULL, B_WIDTH_AS_USUAL, B_STOP_ALERT);
                alert->SetFlags(alert->Flags() | B_CLOSE_ON_ESCAPE);
                int32 result = alert->Go();
                if (result != 0)
                        return false;
        }
        fSettings.SetFrame(Frame());
        be_app->PostMessage(B_QUIT_REQUESTED);
        return BWindow::QuitRequested();
}


void
RepositoriesWindow::MessageReceived(BMessage* message)
{
        switch (message->what)
        {
                case ADD_REPO_WINDOW:
                {
                        BRect frame = Frame();
                        fAddWindow = new AddRepoWindow(frame, fMessenger);
                        break;
                }
                
                case ADD_REPO_URL:
                {
                        BString url;
                        status_t result = message->FindString(key_url, &url);
                        if (result == B_OK)
                                fView->AddManualRepository(url);
                        break;
                }
                
                case ADD_WINDOW_CLOSED:
                {
                        fAddWindow = NULL;
                        break;
                }
                
                case DELETE_KEY_PRESSED:
                {
                        BMessage message(REMOVE_REPOS);
                        fView->MessageReceived(&message);
                        break;
                }
                
                // captures pkgman changes while the Repositories application is running
                case B_NODE_MONITOR:
                {
                        // This preflet is making the changes, so ignore this message
                        if (fView->IsTaskRunning())
                                break;

                        int32 opcode;
                        if (message->FindInt32("opcode", &opcode) == B_OK) {
                                switch (opcode)
                                {
                                        case B_ATTR_CHANGED:
                                        case B_ENTRY_CREATED:
                                        case B_ENTRY_REMOVED:
                                        {
                                                PostMessage(UPDATE_LIST, fView);
                                                break;
                                        }
                                }
                        }
                        break;
                }
                
                default:
                        BWindow::MessageReceived(message);
        }
}