handle date changes

Author: Jesperpaulsen

packages/data-collector/src/Firestore.ts

@@ -24,6 +24,7 @@ export class Firestore {
   private store: typeof Store
   private usageCollection: CollectionReference<DocumentData>
   private userCollection: CollectionReference<DocumentData>
+  private unsubscribeTodaysListener?: () => void
 
   constructor(store: typeof Store) {
     this.store = store
@@ -58,7 +59,8 @@ export class Firestore {
     callback: (props: { size: number; CO2: number }) => void
   ) => {
     const d = doc(this.usageCollection, `${uid}-${date}`)
-    onSnapshot(d, (doc) => {
+    if (this.unsubscribeTodaysListener) this.unsubscribeTodaysListener()
+    this.unsubscribeTodaysListener = onSnapshot(d, (doc) => {
       const data = doc.data() as BaseUsageDoc
       const usage = { size: data.size, CO2: data.CO2 }
       callback(usage)

packages/data-collector/src/UsageCounter.ts

@@ -9,6 +9,7 @@ export class UsageCounter {
   private usageSinceSubscriptionStarted = 0
   private store: typeof Store
   private totalCO2 = 0
+  private currentDate = new Date().setHours(0, 0, 0, 0).valueOf()
 
   constructor(store: typeof Store) {
     this.store = store
@@ -47,19 +48,37 @@ export class UsageCounter {
     )
   }
 
+  private checkIfDateHasChanged = () => {
+    const now = new Date().setHours(0, 0, 0, 0).valueOf()
+    if (this.currentDate !== now) {
+      this.currentDate = now
+      return true
+    }
+    return false
+  }
+
   private handleUsageUpdate = ({
     size,
     CO2
   }: {
     size: number
     CO2: number
   }) => {
+    if (this.checkIfDateHasChanged()) {
+      this.usageSinceSubscriptionStarted = 0
+      this.usageToday = 0
+      this.usageLast7Days = 0
+      this.totalUsage = 0
+      this.totalCO2 = 0
+      this.listenToChanges()
+      return
+    }
     if (!this.usageToday) this.usageToday = size
     this.usageSinceSubscriptionStarted = size - this.usageToday
     this.usageToday += this.usageSinceSubscriptionStarted
     this.usageLast7Days += this.usageSinceSubscriptionStarted
     this.totalUsage += this.usageSinceSubscriptionStarted
-    this.totalCO2 += CO2
+    this.totalCO2 = CO2
     this.sendUsageUpdate()
   }
 

packages/data-presenter/src/components/dashboard/UsageDisplay.tsx

@@ -66,8 +66,8 @@ const UsageDisplay: FunctionalComponent = () => {
       </div>
       <div className="flex justify-center mt-3">
         <div className="text-center">
-          <div className="text-6xl font-medium">{totalCO2.toFixed(2)}</div>
-          <div className="text-xs font-light">kg CO2</div>
+          <div className="text-6xl font-medium">{totalCO2.toFixed(4)}</div>
+          <div className="text-xs font-light">kg CO2 emitted today</div>
         </div>
       </div>
     </div>