From 03989d6b7ac2bd02e583a34f8a8a14773476eb84 Mon Sep 17 00:00:00 2001
From: Luke Leppan <lukelepp@gmail.com>
Date: Sun, 11 Jul 2021 13:18:37 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20Fix=20Broken=20Stats?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/constants.ts      | 1 +
 src/data/stats.ts     | 1 -
 src/status/manager.ts | 5 ++++-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/constants.ts b/src/constants.ts
index e06accd..079cc13 100644
--- a/src/constants.ts
+++ b/src/constants.ts
@@ -9,4 +9,5 @@ export const MATCH_HTML_COMMENT = new RegExp(
     "|<[?][^>]*>?",
   "g"
 );
+export const MATCH_COMMENT = new RegExp("%%[^%%]+%%", "g");
 export const MATCH_PARAGRAPH = new RegExp("\n([^\n]+)\n", "g");
diff --git a/src/data/stats.ts b/src/data/stats.ts
index 89ca39e..8a5b5dc 100644
--- a/src/data/stats.ts
+++ b/src/data/stats.ts
@@ -37,6 +37,5 @@ export function getSentenceCount(text: string): number {
 }
 
 export function cleanComments(text: string): string {
-  console.log(text.match(MATCH_COMMENT));
   return text.replace(MATCH_COMMENT, "").replace(MATCH_HTML_COMMENT, "");
 }
diff --git a/src/status/manager.ts b/src/status/manager.ts
index 6cd2cfb..c04153d 100644
--- a/src/status/manager.ts
+++ b/src/status/manager.ts
@@ -35,7 +35,10 @@ export class BarManager {
   async updateStatusBar(text: string): Promise<void> {
     let newText = "";
     const expression: Expression = parse(this.settings.statusBarQuery);
-    if (this.settings.collectStats) this.dataManager.updateTodayCounts();
+    if (this.settings.collectStats) {
+      this.dataManager.updateToday();
+      this.dataManager.updateTodayCounts();
+    }
     const todayCounts: TodayCounts = this.settings.collectStats
       ? this.dataManager.getTodayCounts()
       : { words: 0, characters: 0, sentences: 0 };