Simplify cycle start detection funtions

db/index.js

@@ -8,7 +8,7 @@ import cycleModule from '../lib/cycle'
 
 let db
 let isMensesStart
-let getMensesDaysAfter
+let getMensesDaysRightAfter
 
 export async function openDb ({ hash, persistConnection }) {
   const realmConfig = {}
@@ -37,7 +37,7 @@ export async function openDb ({ hash, persistConnection }) {
   if (persistConnection) db = connection
   const cycle = cycleModule()
   isMensesStart = cycle.isMensesStart
-  getMensesDaysAfter = cycle.getMensesDaysAfter
+  getMensesDaysRightAfter = cycle.getMensesDaysRightAfter
 }
 
 
@@ -57,51 +57,48 @@ export function getCycleStartsSortedByDate() {
 
 export function saveSymptom(symptom, cycleDay, val) {
   db.write(() => {
-    if (symptom === 'bleeding') {
-      saveBleeding(cycleDay, val)
+    if (bleedingValueDeleted(symptom, val)) {
+      cycleDay.bleeding = val
+      cycleDay.isCycleStart = false
+      maybeSetNewCycleStart(cycleDay, val)
+    } else if (bleedingValueAddedOrChanged(symptom, val)) {
+      cycleDay.bleeding = val
+      cycleDay.isCycleStart = isMensesStart(cycleDay)
+      maybeClearOldCycleStarts(cycleDay)
     } else {
       cycleDay[symptom] = val
     }
   })
+
+  function bleedingValueDeleted(symptom, val) {
+    return symptom === 'bleeding' && !val
   }
 
-// TODO this also needs a test
-export function saveBleeding(cycleDay, bleeding) {
-  if (!bleeding) {
-    updateCycleDayAndMaybeSetNewCycleStart(cycleDay, bleeding)
-  } else {
-    cycleDay.bleeding = bleeding
-    cycleDay.isCycleStart = isMensesStart(cycleDay)
-    maybeClearOldCycleStartsInThisMenses(cycleDay)
+  function bleedingValueAddedOrChanged(symptom, val) {
+    return symptom === 'bleeding' && val
   }
 
-  function updateCycleDayAndMaybeSetNewCycleStart(oldCycleDay, newValue) {
+  function maybeSetNewCycleStart(dayWithDeletedBleeding) {
     // if a bleeding value is deleted, we need to check if
     // there are any following bleeding days and if the
     // next one of them is now a cycle start
-
-    // in order to get the menses days, the cycle day in question still
-    // has to have a bleeding value, so we get those days first and only
-    // then update the cycle day
-    const mensesDaysAfter = getMensesDaysAfter(oldCycleDay)
-    oldCycleDay.bleeding = newValue
-    oldCycleDay.isCycleStart = false
-
+    const mensesDaysAfter = getMensesDaysRightAfter(dayWithDeletedBleeding)
     if (!mensesDaysAfter.length) return
-
     const nextOne = mensesDaysAfter[mensesDaysAfter.length - 1]
     if (isMensesStart(nextOne)) {
       nextOne.isCycleStart = true
     }
   }
 
-  function maybeClearOldCycleStartsInThisMenses(cycleDay) {
+  function maybeClearOldCycleStarts(cycleDay) {
     // if we have a new bleeding day, we need to clear the
     // menses start marker from all following days of this
     // menses that may have been marked as start before
-    const mensesDaysAfter = getMensesDaysAfter(cycleDay)
+    const mensesDaysAfter = getMensesDaysRightAfter(cycleDay)
     mensesDaysAfter.forEach(day => day.isCycleStart = false)
   }
+
+
 }
 
 export function getOrCreateCycleDay(localDate) {

lib/cycle.js

@@ -177,36 +177,6 @@ export default function config(opts) {
   }
 
 
-  // TODO this also needs a test
-  function maybeSetNewCycleStart(oldCycleDay, newValue) {
-    // if a bleeding value is deleted, we need to check if
-    // there are any following bleeding days and if the
-    // next one of them is now a cycle start
-
-    // in order to get the menses days, the cycle day in question still
-    // has to have a bleeding value, so we get those days first and only
-    // then update the cycle day
-    const mensesDaysAfter = getMensesDaysRightAfter(oldCycleDay)
-    oldCycleDay.bleeding = newValue
-    oldCycleDay.isCycleStart = false
-
-    if (!mensesDaysAfter.length) return
-
-    const nextOne = mensesDaysAfter[mensesDaysAfter.length - 1]
-    if (isMensesStart(nextOne)) {
-      nextOne.isCycleStart = true
-    }
-  }
-
-  function maybeClearOldCycleStartsInThisMenses(cycleDay) {
-    // if we have a new bleeding day, we need to clear the
-    // menses start marker from all following days of this
-    // menses that may have been marked as start before
-    const mensesDaysAfter = getMensesDaysRightAfter(cycleDay)
-    mensesDaysAfter.forEach(day => day.isCycleStart = false)
-  }
-
-
   return {
     getCycleDayNumber,
     getCycleForDay,
@@ -215,8 +185,6 @@ export default function config(opts) {
     getAllCycleLengths,
     getPredictedMenses,
     isMensesStart,
-    getMensesDaysRightAfter,
-    maybeSetNewCycleStart,
-    maybeClearOldCycleStartsInThisMenses
+    getMensesDaysRightAfter
   }
 }