/*
 * Copyright (C) 2010 Stephan Aßmus <superstippi@gmx.de>
 *
 * All rights reserved. Distributed under the terms of the MIT License.
 */
#include "AuthenticationPanel.h"

#include <Button.h>
#include <Catalog.h>
#include <CheckBox.h>
#include <ControlLook.h>
#include <GridLayoutBuilder.h>
#include <GroupLayoutBuilder.h>
#include <Locale.h>
#include <Message.h>
#include <Screen.h>
#include <SeparatorView.h>
#include <SpaceLayoutItem.h>
#include <StringView.h>
#include <TextControl.h>
#include <stdio.h>

static const uint32 kMsgPanelOK = 'pnok';
static const uint32 kMsgJitter = 'jitr';
static const uint32 kHidePassword = 'hdpw';


#undef B_TRANSLATION_CONTEXT
#define B_TRANSLATION_CONTEXT "Authentication Panel"

AuthenticationPanel::AuthenticationPanel(BRect parentFrame)
        :
        BWindow(BRect(-1000, -1000, -900, -900),
                B_TRANSLATE("Authentication required"), B_TITLED_WINDOW_LOOK,
                B_MODAL_APP_WINDOW_FEEL, B_ASYNCHRONOUS_CONTROLS | B_NOT_RESIZABLE
                        | B_NOT_ZOOMABLE | B_CLOSE_ON_ESCAPE | B_AUTO_UPDATE_SIZE_LIMITS),
        m_parentWindowFrame(parentFrame),
        m_usernameTextControl(new BTextControl("user", B_TRANSLATE("Username:"),
                "", NULL)),
        m_passwordTextControl(new BTextControl("pass", B_TRANSLATE("Password:"),
                "", NULL)),
        m_hidePasswordCheckBox(new BCheckBox("hide", B_TRANSLATE("Hide password "
                "text"), new BMessage(kHidePassword))),
        m_rememberCredentialsCheckBox(new BCheckBox("remember",
                B_TRANSLATE("Remember username and password for this site"), NULL)),
        m_okButton(new BButton("ok", B_TRANSLATE("OK"),
                new BMessage(kMsgPanelOK))),
        m_cancelButton(new BButton("cancel", B_TRANSLATE("Cancel"),
                new BMessage(B_QUIT_REQUESTED))),
        m_cancelled(false),
        m_exitSemaphore(create_sem(0, "Authentication Panel"))
{
}


AuthenticationPanel::~AuthenticationPanel()
{
        delete_sem(m_exitSemaphore);
}


bool
AuthenticationPanel::QuitRequested()
{
        m_cancelled = true;
        release_sem(m_exitSemaphore);
        return false;
}


void
AuthenticationPanel::MessageReceived(BMessage* message)
{
        switch (message->what) {
        case kMsgPanelOK:
                release_sem(m_exitSemaphore);
                break;
        case kHidePassword: {
                // TODO: Toggling this is broken in BTextView. Workaround is to
                // set the text and selection again.
                BString text = m_passwordTextControl->Text();
                int32 selectionStart;
                int32 selectionEnd;
                m_passwordTextControl->TextView()->GetSelection(&selectionStart,
                        &selectionEnd);
                m_passwordTextControl->TextView()->HideTyping(
                        m_hidePasswordCheckBox->Value() == B_CONTROL_ON);
                m_passwordTextControl->SetText(text.String());
                m_passwordTextControl->TextView()->Select(selectionStart,
                        selectionEnd);
                break;
        }
        case kMsgJitter: {
                UpdateIfNeeded();
                BPoint leftTop = Frame().LeftTop();
                const float jitterOffsets[] = { -10, 0, 10, 0 };
                const int32 jitterOffsetCount = sizeof(jitterOffsets) / sizeof(float);
                for (int32 i = 0; i < 20; i++) {
                        float offset = jitterOffsets[i % jitterOffsetCount];
                        MoveTo(leftTop.x + offset, leftTop.y);
                        snooze(15000);
                }
                MoveTo(leftTop);
                break;
        }
        default:
                BWindow::MessageReceived(message);
        }
}


