From 8e96edca8659bc87cd85072741e6db8aeaf751ff Mon Sep 17 00:00:00 2001
From: Kjetil Orbekk <kjetil.orbekk@gmail.com>
Date: Tue, 4 Feb 2020 06:20:25 -0500
Subject: Add raw data to database

---
 src/db.rs       |  8 ++++++++
 src/importer.rs | 14 ++++++++++++--
 src/models.rs   | 35 +++++++++++++++++++++++++++++++++++
 src/schema.rs   | 18 +++++++++++++++++-
 4 files changed, 72 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/db.rs b/src/db.rs
index 548ca99..5135b91 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -155,3 +155,11 @@ pub fn take_task(
         Ok(task)
     })
 }
+
+pub fn insert_data(conn: &PgConnection, data: &models::RawData) -> Result<usize, Error> {
+    use crate::schema::raw_data;
+    let rows = diesel::insert_into(raw_data::table)
+        .values(data)
+        .execute(conn)?;
+    Ok(rows)
+}
diff --git a/src/importer.rs b/src/importer.rs
index 05d56fd..5f5790a 100644
--- a/src/importer.rs
+++ b/src/importer.rs
@@ -140,15 +140,25 @@ fn import_strava_user<S: strava::StravaApi>(
 
         let result = strava.get("/athlete/activities", &token.access_token, &params[..])?;
 
+        let json_error = || Error::UnexpectedJson(result.clone());
         let result = result
             .as_array()
-            .ok_or(Error::UnexpectedJson(result.clone()))?;
+            .ok_or_else(json_error)?;
 
         for activity in result {
+            let id = activity["id"].as_i64().ok_or_else(json_error)?;
             info!(
                 "activity id: {} start: {}",
-                activity["id"], activity["start_date"]
+                id, activity["start_date"]
             );
+
+            db::insert_data(&shared.conn.lock().unwrap(),
+                            &models::RawData {
+                                data_type: models::DataType::StravaActivity,
+                                id: id,
+                                username: username.to_string(),
+                                payload: activity.clone(),
+                            })?;
         }
 
         if result.len() < per_page {
diff --git a/src/models.rs b/src/models.rs
index 3d7eb24..cd8ddf1 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -1,3 +1,4 @@
+use crate::schema::raw_data;
 use crate::schema::config;
 use crate::schema::strava_tokens;
 use crate::schema::tasks;
@@ -104,3 +105,37 @@ pub struct StravaToken {
     pub access_token: String,
     pub expires_at: DateTime<Utc>,
 }
+
+#[derive(PartialEq, Debug, Clone, Copy, AsExpression, FromSqlRow)]
+#[sql_type = "sql_types::Text"]
+pub enum DataType {
+    StravaActivity = 0,
+}
+
+impl ToSql<sql_types::Text, Pg> for DataType {
+    fn to_sql<W: Write>(&self, out: &mut Output<W, Pg>) -> serialize::Result {
+        let t = match *self {
+            DataType::StravaActivity => "s:activi".to_string(),
+        };
+        <String as ToSql<sql_types::Text, Pg>>::to_sql(&t, out)
+    }
+}
+
+impl FromSql<sql_types::Text, Pg> for DataType {
+    fn from_sql(bytes: Option<&[u8]>) -> deserialize::Result<Self> {
+        let s = <String as FromSql<sql_types::Text, Pg>>::from_sql(bytes)?;
+        match s.as_str() {
+            "s:activi" => Ok(DataType::StravaActivity),
+            &_ => Err("Unrecognized data type".into()),
+        }
+    }
+}
+
+#[derive(Insertable, Queryable)]
+#[table_name = "raw_data"]
+pub struct RawData {
+    pub data_type: DataType,
+    pub id: i64,
+    pub username: String,
+    pub payload: Value,
+}
diff --git a/src/schema.rs b/src/schema.rs
index e605bda..df20c4f 100644
--- a/src/schema.rs
+++ b/src/schema.rs
@@ -7,6 +7,15 @@ table! {
     }
 }
 
+table! {
+    raw_data (data_type, id) {
+        data_type -> Varchar,
+        id -> Int8,
+        username -> Varchar,
+        payload -> Nullable<Jsonb>,
+    }
+}
+
 table! {
     strava_tokens (username) {
         username -> Varchar,
@@ -33,7 +42,14 @@ table! {
     }
 }
 
+joinable!(raw_data -> users (username));
 joinable!(strava_tokens -> users (username));
 joinable!(tasks -> users (username));
 
-allow_tables_to_appear_in_same_query!(config, strava_tokens, tasks, users,);
+allow_tables_to_appear_in_same_query!(
+    config,
+    raw_data,
+    strava_tokens,
+    tasks,
+    users,
+);
-- 
cgit v1.2.3