use url crate

Cargo.lock

@@ -3279,6 +3279,7 @@ dependencies = [
  "form_urlencoded",
  "idna",
  "percent-encoding",
+ "serde",
 ]
 
 [[package]]

Cargo.toml

@@ -14,4 +14,4 @@ rss = "2.0.11"
 sqlx = { version = "0.7", features = ["runtime-tokio", "sqlite", "chrono", "uuid"] }
 uuid = { version = "1.7.0", features = ["v4", "serde"] }
 bcrypt = "0.15"
-url = "2.5.4"
+url = { version = "2.5", features = ["serde"] }

src/feeds.rs

@@ -3,6 +3,7 @@ use rocket::http::Status;
 use rocket::serde::{json::Json, Deserialize, Serialize};
 use rocket_db_pools::Connection;
 use uuid::Uuid;
+use url::Url;
 
 use crate::Db;
 use crate::user::AuthenticatedUser;
@@ -12,14 +13,14 @@ use crate::user::AuthenticatedUser;
 pub struct Feed {
     feed_id: Uuid,
     name: String,
-    url: String,
+    url: Url,
     user_id: Uuid,
     added_time: chrono::DateTime<chrono::Utc>,
     last_checked_time: chrono::DateTime<chrono::Utc>,
 }
 
 impl Feed {
-    pub fn new(name: String, url: String, user_id: Uuid) -> Self {
+    pub fn new(name: String, url: Url, user_id: Uuid) -> Self {
         let now = chrono::Utc::now();
         Feed {
             feed_id: Uuid::new_v4(),
@@ -36,7 +37,7 @@ impl Feed {
 #[serde(crate = "rocket::serde")]
 pub struct NewFeed {
     pub name: String,
-    pub url: String,
+    pub url: Url,  // Changed from String to Url
 }
 
 #[post("/feeds", data = "<new_feed>")]
@@ -46,6 +47,12 @@ pub async fn create_feed(
     user: AuthenticatedUser,
 ) -> Result<Json<Feed>, Status> {
     let new_feed = new_feed.into_inner();
+    
+    // URL validation only needs to check scheme now
+    if !new_feed.url.scheme().eq("http") && !new_feed.url.scheme().eq("https") {
+        return Err(Status::UnprocessableEntity);
+    }
+
     let feed = Feed::new(new_feed.name, new_feed.url, user.user_id);
 
     let query = sqlx::query(
@@ -54,7 +61,7 @@ pub async fn create_feed(
     )
         .bind(feed.feed_id.to_string())
         .bind(&feed.name)
-        .bind(&feed.url)
+        .bind(feed.url.as_str())
         .bind(feed.user_id.to_string())
         .bind(feed.added_time.to_rfc3339())
         .bind(feed.last_checked_time.to_rfc3339())
@@ -106,15 +113,20 @@ pub async fn list_feeds(
 
     let feeds = query
         .into_iter()
-        .map(|row| Feed {
+        .map(|row| {
-            feed_id: Uuid::parse_str(&row.feed_id).unwrap(),
+            // Parse URL from string
-            name: row.name,
+            let url = Url::parse(&row.url).map_err(|_| Status::InternalServerError)?;
-            url: row.url,
+            
-            user_id: Uuid::parse_str(&row.user_id).unwrap(),
+            Ok(Feed {
-            added_time: row.added_time,
+                feed_id: Uuid::parse_str(&row.feed_id).unwrap(),
-            last_checked_time: row.last_checked_time,
+                name: row.name,
+                url,
+                user_id: Uuid::parse_str(&row.user_id).unwrap(),
+                added_time: row.added_time,
+                last_checked_time: row.last_checked_time,
+            })
         })
-        .collect();
+        .collect::<Result<Vec<_>, Status>>()?;
 
     Ok(Json(feeds))
 }