/*
 * Copyright 2008, François Revol, <revol@free.fr>. All rights reserved.
 * Distributed under the terms of the MIT License.
 */
#include <Catalog.h>
#include <ScrollView.h>
#include <String.h>
#include <Window.h>
#ifdef __HAIKU__
#include <Layout.h>
#endif

#include <pwd.h>

#include "LoginApp.h"
#include "LoginView.h"

#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "Login View"

#define LW 120
#define CSEP 15
#define BH 20
#define BW 60

class PwdItem : public BStringItem {
public:
                                        PwdItem(struct passwd *pwd, uint32 level = 0,
                                                bool expanded = true)
                                                 : BStringItem("", level, expanded)
                                                {
                                                        if (pwd) {
                                                                BString name(pwd->pw_gecos);
                                                                // TODO: truncate at first ;
                                                                fLogin = pwd->pw_name;
                                                                SetText(name.String());
                                                        }
                                                };
        virtual                 ~PwdItem() {};
        const char*             Login() const { return fLogin.String(); };
private:
        BString                 fLogin;
};


LoginView::LoginView(BRect frame)
        : BView(frame, "LoginView", B_FOLLOW_ALL, B_PULSE_NEEDED)
{
        // TODO: when I don't need to test in BeOS anymore,
        // rewrite to use layout engine.
        SetViewUIColor(B_PANEL_BACKGROUND_COLOR);
        SetLowUIColor(ViewUIColor());
        BRect r;
        r.Set(CSEP, CSEP, LW, Bounds().Height() - 3 * CSEP - BH);
        fUserList = new BListView(r, "users");
        BScrollView *sv = new BScrollView("userssv", fUserList,
                B_FOLLOW_LEFT | B_FOLLOW_TOP, 0, false, true);
        AddChild(sv);
        fUserList->SetSelectionMessage(new BMessage(kUserSelected));
        fUserList->SetInvocationMessage(new BMessage(kUserInvoked));

        r.Set(LW + 30, Bounds().top + CSEP,
                Bounds().right - CSEP, Bounds().top + CSEP + CSEP);
        fLoginControl = new BTextControl(r, "login", B_TRANSLATE("Login:"), "",
                new BMessage(kLoginEdited));
        AddChild(fLoginControl);

        r.OffsetBySelf(0, CSEP + CSEP);
        fPasswordControl = new BTextControl(r, "password",
                B_TRANSLATE("Password:"), "", new BMessage(kPasswordEdited));
        fPasswordControl->TextView()->HideTyping(true);
        AddChild(fPasswordControl);

        r.OffsetBySelf(0, CSEP + CSEP);
        fHidePasswordCheckBox = new BCheckBox(r, "hidepw",
                B_TRANSLATE("Hide password"), new BMessage(kHidePassword));
        fHidePasswordCheckBox->SetValue(1);
        AddChild(fHidePasswordCheckBox);

        // buttons
        float buttonWidth = BW; //(Bounds().Width() - 4 * CSEP) / 3;
        BRect buttonRect(0, Bounds().bottom - BH,
                buttonWidth, Bounds().bottom);
        buttonRect.OffsetBySelf(CSEP, -CSEP);

        fHaltButton = new BButton(buttonRect, "halt", B_TRANSLATE("Halt"),
                new BMessage(kHaltAction));
        fHaltButton->ResizeToPreferred();
        AddChild(fHaltButton);

        buttonRect.OffsetBySelf(CSEP + fHaltButton->Frame().Width(), 0);
        fRebootButton = new BButton(buttonRect, "reboot", B_TRANSLATE("Reboot"),
                new BMessage(kRebootAction));

        fRebootButton->ResizeToPreferred();
        AddChild(fRebootButton);

        BRect infoRect(buttonRect);
        infoRect.OffsetBySelf(fRebootButton->Frame().Width() + CSEP, 0);

        buttonRect.OffsetToSelf(Bounds().Width() - CSEP - buttonWidth,
                Bounds().Height() - CSEP - BH);
        fLoginButton = new BButton(buttonRect, "ok", B_TRANSLATE("OK"),
                new BMessage(kAttemptLogin));
        fLoginButton->ResizeToPreferred();
        AddChild(fLoginButton);

        infoRect.right = buttonRect.left - CSEP + 5;
        BString info;
        //info << hostname;
        fInfoView = new BStringView(infoRect, "info", info.String());
        AddChild(fInfoView);
}


LoginView::~LoginView()
{
}

