Modulified it

koucha/src/user.rs

@@ -0,0 +1,62 @@
+use sqlx::SqlitePool;
+use crate::{
+  Result,
+  Feed
+};
+
+pub struct UserId(pub i64);
+impl From<i64> for UserId { fn from(id: i64) -> Self { UserId(id) } }
+impl From<UserId> for i64 { fn from(id: UserId) -> Self { id.0 } }
+
+pub struct User {
+  pub id: UserId,
+  pub name: String,
+}
+
+impl User {
+  pub async fn get_all(pool: &SqlitePool) -> Result<Vec<Self>> {
+    let users = sqlx::query_as!(
+      User,
+      "SELECT id, name FROM users"
+    ).fetch_all(pool).await?;
+
+    Ok(users)
+  }
+
+  pub async fn create(pool: &SqlitePool, name: &str) -> Result<Self> {
+    let result = sqlx::query!(
+      "INSERT INTO users (name) 
+       VALUES (?)
+       RETURNING id, name",
+      name
+    ).fetch_one(pool).await?;
+
+    Ok(Self {
+      id: UserId(result.id),
+      name: result.name,
+    })
+  }
+
+  pub async fn update_name(
+      pool: &SqlitePool, id: UserId, new_name: &str
+  ) -> Result<()> {
+    sqlx::query!(
+      "UPDATE users SET name = ? WHERE id = ?",
+      new_name, id.0
+    ).execute(pool).await?;
+
+    Ok(())
+  }
+
+  pub async fn get_feeds(pool: &SqlitePool, id: UserId) -> Result<Vec<Feed>> {
+    let feeds = sqlx::query_as!(
+        Feed,
+        "SELECT id FROM feeds WHERE user_id = ?",
+        id.0
+      ).fetch_all(pool).await?;
+
+    Ok(feeds)
+  }
+
+  pub fn name(&self) -> &str { &self.name }
+}