X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=crawler%2Fkiss%2Fsrc%2Forg%2Fwamblee%2Fcrawler%2Fkiss%2FProgramConfigurationParser.java;h=3237ca79a449d140001b072d95fa266781fa644b;hb=7e86d075a568401dc93f0f58b10dd458357a48fd;hp=5d9f578d6e596d2912de5149017fcd6600b61588;hpb=a641296cc35afce34a55bab521150446ea896798;p=utils diff --git a/crawler/kiss/src/org/wamblee/crawler/kiss/ProgramConfigurationParser.java b/crawler/kiss/src/org/wamblee/crawler/kiss/ProgramConfigurationParser.java index 5d9f578d..3237ca79 100644 --- a/crawler/kiss/src/org/wamblee/crawler/kiss/ProgramConfigurationParser.java +++ b/crawler/kiss/src/org/wamblee/crawler/kiss/ProgramConfigurationParser.java @@ -35,13 +35,54 @@ import org.wamblee.conditions.PropertyRegexCondition; */ class ProgramConfigurationParser { + 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_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 _filters; + + private Notifier _notifier; + + ProgramConfigurationParser() { + _filters = null; + _notifier = null; + } /** * Parses the condition used to match the desired programs. @@ -50,7 +91,7 @@ class ProgramConfigurationParser { * Input stream to parse from. * @return Condition. */ - List parse(InputStream aStream) { + void parse(InputStream aStream) { List filters = new ArrayList(); try { SAXReader reader = new SAXReader(); @@ -61,13 +102,20 @@ class ProgramConfigurationParser { 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); ProgramAction action = new RecordProgramAction(); if (actionElem != null) { if (actionElem.getText().equals(ACTION_NOTIFY)) { - action = new InterestingProgramAction(""); + action = new InterestingProgramAction(category); } } + List> regexConditions = new ArrayList>(); for (Iterator j = program.elementIterator(ELEM_PATTERN); j.hasNext(); ) { @@ -84,9 +132,67 @@ class ProgramConfigurationParser { Condition condition = new AndCondition(regexConditions); filters.add(new ProgramFilter(condition, action)); } - return filters; + _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); + } + + /** + * 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 = 25; + 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 getFilters() { + return _filters; + } + + /** + * Returns the notifier to use. + * @return Notifier. + */ + public Notifier getNotifier() { + return _notifier; + } }