void
LoginView::AttachedToWindow()
{
        fUserList->SetTarget(this);
        fLoginControl->SetTarget(this);
        fPasswordControl->SetTarget(this);
        fHidePasswordCheckBox->SetTarget(this);
        fHaltButton->SetTarget(be_app_messenger);
        fRebootButton->SetTarget(be_app_messenger);
        fLoginButton->SetTarget(this);
        Window()->SetDefaultButton(fLoginButton);
        //fLoginControl->MakeFocus();
        fUserList->MakeFocus();
        // populate user list
        BMessenger(this).SendMessage(kAddNextUser);

        // size window relative to buttons
        BRect bounds = Window()->Bounds();
        float spacing = fHaltButton->Frame().left;
        bounds.bottom = fLoginButton->Frame().bottom + spacing;
        bounds.right = fLoginButton->Frame().right + spacing;
        Window()->ResizeTo(bounds.Width(), bounds.Height());

        // Center info view
        BPoint leftTop = fInfoView->Frame().LeftTop();
        leftTop.y += fHaltButton->Frame().Height() / 2;
        leftTop.y -= fInfoView->Bounds().Height() / 2;
        fInfoView->MoveTo(leftTop);
}


void
LoginView::MessageReceived(BMessage *message)
{
        switch (message->what) {
                case kSetProgress:
                        break;
                case kAddNextUser:
                        AddNextUser();
                        break;
                case kUserSelected:
                {
                        int32 selection = fUserList->CurrentSelection();
                        if (selection > -1) {
                                PwdItem *item = dynamic_cast<PwdItem *>(
                                        fUserList->ItemAt(selection));
                                if (item)
                                        fLoginControl->SetText(item->Login());
                        }
                        break;
                }
                case kUserInvoked:
                        fPasswordControl->MakeFocus();
                        break;
                case kLoginEdited:
                        break;
                case kHidePassword:
                        fPasswordControl->TextView()->HideTyping(
                                fHidePasswordCheckBox->Value());
                        break;
                case kAttemptLogin:
                {
                        // if no pass specified and we were selecting the user,
                        // give a chance to enter the password
                        // else we might want to enter an empty password.
                        if (strlen(fPasswordControl->Text()) < 1
                                && (fUserList->IsFocus() || fLoginControl->IsFocus())) {
                                fPasswordControl->MakeFocus();
                                break;
                        }
                        BMessage *m = new BMessage(kAttemptLogin);
                        m->AddString("login", fLoginControl->Text());
                        m->AddString("password", fPasswordControl->Text());
                        be_app->PostMessage(m, NULL, this);
                        break;
                }
                case kLoginBad:
                        fPasswordControl->SetText("");
                        EnableControls(false);
                        fInfoView->SetText(B_TRANSLATE("Invalid login!"));
                        if (Window()) {
                                BPoint savedPos = Window()->Frame().LeftTop();
                                for (int i = 0; i < 10; i++) {
                                        BPoint p(savedPos);
                                        p.x += (i%2) ? 10 : -10;
                                        Window()->MoveTo(p);
                                        Window()->UpdateIfNeeded();
                                        snooze(30000);
                                }
                                Window()->MoveTo(savedPos);
                        }
                        EnableControls(true);
                        break;
                case kLoginOk:
                        // XXX: quit ?
                        if (Window()) {
                                Window()->Hide();
                        }
                        break;
                default:
                        message->PrintToStream();
                        BView::MessageReceived(message);
        }
}


void
LoginView::Pulse()
{
        BString info;
        time_t now = time(NULL);
        struct tm *t = localtime(&now);
        // TODO: use strftime and locale settings
        info << asctime(t);
        info.RemoveSet("\r\n");
        fInfoView->SetText(info.String());
}


void
LoginView::AddNextUser()
{
        struct passwd *pwd;
        if (fUserList->CountItems() < 1)
                setpwent();

        pwd = getpwent();

        if (pwd && pwd->pw_shell &&
                strcmp(pwd->pw_shell, "false") &&
                strcmp(pwd->pw_shell, "true") &&
                strcmp(pwd->pw_shell, "/bin/false") &&
                strcmp(pwd->pw_shell, "/bin/true")) {
                // not disabled
                PwdItem *item = new PwdItem(pwd);
                fUserList->AddItem(item);
        }
        if (pwd)
                BMessenger(this).SendMessage(kAddNextUser);
        else
                endpwent();
}


void
LoginView::EnableControls(bool enable)
{
        int32 i;
        for (i = 0; i < fUserList->CountItems(); i++) {
                fUserList->ItemAt(i)->SetEnabled(enable);
        }
        fLoginControl->SetEnabled(enable);
        fPasswordControl->SetEnabled(enable);
        fHidePasswordCheckBox->SetEnabled(enable);
        fHaltButton->SetEnabled(enable);
        fRebootButton->SetEnabled(enable);
        fLoginButton->SetEnabled(enable);
}