hevy-shared 1.0.960 → 1.0.962

package/src/filterExercises.ts

@@ -0,0 +1,222 @@
+import { intersectionWith } from 'lodash';
+import {
+  EquipmentFilter,
+  ExerciseTemplate,
+  Language,
+  MuscleGroupFilter,
+} from './index';
+import { removeAccents } from './utils';
+
+export interface SearchableExerciseTemplate extends ExerciseTemplate {
+  localized_muscle_group_name: string;
+}
+
+interface FilterExercisesProps {
+  exercises: SearchableExerciseTemplate[];
+  muscleGroupFilter: MuscleGroupFilter;
+  equipmentType: EquipmentFilter;
+  searchText: string;
+  language: Language;
+}
+
+export const filterExercises = ({
+  exercises,
+  muscleGroupFilter,
+  equipmentType,
+  searchText,
+  language,
+}: FilterExercisesProps): ExerciseTemplate[] => {
+  let result = exercises;
+
+  result = filterExercisesWithMuscleGroup(muscleGroupFilter, result);
+  result = filterExercisesWithEquipment(equipmentType, result);
+
+  const normalizedSearchText = normalizeSearchText(searchText, language);
+
+  result = filterExercisesWithSearchText(
+    normalizedSearchText,
+    result,
+    language,
+  );
+  result = orderExercisesWithSearchText(normalizedSearchText, result);
+
+  return result;
+};
+
+const filterExercisesWithMuscleGroup = (
+  muscleGroup: MuscleGroupFilter,
+  exercises: SearchableExerciseTemplate[],
+): SearchableExerciseTemplate[] => {
+  if (muscleGroup === 'all_muscles') {
+    return exercises;
+  }
+
+  return exercises.filter((rd) => rd.muscle_group === muscleGroup);
+};
+
+const filterExercisesWithEquipment = (
+  equipmentType: EquipmentFilter,
+  exercises: SearchableExerciseTemplate[],
+): SearchableExerciseTemplate[] => {
+  if (equipmentType === 'all_equipment') {
+    return exercises;
+  }
+
+  return exercises.filter((rd) => rd.equipment_category === equipmentType);
+};
+
+const filterExercisesWithSearchText = (
+  searchText: string,
+  exercises: SearchableExerciseTemplate[],
+  language: Language,
+): SearchableExerciseTemplate[] => {
+  if (searchText.length === 0) {
+    return exercises;
+  }
+
+  return exercises.filter((e) => {
+    const searchableSearchText = searchable(searchText);
+    const searchableEnglishTitle = searchable(e.title);
+
+    const searchableLocalizedTitleMap = {
+      en: searchable(e.title),
+      es: e.es_title ? searchable(e.es_title) : undefined,
+      de: e.de_title ? searchable(e.de_title) : undefined,
+      fr: e.fr_title ? searchable(e.fr_title) : undefined,
+      it: e.it_title ? searchable(e.it_title) : undefined,
+      pt: e.pt_title ? searchable(e.pt_title) : undefined,
+      tr: e.tr_title ? searchable(e.tr_title) : undefined,
+      zh_CN: e.zh_cn_title ? searchable(e.zh_cn_title) : undefined,
+      zh_TW: e.zh_tw_title ? searchable(e.zh_tw_title) : undefined,
+      ru: e.ru_title ? searchable(e.ru_title) : undefined,
+      ja: e.ja_title ? searchable(e.ja_title) : undefined,
+      ko: e.ko_title ? searchable(e.ko_title) : undefined,
+    };
+
+    const searchableLocalizedTitle =
+      searchableLocalizedTitleMap[language] || searchableLocalizedTitleMap.en;
+
+    const searchableLocalizedMuscleGroupName = searchable(
+      e.localized_muscle_group_name,
+    );
+    const searchableMuscleGroup = searchable(e.muscle_group);
+    const searchableMisspelledDumbbellEquipment =
+      e.equipment_category === 'dumbbell' ? 'dumbell' : '';
+    const searchableTags = e.manual_tag ? searchable(e.manual_tag) : '';
+    const searchableAka = e.aka ? searchable(e.aka) : '';
+
+    if (
+      searchableLocalizedTitle.includes(searchableSearchText) ||
+      searchableEnglishTitle.includes(searchableSearchText) ||
+      searchableLocalizedMuscleGroupName.includes(searchableSearchText) ||
+      searchableMuscleGroup.includes(searchableSearchText)
+    ) {
+      return true;
+    }
+
+    if (e.is_custom && 'custom'.includes(searchableSearchText)) {
+      return true;
+    }
+
+    const splitSearch = searchableSearchText.split(' ');
+    const splitTitle = [
+      searchableLocalizedTitle,
+      searchableEnglishTitle,
+      searchableTags,
+      searchableAka,
+      searchableMisspelledDumbbellEquipment,
+    ]
+      .join(' ')
+      .split(' ');
+
+    // Find:
+    // title ['arnold', 'press', '(barbell)']
+    // search ['press', 'bar']
+    if (
+      intersectionWith(splitSearch, splitTitle, (search, title) =>
+        title.includes(search),
+      ).length === splitSearch.length
+    ) {
+      return true;
+    }
+
+    return false;
+  });
+};
+
+const orderExercisesWithSearchText = (
+  searchText: string,
+  exercises: SearchableExerciseTemplate[],
+): SearchableExerciseTemplate[] => {
+  if (searchText.length === 0) {
+    return exercises;
+  }
+
+  return exercises
+    .reduce((accu: any[], currentExercise) => {
+      const lowercaseTitle = currentExercise.title.toLowerCase();
+
+      if (lowercaseTitle.startsWith(searchText)) {
+        accu.push({
+          exercise: currentExercise,
+          weight: currentExercise.priority * 2,
+        });
+        return accu;
+      }
+
+      accu.push({
+        exercise: currentExercise,
+        weight: currentExercise.priority,
+      });
+      return accu;
+    }, [])
+    .sort((a, b) => b.weight - a.weight)
+    .map((a) => a.exercise);
+};
+
+/**
+ * trims, lowercases and removes trailing "s" character to
+ * simplify english searches
+ */
+const normalizeSearchText = (searchText: string, language: Language) => {
+  const searchableSearchText = searchable(searchText).replace('-', ' ').trim();
+
+  if (language !== 'en') return searchableSearchText;
+  if (searchableSearchText.length < 3) return searchableSearchText;
+
+  const lastCharacter = searchableSearchText.slice(-1);
+  if (lastCharacter !== 's') return searchableSearchText;
+
+  return searchableSearchText.slice(0, -1);
+};
+
+const searchable = (str: string): string => {
+  return removeAccents(str.toLowerCase());
+};
+
+/**
+ * Exercise templates that are shown at the top of the Exercise Library to
+ * users who don't have any workouts logged.
+ */
+export const popularExerciseTemplateIds = [
+  'B8127AD1', // Lying Leg Curl (Machine)"
+  '11A123F3', // Seated Leg Curl (Machine)"
+  'D04AC939', // Squat (Barbell)"
+  '75A4F6C4', // Leg Extension (Machine)"
+  '422B08F1', // Lateral Raise (Dumbbell)"
+  'C7973E0E', // Leg Press (Machine)"
+  '07B38369', // Incline Bench Press (Dumbbell)"
+  '3601968B', // Bench Press (Dumbbell)"
+  '6A6C31A5', // Lat Pulldown (Cable)"
+  '0393F233', // Seated Cable Row - V Grip (Cable)"
+  '37FCC2BB', // Bicep Curl (Dumbbell)"
+  '55E6546F', // Bent Over Row (Barbell)"
+  '94B7239B', // Triceps Rope Pushdown"
+  'C6272009', // Deadlift (Barbell)"
+  '878CD1D0', // Shoulder Press (Dumbbell)"
+  '651F844C', // Cable Fly Crossovers"
+  '79D0BB3A', // Bench Press (Barbell)"
+  '93A552C6', // Triceps Pushdown"
+  '7E3BC8B6', // Hammer Curl (Dumbbell)"
+  'BE640BA0', // Face Pull
+] as const;