camerin
/
request-mirror
mirror of https://github.com/RaspberryProgramming/request-mirror.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed cookie bug 

When connecting for the first time, if a request uses the cookie jar the cookie may be pending.
Made a new function that will account for that and return the pending cookie if it exists. Otherwise it will return the cookie using the .get function of the cookiejar.
This commit is contained in:
RaspberryProgramming 2024-07-19 14:20:02 +00:00
parent 8eccd1ed24
commit 355066b197
2 changed files with 58 additions and 54 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/lib.rs
View File

@ -3,6 +3,7 @@ pub mod schema;
use chrono::offset::Utc;
use diesel::prelude::*;
use rocket::http::CookieJar;
use std::env;
use dotenvy::dotenv;
use models::{
@ -129,3 +130,12 @@ pub fn ownership_exists(client_id: &str, owner_id: &str, connection: &mut PgConn
ownerships.len() > 0
}
/// Looks for a cookie in the cookiejar. If it is a pending cookie, the pending will be returned as a String.
/// This accounts for if this client hasn't connected before and is making its first request
pub fn get_cookie(key: &str, cookies: &CookieJar<'_>) -> String {
match cookies.get_pending(key) {
Some(v) => v.value().to_string(),
None => cookies.get(key).unwrap().value().to_string()
}
}

102
src/main.rs
View File

@ -152,30 +152,27 @@ fn index(_info: RequestInfo) -> Template {
#[get("/test")]
fn test_get(request: RequestInfo, cookies: &CookieJar<'_>) -> Template {
let client_id = cookies.get("mirror-id");
let client_id = get_cookie("mirror-id", cookies);
// If the cookie exists, create new database records for this request
if client_id.is_some() {
let connection = &mut establish_connection();
// create new database records for this request
let connection = &mut establish_connection();
// Create new history record and retrieve new history_id
let history_id = create_history_record(connection, client_id.unwrap().value(), "Get");
// Create new history record and retrieve new history_id
let history_id = create_history_record(connection, &client_id, "Get");
// Create header pair records
for row in &request.header {
create_pair_record(connection, history_id, PairType::Header, &row.key, &row.value);
}
// Create cookie pair records
for row in &request.cookies {
create_pair_record(connection, history_id, PairType::Cookie, &row.key, &row.value);
}
// Create query pair records
for row in &request.query {
create_pair_record(connection, history_id, PairType::Query, &row.key, &row.value);
}
// Create header pair records
for row in &request.header {
create_pair_record(connection, history_id, PairType::Header, &row.key, &row.value);
}
// Create cookie pair records
for row in &request.cookies {
create_pair_record(connection, history_id, PairType::Cookie, &row.key, &row.value);
}
// Create query pair records
for row in &request.query {
create_pair_record(connection, history_id, PairType::Query, &row.key, &row.value);
}
// Define context for the template
@ -204,36 +201,33 @@ fn test_get(request: RequestInfo, cookies: &CookieJar<'_>) -> Template {
#[post("/test", data = "<body>")]
fn test_post(body: RequestBody, request: RequestInfo, cookies: &CookieJar<'_>) -> Template {
let client_id = cookies.get("mirror-id");
let client_id = get_cookie("mirror-id", cookies);
// If the cookie exists, create new database records for this request
if client_id.is_some() {
let connection = &mut establish_connection();
// Create new history record and retrieve new history_id
let history_id = create_history_record(connection, client_id.unwrap().value(), "Post");
// Create header pair records
for row in &request.header {
create_pair_record(connection, history_id, PairType::Header, &row.key, &row.value);
}
// Create cookie pair records
for row in &request.cookies {
create_pair_record(connection, history_id, PairType::Cookie, &row.key, &row.value);
}
// Create query pair records
for row in &request.query {
create_pair_record(connection, history_id, PairType::Query, &row.key, &row.value);
}
// Create a pair record for body of the request
println!("Creating body Records for {history_id}");
create_pair_record(connection, history_id, PairType::Body, "body", &body.0.clone());
// Create new database records for this request
let connection = &mut establish_connection();
// Create new history record and retrieve new history_id
let history_id = create_history_record(connection, &client_id, "Post");
// Create header pair records
for row in &request.header {
create_pair_record(connection, history_id, PairType::Header, &row.key, &row.value);
}
// Create cookie pair records
for row in &request.cookies {
create_pair_record(connection, history_id, PairType::Cookie, &row.key, &row.value);
}
// Create query pair records
for row in &request.query {
create_pair_record(connection, history_id, PairType::Query, &row.key, &row.value);
}
// Create a pair record for body of the request
println!("Creating body Records for {history_id}");
create_pair_record(connection, history_id, PairType::Body, "body", &body.0.clone());
// Define context for the template
#[derive(Serialize)]
struct Context<'a> {
@ -260,10 +254,10 @@ fn test_post(body: RequestBody, request: RequestInfo, cookies: &CookieJar<'_>) -
/// Request function that returns a history of requests that the current client has made
/// The user can click a history_id and view the request itself
#[get("/history")]
fn history_req(cookies: &CookieJar<'_>) -> Template {
fn history_req(_info: RequestInfo, cookies: &CookieJar<'_>) -> Template {
// Get the client_id from the cookies
let client_id = cookies.get("mirror-id").unwrap().value();
let client_id = get_cookie("mirror-id", cookies);
let connection = &mut establish_connection();
@ -275,7 +269,7 @@ fn history_req(cookies: &CookieJar<'_>) -> Template {
.expect("Error loading clients");
// Get ownership relationships
let ownerships: Vec<Ownership> = get_ownerships(client_id, connection);
let ownerships: Vec<Ownership> = get_ownerships(&client_id, connection);
// Add any records that owned clients have
for ownership in ownerships {
@ -325,14 +319,14 @@ fn history_req(cookies: &CookieJar<'_>) -> Template {
/// request that was recorded to the database.
/// This includes the body of a post request, headers, cookies and query parameters.
#[get("/history/<history_id>")]
fn history_details(history_id: i64, cookies: &CookieJar<'_>) -> Template {
fn history_details(history_id: i64, _info: RequestInfo, cookies: &CookieJar<'_>) -> Template {
let client_id = cookies.get("mirror-id").unwrap().value();
let client_id = get_cookie("mirror-id", cookies);
let connection: &mut PgConnection = &mut establish_connection();
// Get owned client ids
let owned_clients: Vec<String> = get_ownerships(client_id, connection)
let owned_clients: Vec<String> = get_ownerships(&client_id, connection)
.iter()
.map(|x|x.client_id.to_string())
.collect::<Vec<String>>();
@ -415,7 +409,7 @@ fn history_details(history_id: i64, cookies: &CookieJar<'_>) -> Template {
#[get("/ownership_registration")]
fn ownership_registration(info: RequestInfo, cookies: &CookieJar<'_>) -> Template {
let client_id = cookies.get("mirror-id").unwrap().value().to_string();
let client_id = get_cookie("mirror-id", cookies);
let owner_id_pair = info.find_query_key("owner_id");
let mut disp_owner_reg = false;
let re = Regex::new(r"^[{]?[0-9a-fA-F]{8}-([0-9a-fA-F]{4}-){3}[0-9a-fA-F]{12}[}]?$").unwrap();