/*
 * Copyright 2006, Haiku, Inc. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *              Axel Dörfler, axeld@pinc-software.de
 */


#include "PowerStatus.h"

#include <Alert.h>
#include <Application.h>
#include <Catalog.h>
#include <Deskbar.h>
#include <Entry.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "ACPIDriverInterface.h"
#include "APMDriverInterface.h"
#include "PowerStatusWindow.h"


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "PowerStatus"


class PowerStatus : public BApplication {
        public:
                PowerStatus();
                virtual ~PowerStatus();

                virtual void    ArgvReceived(int32 argc, char** argv);
                virtual void    ReadyToRun();
                virtual void    AboutRequested();

        private:
                bool fAutoInstallInDeskbar;
};


const char* kSignature = "application/x-vnd.Haiku-PowerStatus";
const char* kDeskbarSignature = "application/x-vnd.Be-TSKB";
const char* kDeskbarItemName = "PowerStatus";


status_t
our_image(image_info& image)
{
        int32 cookie = 0;
        while (get_next_image_info(B_CURRENT_TEAM, &cookie, &image) == B_OK) {
                if ((char *)our_image >= (char *)image.text
                        && (char *)our_image <= (char *)image.text + image.text_size)
                        return B_OK;
        }

        return B_ERROR;
}


//      #pragma mark -


PowerStatus::PowerStatus()
        :
        BApplication(kSignature),
        fAutoInstallInDeskbar(false)
{
}


PowerStatus::~PowerStatus()
{
}


void
PowerStatus::ArgvReceived(int32 argc, char** argv)
{
        if (argc <= 1)
                return;

        if (strcmp(argv[1], "--help") == 0
                || strcmp(argv[1], "-h") == 0) {
                const char* str = "PowerStatus options:\n"
                        "\t--deskbar\tautomatically add replicant to Deskbar\n"
                        "\t--help\t\tprint this info and exit";
                puts(str);
                Quit();
                return;
        }

        if (strcmp(argv[1], "--deskbar") == 0)
                fAutoInstallInDeskbar = true;
}


void
PowerStatus::ReadyToRun()
{
        bool isInstalled = false;
        bool isDeskbarRunning = true;

        if (ACPIDriverInterface().Connect() != B_OK) {
                if (APMDriverInterface().Connect() != B_OK) {
                        BString text(B_TRANSLATE("No supported battery detected. %appname% "
                                "cannot be used on your system."));
                        text.ReplaceFirst("%appname%", B_TRANSLATE_SYSTEM_NAME("PowerStatus"));
                        BAlert* alert = new BAlert("", text, B_TRANSLATE("Too bad!"),
                                NULL, NULL, B_WIDTH_AS_USUAL, B_WARNING_ALERT);
                        if (!fAutoInstallInDeskbar)
                                alert->Go();
                        Quit();
                        return;
                }
        }

        {
                // if the Deskbar is not alive at this point, it might be after having
                // acknowledged the requester below
                BDeskbar deskbar;
                isDeskbarRunning = deskbar.IsRunning();
                isInstalled = deskbar.HasItem(kDeskbarItemName);
        }

        if (isInstalled && fAutoInstallInDeskbar) {
                Quit();
                return;
        }

        if (isDeskbarRunning && !isInstalled) {
                BString text(B_TRANSLATE("You can run %appname% in a window "
                        "or install it in the Deskbar."));
                text.ReplaceFirst("%appname%", B_TRANSLATE_SYSTEM_NAME("PowerStatus"));
                BAlert* alert = new BAlert("", text, B_TRANSLATE("Run in window"),
                        B_TRANSLATE("Install in Deskbar"), NULL, B_WIDTH_AS_USUAL,
                        B_WARNING_ALERT);

                if (fAutoInstallInDeskbar || alert->Go()) {
                        image_info info;
                        entry_ref ref;

                        if (our_image(info) == B_OK
                                && get_ref_for_path(info.name, &ref) == B_OK) {
                                BDeskbar deskbar;
                                deskbar.AddItem(&ref);
                        }

                        Quit();
                        return;
                }
        }

        BWindow* window = new PowerStatusWindow();
        window->Show();
}


void
PowerStatus::AboutRequested()
{
        BWindow* window = WindowAt(0);
        if (window == NULL)
                return;

        BView* view = window->FindView(kDeskbarItemName);
        if (view == NULL)
                return;

        BMessenger target((BHandler*)view);
        BMessage about(B_ABOUT_REQUESTED);
        target.SendMessage(&about);
}


//      #pragma mark -


int
main(int argc, char* argv[])
{
        PowerStatus app;
        app.Run();

        return 0;
}