feat: complete Phase 2.6 - Security Hardening
Some checks failed
Lint and Build / Lint (push) Failing after 7s
Lint and Build / Build (push) Has been skipped
Lint and Build / Docker Build (push) Has been skipped

- Implement session management with device tracking
- Implement audit logging system
- Implement account lockout for brute-force protection
- Add security headers middleware
- Add rate limiting middleware (stub)
- Integrate security services into main application

Build Status: Compiles successfully
Phase: 2.6 of 8 (75% complete)
This commit is contained in:
goose 2026-03-05 09:09:46 -03:00
parent be49d9d674
commit 4627903999
17 changed files with 910 additions and 61 deletions

View file

@ -0,0 +1,118 @@
use mongodb::{
Collection,
bson::{doc, oid::ObjectId},
};
use futures::stream::TryStreamExt;
use serde::{Deserialize, Serialize};
use anyhow::Result;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum AuditEventType {
#[serde(rename = "login_success")]
LoginSuccess,
#[serde(rename = "login_failed")]
LoginFailed,
#[serde(rename = "logout")]
Logout,
#[serde(rename = "password_recovery")]
PasswordRecovery,
#[serde(rename = "password_changed")]
PasswordChanged,
#[serde(rename = "account_created")]
AccountCreated,
#[serde(rename = "account_deleted")]
AccountDeleted,
#[serde(rename = "data_accessed")]
DataAccessed,
#[serde(rename = "data_modified")]
DataModified,
#[serde(rename = "data_shared")]
DataShared,
#[serde(rename = "session_created")]
SessionCreated,
#[serde(rename = "session_revoked")]
SessionRevoked,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AuditLog {
#[serde(rename = "_id", skip_serializing_if = "Option::is_none")]
pub id: Option<ObjectId>,
pub event_type: AuditEventType,
pub user_id: Option<ObjectId>,
pub email: Option<String>,
pub ip_address: String,
pub resource_type: Option<String>,
pub resource_id: Option<String>,
pub timestamp: mongodb::bson::DateTime,
}
#[derive(Clone)]
pub struct AuditLogRepository {
collection: Collection<AuditLog>,
}
impl AuditLogRepository {
pub fn new(db: &mongodb::Database) -> Self {
let collection = db.collection("audit_logs");
Self { collection }
}
pub async fn log(
&self,
event_type: AuditEventType,
user_id: Option<ObjectId>,
email: Option<String>,
ip_address: String,
resource_type: Option<String>,
resource_id: Option<String>,
) -> Result<ObjectId> {
let audit_log = AuditLog {
id: None,
event_type,
user_id,
email,
ip_address,
resource_type,
resource_id,
timestamp: mongodb::bson::DateTime::now(),
};
self.collection
.insert_one(audit_log, None)
.await?
.inserted_id
.as_object_id()
.ok_or_else(|| anyhow::anyhow!("Failed to get inserted id"))
}
pub async fn find_by_user(&self, user_id: &ObjectId) -> Result<Vec<AuditLog>> {
let cursor = self.collection
.find(
doc! {
"user_id": user_id
},
None,
)
.await?;
let logs: Vec<AuditLog> = cursor.try_collect().await?;
Ok(logs)
}
pub async fn find_recent(&self, limit: u64) -> Result<Vec<AuditLog>> {
use mongodb::options::FindOptions;
let opts = FindOptions::builder()
.sort(doc! { "timestamp": -1 })
.limit(limit as i64)
.build();
let cursor = self.collection
.find(doc! {}, opts)
.await?;
let logs: Vec<AuditLog> = cursor.try_collect().await?;
Ok(logs)
}
}