/**
 * $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.spark.ui;

import org.jdesktop.jdic.browser.WebBrowser;
import org.jdesktop.jdic.browser.WebBrowserEvent;
import org.jdesktop.jdic.browser.WebBrowserListener;
import org.jivesoftware.Spark;
import org.jivesoftware.smack.packet.Message;
import org.jivesoftware.smack.util.StringUtils;
import org.jivesoftware.smackx.packet.DelayInformation;
import org.jivesoftware.spark.SparkManager;
import org.jivesoftware.spark.component.VerticalFlowLayout;
import org.jivesoftware.spark.ui.themes.ThemeManager;
import org.jivesoftware.spark.util.ModelUtil;
import org.jivesoftware.spark.util.URLFileSystem;
import org.jivesoftware.spark.util.log.Log;
import org.jivesoftware.sparkimpl.plugin.emoticons.Emoticon;
import org.jivesoftware.sparkimpl.plugin.emoticons.EmoticonManager;
import org.jivesoftware.sparkimpl.profile.VCardManager;
import org.jivesoftware.sparkimpl.settings.local.LocalPreferences;
import org.jivesoftware.sparkimpl.settings.local.SettingsManager;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Dimension;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.StringTokenizer;
import java.util.Timer;
import java.util.TimerTask;

import javax.swing.JComponent;
import javax.swing.JFileChooser;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;

/**
 * The <CODE>TranscriptWindow</CODE> class. Provides a default implementation
 * of a Chat Window. In general, extensions could override this class
 * to offer more support within the chat, but should not be necessary.
 */
public class TranscriptWindow extends JPanel {

    private List<TranscriptWindowInterceptor> interceptors = new ArrayList<TranscriptWindowInterceptor>();

    private Date lastUpdated;

    private WebBrowser browser;

    private ThemeManager themeManager;

    private String activeUser;

    private boolean documentLoaded;

    private JPanel extraPanel;

    private VCardManager vcardManager;

    private List scriptList = new ArrayList();


    /**
     * Creates a default instance of <code>TranscriptWindow</code>.
     */
    public TranscriptWindow() {
        setLayout(new BorderLayout());

        themeManager = ThemeManager.getInstance();
        vcardManager = SparkManager.getVCardManager();

        extraPanel = new JPanel();

        browser = new WebBrowser();
        browser.setURL(themeManager.getTemplateURL());


        browser.addWebBrowserListener(new WebBrowserListener() {
            public void downloadStarted(WebBrowserEvent webBrowserEvent) {
            }

            public void downloadCompleted(WebBrowserEvent webBrowserEvent) {

            }

            public void downloadProgress(WebBrowserEvent webBrowserEvent) {
            }

            public void downloadError(WebBrowserEvent webBrowserEvent) {
            }

            public void documentCompleted(WebBrowserEvent webBrowserEvent) {
                documentLoaded = true;
            }

            public void titleChange(WebBrowserEvent webBrowserEvent) {
            }

            public void statusTextChange(WebBrowserEvent webBrowserEvent) {
            }


            public void initializationCompleted(WebBrowserEvent webBrowserEvent) {
            }
        });

        final JScrollPane pane = new JScrollPane(browser);
        pane.setBorder(null);
        add(pane, BorderLayout.CENTER);

        extraPanel.setBackground(Color.white);
        extraPanel.setLayout(new VerticalFlowLayout(VerticalFlowLayout.TOP, 0, 0, true, false));

        add(extraPanel, BorderLayout.SOUTH);

        startCommandListener();

        browser.setFocusable(false);
        this.setFocusable(false);
        setBorder(null);
    }