bool AuthenticationPanel::getAuthentication(const BString& text,
        const BString& previousUser, const BString& previousPass,
        bool previousRememberCredentials, bool badPassword,
        BString& user, BString&  pass, bool* rememberCredentials)
{
        // Configure panel and layout controls.
        rgb_color infoColor = ui_color(B_PANEL_TEXT_COLOR);
        BRect textBounds(0, 0, 250, 200);
        BTextView* textView = new BTextView(textBounds, "text", textBounds,
                be_plain_font, &infoColor, B_FOLLOW_NONE, B_WILL_DRAW
                        | B_SUPPORTS_LAYOUT);
        textView->SetViewUIColor(B_PANEL_BACKGROUND_COLOR);
        textView->SetText(text.String());
        textView->MakeEditable(false);
        textView->MakeSelectable(false);

        m_usernameTextControl->SetText(previousUser.String());
        m_passwordTextControl->TextView()->HideTyping(true);
        // Ignore the previous password, if it didn't work.
        if (!badPassword)
                m_passwordTextControl->SetText(previousPass.String());
        m_hidePasswordCheckBox->SetValue(B_CONTROL_ON);
        m_rememberCredentialsCheckBox->SetValue(previousRememberCredentials);

        // create layout
        SetLayout(new BGroupLayout(B_VERTICAL, 0.0));
        float spacing = be_control_look->DefaultItemSpacing();
        AddChild(BGroupLayoutBuilder(B_VERTICAL, 0.0)
                .Add(BGridLayoutBuilder(0, spacing)
                        .Add(textView, 0, 0, 2)
                        .Add(m_usernameTextControl->CreateLabelLayoutItem(), 0, 1)
                        .Add(m_usernameTextControl->CreateTextViewLayoutItem(), 1, 1)
                        .Add(m_passwordTextControl->CreateLabelLayoutItem(), 0, 2)
                        .Add(m_passwordTextControl->CreateTextViewLayoutItem(), 1, 2)
                        .Add(BSpaceLayoutItem::CreateGlue(), 0, 3)
                        .Add(m_hidePasswordCheckBox, 1, 3)
                        .Add(m_rememberCredentialsCheckBox, 0, 4, 2)
                        .SetInsets(spacing, spacing, spacing, spacing)
                )
                .Add(new BSeparatorView(B_HORIZONTAL, B_PLAIN_BORDER))
                .Add(BGroupLayoutBuilder(B_HORIZONTAL, spacing)
                        .AddGlue()
                        .Add(m_cancelButton)
                        .Add(m_okButton)
                        .SetInsets(spacing, spacing, spacing, spacing)
                )
        );

        float textHeight = textView->LineHeight(0) * textView->CountLines();
        textView->SetExplicitMinSize(BSize(B_SIZE_UNSET, textHeight));

        SetDefaultButton(m_okButton);
        if (badPassword && previousUser.Length())
                m_passwordTextControl->MakeFocus(true);
        else
                m_usernameTextControl->MakeFocus(true);

        if (m_parentWindowFrame.IsValid())
                CenterIn(m_parentWindowFrame);
        else
                CenterOnScreen();

        // Start AuthenticationPanel window thread
        Show();

        // Let the window jitter, if the previous password was invalid
        if (badPassword)
                PostMessage(kMsgJitter);

        // Block calling thread
        // Get the originating window, if it exists, to let it redraw itself.
        BWindow* window = dynamic_cast<BWindow*>
                (BLooper::LooperForThread(find_thread(NULL)));
        if (window) {
                status_t err;
                for (;;) {
                        do {
                                err = acquire_sem_etc(m_exitSemaphore, 1, B_RELATIVE_TIMEOUT,
                                        10000);
                                // We've (probably) had our time slice taken away from us
                        } while (err == B_INTERRUPTED);

                        if (err != B_TIMED_OUT) {
                                // Semaphore was finally released or nuked.
                                break;
                        }
                        window->UpdateIfNeeded();
                }
        } else {
                // No window to update, so just hang out until we're done.
                while (acquire_sem(m_exitSemaphore) == B_INTERRUPTED) {
                }
        }

        // AuthenticationPanel wants to quit.
        Lock();

        user = m_usernameTextControl->Text();
        pass = m_passwordTextControl->Text();
        if (rememberCredentials)
                *rememberCredentials = m_rememberCredentialsCheckBox->Value()
                        == B_CONTROL_ON;

        bool canceled = m_cancelled;
        Quit();
        // AuthenticationPanel object is TOAST here.
        return !canceled;
}