Jędrzej Stuczyński
117 lines
3.9 KiB
Rust
117 lines
3.9 KiB
Rust
// Copyright 2022-2024 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::models::StoredIssuedCredential;
|
|
|
|
#[derive(Clone)]
|
|
pub struct CoconutCredentialManager {
|
|
connection_pool: sqlx::SqlitePool,
|
|
}
|
|
|
|
impl CoconutCredentialManager {
|
|
/// Creates new instance of the `CoconutCredentialManager` with the provided sqlite connection pool.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `connection_pool`: database connection pool to use.
|
|
pub fn new(connection_pool: sqlx::SqlitePool) -> Self {
|
|
CoconutCredentialManager { connection_pool }
|
|
}
|
|
|
|
pub async fn insert_issued_credential(
|
|
&self,
|
|
credential_type: String,
|
|
serialization_revision: u8,
|
|
credential_data: &[u8],
|
|
epoch_id: u32,
|
|
) -> Result<(), sqlx::Error> {
|
|
sqlx::query!(
|
|
r#"
|
|
INSERT INTO coconut_credentials(serialization_revision, credential_type, credential_data, epoch_id, expired)
|
|
VALUES (?, ?, ?, ?, false)
|
|
"#,
|
|
serialization_revision, credential_type, credential_data, epoch_id
|
|
).execute(&self.connection_pool).await?;
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn get_next_unspect_freepass(
|
|
&self,
|
|
gateway_id: &str,
|
|
) -> Result<Option<StoredIssuedCredential>, sqlx::Error> {
|
|
// get a credential of freepass type that doesn't appear in `credential_usage` for the provided gateway_id
|
|
sqlx::query_as(
|
|
r#"
|
|
SELECT *
|
|
FROM coconut_credentials
|
|
WHERE coconut_credentials.credential_type == "FreeBandwidthPass" AND coconut_credentials.expired = false
|
|
AND NOT EXISTS (SELECT 1
|
|
FROM credential_usage
|
|
WHERE credential_usage.credential_id = coconut_credentials.id
|
|
AND credential_usage.gateway_id_bs58 == ?)
|
|
ORDER BY coconut_credentials.id
|
|
LIMIT 1
|
|
"#,
|
|
)
|
|
.bind(gateway_id)
|
|
.fetch_optional(&self.connection_pool)
|
|
.await
|
|
}
|
|
|
|
pub async fn get_next_unspect_bandwidth_voucher(
|
|
&self,
|
|
) -> Result<Option<StoredIssuedCredential>, sqlx::Error> {
|
|
// get a credential of bandwidth voucher type that doesn't appear in `credential_usage` for any gateway_id
|
|
sqlx::query_as(
|
|
r#"
|
|
SELECT *
|
|
FROM coconut_credentials
|
|
WHERE coconut_credentials.credential_type == "BandwidthVoucher"
|
|
AND NOT EXISTS (SELECT 1
|
|
FROM credential_usage
|
|
WHERE credential_usage.credential_id = coconut_credentials.id)
|
|
ORDER BY coconut_credentials.id
|
|
LIMIT 1
|
|
"#,
|
|
)
|
|
.fetch_optional(&self.connection_pool)
|
|
.await
|
|
}
|
|
|
|
/// Consumes in the database the specified credential.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `id`: Database id.
|
|
/// * `gateway_id`: id of the gateway that received the credential
|
|
pub async fn consume_coconut_credential(
|
|
&self,
|
|
id: i64,
|
|
gateway_id: &str,
|
|
) -> Result<(), sqlx::Error> {
|
|
sqlx::query!(
|
|
"INSERT INTO credential_usage (credential_id, gateway_id_bs58) VALUES (?, ?)",
|
|
id,
|
|
gateway_id
|
|
)
|
|
.execute(&self.connection_pool)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// Marks the specified credential as expired
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `id`: Id of the credential to mark as expired.
|
|
pub async fn mark_expired(&self, id: i64) -> Result<(), sqlx::Error> {
|
|
sqlx::query!(
|
|
"UPDATE coconut_credentials SET expired = TRUE WHERE id = ?",
|
|
id
|
|
)
|
|
.execute(&self.connection_pool)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
}
|