    /**
     * Create and insert a message from the current user.
     *
     * @param userid  the userid of the current user.
     * @param message the message to insert.
     */
    public void insertMessage(String userid, Message message) {
        // Check interceptors.
        for (TranscriptWindowInterceptor interceptor : interceptors) {
            boolean handled = interceptor.handleInsertMessage(userid, message);
            if (handled) {
                // Do nothing.
                return;
            }
        }

        String body = message.getBody();
        body = org.jivesoftware.spark.util.StringUtils.escapeHTMLTags(body);
        body = filterBody(body);
        String date = getDate(null);

        String jid = SparkManager.getSessionManager().getJID();


        if (userid.equals(activeUser)) {
            String text = themeManager.getNextOutgoingMessage(body, date);
            executeScript("appendNextMessage('" + text + "')");
        }
        else {
            String text = themeManager.getOutgoingMessage(userid, date, body, vcardManager.getAvatar(jid));
            executeScript("appendMessage('" + text + "')");
        }

        activeUser = userid;
    }


    public void insertCustomMessage(String prefix, String message) {
        message = filterBody(message);
        String text = themeManager.getOutgoingMessage(prefix, "", message, vcardManager.getAvatar(""));
        executeScript("appendMessage('" + text + "')");
    }

    public void insertCustomOtherMessage(String prefix, String message) {
        message = filterBody(message);
        String text = themeManager.getIncomingMessage(prefix, "", message, vcardManager.getAvatar(""));
        executeScript("appendMessage('" + text + "')");
    }

    /**
     * Create and insert a message from a customer.
     *
     * @param userid  the userid of the customer.
     * @param message the message from the customer.
     */
    public void insertOthersMessage(String userid, Message message) {
        // Check interceptors.
        for (TranscriptWindowInterceptor in : interceptors) {
            boolean handled = in.handleOtherMessage(userid, message);
            if (handled) {
                // Do nothing.
                return;
            }
        }

        String body = message.getBody();


        try {
            DelayInformation inf = (DelayInformation)message.getExtension("x", "jabber:x:delay");
            Date sentDate = null;
            if (inf != null) {
                sentDate = inf.getStamp();

                body = "(Offline) " + body;
            }
            else {
                sentDate = new Date();
            }

            String theDate = getDate(sentDate);

            body = org.jivesoftware.spark.util.StringUtils.escapeHTMLTags(body);
            body = filterBody(body);

            if (userid.equals(activeUser)) {
                String text = themeManager.getNextIncomingMessage(body, theDate);
                executeScript("appendNextMessage('" + text + "')");
            }
            else {
                String text = themeManager.getIncomingMessage(userid, theDate, body, vcardManager.getAvatar(message.getFrom()));
                executeScript("appendMessage('" + text + "')");
            }

            activeUser = userid;
        }
        catch (Exception ex) {
            Log.error("Error message.", ex);
        }
    }


    /**
     * Create and insert a notification message. A notification message generally is a
     * presence update, but can be used for most anything related to the room.
     *
     * @param message the information message to insert.
     */
    public synchronized void insertNotificationMessage(String message) {
        message = filterBody(message);
        String text = themeManager.getStatusMessage(message, "");
        executeScript("appendMessage('" + text + "')");
    }

    /**
     * Creates and inserts an error message.
     *
     * @param message the information message to insert.
     */
    public void insertErrorMessage(String message) {
        message = filterBody(message);
        String text = themeManager.getStatusMessage(message, "");
        executeScript("appendMessage('" + text + "')");
    }

    /**
     * Create and insert a question message. A question message is specified by the
     * end customer during the initial request.
     *
     * @param question the question asked by the customer.
     */
    public void insertQuestionMessage(String question) {
        String text = themeManager.getStatusMessage(question, "");
        executeScript("appendMessage('" + text + "')");
    }

    public void insertHTML(String html) {
        executeScript("appendMessage('" + html + "')");
    }


    /**
     * Returns the formatted date.
     *
     * @param insertDate the date to format.
     * @return the formatted date.
     */
    private String getDate(Date insertDate) {
        final LocalPreferences pref = SettingsManager.getLocalPreferences();

        if (insertDate == null) {
            insertDate = new Date();
        }


        if (pref.isTimeDisplayedInChat()) {
            final SimpleDateFormat formatter = new SimpleDateFormat("h:mm");
            return formatter.format(insertDate);
        }
        lastUpdated = insertDate;
        return "";
    }


