/**
 * $Revision: $
 * $Date: $
 *
 * Copyright (C) 2006 Jive Software. All rights reserved.
 *
 * This software is published under the terms of the GNU Lesser Public License (LGPL),
 * a copy of which is included in this distribution.
 */

package org.jivesoftware.sparkimpl.search.users;

import org.jivesoftware.resource.SparkRes;
import org.jivesoftware.smack.XMPPException;
import org.jivesoftware.smackx.Form;
import org.jivesoftware.smackx.search.UserSearchManager;
import org.jivesoftware.spark.SparkManager;
import org.jivesoftware.spark.component.TitlePanel;
import org.jivesoftware.spark.ui.DataFormUI;
import org.jivesoftware.spark.util.ModelUtil;
import org.jivesoftware.spark.util.ResourceUtils;
import org.jivesoftware.spark.util.SwingWorker;
import org.jivesoftware.spark.util.log.Log;

import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;

import java.awt.CardLayout;
import java.awt.Component;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.Insets;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

/**
 * UserSearchForm is used to do explicit searching of users using the JEP 55 Search Service.
 */
public class UserSearchForm extends JPanel {
    private JComboBox servicesBox;
    private UserSearchManager searchManager;

    private Collection searchServices;

    private CardLayout cardLayout = new CardLayout();
    private JPanel cardPanel = new JPanel();

    private TitlePanel titlePanel;

    private Map serviceMap = new HashMap();

    /**
     * Initializes the UserSearchForm with all available search services.
     *
     * @param searchServices a Collection of all search services found.
     */
    public UserSearchForm(Collection searchServices) {
        setLayout(new GridBagLayout());

        cardPanel.setLayout(cardLayout);

        this.searchServices = searchServices;

        searchManager = new UserSearchManager(SparkManager.getConnection());

        addSearchServices();

        showService(getSearchService());
    }

    private void addSearchServices() {
        // Populate with Search Services
        servicesBox = new JComboBox();

        Iterator services = searchServices.iterator();
        while (services.hasNext()) {
            String service = (String)services.next();
            servicesBox.addItem(service);
        }

        if (servicesBox.getItemCount() > 0) {
            servicesBox.setSelectedIndex(0);
        }


        titlePanel = new TitlePanel("", "", SparkRes.getImageIcon(SparkRes.BLANK_IMAGE), true);
        add(titlePanel, new GridBagConstraints(0, 0, 3, 1, 1.0, 0.0, GridBagConstraints.CENTER, GridBagConstraints.HORIZONTAL, new Insets(0, 0, 0, 0), 0, 0));

        // Add Search Service ComboBox
        final JLabel serviceLabel = new JLabel("Search Service:");
        add(serviceLabel, new GridBagConstraints(0, 1, 1, 1, 0.0, 0.0, GridBagConstraints.WEST, GridBagConstraints.NONE, new Insets(5, 5, 5, 5), 0, 0));
        add(servicesBox, new GridBagConstraints(1, 1, 1, 1, 0.0, 0.0, GridBagConstraints.WEST, GridBagConstraints.NONE, new Insets(5, 5, 5, 5), 150, 0));

        final JButton addService = new JButton();
        ResourceUtils.resButton(addService, "&Add Service");
        add(addService, new GridBagConstraints(2, 1, 1, 1, 0.0, 0.0, GridBagConstraints.WEST, GridBagConstraints.NONE, new Insets(5, 5, 5, 5), 0, 0));
        addService.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent actionEvent) {
                final String serviceName = JOptionPane.showInputDialog(getRootPane(), "Name of search service?", "Add Search Service", JOptionPane.QUESTION_MESSAGE);
                if (ModelUtil.hasLength(serviceName)) {

                    SwingWorker findServiceThread = new SwingWorker() {
                        Form newForm;

                        public Object construct() {
                            try {
                                newForm = searchManager.getSearchForm(serviceName);
                            }
                            catch (XMPPException e) {
                            }
                            return newForm;
                        }

                        public void finished() {
                            if (newForm == null) {
                                JOptionPane.showMessageDialog(getGUI(), "Unable to contact search service.", "Search Service Not Available", JOptionPane.ERROR_MESSAGE);
                                return;
                            }
                            else {
                                servicesBox.addItem(serviceName);
                                servicesBox.setSelectedItem(serviceName);
                            }
                        }

                    };
                    findServiceThread.start();
                }
            }
        });

        servicesBox.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent actionEvent) {
                SwingWorker worker = new SwingWorker() {
                    public Object construct() {
                        try {
                            Thread.sleep(50);
                        }
                        catch (Exception e) {
                            Log.error("Problem sleeping thread.", e);
                        }
                        return "ok";
                    }

                    public void finished() {
                        showService(getSearchService());
                    }
                };
                worker.start();
            }
        });

        add(cardPanel, new GridBagConstraints(0, 3, 3, 1, 1.0, 1.0, GridBagConstraints.WEST, GridBagConstraints.BOTH, new Insets(5, 5, 5, 5), 0, 0));


    }

    /**
     * Displays the specified search service.
     *
     * @param service the search service to display.
     */
    public void showService(String service) {
        if (serviceMap.containsKey(service)) {
            cardLayout.show(cardPanel, service);
        }
        else {
            // Create new Form
            SearchForm searchForm = new SearchForm(service);
            cardPanel.add(searchForm, service);
            serviceMap.put(service, searchForm);
            cardLayout.show(cardPanel, service);
        }

        SearchForm searchForm = (SearchForm)serviceMap.get(service);
        Form form = searchForm.getSearchForm();
        String title = form.getTitle();
        String description = form.getInstructions();
        titlePanel.setTitle(title);
        titlePanel.setDescription(description);
    }


    /**
     * Returns the selected search service.
     *
     * @return the selected search service.
     */
    public String getSearchService() {
        return (String)servicesBox.getSelectedItem();
    }

    /**
     * Return the QuestionForm retrieved by the search service.
     *
     * @return the QuestionForm retrieved by the search service.
     */
    public DataFormUI getQuestionForm() {
        SearchForm searchForm = (SearchForm)serviceMap.get(getSearchService());
        return searchForm.getQuestionForm();
    }

    /**
     * Performs a search on the specified search service.
     */
    public void performSearch() {
        SearchForm searchForm = (SearchForm)serviceMap.get(getSearchService());
        searchForm.performSearch();
    }

    /**
     * Returns the UI that represent the UserSearchForm
     *
     * @return the UserSearchForm
     */
    public Component getGUI() {
        return this;
    }

}
