From 198858de3e3cd353674d8669b3e2c9017b38212f Mon Sep 17 00:00:00 2001
From: Christine Dodrill <me@christine.website>
Date: Tue, 27 Oct 2020 12:21:28 -0400
Subject: [PATCH] what

---
 src/main.rs   | 42 +++++++++++++++++++++++++++++++++++++-----
 src/models.rs | 23 ++++++++++++++++-------
 2 files changed, 53 insertions(+), 12 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index e833d57..9510d15 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -54,17 +54,49 @@ fn gitea_callback(
     let tok = token.access_token().to_string();
     let refresh = token.refresh_token().unwrap().to_string();
 
-    let gitea_user = gitea::user(tok.clone()).exepct("gitea api call to work");
+    let gitea_user = gitea::user(tok.clone()).expect("gitea api call to work");
 
-    use schema::users::dsl::{users, email};
-    let user: models::User = match users.filter(email.eq(gitea_user.email)).limit(1).load::<models::User>(&conn) {
-        Ok(u) => u,
+    use schema::{
+        gitea_tokens,
+        users::{
+            dsl::{email, users},
+            table as users_table,
+        },
+    };
+    let user: models::User = match users
+        .filter(email.eq(gitea_user.email.clone()))
+        .limit(1)
+        .load::<models::User>(&*conn)
+    {
+        Ok(u) => u[0].clone(),
         Err(why) => {
             let u = models::User {
+                id: uuid::Uuid::new_v4(),
                 salutation: gitea_user.full_name,
                 email: gitea_user.email,
                 is_admin: gitea_user.is_admin,
-            }
+                is_locked: false,
+                tier: 0,
+            };
+
+            diesel::insert_into(users_table)
+                .values(&u)
+                .get_result(&*conn)
+                .expect("able to insert user");
+
+            let tok = models::GiteaToken {
+                id: uuid::Uuid::new_v4(),
+                user_id: u.id.clone(),
+                access_token: tok,
+                refresh_token: refresh,
+            };
+
+            diesel::insert_into(gitea_tokens::table)
+                .values(&tok)
+                .get_result(&*conn)
+                .expect("able to insert token");
+
+            u
         }
     };
 
diff --git a/src/models.rs b/src/models.rs
index 1139a04..049ff44 100644
--- a/src/models.rs
+++ b/src/models.rs
@@ -1,14 +1,23 @@
 use serde::Serialize;
 use uuid::Uuid;
-use crate::schema::users;
+use crate::schema::{gitea_tokens, users};
 
 #[derive(Insertable, Queryable, Serialize, Debug, Clone)]
 #[table_name="users"]
 pub struct User {
-    id: Uuid,
-    email: String,
-    salutation: String,
-    is_admin: bool,
-    is_locked: bool,
-    tier: i32,
+    pub id: Uuid,
+    pub email: String,
+    pub salutation: String,
+    pub is_admin: bool,
+    pub is_locked: bool,
+    pub tier: i32,
+}
+
+#[derive(Insertable, Queryable, Debug, Clone)]
+#[table_name="gitea_tokens"]
+pub struct GiteaToken {
+    pub id: Uuid,
+    pub user_id: Uuid,
+    pub access_token: String,
+    pub refresh_token: String,
 }