    /**
     * Return the last time the <code>TranscriptWindow</code> was updated.
     *
     * @return the last time the <code>TranscriptWindow</code> was updated.
     */
    public Date getLastUpdated() {
        return lastUpdated;
    }

    /**
     * Inserts a history message.
     *
     * @param jid     the users jid.
     * @param userid  the userid of the sender.
     * @param message the message to insert.
     * @param date    the Date object created when the message was delivered.
     */
    public void insertHistoryMessage(String jid, String userid, String message, Date date) {
        final String sessionJID = SparkManager.getSessionManager().getJID();
        boolean outgoingMessage = false;
        if (StringUtils.parseBareAddress(sessionJID).equals(jid)) {
            outgoingMessage = true;
        }

        final SimpleDateFormat formatter = new SimpleDateFormat("h:mm");
        String time = formatter.format(date);

        message = filterBody(message);

        if (userid.equals(activeUser)) {
            if (outgoingMessage) {
                String text = themeManager.getNextOutgoingHistoryString(message, time);
                executeScript("appendNextMessage('" + text + "')");
            }
            else {
                String text = themeManager.getNextIncomingHistoryMessage(message, time);
                executeScript("appendNextMessage('" + text + "')");
            }
        }
        else {
            if (outgoingMessage) {
                String text = themeManager.getOutgoingHistoryMessage(userid, time, message, vcardManager.getAvatar(jid));
                executeScript("appendMessage('" + text + "')");
            }
            else {
                String text = themeManager.getIncomingHistoryMessage(userid, time, message, vcardManager.getAvatar(jid));
                executeScript("appendMessage('" + text + "')");
            }
        }

        activeUser = userid;
    }

    /**
     * Disable the entire <code>TranscriptWindow</code> and visually represent
     * it as disabled.
     */
    public void showDisabledWindowUI() {

    }

    /**
     * Persist a current transcript.
     *
     * @param fileName   the name of the file to save the transcript as. Note: This can be modified by the user.
     * @param transcript the collection of transcript.
     * @param headerData the string to prepend to the transcript.
     * @see ChatRoom#getTranscripts()
     */
    public void saveTranscript(String fileName, List transcript, String headerData) {
        final LocalPreferences pref = SettingsManager.getLocalPreferences();

        try {
            SimpleDateFormat formatter;

            File defaultSaveFile = new File(new File(Spark.getUserHome()), fileName);
            final JFileChooser fileChooser = new JFileChooser(defaultSaveFile);
            fileChooser.setSelectedFile(defaultSaveFile);

            // Show save dialog; this method does not return until the dialog is closed
            int result = fileChooser.showSaveDialog(this);
            final File selFile = fileChooser.getSelectedFile();

            if (selFile != null && result == JFileChooser.APPROVE_OPTION) {
                final StringBuffer buf = new StringBuffer();
                final Iterator transcripts = transcript.iterator();
                buf.append("<html><body>");
                if (headerData != null) {
                    buf.append(headerData);
                }

                buf.append("<table width=600>");
                while (transcripts.hasNext()) {
                    final Message message = (Message)transcripts.next();
                    String from = message.getFrom();
                    if (from == null) {
                        from = pref.getNickname();
                    }

                    if (Message.Type.GROUP_CHAT == message.getType()) {
                        if (ModelUtil.hasLength(StringUtils.parseResource(from))) {
                            from = StringUtils.parseResource(from);
                        }
                    }

                    final String body = message.getBody();
                    final Date insertionDate = (Date)message.getProperty("insertionDate");
                    formatter = new SimpleDateFormat("hh:mm:ss");

                    String value = "";
                    if (insertionDate != null) {
                        value = "[" + formatter.format(insertionDate) + "] ";
                    }
                    buf.append("<tr><td nowrap><font size=2>").append(value).append("<strong>").append(from).append(":</strong>&nbsp;").append(body).append("</font></td></tr>");

                }
                buf.append("</table></body></html>");
                final BufferedWriter writer = new BufferedWriter(new FileWriter(selFile));
                writer.write(buf.toString());
                writer.close();
                JOptionPane.showMessageDialog(SparkManager.getMainWindow(), "Chat transcript has been saved.",
                    "Chat Transcript Saved", JOptionPane.INFORMATION_MESSAGE);
            }
        }
        catch (Exception ex) {
            Log.error("Unable to save chat transcript.", ex);
            JOptionPane.showMessageDialog(SparkManager.getMainWindow(), "Could not save transcript.", "Error", JOptionPane.ERROR_MESSAGE);
        }
    }

