
/**
 * Copyright (c) 2011-2019, James Zhan 詹波 (jfinal@126.com).
 *
 * 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.tio.jfinal.template.expr.ast;

import org.tio.jfinal.template.TemplateException;
import org.tio.jfinal.template.expr.Sym;
import org.tio.jfinal.template.stat.Location;
import org.tio.jfinal.template.stat.ParseException;
import org.tio.jfinal.template.stat.Scope;

import java.math.BigDecimal;
import java.math.BigInteger;
import java.math.RoundingMode;
import java.util.Objects;

/**
 * Arithmetic
 * 1：支持 byte short int long float double BigDecimal 的 + - * / % 运算
 * 2：支持字符串加法运算
 */
public class Arith extends Expr {
	
	public static final int INT = 0;	// byte、short 用 int 类型支持，java 表达式亦如此
	public static final int LONG = 1;
	public static final int FLOAT = 2;
	public static final int DOUBLE = 3;
	public static final int BIGINTEGER = 4;
	public static final int BIGDECIMAL = 5;
	public static final int UNKNOWN = 99;
	
	private Sym op;
	private Expr left;
	private Expr right;
	
	// BigDecimal 除法使用的最小 scale 值，默认为 5
	protected static int bigDecimalDivideMinScale = 5;
	// BigDecimal 除法使用的舍入模式，默认为四舍五入
	protected static RoundingMode bigDecimalDivideRoundingMode = RoundingMode.HALF_UP;
	
	/**
	 * 设置 BigDecimal 除法使用的最小 scale 值，默认为 5
	 */
	public static void setBigDecimalDivideMinScale(int scale) {
		Arith.bigDecimalDivideMinScale = scale;
	}
	
	/**
	 * 设置 BigDecimal 除法使用的舍入模式，默认为四舍五入
	 */
	public static void setBigDecimalDivideRoundingMode(RoundingMode roundingMode) {
		Objects.requireNonNull(roundingMode, "roundingMode can not be null");
		Arith.bigDecimalDivideRoundingMode = roundingMode;
	}
	
	public Arith(Sym op, Expr left, Expr right, Location location) {
		if (left == null || right == null) {
			throw new ParseException("The target of \"" + op.value() + "\" operator can not be blank", location);
		}
		this.op = op;
		this.left = left;
		this.right = right;
		this.location = location;
	}
	
	public Object eval(Scope scope) {
		try {
			return doEval(scope);
		} catch (TemplateException e) {
			throw e;
		} catch (Exception e) {
			throw new TemplateException(e.getMessage(), location, e);
		}
	}
	
	private Object doEval(Scope scope) {
		Object leftValue = left.eval(scope);
		Object rightValue = right.eval(scope);
		
		if (leftValue instanceof Number && rightValue instanceof Number) {
			Number l = (Number)leftValue;
			Number r = (Number)rightValue;
			int maxType = getMaxType(l, r);
			if (maxType == UNKNOWN) {
				throw unsupportedTypeException(l, r, location);
			}
			
			switch (op) {
			case ADD:
				return add(maxType, l, r);
			case SUB:
				return sub(maxType, l, r);
			case MUL:
				return mul(maxType, l, r);
			case DIV:
				return div(maxType, l, r);
			case MOD:
				return remainder(maxType, l, r);
			default :
				throw new TemplateException("Unsupported operator: " + op.value(), location);
			}
		}
		
		// 字符串加法运算
		if (op == Sym.ADD) {
			if (leftValue instanceof String || rightValue instanceof String) {
				return String.valueOf(leftValue).concat(String.valueOf(rightValue));
			}
		}
		
		String leftObj = leftValue != null ? leftValue.getClass().getName() : "null";
		String rightObj = rightValue != null ? rightValue.getClass().getName() : "null";
		throw new TemplateException("Unsupported operation type: " + leftObj + " " +  op.value() + " " + rightObj, location);
	}
	
	static BigDecimal[] toBigDecimals(Number left, Number right) {
		BigDecimal[] ret = new BigDecimal[2];
		
		if (left instanceof BigDecimal) {
			ret[0] = (BigDecimal)left;
		} else {
			ret[0] = new BigDecimal(left.toString());
		}
		
		if (right instanceof BigDecimal) {
			ret[1] = (BigDecimal)right;
		} else {
			ret[1] = new BigDecimal(right.toString());
		}
		
		return ret;
	}
	
	static BigInteger[] toBigIntegers(Number left, Number right) {
		BigInteger[] ret = new BigInteger[2];
		
		if (left instanceof BigInteger) {
			ret[0] = (BigInteger)left;
		} else {
			ret[0] = new BigInteger(left.toString());
		}
		
		if (right instanceof BigInteger) {
			ret[1] = (BigInteger)right;
		} else {
			ret[1] = new BigInteger(right.toString());
		}
		
		return ret;
	}
	
	static int getMaxType(Number obj1, Number obj2) {
		int t1 = getType(obj1);
		int t2 = getType(obj2);
		int ret = t1 > t2 ? t1 : t2;
		if (ret != BIGINTEGER) {
			return ret;
		}
		
		// BigInteger 在与 Double、Float 运算时，需要升级为 BigDecimal
		if (t1 == BIGINTEGER) {
			if (t2 == DOUBLE || t2 == FLOAT) {
				return BIGDECIMAL;	// 升级为 BigDecimal
			}
		} else {
			if (t1 == DOUBLE || t1 == FLOAT) {
				return BIGDECIMAL;	// 升级为 BigDecimal
			}
		}
		
		return ret;
	}
	
