--- /dev/null
+/*
+ * Copyright 2006 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.notification;
+
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Date;
+
+import javax.mail.internet.AddressException;
+import javax.mail.internet.InternetAddress;
+import javax.xml.transform.TransformerException;
+
+import org.apache.commons.mail.EmailException;
+import org.apache.commons.mail.HtmlEmail;
+import org.dom4j.Element;
+import org.wamblee.xml.XslTransformer;
+
+/**
+ * A notifier that uses SMTP to notify users by mail.
+ *
+ */
+public class MailNotifier implements Notifier {
+
+ private String _from;
+
+ private String _to;
+
+ private String _subject;
+
+ private String _htmlXslt;
+
+ private String _textXslt;
+
+ private MailServer _server;
+
+ private XslTransformer _transformer;
+
+ /**
+ * Constructs the notifier.
+ *
+ * @param aFrom
+ * Sender mail address to use.
+ * @param aTo
+ * Recipient mail address to use.
+ * @param aSubject
+ * Subject to use in the email.
+ * @param aHtmlXslt
+ * XSLT file to transform the report into HTML.
+ * @param aTextXslt
+ * XSLT file to transform the report into text.
+ * @param aServer
+ * Mail server to use.
+ * @param aTransformer Transformer to use.
+ */
+ public MailNotifier(String aFrom, String aTo, String aSubject,
+ String aHtmlXslt, String aTextXslt, MailServer aServer, XslTransformer aTransformer) {
+ _from = aFrom;
+ _to = aTo;
+ _subject = aSubject;
+ _htmlXslt = aHtmlXslt;
+ _textXslt = aTextXslt;
+ _server = aServer;
+ _transformer = aTransformer;
+ }
+
+ /*
+ * (non-Javadoc)
+ *
+ * @see org.wamblee.crawler.kiss.Notifier#send(org.dom4j.Element)
+ */
+ public void send(Element aReport) throws NotificationException {
+ HtmlEmail mail = new HtmlEmail();
+ try {
+ mail.setFrom(_from);
+ mail
+ .setTo(Arrays
+ .asList(new InternetAddress[] { new InternetAddress(
+ _to) }));
+ mail.setSentDate(new Date());
+ mail.setSubject(_subject);
+
+ String htmlText = transformReport(aReport, _htmlXslt);
+ String plainText = transformReport(aReport, _textXslt);
+
+ mail.setHtmlMsg(htmlText);
+ mail.setTextMsg(plainText);
+
+ _server.send(mail);
+ } catch (EmailException e) {
+ throw new NotificationException(e.getMessage(), e);
+ } catch (TransformerException e) {
+ throw new NotificationException(e.getMessage(), e);
+ } catch (IOException e) {
+ throw new NotificationException(e.getMessage(), e);
+ } catch (AddressException e) {
+ throw new NotificationException(e.getMessage(), e);
+ }
+ }
+
+ /**
+ * Transforms a report into a destination format.
+ *
+ * @param aReport
+ * Report to transform
+ * @param aXslt
+ * XSLT to use.
+ * @return Transformed result.
+ * @throws IOException
+ * In case of IO problems.
+ * @throws TransformerException
+ * In case of problems transforming.
+ */
+ private String transformReport(Element aReport, String aXslt)
+ throws IOException, TransformerException {
+ String reportXmlText = aReport.asXML();
+ return _transformer.textTransform(reportXmlText.getBytes(), _transformer.resolve(aXslt));
+ }
+
+ /* (non-Javadoc)
+ * @see org.wamblee.crawler.kiss.notification.Notifier#asHtml(org.dom4j.Element)
+ */
+ public String asHtml(Element aReport) throws IOException, TransformerException {
+ return transformReport(aReport, _htmlXslt);
+ }
+
+ /* (non-Javadoc)
+ * @see org.wamblee.crawler.kiss.notification.Notifier#asText(org.dom4j.Element)
+ */
+ public String asText(Element aReport) throws IOException, TransformerException {
+ return transformReport(aReport, _textXslt);
+ }
+}