From ed85145490d23b4301350c5db7179e168190ec6b Mon Sep 17 00:00:00 2001
From: brayo <vukubrian@gmail.com>
Date: Wed, 26 Jun 2024 11:14:36 +0300
Subject: [PATCH] feat: run every 5 mins

---
 aw-firebase-sync/src/main.rs | 59 ++++++++++++++++++++----------------
 1 file changed, 33 insertions(+), 26 deletions(-)

diff --git a/aw-firebase-sync/src/main.rs b/aw-firebase-sync/src/main.rs
index 5eb4c2bd..09a9922b 100644
--- a/aw-firebase-sync/src/main.rs
+++ b/aw-firebase-sync/src/main.rs
@@ -4,13 +4,15 @@ use dirs::config_dir;
 use reqwest;
 use serde_json::{json, Value};
 use serde_yaml;
+use std::env;
 use std::fs::{DirBuilder, File};
 use std::io::prelude::*;
-use std::env;
+use tracing::info;
+use tracing_subscriber;
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
-
+    tracing_subscriber::fmt::init();
     let args: Vec<String> = env::args().collect();
     let mut port: u16 = 5600;
     if args.len() > 1 {
@@ -30,9 +32,6 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     }
     let aw_client = AwClient::new("localhost", port, "aw-firebase-sync").unwrap();
 
-    let start = Utc::now().date().and_hms_opt(0, 0, 0).unwrap() - chrono::Duration::days(6);
-    let end = Utc::now().date().and_hms_opt(0, 0, 0).unwrap() + chrono::Duration::days(1);
-
     let path = config_dir()
         .map(|mut path| {
             path.push("activitywatch");
@@ -77,30 +76,38 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
             RETURN = events;
         ";
 
-    let timeperiods = vec![(start, end)];
-
-    let query_result = aw_client.query(&query, timeperiods).await.expect("Failed to query data");
-
-    let query_data = serde_json::to_string(&query_result[0]).unwrap();
-
     let firebase_url = "https://us-central1-aw-mockup.cloudfunctions.net/uploadData";
     // let firebase_url = "http://localhost:5001/aw-mockup/us-central1/uploadData";
 
-    let payload = json!({
-        "apiKey": apikey,
-        "data": query_data
-    });
-
     let firebase_client = reqwest::Client::new();
-    let response = firebase_client
-        .post(firebase_url)
-        .json(&payload)
-        .send()
-        .await?
-        .json::<Value>()
-        .await?;
 
-    println!("Response: {:?}", response);
-
-    Ok(())
+    loop {
+        #[allow(deprecated)]
+        let start =
+            Utc::now().date().and_hms_opt(0, 0, 0).unwrap() - chrono::Duration::seconds(300);
+        #[allow(deprecated)]
+        let end = Utc::now().date().and_hms_opt(0, 0, 0).unwrap();
+        let timeperiods = vec![(start, end)];
+
+        let query_result = aw_client
+            .query(&query, timeperiods)
+            .await
+            .expect("Failed to query data");
+        let query_data =
+            serde_json::to_string(&query_result[0]).expect("Failed to serialize query data");
+        let payload = json!({
+            "apiKey": apikey,
+            "data": query_data
+        });
+        let response = firebase_client
+            .post(firebase_url)
+            .json(&payload)
+            .send()
+            .await?
+            .json::<Value>()
+            .await?;
+        info!("Response: {:?}", response);
+        std::thread::sleep(std::time::Duration::from_secs(300));
+    }
+    // Ok(())
 }