add support for reading http cookies using the new sqlpage.cookie('co…

…okie_name') function
sqlpage · Jun 20, 2023 · de948fb · de948fb
1 parent 2ce4dbe
commit de948fb
Show file tree

Hide file tree

Showing 5 changed files with 207 additions and 94 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -19,7 +19,7 @@ codegen-units = 2
 [dependencies]
 sqlx = { version = "0.6", features = ["any", "runtime-actix-rustls", "sqlite", "postgres", "mysql", "chrono"] }
 chrono = "0.4.23"
-actix-web = { version = "4", features = ["rustls"] }
+actix-web = { version = "4", features = ["rustls", "cookies"] }
 percent-encoding = "2.2.0"
 handlebars = "5.0.0-beta.0"
 log = "0.4.17"

diff --git a/src/webserver/database/mod.rs b/src/webserver/database/mod.rs
@@ -151,6 +151,7 @@ fn bind_parameters<'a>(
                 .post_variables
                 .get(x)
                 .or_else(|| request.get_variables.get(x)),
+            StmtParam::Cookie(x) => request.cookies.get(x),
         };
         log::debug!("Binding value {:?} in statement {}", &argument, stmt);
         match argument {
@@ -293,10 +294,12 @@ impl Display for PreparedStatement {
     }
 }
 
+#[derive(Debug, PartialEq, Eq)]
 enum StmtParam {
     Get(String),
     Post(String),
     GetOrPost(String),
+    Cookie(String),
 }
 
 #[actix_web::test]