framework/service/src/org/ofbiz/service/mail/ServiceMcaCondition.java - ofbiz - Git at Google

 /*******************************************************************************
  * Licensed to the Apache Software Foundation (ASF) under one
  * or more contributor license agreements.  See the NOTICE file
  * distributed with this work for additional information
  * regarding copyright ownership.  The ASF licenses this file
  * to you 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.ofbiz.service.mail;

 import java.io.IOException;
 import java.util.List;
 import java.util.Map;
 import javax.mail.Address;
 import javax.mail.BodyPart;
 import javax.mail.MessagingException;
 import javax.mail.Multipart;
 import javax.mail.Part;
 import javax.mail.internet.MimeMessage;

 import javolution.util.FastList;

 import org.ofbiz.base.util.Debug;
 import org.ofbiz.base.util.UtilMisc;
 import org.ofbiz.service.LocalDispatcher;
 import org.ofbiz.service.GenericServiceException;
 import org.ofbiz.service.ServiceUtil;
 import org.ofbiz.entity.GenericValue;

 import org.w3c.dom.Element;

 @SuppressWarnings("serial")
 public class ServiceMcaCondition implements java.io.Serializable {

     public static final String module = ServiceMcaCondition.class.getName();
     public static final int CONDITION_FIELD = 1;
     public static final int CONDITION_HEADER = 2;
     public static final int CONDITION_SERVICE = 3;

     protected String serviceName = null;
     protected String headerName = null;
     protected String fieldName = null;
     protected String operator = null;
     protected String value = null;

     public ServiceMcaCondition(Element condElement, int condType) {
         switch (condType) {
             case CONDITION_FIELD:
                 // fields: from|to|subject|body|sent-date|receieved-date
                 this.fieldName = condElement.getAttribute("field-name");
                 // operators: equals|not-equals|empty|not-empty|matches|not-matches
                 this.operator = condElement.getAttribute("operator");
                 // value to compare
                 this.value = condElement.getAttribute("value");
                 break;
             case CONDITION_HEADER:
                 // free form header name
                 this.headerName = condElement.getAttribute("header-name");
                 // operators: equals|not-equals|empty|not-empty|matches|not-matches
                 this.operator = condElement.getAttribute("operator");
                 // value to compare
                 this.value = condElement.getAttribute("value");
                 break;
             case CONDITION_SERVICE:
                 this.serviceName = condElement.getAttribute("service-name");
                 break;
         }
     }

     public boolean eval(LocalDispatcher dispatcher, MimeMessageWrapper messageWrapper, GenericValue userLogin) {
         boolean passedCondition = false;
         if (serviceName != null) {
             Map<String, Object> result = null;
             try {
                 result = dispatcher.runSync(serviceName, UtilMisc.<String, Object>toMap("messageWrapper", messageWrapper, "userLogin", userLogin));
             } catch (GenericServiceException e) {
                 Debug.logError(e, module);
                 return false;
             }
             if (result == null) {
                 Debug.logError("Service MCA Condition Service [" + serviceName + "] returned null!", module);
                 return false;
             } else {
                 if (ServiceUtil.isError(result)) {
                     Debug.logError(ServiceUtil.getErrorMessage(result), module);
                     return false;
                 } else {
                     Boolean reply = (Boolean) result.get("conditionReply");
                     if (reply == null) {
                         reply = Boolean.FALSE;
                     }
                     return reply.booleanValue();
                 }
             }
             // invoke the condition service
         } else if (headerName != null) {
             // compare the header field
             MimeMessage message = messageWrapper.getMessage();
             String[] headerValues = null;
             try {
                 headerValues = message.getHeader(headerName);
             } catch (MessagingException e) {
                 Debug.logError(e, module);
             }

             if (headerValues != null) {
                 for (String headerValue: headerValues) {
                     if ("equals".equals(operator)) {
                         if (headerValue.equals(value)) {
                             passedCondition = true;
                             break;
                         }
                     } else if ("not-equals".equals(operator)) {
                         if (!headerValue.equals(value)) {
                             passedCondition = true;
                         } else {
                             passedCondition = false;
                         }
                     } else if ("matches".equals(operator)) {
                         if (headerValue.matches(value)) {
                             passedCondition = true;
                             break;
                         }
                     } else if ("not-matches".equals(operator)) {
                         if (!headerValue.matches(value)) {
                             passedCondition = true;
                         } else {
                             passedCondition = false;
                         }
                     } else if ("not-empty".equals(operator)) {
                         passedCondition = true;
                         break;
                     }
                 }
             } else if ("empty".equals(operator)) {
                 passedCondition = true;
             }
         } else if (fieldName != null) {
             MimeMessage message = messageWrapper.getMessage();
             String[] fieldValues = null;
             try {
                 fieldValues = this.getFieldValue(message, fieldName);
             } catch (MessagingException e) {
                 Debug.logError(e, module);
             } catch (IOException e) {
                 Debug.logError(e, module);
             }

             if (fieldValues != null) {
                 for (String fieldValue: fieldValues) {
                     if ("equals".equals(operator)) {
                         if (fieldValue.equals(value)) {
                             passedCondition = true;
                             break;
                         }
                     } else if ("not-equals".equals(operator)) {
                         if (!fieldValue.equals(value)) {
                             passedCondition = true;
                         } else {
                             passedCondition = false;
                         }
                     } else if ("matches".equals(operator)) {
                         if (fieldValue.matches(value)) {
                             passedCondition = true;
                             break;
                         }
                     } else if ("not-matches".equals(operator)) {
                         if (!fieldValue.matches(value)) {
                             passedCondition = true;
                         } else {
                             passedCondition = false;
                         }
                     } else if ("not-empty".equals(operator)) {
                         passedCondition = true;
                         break;
                     }
                 }
             } else if ("empty".equals(operator)) {
                 passedCondition = true;
             }
         } else {
             passedCondition = false;
         }

         return passedCondition;
     }

     protected String[] getFieldValue(MimeMessage message, String fieldName) throws MessagingException, IOException {
         String[] values = null;
         if ("to".equals(fieldName)) {
             Address[] addrs = message.getRecipients(MimeMessage.RecipientType.TO);
             if (addrs != null) {
                 values = new String[addrs.length];
                 for (int i = 0; i < addrs.length; i++) {
                     values[i] = addrs[i].toString();
                 }
             }
         } else if ("cc".equals(fieldName)) {
             Address[] addrs = message.getRecipients(MimeMessage.RecipientType.CC);
             if (addrs != null) {
                 values = new String[addrs.length];
                 for (int i = 0; i < addrs.length; i++) {
                     values[i] = addrs[i].toString();
                 }
             }
         } else if ("bcc".equals(fieldName)) {
             Address[] addrs = message.getRecipients(MimeMessage.RecipientType.BCC);
             if (addrs != null) {
                 values = new String[addrs.length];
                 for (int i = 0; i < addrs.length; i++) {
                     values[i] = addrs[i].toString();
                 }
             }
         } else if ("from".equals(fieldName)) {
             Address[] addrs = message.getFrom();
             if (addrs != null) {
                 values = new String[addrs.length];
                 for (int i = 0; i < addrs.length; i++) {
                     values[i] = addrs[i].toString();
                 }
             }
         } else if ("subject".equals(fieldName)) {
             values = new String[1];
             values[0] = message.getSubject();
         } else if ("send-date".equals(fieldName)) {
             values = new String[1];
             values[0] = message.getSentDate().toString();
         } else if ("received-date".equals(fieldName)) {
             values = new String[1];
             values[0] = message.getReceivedDate().toString();
         } else if ("body".equals(fieldName)) {
             List<String> bodyParts = this.getBodyText(message);
             values = bodyParts.toArray(new String[bodyParts.size()]);
         }
         return values;
     }

     private List<String> getBodyText(Part part) throws MessagingException, IOException {
         Object c = part.getContent();
         if (c instanceof String) {
             return UtilMisc.toList((String) c);
         } else if (c instanceof Multipart) {
             List<String> textContent = FastList.newInstance();
             int count = ((Multipart) c).getCount();
             for (int i = 0; i < count; i++) {
                 BodyPart bp = ((Multipart) c).getBodyPart(i);
                 textContent.addAll(this.getBodyText(bp));
             }
             return textContent;
         } else {
             return FastList.newInstance();
         }
     }
 }
	/*******************************************************************************
	* Licensed to the Apache Software Foundation (ASF) under one
	* or more contributor license agreements. See the NOTICE file
	* distributed with this work for additional information
	* regarding copyright ownership. The ASF licenses this file
	* to you 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.ofbiz.service.mail;

	import java.io.IOException;
	import java.util.List;
	import java.util.Map;
	import javax.mail.Address;
	import javax.mail.BodyPart;
	import javax.mail.MessagingException;
	import javax.mail.Multipart;
	import javax.mail.Part;
	import javax.mail.internet.MimeMessage;

	import javolution.util.FastList;

	import org.ofbiz.base.util.Debug;
	import org.ofbiz.base.util.UtilMisc;
	import org.ofbiz.service.LocalDispatcher;
	import org.ofbiz.service.GenericServiceException;
	import org.ofbiz.service.ServiceUtil;
	import org.ofbiz.entity.GenericValue;

	import org.w3c.dom.Element;

	@SuppressWarnings("serial")
	public class ServiceMcaCondition implements java.io.Serializable {

	public static final String module = ServiceMcaCondition.class.getName();
	public static final int CONDITION_FIELD = 1;
	public static final int CONDITION_HEADER = 2;
	public static final int CONDITION_SERVICE = 3;

	protected String serviceName = null;
	protected String headerName = null;
	protected String fieldName = null;
	protected String operator = null;
	protected String value = null;

	public ServiceMcaCondition(Element condElement, int condType) {
	switch (condType) {
	case CONDITION_FIELD:
	// fields: from\|to\|subject\|body\|sent-date\|receieved-date
	this.fieldName = condElement.getAttribute("field-name");
	// operators: equals\|not-equals\|empty\|not-empty\|matches\|not-matches
	this.operator = condElement.getAttribute("operator");
	// value to compare
	this.value = condElement.getAttribute("value");
	break;
	case CONDITION_HEADER:
	// free form header name
	this.headerName = condElement.getAttribute("header-name");
	// operators: equals\|not-equals\|empty\|not-empty\|matches\|not-matches
	this.operator = condElement.getAttribute("operator");
	// value to compare
	this.value = condElement.getAttribute("value");
	break;
	case CONDITION_SERVICE:
	this.serviceName = condElement.getAttribute("service-name");
	break;
	}
	}

	public boolean eval(LocalDispatcher dispatcher, MimeMessageWrapper messageWrapper, GenericValue userLogin) {
	boolean passedCondition = false;
	if (serviceName != null) {
	Map<String, Object> result = null;
	try {
	result = dispatcher.runSync(serviceName, UtilMisc.<String, Object>toMap("messageWrapper", messageWrapper, "userLogin", userLogin));
	} catch (GenericServiceException e) {
	Debug.logError(e, module);
	return false;
	}
	if (result == null) {
	Debug.logError("Service MCA Condition Service [" + serviceName + "] returned null!", module);
	return false;
	} else {
	if (ServiceUtil.isError(result)) {
	Debug.logError(ServiceUtil.getErrorMessage(result), module);
	return false;
	} else {
	Boolean reply = (Boolean) result.get("conditionReply");
	if (reply == null) {
	reply = Boolean.FALSE;
	}
	return reply.booleanValue();
	}
	}
	// invoke the condition service
	} else if (headerName != null) {
	// compare the header field
	MimeMessage message = messageWrapper.getMessage();
	String[] headerValues = null;
	try {
	headerValues = message.getHeader(headerName);
	} catch (MessagingException e) {
	Debug.logError(e, module);
	}

	if (headerValues != null) {
	for (String headerValue: headerValues) {
	if ("equals".equals(operator)) {
	if (headerValue.equals(value)) {
	passedCondition = true;
	break;
	}
	} else if ("not-equals".equals(operator)) {
	if (!headerValue.equals(value)) {
	passedCondition = true;
	} else {
	passedCondition = false;
	}
	} else if ("matches".equals(operator)) {
	if (headerValue.matches(value)) {
	passedCondition = true;
	break;
	}
	} else if ("not-matches".equals(operator)) {
	if (!headerValue.matches(value)) {
	passedCondition = true;
	} else {
	passedCondition = false;
	}
	} else if ("not-empty".equals(operator)) {
	passedCondition = true;
	break;
	}
	}
	} else if ("empty".equals(operator)) {
	passedCondition = true;
	}
	} else if (fieldName != null) {
	MimeMessage message = messageWrapper.getMessage();
	String[] fieldValues = null;
	try {
	fieldValues = this.getFieldValue(message, fieldName);
	} catch (MessagingException e) {
	Debug.logError(e, module);
	} catch (IOException e) {
	Debug.logError(e, module);
	}

	if (fieldValues != null) {
	for (String fieldValue: fieldValues) {
	if ("equals".equals(operator)) {
	if (fieldValue.equals(value)) {
	passedCondition = true;
	break;
	}
	} else if ("not-equals".equals(operator)) {
	if (!fieldValue.equals(value)) {
	passedCondition = true;
	} else {
	passedCondition = false;
	}
	} else if ("matches".equals(operator)) {
	if (fieldValue.matches(value)) {
	passedCondition = true;
	break;
	}
	} else if ("not-matches".equals(operator)) {
	if (!fieldValue.matches(value)) {
	passedCondition = true;
	} else {
	passedCondition = false;
	}
	} else if ("not-empty".equals(operator)) {
	passedCondition = true;
	break;
	}
	}
	} else if ("empty".equals(operator)) {
	passedCondition = true;
	}
	} else {
	passedCondition = false;
	}

	return passedCondition;
	}

	protected String[] getFieldValue(MimeMessage message, String fieldName) throws MessagingException, IOException {
	String[] values = null;
	if ("to".equals(fieldName)) {
	Address[] addrs = message.getRecipients(MimeMessage.RecipientType.TO);
	if (addrs != null) {
	values = new String[addrs.length];
	for (int i = 0; i < addrs.length; i++) {
	values[i] = addrs[i].toString();
	}
	}
	} else if ("cc".equals(fieldName)) {
	Address[] addrs = message.getRecipients(MimeMessage.RecipientType.CC);
	if (addrs != null) {
	values = new String[addrs.length];
	for (int i = 0; i < addrs.length; i++) {
	values[i] = addrs[i].toString();
	}
	}
	} else if ("bcc".equals(fieldName)) {
	Address[] addrs = message.getRecipients(MimeMessage.RecipientType.BCC);
	if (addrs != null) {
	values = new String[addrs.length];
	for (int i = 0; i < addrs.length; i++) {
	values[i] = addrs[i].toString();
	}
	}
	} else if ("from".equals(fieldName)) {
	Address[] addrs = message.getFrom();
	if (addrs != null) {
	values = new String[addrs.length];
	for (int i = 0; i < addrs.length; i++) {
	values[i] = addrs[i].toString();
	}
	}
	} else if ("subject".equals(fieldName)) {
	values = new String[1];
	values[0] = message.getSubject();
	} else if ("send-date".equals(fieldName)) {
	values = new String[1];
	values[0] = message.getSentDate().toString();
	} else if ("received-date".equals(fieldName)) {
	values = new String[1];
	values[0] = message.getReceivedDate().toString();
	} else if ("body".equals(fieldName)) {
	List<String> bodyParts = this.getBodyText(message);
	values = bodyParts.toArray(new String[bodyParts.size()]);
	}
	return values;
	}

	private List<String> getBodyText(Part part) throws MessagingException, IOException {
	Object c = part.getContent();
	if (c instanceof String) {
	return UtilMisc.toList((String) c);
	} else if (c instanceof Multipart) {
	List<String> textContent = FastList.newInstance();
	int count = ((Multipart) c).getCount();
	for (int i = 0; i < count; i++) {
	BodyPart bp = ((Multipart) c).getBodyPart(i);
	textContent.addAll(this.getBodyText(bp));
	}
	return textContent;
	} else {
	return FastList.newInstance();
	}
	}
	}