PQLOperatorFactory.java

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package com.liferay.poshi.runner.pql;

import java.util.Properties;

/**
 * @author Michael Hashimoto
 */
public class PQLOperatorFactory {

	public static PQLOperator newPQLOperator(String operator) throws Exception {
		PQLOperator.validateOperator(operator);

		if (operator.equals("AND") || operator.equals("OR")) {
			return new PQLOperator(operator) {

				public Object getPQLResult(
						PQLEntity pqlEntity1, PQLEntity pqlEntity2,
						Properties properties)
					throws Exception {

					String operator = getOperator();
					Object pqlResultObject1 = pqlEntity1.getPQLResult(
						properties);
					Object pqlResultObject2 = pqlEntity2.getPQLResult(
						properties);

					if ((pqlResultObject1 == null) ||
						(pqlResultObject2 == null)) {

						throw new Exception(
							"Operators must be surrounded by 2 boolean " +
								"values: " + operator);
					}

					if (!(pqlResultObject1 instanceof Boolean) ||
						!(pqlResultObject2 instanceof Boolean)) {

						throw new Exception(
							"Operators must be surrounded by 2 boolean " +
								"values: " + operator);
					}

					Boolean pqlResultBoolean1 = (Boolean)pqlResultObject1;
					Boolean pqlResultBoolean2 = (Boolean)pqlResultObject2;

					if (operator.equals("AND")) {
						return pqlResultBoolean1 && pqlResultBoolean2;
					}
					else if (operator.equals("OR")) {
						return pqlResultBoolean1 || pqlResultBoolean2;
					}

					throw new Exception("Unsupported operator: " + operator);
				}

			};
		}
		else if (operator.equals("~") || operator.equals("!~")) {
			return new PQLOperator(operator) {

				public Object getPQLResult(
						PQLEntity pqlEntity1, PQLEntity pqlEntity2,
						Properties properties)
					throws Exception {

					String operator = getOperator();
					Object pqlResultObject1 = pqlEntity1.getPQLResult(
						properties);
					Object pqlResultObject2 = pqlEntity2.getPQLResult(
						properties);

					if ((pqlResultObject1 == null) ||
						(pqlResultObject2 == null)) {

						return false;
					}

					if (!(pqlResultObject1 instanceof String) ||
						!(pqlResultObject2 instanceof String)) {

						throw new Exception(
							"Operator only works for string values: " +
								operator);
					}

					String pqlResultString1 = (String)pqlResultObject1;
					String pqlResultString2 = (String)pqlResultObject2;

					if (operator.equals("~")) {
						return pqlResultString1.contains(pqlResultString2);
					}
					else if (operator.equals("!~")) {
						return !pqlResultString1.contains(pqlResultString2);
					}

					throw new Exception("Unsupported operator: " + operator);
				}

			};
		}
		else if (operator.equals("==") || operator.equals("!=")) {
			return new PQLOperator(operator) {

				public Object getPQLResult(
						PQLEntity pqlEntity1, PQLEntity pqlEntity2,
						Properties properties)
					throws Exception {

					String operator = getOperator();
					Object pqlResultObject1 = pqlEntity1.getPQLResult(
						properties);
					Object pqlResultObject2 = pqlEntity2.getPQLResult(
						properties);

					if ((pqlResultObject1 == null) ||
						(pqlResultObject2 == null)) {

						if (operator.equals("==")) {
							return pqlResultObject1 == pqlResultObject2;
						}
						else if (operator.equals("!=")) {
							return pqlResultObject1 != pqlResultObject2;
						}
					}
					else {
						if (operator.equals("==")) {
							return pqlResultObject1.equals(pqlResultObject2);
						}
						else if (operator.equals("!=")) {
							return !pqlResultObject1.equals(pqlResultObject2);
						}
					}

					throw new Exception("Unsupported operator: " + operator);
				}

			};
		}
		else if (operator.equals("<") || operator.equals("<=") ||
				 operator.equals(">") || operator.equals(">=")) {

			return new PQLOperator(operator) {

				public Object getPQLResult(
						PQLEntity pqlEntity1, PQLEntity pqlEntity2,
						Properties properties)
					throws Exception {

					String operator = getOperator();
					Object pqlResultObject1 = pqlEntity1.getPQLResult(
						properties);
					Object pqlResultObject2 = pqlEntity2.getPQLResult(
						properties);

					if ((pqlResultObject1 == null) ||
						(pqlResultObject2 == null)) {

						throw new Exception(
							"Operator only works for number values: " +
								operator);
					}

					if ((pqlResultObject1 instanceof Double ||
						 pqlResultObject1 instanceof Integer) &&
						(pqlResultObject2 instanceof Double ||
						 pqlResultObject2 instanceof Integer)) {

						Double pqlResultDouble1 = null;
						Double pqlResultDouble2 = null;

						if (pqlResultObject1 instanceof Integer) {
							Integer pqlResultInteger1 =
								(Integer)pqlResultObject1;

							pqlResultDouble1 = pqlResultInteger1.doubleValue();
						}
						else {
							pqlResultDouble1 = (Double)pqlResultObject1;
						}

						if (pqlResultObject2 instanceof Integer) {
							Integer pqlResultInteger2 =
								(Integer)pqlResultObject2;

							pqlResultDouble2 = pqlResultInteger2.doubleValue();
						}
						else {
							pqlResultDouble2 = (Double)pqlResultObject2;
						}

						if (operator.equals("<")) {
							return pqlResultDouble1 < pqlResultDouble2;
						}
						else if (operator.equals("<=")) {
							return pqlResultDouble1 <= pqlResultDouble2;
						}
						else if (operator.equals(">")) {
							return pqlResultDouble1 > pqlResultDouble2;
						}
						else if (operator.equals(">=")) {
							return pqlResultDouble1 >= pqlResultDouble2;
						}

						throw new Exception(
							"Unsupported operator: " + operator);
					}

					throw new Exception(
						"Operator only works for number values: " + operator);
				}

			};
		}

		throw new Exception("Unsupported operator: " + operator);
	}

}