    public void scrollToBottom() {
        executeScript("scrollToBottom();");
    }


    public void addTranscriptWindowInterceptor(TranscriptWindowInterceptor interceptor) {
        interceptors.add(interceptor);
    }

    public void removeTranscriptWindowInterceptor(TranscriptWindowInterceptor interceptor) {
        interceptors.remove(interceptor);
    }

    public boolean isDocumentLoaded() {
        return documentLoaded;
    }

    public void setInnerHTML(String elementID, String value) {
        final StringBuilder builder = new StringBuilder();
        builder.append("var myVar = document.getElementById(\"" + elementID + "\");");
        builder.append("if(myVar){ myVar.innerHTML = '" + value + "';}");
        executeScript(builder.toString());
    }

    public void executeScript(final String script) {
        scriptList.add(script);
    }

    public void setURL(URL url) {
        documentLoaded = false;

        browser.setURL(url);
    }

    private void startCommandListener() {
        Timer timer = new Timer();
        timer.scheduleAtFixedRate(new TimerTask() {
            public void run() {
                if (documentLoaded) {
                    if (scriptList.size() > 0) {
                        String script = (String)scriptList.get(0);
                        scriptList.remove(0);
                        browser.executeScript(script);
                    }
                }
            }
        }, 50, 50);
    }

    public void addComponent(JComponent component) {
        extraPanel.add(component);
        extraPanel.setVisible(true);
        extraPanel.invalidate();
        extraPanel.validate();
        extraPanel.repaint();
    }

    public void removeComponent(JComponent component) {
        if (extraPanel.getComponentCount() == 0) {
            extraPanel.setVisible(false);
        }

        extraPanel.remove(component);
        extraPanel.invalidate();
        extraPanel.validate();
        extraPanel.repaint();
    }

    public Dimension getPreferredSize() {
        final Dimension size = super.getPreferredSize();
        size.width = 0;
        return size;
    }

    private String filterBody(String text) {
        EmoticonManager emoticonManager = EmoticonManager.getInstance();
        StringBuilder builder = new StringBuilder();


        final StringTokenizer tokenizer = new StringTokenizer(text, " \n \t", true);
        while (tokenizer.hasMoreTokens()) {
            String textFound = tokenizer.nextToken();
            if (textFound.startsWith("http://") || textFound.startsWith("ftp://")
                || textFound.startsWith("https://") || textFound.startsWith("www.") || textFound.startsWith("\\") || textFound.indexOf("://") != -1) {
                builder.append("<a href=\"").append(textFound).append("\" target=_blank>").append(textFound).append("</a>");
            }
            else if (emoticonManager.getEmoticon(textFound) != null) {
                Emoticon emot = emoticonManager.getEmoticon(textFound);
                URL url = emoticonManager.getEmoticonURL(emot);
                File file = URLFileSystem.url2File(url);
                builder.append("<img src=\"").append(url.toExternalForm()).append("\" />");
            }
            else {
                builder.append(textFound);
            }
        }

        return builder.toString();

    }


}