/*
 * Copyright 2010, The Android Open Source Project
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *  * Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  * Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "config.h"
#include "WebAutoFill.h"

#if ENABLE(WEB_AUTOFILL)

#include "AutoFillHostAndroid.h"
#include "Document.h"
#include "Frame.h"
#include "FormData.h"
#include "FormManagerAndroid.h"
#include "FrameLoader.h"
#include "HTMLFormControlElement.h"
#include "MainThreadProxy.h"
#include "Node.h"
#include "Page.h"
#include "Settings.h"
#include "WebFrame.h"
#include "WebRequestContext.h"
#include "WebUrlLoaderClient.h"
#include "WebViewCore.h"

#define FORM_NOT_AUTOFILLABLE -1

namespace android
{

static URLRequestContext* webAutoFillContextGetter()
{
    return WebRequestContext::get(false /* isPrivateBrowsing */);
}

WebAutoFill::WebAutoFill()
    : mWebViewCore(0)
{
    mFormManager = new FormManager();
    mQueryId = 1;

    AndroidURLRequestContextGetter::Get()->SetURLRequestContextGetterFunction(&webAutoFillContextGetter);
    AndroidURLRequestContextGetter::Get()->SetIOThread(WebUrlLoaderClient::ioThread());
    mTabContents = new TabContents();
    mAutoFillManager = new AutoFillManager(mTabContents.get());
    mAutoFillHost = new AutoFillHostAndroid(this);
    mTabContents->SetAutoFillHost(mAutoFillHost.get());
}

WebAutoFill::~WebAutoFill()
{
    mQueryMap.clear();
    mUniqueIdMap.clear();
}

void WebAutoFill::searchDocument(WebCore::Document* document)
{
    // TODO: This method is called when the main frame finishes loading and
    // scans the document for forms and tries to work out the type of the
    // fields in those forms. It's currently synchronous and might be slow
    // if the page has many or complex forms. Might want to make this an
    // async method.

    if (!enabled())
        return;

    if (!mFormManager)
        return;

    mQueryMap.clear();
    mUniqueIdMap.clear();
    mQueryId = 1;
    mAutoFillManager->Reset();
    mFormManager->Reset();
    mFormManager->ExtractForms(document);
    mForms.clear();
    mFormManager->GetForms(FormManager::REQUIRE_AUTOCOMPLETE, &mForms);
    mAutoFillManager->FormsSeen(mForms);
}

void WebAutoFill::formFieldFocused(WebCore::HTMLFormControlElement* formFieldElement)
{
    if (!enabled()) {
        // In case that we've just been disabled and the last time we got autofill
        // suggestions and told Java about them, clear that bit Java side now
        // we're disabled.
        mWebViewCore->setWebTextViewAutoFillable(FORM_NOT_AUTOFILLABLE);
        return;
    }

    // Get the FormField from the Node.
    webkit_glue::FormField formField;
    FormManager::HTMLFormControlElementToFormField(*formFieldElement, false, &formField);
    formField.set_label(FormManager::LabelForElement(*formFieldElement));

    webkit_glue::FormData* form = new webkit_glue::FormData;
    mFormManager->FindFormWithFormControlElement(*formFieldElement, FormManager::REQUIRE_AUTOCOMPLETE, form);
    mQueryMap[mQueryId] = form;

    bool suggestions = mAutoFillManager->GetAutoFillSuggestions(mQueryId, false, formField);
    mQueryId++;
    if (!suggestions) {
        ASSERT(mWebViewCore);
        // Tell Java no autofill suggestions for this form.
        mWebViewCore->setWebTextViewAutoFillable(FORM_NOT_AUTOFILLABLE);
        return;
    }
}

void WebAutoFill::querySuccessful(int queryId, const string16& value, const string16& label, int uniqueId)
{
    if (!enabled())
        return;

    // Store the unique ID for the query and inform java that autofill suggestions for this form are available.
    // Pass java the queryId so that it can pass it back if the user decides to use autofill.
    mUniqueIdMap[queryId] = uniqueId;

    ASSERT(mWebViewCore);
    mWebViewCore->setWebTextViewAutoFillable(queryId);
}

void WebAutoFill::fillFormFields(int queryId)
{
    if (!enabled())
        return;

    webkit_glue::FormData* form = mQueryMap[queryId];
    AutoFillQueryToUniqueIdMap::iterator iter = mUniqueIdMap.find(queryId);
    ASSERT(iter != mUniqueIdMap.end());
    mAutoFillManager->FillAutoFillFormData(queryId, *form, iter->second);
    mUniqueIdMap.erase(iter);
}

void WebAutoFill::fillFormInPage(int queryId, const webkit_glue::FormData& form)
{
    if (!enabled())
        return;

    mFormManager->FillForm(form);
}

bool WebAutoFill::enabled() const
{
    Page* page = mWebViewCore->mainFrame()->page();
    return page ? page->settings()->autoFillEnabled() : false;
}

void WebAutoFill::setProfile(int id, const string16& fullName, const string16& emailAddress, const string16& companyName,
                             const string16& addressLine1, const string16& addressLine2, const string16& city,
                             const string16& state, const string16& zipCode, const string16& country, const string16& phoneNumber)
{
    AutoFillProfile autoFillProfile;
    autoFillProfile.set_unique_id(id);

    // Constants for AutoFill field types are found in external/chromium/chrome/browser/autofill/field_types.h.
    autoFillProfile.SetInfo(AutoFillType(NAME_FULL), fullName);
    autoFillProfile.SetInfo(AutoFillType(EMAIL_ADDRESS), emailAddress);
    autoFillProfile.SetInfo(AutoFillType(COMPANY_NAME), companyName);
    autoFillProfile.SetInfo(AutoFillType(ADDRESS_HOME_LINE1), addressLine1);
    autoFillProfile.SetInfo(AutoFillType(ADDRESS_HOME_LINE2), addressLine2);
    autoFillProfile.SetInfo(AutoFillType(ADDRESS_HOME_CITY), city);
    autoFillProfile.SetInfo(AutoFillType(ADDRESS_HOME_STATE), state);
    autoFillProfile.SetInfo(AutoFillType(ADDRESS_HOME_ZIP), zipCode);
    autoFillProfile.SetInfo(AutoFillType(ADDRESS_HOME_COUNTRY), country);
    autoFillProfile.SetInfo(AutoFillType(PHONE_HOME_WHOLE_NUMBER), phoneNumber);

    std::vector<AutoFillProfile> profiles;
    profiles.push_back(autoFillProfile);
    mTabContents->profile()->GetPersonalDataManager()->SetProfiles(&profiles);
}

}

#endif