/*
 * 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 "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 NO_PROFILE_SET 0
#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::Frame* frame)
{
    // 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(frame);
    mForms.clear();
    mFormManager->GetFormsInFrame(frame, 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, string16());
        return;
    }

    // Get the FormField from the Node.
    webkit_glue::FormField formField;
    FormManager::HTMLFormControlElementToFormField(formFieldElement, FormManager::EXTRACT_NONE, &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, string16());
        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, mAutoFillProfile->Label());
}

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;

    // FIXME: Pass a pointer to the Node that triggered the AutoFill flow here instead of 0.
    // The consquence of passing 0 is that we should always fail the test in FormManader::ForEachMathcingFormField():169
    // that says "only overwrite an elements current value if the user triggered autofill through that element"
    // for elements that have a value already. But by a quirk of Android text views we are OK. We should still
    // fix this though.
    mFormManager->FillForm(form, 0);
}

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

void WebAutoFill::setProfile(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)
{
    mAutoFillProfile.set(new AutoFillProfile());

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

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

bool WebAutoFill::updateProfileLabel()
{
    std::vector<AutoFillProfile*> profiles;
    profiles.push_back(mAutoFillProfile.get());
    return AutoFillProfile::AdjustInferredLabels(&profiles);
}

void WebAutoFill::clearProfiles()
{
    if (!mAutoFillProfile)
        return;
    // For now Chromium only ever knows about one profile, so we can just
    // remove it. If we support multiple profiles in the future
    // we need to remove them all here.
    std::string profileGuid = mAutoFillProfile->guid();
    mTabContents->profile()->GetPersonalDataManager()->RemoveProfile(profileGuid);
}

}

#endif