This commit is contained in:
2025-11-08 15:06:30 +08:00
parent 27614f695f
commit e28e885e66
14 changed files with 99 additions and 78 deletions

View File

@@ -78,9 +78,9 @@ pub(crate) async fn create_logic(domain: Domain, user_id: i32, app: &AppState) -
}
let existing_count: i64 =
sqlx::query_scalar("SELECT COUNT(*) FROM domains WHERE name = $1 AND tld = $2")
.bind(&domain.name)
.bind(&domain.tld)
sqlx::query_scalar("SELECT COUNT(*) FROM domains WHERE name = ? AND tld = ?")
.bind(&domain.name)
.bind(&domain.tld)
.fetch_one(&app.db)
.await
.map_err(|_| GurtError::invalid_message("Database error"))?;
@@ -89,7 +89,7 @@ pub(crate) async fn create_logic(domain: Domain, user_id: i32, app: &AppState) -
return Err(GurtError::invalid_message("Domain already exists"));
}
let user: (String,) = sqlx::query_as("SELECT username FROM users WHERE id = $1")
let user: (String,) = sqlx::query_as("SELECT username FROM users WHERE id = ?")
.bind(user_id)
.fetch_one(&app.db)
.await
@@ -98,7 +98,7 @@ pub(crate) async fn create_logic(domain: Domain, user_id: i32, app: &AppState) -
let username = user.0;
let domain_row: (i32,) = sqlx::query_as(
"INSERT INTO domains (name, tld, user_id, status) VALUES ($1, $2, $3, 'pending') RETURNING id"
"INSERT INTO domains (name, tld, user_id, status) VALUES (?, ?, ?, 'pending') RETURNING id"
)
.bind(&domain.name)
.bind(&domain.tld)
@@ -110,7 +110,7 @@ pub(crate) async fn create_logic(domain: Domain, user_id: i32, app: &AppState) -
let domain_id = domain_row.0;
let affected_rows = sqlx::query(
"UPDATE users SET registrations_remaining = registrations_remaining - 1 WHERE id = $1 AND registrations_remaining > 0",
"UPDATE users SET registrations_remaining = registrations_remaining - 1 WHERE id = ? AND registrations_remaining > 0",
)
.bind(user_id)
.execute(&app.db)
@@ -119,8 +119,8 @@ pub(crate) async fn create_logic(domain: Domain, user_id: i32, app: &AppState) -
.rows_affected();
if affected_rows == 0 {
sqlx::query("DELETE FROM domains WHERE id = $1")
.bind(domain_id)
sqlx::query("DELETE FROM domains WHERE id = ?")
.bind(domain_id)
.execute(&app.db)
.await
.map_err(|_| GurtError::invalid_message("Database cleanup error"))?;
@@ -321,7 +321,7 @@ pub(crate) async fn delete_domain(
return Ok(GurtResponse::not_found().with_string_body("Domain not found or access denied"));
}
sqlx::query("DELETE FROM domains WHERE name = $1 AND tld = $2 AND user_id = $3")
sqlx::query("DELETE FROM domains WHERE name = ? AND tld = ? AND user_id = ?")
.bind(name)
.bind(tld)
.bind(claims.user_id)
@@ -620,7 +620,7 @@ pub(crate) async fn delete_domain_record(
}
};
let rows_affected = sqlx::query("DELETE FROM dns_records WHERE id = $1 AND domain_id = $2")
let rows_affected = sqlx::query("DELETE FROM dns_records WHERE id = ? AND domain_id = ?")
.bind(record_id)
.bind(domain.id.unwrap())
.execute(&app_state.db)
@@ -1066,7 +1066,7 @@ pub(crate) async fn get_certificate(
});
// Delete the challenge as it's completed
sqlx::query("DELETE FROM certificate_challenges WHERE token = $1")
sqlx::query("DELETE FROM certificate_challenges WHERE token = ?")
.bind(token)
.execute(&app_state.db)
.await