Config polling

migrations/20240906090729_polling_token.down.sql

@@ -0,0 +1 @@
+DROP TABLE pollingtoken;

migrations/20240906090729_polling_token.up.sql

@@ -0,0 +1,7 @@
+CREATE TABLE pollingtoken (
+    id bigserial PRIMARY KEY,
+    token TEXT NOT NULL,
+    device_id bigint NOT NULL,
+    created_at timestamp without time zone NOT NULL DEFAULT now(),
+    FOREIGN KEY(device_id) REFERENCES "device"(id) ON DELETE CASCADE
+);

src/db/models/mod.rs

@@ -12,6 +12,7 @@ pub mod oauth2authorizedapp;
 pub mod oauth2client;
 #[cfg(feature = "openid")]
 pub mod oauth2token;
+pub mod polling_token;
 pub mod session;
 pub mod settings;
 pub mod user;

src/db/models/polling_token.rs

@@ -0,0 +1,38 @@
+use chrono::{NaiveDateTime, Utc};
+use model_derive::Model;
+use sqlx::{query_as, Error as SqlxError};
+
+use crate::random::gen_alphanumeric;
+
+use super::DbPool;
+
+// Token used for polling requests.
+#[derive(Clone, Debug, Model)]
+pub struct PollingToken {
+    pub id: Option<i64>,
+    pub token: String,
+    pub device_id: i64,
+    pub created_at: NaiveDateTime,
+}
+
+impl PollingToken {
+    pub fn new(device_id: i64) -> Self {
+        Self {
+            id: None,
+            device_id,
+            token: gen_alphanumeric(32),
+            created_at: Utc::now().naive_utc(),
+        }
+    }
+
+    pub async fn find(pool: &DbPool, token: &str) -> Result<Option<Self>, SqlxError> {
+        query_as!(
+            Self,
+            "SELECT id, token, device_id, created_at
+            FROM pollingtoken WHERE token = $1",
+            token
+        )
+        .fetch_optional(pool)
+        .await
+    }
+}

src/db/models/user.rs

@@ -871,6 +871,27 @@ impl User {
         }
         Ok(())
     }
+
+    pub async fn find_by_device_id<'e, E>(
+        executor: E,
+        device_id: i64,
+    ) -> Result<Option<Self>, SqlxError>
+    where
+        E: PgExecutor<'e>,
+    {
+        query_as!(
+            Self,
+            "SELECT u.id \"id?\", u.username, u.password_hash, u.last_name, u.first_name, u.email, \
+            u.phone, u.mfa_enabled, u.totp_enabled, u.email_mfa_enabled, \
+            u.totp_secret, u.email_mfa_secret, u.mfa_method \"mfa_method: _\", u.recovery_codes, u.is_active, u.openid_login \
+            FROM \"user\" as u \
+            JOIN \"device\" as d ON u.id = d.user_id \
+            WHERE d.id = $1",
+            device_id
+        )
+        .fetch_optional(executor)
+        .await
+    }
 }
 
 #[cfg(test)]

src/enterprise/grpc/mod.rs

@@ -0,0 +1 @@
+pub mod polling;