/*
 * Copyright 2005 the original author or authors.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.wamblee.crawler.kiss.main;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.wamblee.conditions.AndCondition;
import org.wamblee.conditions.Condition;
import org.wamblee.conditions.PropertyRegexCondition;
import org.wamblee.crawler.kiss.guide.Program;
import org.wamblee.crawler.kiss.notification.MailNotifier;
import org.wamblee.crawler.kiss.notification.MailServer;
import org.wamblee.crawler.kiss.notification.Notifier;
import org.wamblee.xml.XslTransformer;

/**
 * Parse the configuration of desired programs.
 */
class ProgramConfigurationParser {
    private static final int DEFAULT_SMTP_PORT = 25;
    
    private static final int DEFAULT_PRIORITY = 1; 

    private static final String ELEM_PASSWORD = "password";

    private static final String ELEM_USERNAME = "username";

    private static final String ELEM_PORT = "port";

    private static final String ELEM_HOST = "host";

    // Formatting configuration.
    private static final String ELEM_FORMAT = "format";

    private static final String ELEM_TEXT = "text";

    private static final String ELEM_HTML = "html";

    // Mail server configuration.

    private static final String ELEM_NOTIFICATION = "notification";

    private static final String ELEM_SMTP = "smtp";

    private static final String ELEM_SUBJECT = "subject";

    private static final String ELEM_TO = "to";

    private static final String ELEM_FROM = "from";

    // Configuration of interesting programs.

    private static final String ELEM_PROGRAM = "program";
    
    private static final String ELEM_PRIORITY = "priority";

    private static final String ELEM_PATTERN = "match";

    private static final String ELEM_ACTION = "action";

    private static final String ELEM_CATEGORY = "category";

    private static final String ACTION_NOTIFY = "notify";

    private List<ProgramFilter> _filters;
    
    private XslTransformer _transformer;

    private Notifier _notifier;

    ProgramConfigurationParser(XslTransformer aTransformer) {
        _filters = null;
        _notifier = null;
        _transformer = aTransformer;
    }

    /**
     * Parses the condition used to match the desired programs.
     * 
     * @param aStream
     *            Input stream to parse from.
     * @return Condition.
     */
    void parse(InputStream aStream) {
        List<ProgramFilter> filters = new ArrayList<ProgramFilter>();
        try {
            SAXReader reader = new SAXReader();
            Document document = reader.read(aStream);

            Element root = document.getRootElement();

            for (Iterator i = root.elementIterator(ELEM_PROGRAM); i.hasNext();) {
                Element program = (Element) i.next();

                Element categoryElem = program.element(ELEM_CATEGORY);
                String category = "";
                if (categoryElem != null) {
                    category = categoryElem.getText().trim();
                }

                Element actionElem = program.element(ELEM_ACTION);
                int priority = DEFAULT_PRIORITY; 
                String priorityString = program.elementTextTrim(ELEM_PRIORITY);
                if ( priorityString != null ) { 
                    priority = Integer.valueOf(priorityString);
                }
                ProgramAction action = new RecordProgramAction(priority);
                if (actionElem != null) {
                    if (actionElem.getText().equals(ACTION_NOTIFY)) {
                        action = new InterestingProgramAction(category);
                    }
                }

                List<Condition<Program>> regexConditions = new ArrayList<Condition<Program>>();
                for (Iterator j = program.elementIterator(ELEM_PATTERN); j
                        .hasNext();) {
                    Element patternElem = (Element) j.next();
                    String fieldName = "name";
                    Attribute fieldAttribute = patternElem.attribute("field");
                    if (fieldAttribute != null) {
                        fieldName = fieldAttribute.getText();
                    }
                    String pattern = ".*(" + patternElem.getText() + ").*";
                    regexConditions.add(new PropertyRegexCondition<Program>(
                            fieldName, pattern, true));
                }
                Condition<Program> condition = new AndCondition<Program>(
                        regexConditions);
                filters.add(new ProgramFilter(condition, action));
            }
            _filters = filters;

            Element notifier = root.element(ELEM_NOTIFICATION);
            _notifier = parseNotifier(notifier);

        } catch (DocumentException e) {
            throw new RuntimeException("Error parsing program configuraiton", e);
        }
    }

    /**
     * Parses the notifier
     * 
     * @return Notifier
     */
    private Notifier parseNotifier(Element aNotifier) {
        String from = aNotifier.elementTextTrim(ELEM_FROM);
        String to = aNotifier.elementTextTrim(ELEM_TO);
        String subject = aNotifier.elementTextTrim(ELEM_SUBJECT);

        Element smtp = aNotifier.element(ELEM_SMTP);
        MailServer server = parseMailServer(smtp);

        Element format = aNotifier.element(ELEM_FORMAT);
        String htmlXslt = format.elementTextTrim(ELEM_HTML);
        String textXslt = format.elementTextTrim(ELEM_TEXT);

        return new MailNotifier(from, to, subject, htmlXslt, textXslt, server, _transformer);
    }

    /**
     * Parses the mail server from the XML.
     * 
     * @param aSmtp
     *            Mail server configuration.
     * @return Mail server.
     */
    private MailServer parseMailServer(Element aSmtp) {
        String host = aSmtp.elementTextTrim(ELEM_HOST);
        Element portElem = aSmtp.element(ELEM_PORT);
        int port = DEFAULT_SMTP_PORT;
        if (portElem != null) {
            port = Integer.valueOf(portElem.getTextTrim());
        }
        String username = aSmtp.elementTextTrim(ELEM_USERNAME);
        String password = aSmtp.elementTextTrim(ELEM_PASSWORD);

        MailServer server = new MailServer(host, port, username, password);
        return server;
    }

    /**
     * Returns the list of program filters.
     * 
     * @return Filter list.
     */
    public List<ProgramFilter> getFilters() {
        return _filters;
    }

    /**
     * Returns the notifier to use.
     * 
     * @return Notifier.
     */
    public Notifier getNotifier() {
        return _notifier;
    }
}