	static int getType(Number obj) {
		if (obj instanceof Integer) {
			return INT;
		} else if (obj instanceof Long) {
			return LONG;
		} else if (obj instanceof Float) {
			return FLOAT;
		} else if (obj instanceof Double) {
			return DOUBLE;
		} else if (obj instanceof BigDecimal) {
			return BIGDECIMAL;
		} else if (obj instanceof Short || obj instanceof Byte) {
			return INT;			// short byte 用 int 支持，java 表达式亦如此
		} else if (obj instanceof BigInteger) {
			return BIGINTEGER;	// 新增 BigInteger 支持
		}
		
		// throw new TemplateException("Unsupported data type: " + obj.getClass().getName(), location);
		return UNKNOWN;
	}
	
	private Number add(int maxType, Number left, Number right) {
		switch (maxType) {
		case INT:
			return Integer.valueOf(left.intValue() + right.intValue());
		case LONG:
			return Long.valueOf(left.longValue() + right.longValue());
		case FLOAT:
			return Float.valueOf(left.floatValue() + right.floatValue());
		case DOUBLE:
			return Double.valueOf(left.doubleValue() + right.doubleValue());
		case BIGDECIMAL:
			BigDecimal[] bd = toBigDecimals(left, right);
			return (bd[0]).add(bd[1]);
		case BIGINTEGER:	// 新增 BigInteger 支持
			BigInteger[] bi = toBigIntegers(left, right);
			return (bi[0]).add(bi[1]);
		}
		throw new TemplateException("Unsupported data type", location);
	}
	
	private Number sub(int maxType, Number left, Number right) {
		switch (maxType) {
		case INT:
			return Integer.valueOf(left.intValue() - right.intValue());
		case LONG:
			return Long.valueOf(left.longValue() - right.longValue());
		case FLOAT:
			return Float.valueOf(left.floatValue() - right.floatValue());
		case DOUBLE:
			return Double.valueOf(left.doubleValue() - right.doubleValue());
		case BIGDECIMAL:
			BigDecimal[] bd = toBigDecimals(left, right);
			return (bd[0]).subtract(bd[1]);
		case BIGINTEGER:	// 新增 BigInteger 支持
			BigInteger[] bi = toBigIntegers(left, right);
			return (bi[0]).subtract(bi[1]);
		}
		throw new TemplateException("Unsupported data type", location);
	}
	
	private Number mul(int maxType, Number left, Number right) {
		switch (maxType) {
		case INT:
			return Integer.valueOf(left.intValue() * right.intValue());
		case LONG:
			return Long.valueOf(left.longValue() * right.longValue());
		case FLOAT:
			return Float.valueOf(left.floatValue() * right.floatValue());
		case DOUBLE:
			return Double.valueOf(left.doubleValue() * right.doubleValue());
		case BIGDECIMAL:
			BigDecimal[] bd = toBigDecimals(left, right);
			return (bd[0]).multiply(bd[1]);
		case BIGINTEGER:	// 新增 BigInteger 支持
			BigInteger[] bi = toBigIntegers(left, right);
			return (bi[0]).multiply(bi[1]);
		}
		throw new TemplateException("Unsupported data type", location);
	}
	
	private Number div(int maxType, Number left, Number right) {
		switch (maxType) {
		case INT:
			return Integer.valueOf(left.intValue() / right.intValue());
		case LONG:
			return Long.valueOf(left.longValue() / right.longValue());
		case FLOAT:
			return Float.valueOf(left.floatValue() / right.floatValue());
		case DOUBLE:
			return Double.valueOf(left.doubleValue() / right.doubleValue());
		case BIGDECIMAL:
			BigDecimal[] bd = toBigDecimals(left, right);
			// return (bd[0]).divide(bd[1]);
			int scale = Math.max(bigDecimalDivideMinScale, bd[0].scale());
			return (bd[0]).divide(bd[1], scale, bigDecimalDivideRoundingMode);
		case BIGINTEGER:	// 新增 BigInteger 支持
			BigInteger[] bi = toBigIntegers(left, right);
			return (bi[0]).divide(bi[1]);
		}
		throw new TemplateException("Unsupported data type", location);
	}
	
	private Number remainder(int maxType, Number left, Number right) {
		switch (maxType) {
		case INT:
			return Integer.valueOf(left.intValue() % right.intValue());
		case LONG:
			return Long.valueOf(left.longValue() % right.longValue());
		case FLOAT:
			return Float.valueOf(left.floatValue() % right.floatValue());
		case DOUBLE:
			return Double.valueOf(left.doubleValue() % right.doubleValue());
		case BIGDECIMAL:
			BigDecimal[] bd = toBigDecimals(left, right);
			return (bd[0]).divideAndRemainder(bd[1])[1];
		case BIGINTEGER:	// 新增 BigInteger 支持
			BigInteger[] bi = toBigIntegers(left, right);
			return (bi[0]).divideAndRemainder(bi[1])[1];
		}
		throw new TemplateException("Unsupported data type", location);
	}
	
	static TemplateException unsupportedTypeException(Number left, Number right, Location location) {
		Number unsupportedType;
		if (left instanceof Integer
			|| left instanceof Long
			|| left instanceof Float
			|| left instanceof Double
			|| left instanceof BigDecimal
			|| left instanceof Short
			|| left instanceof Byte
			|| left instanceof BigInteger) {
			unsupportedType = right;
		} else {
			unsupportedType = left;
		}
		return new TemplateException("Unsupported data type: " + unsupportedType.getClass().getName(), location);
	}
}




