1use chrono::{DateTime, Duration, Utc};
12use serde::{Deserialize, Serialize};
13use std::collections::HashMap;
14use uuid::Uuid;
15
16#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
18#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
19#[serde(rename_all = "lowercase")]
20pub enum ReviewFrequency {
21 Monthly,
23 Quarterly,
25 Annually,
27}
28
29impl ReviewFrequency {
30 pub fn duration(&self) -> Duration {
32 match self {
33 ReviewFrequency::Monthly => Duration::days(30),
34 ReviewFrequency::Quarterly => Duration::days(90),
35 ReviewFrequency::Annually => Duration::days(365),
36 }
37 }
38
39 pub fn next_review_date(&self, from: DateTime<Utc>) -> DateTime<Utc> {
41 from + self.duration()
42 }
43}
44
45#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
47#[serde(rename_all = "lowercase")]
48pub enum ReviewStatus {
49 Pending,
51 InProgress,
53 Completed,
55 Cancelled,
57}
58
59#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize)]
61#[serde(rename_all = "snake_case")]
62pub enum ReviewType {
63 UserAccess,
65 PrivilegedAccess,
67 ApiToken,
69 ResourceAccess,
71}
72
73#[derive(Debug, Clone, Serialize, Deserialize)]
75pub struct UserAccessInfo {
76 pub user_id: Uuid,
78 pub username: String,
80 pub email: String,
82 pub roles: Vec<String>,
84 pub permissions: Vec<String>,
86 pub last_login: Option<DateTime<Utc>>,
88 pub access_granted: DateTime<Utc>,
90 pub days_inactive: Option<u64>,
92 pub is_active: bool,
94}
95
96#[derive(Debug, Clone, Serialize, Deserialize)]
98pub struct PrivilegedAccessInfo {
99 pub user_id: Uuid,
101 pub username: String,
103 pub roles: Vec<String>,
105 pub mfa_enabled: bool,
107 pub justification: Option<String>,
109 pub justification_expires: Option<DateTime<Utc>>,
111 pub recent_actions_count: u64,
113 pub last_privileged_action: Option<DateTime<Utc>>,
115}
116
117#[derive(Debug, Clone, Serialize, Deserialize)]
119pub struct ApiTokenInfo {
120 pub token_id: String,
122 pub name: Option<String>,
124 pub owner_id: Uuid,
126 pub scopes: Vec<String>,
128 pub created_at: DateTime<Utc>,
130 pub last_used: Option<DateTime<Utc>>,
132 pub expires_at: Option<DateTime<Utc>>,
134 pub days_unused: Option<u64>,
136 pub is_active: bool,
138}
139
140#[derive(Debug, Clone, Serialize, Deserialize)]
142pub struct ResourceAccessInfo {
143 pub resource_type: String,
145 pub resource_id: String,
147 pub users_with_access: Vec<Uuid>,
149 pub access_levels: HashMap<Uuid, String>,
151 pub last_access: HashMap<Uuid, Option<DateTime<Utc>>>,
153}
154
155#[derive(Debug, Clone, Serialize, Deserialize)]
157pub struct ReviewFindings {
158 pub inactive_users: u32,
160 pub excessive_permissions: u32,
162 pub no_recent_access: u32,
164 pub privileged_without_mfa: u32,
166 pub unused_tokens: u32,
168 pub excessive_scopes: u32,
170 pub expiring_soon: u32,
172 pub custom: HashMap<String, u32>,
174}
175
176#[derive(Debug, Clone, Serialize, Deserialize)]
178pub struct ReviewActions {
179 pub users_revoked: u32,
181 pub permissions_reduced: u32,
183 pub mfa_enforced: u32,
185 pub tokens_revoked: u32,
187 pub tokens_rotated: u32,
189 pub scopes_reduced: u32,
191 pub custom: HashMap<String, u32>,
193}
194
195#[derive(Debug, Clone, Serialize, Deserialize)]
197pub struct AccessReview {
198 pub review_id: String,
200 pub review_type: ReviewType,
202 pub status: ReviewStatus,
204 pub review_date: DateTime<Utc>,
206 pub due_date: DateTime<Utc>,
208 pub total_items: u32,
210 pub items_reviewed: u32,
212 pub findings: ReviewFindings,
214 pub actions_taken: ReviewActions,
216 pub pending_approvals: u32,
218 pub next_review_date: DateTime<Utc>,
220 pub metadata: HashMap<String, serde_json::Value>,
222}
223
224#[derive(Debug, Clone, Serialize, Deserialize)]
226pub struct UserReviewItem {
227 pub review_id: String,
229 pub user_id: Uuid,
231 pub access_info: UserAccessInfo,
233 pub status: String,
235 pub manager_id: Option<Uuid>,
237 pub approval_deadline: Option<DateTime<Utc>>,
239 pub approved_by: Option<Uuid>,
241 pub approved_at: Option<DateTime<Utc>>,
243 pub rejection_reason: Option<String>,
245}
246
247#[derive(Debug, Clone, Serialize, Deserialize)]
249#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
250pub struct AccessReviewConfig {
251 pub enabled: bool,
253 pub user_review: UserReviewConfig,
255 pub privileged_review: PrivilegedReviewConfig,
257 pub token_review: TokenReviewConfig,
259 pub resource_review: ResourceReviewConfig,
261 pub notifications: NotificationConfig,
263}
264
265#[derive(Debug, Clone, Serialize, Deserialize)]
267#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
268pub struct UserReviewConfig {
269 pub enabled: bool,
271 pub frequency: ReviewFrequency,
273 pub inactive_threshold_days: u64,
275 pub auto_revoke_inactive: bool,
277 pub require_manager_approval: bool,
279 pub approval_timeout_days: u64,
281}
282
283#[derive(Debug, Clone, Serialize, Deserialize)]
285#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
286pub struct PrivilegedReviewConfig {
287 pub enabled: bool,
289 pub frequency: ReviewFrequency,
291 pub require_mfa: bool,
293 pub require_justification: bool,
295 pub alert_on_escalation: bool,
297}
298
299#[derive(Debug, Clone, Serialize, Deserialize)]
301#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
302pub struct TokenReviewConfig {
303 pub enabled: bool,
305 pub frequency: ReviewFrequency,
307 pub unused_threshold_days: u64,
309 pub auto_revoke_unused: bool,
311 pub rotation_threshold_days: u64,
313}
314
315#[derive(Debug, Clone, Serialize, Deserialize)]
317#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
318pub struct ResourceReviewConfig {
319 pub enabled: bool,
321 pub frequency: ReviewFrequency,
323 pub sensitive_resources: Vec<String>,
325}
326
327#[derive(Debug, Clone, Serialize, Deserialize)]
329#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
330pub struct NotificationConfig {
331 pub enabled: bool,
333 pub channels: Vec<String>,
335 pub recipients: Vec<String>,
337}
338
339impl Default for AccessReviewConfig {
340 fn default() -> Self {
341 Self {
342 enabled: false,
343 user_review: UserReviewConfig {
344 enabled: true,
345 frequency: ReviewFrequency::Quarterly,
346 inactive_threshold_days: 90,
347 auto_revoke_inactive: true,
348 require_manager_approval: true,
349 approval_timeout_days: 30,
350 },
351 privileged_review: PrivilegedReviewConfig {
352 enabled: true,
353 frequency: ReviewFrequency::Monthly,
354 require_mfa: true,
355 require_justification: true,
356 alert_on_escalation: true,
357 },
358 token_review: TokenReviewConfig {
359 enabled: true,
360 frequency: ReviewFrequency::Monthly,
361 unused_threshold_days: 90,
362 auto_revoke_unused: true,
363 rotation_threshold_days: 30,
364 },
365 resource_review: ResourceReviewConfig {
366 enabled: true,
367 frequency: ReviewFrequency::Quarterly,
368 sensitive_resources: vec![
369 "billing".to_string(),
370 "user_data".to_string(),
371 "audit_logs".to_string(),
372 "security_settings".to_string(),
373 ],
374 },
375 notifications: NotificationConfig {
376 enabled: true,
377 channels: vec!["email".to_string()],
378 recipients: vec!["security_team".to_string(), "compliance_team".to_string()],
379 },
380 }
381 }
382}
383
384pub struct AccessReviewEngine {
390 config: AccessReviewConfig,
391 active_reviews: HashMap<String, AccessReview>,
393 user_review_items: HashMap<String, HashMap<Uuid, UserReviewItem>>,
395}
396
397impl AccessReviewEngine {
398 pub fn new(config: AccessReviewConfig) -> Self {
400 Self {
401 config,
402 active_reviews: HashMap::new(),
403 user_review_items: HashMap::new(),
404 }
405 }
406
407 pub fn generate_review_id(&self, review_type: ReviewType, date: DateTime<Utc>) -> String {
409 let type_str = match review_type {
410 ReviewType::UserAccess => "user",
411 ReviewType::PrivilegedAccess => "privileged",
412 ReviewType::ApiToken => "token",
413 ReviewType::ResourceAccess => "resource",
414 };
415
416 let date_str = date.format("%Y-%m-%d");
417 format!("review-{}-{}", date_str, type_str)
418 }
419
420 pub async fn start_user_access_review(
425 &mut self,
426 users: Vec<UserAccessInfo>,
427 ) -> Result<AccessReview, crate::Error> {
428 if !self.config.enabled || !self.config.user_review.enabled {
429 return Err(crate::Error::Generic("User access review is not enabled".to_string()));
430 }
431
432 let now = Utc::now();
433 let review_id = self.generate_review_id(ReviewType::UserAccess, now);
434 let due_date =
435 now + chrono::Duration::days(self.config.user_review.approval_timeout_days as i64);
436 let next_review = self.config.user_review.frequency.next_review_date(now);
437
438 let mut findings = ReviewFindings {
440 inactive_users: 0,
441 excessive_permissions: 0,
442 no_recent_access: 0,
443 privileged_without_mfa: 0,
444 unused_tokens: 0,
445 excessive_scopes: 0,
446 expiring_soon: 0,
447 custom: HashMap::new(),
448 };
449
450 let mut review_items = HashMap::new();
451
452 for user in &users {
453 if let Some(days) = user.days_inactive {
455 if days > self.config.user_review.inactive_threshold_days {
456 findings.inactive_users += 1;
457 }
458 }
459
460 if user.last_login.is_none()
462 || user.last_login.unwrap() < now - chrono::Duration::days(90)
463 {
464 findings.no_recent_access += 1;
465 }
466
467 if user.permissions.len() > 10 {
469 findings.excessive_permissions += 1;
470 }
471
472 let review_item = UserReviewItem {
474 review_id: review_id.clone(),
475 user_id: user.user_id,
476 access_info: user.clone(),
477 status: "pending".to_string(),
478 manager_id: None, approval_deadline: Some(due_date),
480 approved_by: None,
481 approved_at: None,
482 rejection_reason: None,
483 };
484
485 review_items.insert(user.user_id, review_item);
486 }
487
488 let review = AccessReview {
489 review_id: review_id.clone(),
490 review_type: ReviewType::UserAccess,
491 status: ReviewStatus::InProgress,
492 review_date: now,
493 due_date,
494 total_items: users.len() as u32,
495 items_reviewed: 0,
496 findings: findings.clone(),
497 actions_taken: ReviewActions {
498 users_revoked: 0,
499 permissions_reduced: 0,
500 mfa_enforced: 0,
501 tokens_revoked: 0,
502 tokens_rotated: 0,
503 scopes_reduced: 0,
504 custom: HashMap::new(),
505 },
506 pending_approvals: review_items.len() as u32,
507 next_review_date: next_review,
508 metadata: HashMap::new(),
509 };
510
511 self.active_reviews.insert(review_id.clone(), review.clone());
512 self.user_review_items.insert(review_id, review_items);
513
514 Ok(review)
515 }
516
517 pub fn approve_user_access(
519 &mut self,
520 review_id: &str,
521 user_id: Uuid,
522 approved_by: Uuid,
523 justification: Option<String>,
524 ) -> Result<(), crate::Error> {
525 let review = self
526 .active_reviews
527 .get_mut(review_id)
528 .ok_or_else(|| crate::Error::Generic(format!("Review {} not found", review_id)))?;
529
530 let items = self.user_review_items.get_mut(review_id).ok_or_else(|| {
531 crate::Error::Generic(format!("Review items for {} not found", review_id))
532 })?;
533
534 let item = items.get_mut(&user_id).ok_or_else(|| {
535 crate::Error::Generic(format!("User {} not found in review", user_id))
536 })?;
537
538 item.status = "approved".to_string();
539 item.approved_by = Some(approved_by);
540 item.approved_at = Some(Utc::now());
541
542 review.items_reviewed += 1;
543 review.pending_approvals = review.pending_approvals.saturating_sub(1);
544
545 if let Some(just) = justification {
547 review
548 .metadata
549 .insert(format!("justification_{}", user_id), serde_json::json!(just));
550 }
551
552 Ok(())
553 }
554
555 pub fn revoke_user_access(
557 &mut self,
558 review_id: &str,
559 user_id: Uuid,
560 revoked_by: Uuid,
561 reason: String,
562 ) -> Result<(), crate::Error> {
563 let review = self
564 .active_reviews
565 .get_mut(review_id)
566 .ok_or_else(|| crate::Error::Generic(format!("Review {} not found", review_id)))?;
567
568 let items = self.user_review_items.get_mut(review_id).ok_or_else(|| {
569 crate::Error::Generic(format!("Review items for {} not found", review_id))
570 })?;
571
572 let item = items.get_mut(&user_id).ok_or_else(|| {
573 crate::Error::Generic(format!("User {} not found in review", user_id))
574 })?;
575
576 item.status = "revoked".to_string();
577 item.rejection_reason = Some(reason.clone());
578
579 review.items_reviewed += 1;
580 review.pending_approvals = review.pending_approvals.saturating_sub(1);
581 review.actions_taken.users_revoked += 1;
582
583 review
585 .metadata
586 .insert(format!("revocation_reason_{}", user_id), serde_json::json!(reason));
587
588 Ok(())
589 }
590
591 pub fn get_review(&self, review_id: &str) -> Option<&AccessReview> {
593 self.active_reviews.get(review_id)
594 }
595
596 pub fn get_all_reviews(&self) -> Vec<&AccessReview> {
598 self.active_reviews.values().collect()
599 }
600
601 pub fn get_review_items(&self, review_id: &str) -> Option<&HashMap<Uuid, UserReviewItem>> {
603 self.user_review_items.get(review_id)
604 }
605
606 pub fn check_auto_revocation(&mut self) -> Vec<(String, Uuid)> {
611 let now = Utc::now();
612 let mut revoked = Vec::new();
613
614 for (review_id, items) in &mut self.user_review_items {
615 let review = match self.active_reviews.get_mut(review_id) {
616 Some(r) => r,
617 None => continue,
618 };
619
620 if !self.config.user_review.auto_revoke_inactive {
621 continue;
622 }
623
624 for (user_id, item) in items.iter_mut() {
625 if item.status == "pending" {
626 if let Some(deadline) = item.approval_deadline {
627 if now > deadline {
628 item.status = "auto_revoked".to_string();
630 item.rejection_reason = Some(
631 "Access automatically revoked due to missing approval within deadline".to_string(),
632 );
633
634 review.items_reviewed += 1;
635 review.pending_approvals = review.pending_approvals.saturating_sub(1);
636 review.actions_taken.users_revoked += 1;
637
638 revoked.push((review_id.clone(), *user_id));
639 }
640 }
641 }
642 }
643 }
644
645 revoked
646 }
647
648 pub fn complete_review(&mut self, review_id: &str) -> Result<(), crate::Error> {
650 let review = self
651 .active_reviews
652 .get_mut(review_id)
653 .ok_or_else(|| crate::Error::Generic(format!("Review {} not found", review_id)))?;
654
655 review.status = ReviewStatus::Completed;
656
657 Ok(())
658 }
659}
660
661#[cfg(test)]
662mod tests {
663 use super::*;
664
665 #[test]
666 fn test_review_frequency_duration() {
667 assert_eq!(ReviewFrequency::Monthly.duration(), Duration::days(30));
668 assert_eq!(ReviewFrequency::Quarterly.duration(), Duration::days(90));
669 assert_eq!(ReviewFrequency::Annually.duration(), Duration::days(365));
670 }
671
672 #[test]
673 fn test_generate_review_id() {
674 let config = AccessReviewConfig::default();
675 let engine = AccessReviewEngine::new(config);
676 let date = Utc::now();
677 let id = engine.generate_review_id(ReviewType::UserAccess, date);
678 assert!(id.starts_with("review-"));
679 assert!(id.contains("user"));
680 }
681
682 #[tokio::test]
683 async fn test_start_user_access_review() {
684 let mut config = AccessReviewConfig::default();
685 config.enabled = true;
686 config.user_review.enabled = true;
687
688 let mut engine = AccessReviewEngine::new(config);
689
690 let users = vec![
691 UserAccessInfo {
692 user_id: Uuid::new_v4(),
693 username: "user1".to_string(),
694 email: "user1@example.com".to_string(),
695 roles: vec!["editor".to_string()],
696 permissions: vec!["read".to_string(), "write".to_string()],
697 last_login: Some(Utc::now() - Duration::days(10)),
698 access_granted: Utc::now() - Duration::days(100),
699 days_inactive: Some(10),
700 is_active: true,
701 },
702 UserAccessInfo {
703 user_id: Uuid::new_v4(),
704 username: "user2".to_string(),
705 email: "user2@example.com".to_string(),
706 roles: vec!["admin".to_string()],
707 permissions: (0..15).map(|i| format!("perm{}", i)).collect(),
708 last_login: Some(Utc::now() - Duration::days(120)),
709 access_granted: Utc::now() - Duration::days(200),
710 days_inactive: Some(120),
711 is_active: true,
712 },
713 ];
714
715 let review = engine.start_user_access_review(users).await.unwrap();
716 assert_eq!(review.review_type, ReviewType::UserAccess);
717 assert_eq!(review.total_items, 2);
718 assert!(review.findings.inactive_users > 0);
719 assert!(review.findings.excessive_permissions > 0);
720 }
721
722 #[test]
723 fn test_approve_user_access() {
724 let mut config = AccessReviewConfig::default();
725 config.enabled = true;
726 config.user_review.enabled = true;
727
728 let mut engine = AccessReviewEngine::new(config);
729
730 let user = UserAccessInfo {
731 user_id: Uuid::new_v4(),
732 username: "user1".to_string(),
733 email: "user1@example.com".to_string(),
734 roles: vec!["editor".to_string()],
735 permissions: vec!["read".to_string()],
736 last_login: Some(Utc::now()),
737 access_granted: Utc::now() - Duration::days(10),
738 days_inactive: Some(0),
739 is_active: true,
740 };
741
742 let review =
744 futures::executor::block_on(engine.start_user_access_review(vec![user.clone()]))
745 .unwrap();
746 let review_id = review.review_id.clone();
747
748 let approver_id = Uuid::new_v4();
750 engine.approve_user_access(&review_id, user.user_id, approver_id, None).unwrap();
751
752 let review = engine.get_review(&review_id).unwrap();
753 assert_eq!(review.items_reviewed, 1);
754 assert_eq!(review.pending_approvals, 0);
755 }
756
757 #[test]
758 fn test_revoke_user_access() {
759 let mut config = AccessReviewConfig::default();
760 config.enabled = true;
761 config.user_review.enabled = true;
762
763 let mut engine = AccessReviewEngine::new(config);
764
765 let user = UserAccessInfo {
766 user_id: Uuid::new_v4(),
767 username: "user1".to_string(),
768 email: "user1@example.com".to_string(),
769 roles: vec!["editor".to_string()],
770 permissions: vec!["read".to_string()],
771 last_login: Some(Utc::now()),
772 access_granted: Utc::now() - Duration::days(10),
773 days_inactive: Some(0),
774 is_active: true,
775 };
776
777 let review =
779 futures::executor::block_on(engine.start_user_access_review(vec![user.clone()]))
780 .unwrap();
781 let review_id = review.review_id.clone();
782
783 let revoker_id = Uuid::new_v4();
785 engine
786 .revoke_user_access(
787 &review_id,
788 user.user_id,
789 revoker_id,
790 "No longer needed".to_string(),
791 )
792 .unwrap();
793
794 let review = engine.get_review(&review_id).unwrap();
795 assert_eq!(review.actions_taken.users_revoked, 1);
796 }
797}