import mockery from 'mockery';
import AnimatedValue from './_AnimatedValue';

mockery.enable({
	warnOnUnregistered: false
});

type ValueArray = (number | AnimatedValue)[];
type SingleValue = number | AnimatedValue;
type Value = SingleValue | ValueArray;

const getValue = (node: number | AnimatedValue) => {
	if (typeof node === 'number') {
		return node;
	}
	return node.__value;
};

mockery.registerMock('react-native-reanimated', {
	Value: AnimatedValue,
	Node: AnimatedValue,
	add: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) + getValue(b)),
	sub: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) - getValue(b)),
	divide: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) / getValue(b)),
	multiply: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) * getValue(b)),
	sin: (a: SingleValue) => new AnimatedValue(Math.sin(getValue(a))),
	tan: (a: SingleValue) => new AnimatedValue(Math.tan(getValue(a))),
	cos: (a: SingleValue) => new AnimatedValue(Math.cos(getValue(a))),
	sqrt: (a: SingleValue) => new AnimatedValue(Math.sqrt(getValue(a))),
	min: (...a: ValueArray) =>
		new AnimatedValue(Math.min(...a.map(b => getValue(b)))),
	max: (...a: ValueArray) =>
		new AnimatedValue(Math.max(...a.map(b => getValue(b)))),
	pow: (...a: ValueArray) =>
		new AnimatedValue(a.reduce(
			(b, c, index) => (index > 0 ? Math.pow(getValue(b), getValue(c)) : c),
			0
		) as number),
	and: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) && getValue(b)),
	set: (a: SingleValue, b: SingleValue) => getValue(b),
	or: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) || getValue(b)),
	modulo: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(getValue(a) % getValue(b)),
	exp: (a: SingleValue) => new AnimatedValue(Math.exp(getValue(a))),
	asin: (a: SingleValue) => new AnimatedValue(Math.asin(getValue(a))),
	atan: (a: SingleValue) => new AnimatedValue(Math.atan(getValue(a))),
	acos: (a: SingleValue) => new AnimatedValue(Math.acos(getValue(a))),
	floor: (a: SingleValue) => new AnimatedValue(Math.floor(getValue(a))),
	abs: (a: SingleValue) => new AnimatedValue(Math.abs(getValue(a))),
	round: (a: SingleValue) => new AnimatedValue(Math.round(getValue(a))),
	ceil: (a: SingleValue) => new AnimatedValue(Math.ceil(getValue(a))),
	acc: (a: SingleValue) => 'acc_mock' + getValue(a),
	diff: (a: SingleValue) => 'diff_mock' + getValue(a),
	concat: (a: SingleValue, b: string) => getValue(a) + b,
	defined: (a: SingleValue) => 'defined_mock' + getValue(a),
	eq: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(Number(getValue(a) === getValue(b))),
	neq: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(Number(getValue(a) !== getValue(b))),
	lessThan: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(Number(getValue(a) < getValue(b))),
	greaterThan: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(Number(getValue(a) > getValue(b))),
	greaterOrEq: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(Number(getValue(a) >= getValue(b))),
	lessOrEq: (a: SingleValue, b: SingleValue) =>
		new AnimatedValue(Number(getValue(a) <= getValue(b))),
	not: (a: SingleValue) => new AnimatedValue(Number(!getValue(a))),
	cond: (a: SingleValue, b: SingleValue, c: SingleValue) =>
		new AnimatedValue(getValue(a) ? getValue(b) : getValue(c))
});