/*
 * Copyright 2017, Haiku, Inc. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Brian Hill <supernova@tycho.email>
 */


#include "UpdateAction.h"

#include <Application.h>
#include <Catalog.h>
#include <package/manager/Exceptions.h>


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "UpdateAction"


using namespace BPackageKit;
//using namespace BPackageKit::BPrivate;
using namespace BPackageKit::BManager::BPrivate;


UpdateAction::UpdateAction(bool verbose)
        :
        fVerbose(verbose)
{
        fUpdateManager = new(std::nothrow)
                UpdateManager(B_PACKAGE_INSTALLATION_LOCATION_SYSTEM, verbose);
}


UpdateAction::~UpdateAction()
{
        delete fUpdateManager;
}


status_t
UpdateAction::Perform(update_type action_request)
{
        try {
                fUpdateManager->CheckNetworkConnection();
                
                update_type action = action_request;
                // Prompt the user if needed
                if (action == USER_SELECTION_NEEDED)
                        action = fUpdateManager->GetUpdateType();
                
                if (action == CANCEL_UPDATE)
                        throw BAbortedByUserException();
                else if (action <= INVALID_SELECTION || action >= UPDATE_TYPE_END)
                        throw BException(B_TRANSLATE(
                                "Invalid update type, cannot continue with updates"));
                
                fUpdateManager->Init(BPackageManager::B_ADD_INSTALLED_REPOSITORIES
                        | BPackageManager::B_ADD_REMOTE_REPOSITORIES
                        | BPackageManager::B_REFRESH_REPOSITORIES);
                fUpdateManager->CheckRepositories();
                
//              fUpdateManager->SetDebugLevel(1);
                if(action == UPDATE) {
                        // These values indicate that all updates should be installed
                        int packageCount = 0;
                        const char* const packages = "";
                        fUpdateManager->Update(&packages, packageCount);
                } else if (action == FULLSYNC)
                        fUpdateManager->FullSync();
                else
                        // Should not happen but just in case
                        throw BException(B_TRANSLATE(
                                "Invalid update type, cannot continue with updates"));
        
        } catch (BFatalErrorException& ex) {
                fUpdateManager->FinalUpdate(B_TRANSLATE("Updates did not complete"),
                        ex.Message());
                return ex.Error();
        } catch (BAbortedByUserException& ex) {
                if (fVerbose)
                        fprintf(stderr, "Updates aborted by user: %s\n",
                                ex.Message().String());
                // No need for a final message since user initiated cancel request
                be_app->PostMessage(kMsgFinalQuit);
                return B_OK;
        } catch (BNothingToDoException& ex) {
                if (fVerbose)
                        fprintf(stderr, "Nothing to do while updating packages : %s\n",
                                ex.Message().String());
                fUpdateManager->FinalUpdate(B_TRANSLATE("No updates available"),
                        B_TRANSLATE("There were no updates found."));
                return B_OK;
        } catch (BException& ex) {
                if (fVerbose)
                        fprintf(stderr, B_TRANSLATE(
                                "Exception occurred while updating packages : %s\n"),
                                ex.Message().String());
                fUpdateManager->FinalUpdate(B_TRANSLATE("Updates did not complete"),
                        ex.Message());
                return B_ERROR;
        }
        
        return B_OK;
}