|
|
|
use crate::database::models::Role;
|
|
|
|
use crate::database::{DatabaseResult, PostgresPool, Table};
|
|
|
|
use crate::utils::error::DBError;
|
|
|
|
|
|
|
|
#[derive(Clone)]
|
|
|
|
pub struct UserRoles {
|
|
|
|
pool: PostgresPool,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Table for UserRoles {
|
|
|
|
fn new(pool: PostgresPool) -> Self {
|
|
|
|
Self { pool }
|
|
|
|
}
|
|
|
|
|
|
|
|
fn init(&self) -> DatabaseResult<()> {
|
|
|
|
self.pool
|
|
|
|
.get()?
|
|
|
|
.batch_execute(
|
|
|
|
"
|
|
|
|
CREATE TABLE IF NOT EXISTS user_roles (
|
|
|
|
user_id INT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
|
|
|
role_id INT NOT NULL REFERENCES roles(id) ON DELETE CASCADE,
|
|
|
|
PRIMARY KEY (user_id, role_id)
|
|
|
|
);",
|
|
|
|
)
|
|
|
|
.map_err(DBError::from)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl UserRoles {
|
|
|
|
pub fn by_user(&self, user_id: i32) -> DatabaseResult<Vec<Role>> {
|
|
|
|
let mut connection = self.pool.get()?;
|
|
|
|
let rows = connection.query(
|
|
|
|
"SELECT * FROM user_roles, roles WHERE user_id = $1 AND roles.id = user_roles.role_id",
|
|
|
|
&[&user_id],
|
|
|
|
)?;
|
|
|
|
|
|
|
|
serde_postgres::from_rows(&rows).map_err(DBError::from)
|
|
|
|
}
|
|
|
|
}
|