58 lines
2.3 KiB
TypeScript
58 lines
2.3 KiB
TypeScript
import { every, get, isEqual, uniq, gt, lt, isNumber } from "lodash";
|
|
import { TaggedPoint, TaggedPointGroup } from "farmbot";
|
|
import { PointGroup } from "farmbot/dist/resources/api_resources";
|
|
import moment from "moment";
|
|
import { DEFAULT_CRITERIA } from "./interfaces";
|
|
|
|
const eqCriteriaEmpty =
|
|
(eqCriteria: Record<string, (string | number)[] | undefined>) =>
|
|
every(Object.values(eqCriteria).map(values => !values?.length));
|
|
|
|
const checkCriteria =
|
|
(criteria: PointGroup["criteria"], now: moment.Moment) =>
|
|
(point: TaggedPoint, criteriaKey: keyof PointGroup["criteria"]) => {
|
|
switch (criteriaKey) {
|
|
case "string_eq":
|
|
case "number_eq":
|
|
return every(Object.entries(criteria[criteriaKey])
|
|
.map(([k, values]: [string, (string | number)[]]) =>
|
|
values?.includes(get(point.body, k))))
|
|
|| eqCriteriaEmpty(criteria[criteriaKey]);
|
|
case "number_gt":
|
|
case "number_lt":
|
|
const compare = { number_gt: gt, number_lt: lt };
|
|
return every(Object.entries(criteria[criteriaKey])
|
|
.map(([key, value]) => isNumber(value) &&
|
|
compare[criteriaKey](get(point.body, key), value)));
|
|
case "day":
|
|
const pointDate = moment(point.body.pointer_type == "Plant"
|
|
&& point.body.planted_at
|
|
? point.body.planted_at
|
|
: point.body.created_at);
|
|
const compareDate = moment(now)
|
|
.subtract(criteria[criteriaKey].days_ago, "days");
|
|
const matchesDays = criteria[criteriaKey].op == "<"
|
|
? pointDate.isAfter(compareDate)
|
|
: pointDate.isBefore(compareDate);
|
|
return matchesDays || !criteria[criteriaKey].days_ago;
|
|
}
|
|
};
|
|
|
|
export const selectPointsByCriteria = (
|
|
criteria: PointGroup["criteria"] | undefined,
|
|
allPoints: TaggedPoint[],
|
|
now = moment(),
|
|
): TaggedPoint[] => {
|
|
if (!criteria || isEqual(criteria, DEFAULT_CRITERIA)) { return []; }
|
|
const check = checkCriteria(criteria, now);
|
|
return allPoints.filter(point =>
|
|
every(Object.keys(criteria).map((key: keyof PointGroup["criteria"]) =>
|
|
check(point, key))));
|
|
};
|
|
|
|
export const pointsSelectedByGroup =
|
|
(group: TaggedPointGroup, allPoints: TaggedPoint[]) =>
|
|
uniq(allPoints
|
|
.filter(p => group.body.point_ids.includes(p.body.id || 0))
|
|
.concat(selectPointsByCriteria(group.body.criteria, allPoints)));
|