feat: mega bot identity (#2004)

* feat: add bots related model

* feat: add ActorType enum in AuditLogs

* feat: add bot_keys table

* feat: change Option<String> to String

* fix: code review mentioned problem

* fix: clippy error

Author: ztygod

Cargo.toml

@@ -79,6 +79,7 @@ chrono = { version = "0.4.43", features = ["serde"] }
 hex = "0.4.3"
 sha1 = "0.10"
 sha2 = "0.10.9"
+rsa = "0.9.10"
 
 idgenerator = "2.0.0"
 config = "0.15.19"

jupiter/Cargo.toml

@@ -49,6 +49,8 @@ rustls = { workspace = true }
 sha1 = { workspace = true }
 sha2 = { workspace = true }
 hex = { workspace = true }
+rand = { workspace = true }
+rsa = { workspace = true }
 
 [dev-dependencies]
 redis-test = { workspace = true, features = ["aio"] }

jupiter/callisto/src/audit_logs.rs

@@ -0,0 +1,25 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+use super::sea_orm_active_enums::{ActorTypeEnum, AuditActionEnum, TargetTypeEnum};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "audit_logs")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: i64,
+    pub actor_id: i64,
+    pub actor_type: ActorTypeEnum,
+    pub action: AuditActionEnum,
+    pub target_type: TargetTypeEnum,
+    pub target_id: i64,
+    pub metadata: Option<Json>,
+    pub created_at: DateTimeWithTimeZone,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {}
+
+impl ActiveModelBehavior for ActiveModel {}

jupiter/callisto/src/bot_installations.rs

@@ -0,0 +1,39 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+use super::sea_orm_active_enums::{InstallationBotStatusEnum, InstallationTargetTypeEnum};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "bot_installations")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: i64,
+    pub bot_id: i64,
+    pub target_type: InstallationTargetTypeEnum,
+    pub target_id: i64,
+    pub status: InstallationBotStatusEnum,
+    pub installed_by: i64,
+    pub installed_at: DateTimeWithTimeZone,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::bots::Entity",
+        from = "Column::BotId",
+        to = "super::bots::Column::Id",
+        on_update = "Cascade",
+        on_delete = "Cascade"
+    )]
+    Bots,
+}
+
+impl Related<super::bots::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Bots.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

jupiter/callisto/src/bot_keys.rs

@@ -0,0 +1,35 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "bot_keys")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: i64,
+    pub bot_id: i64,
+    pub private_key: String,
+    pub public_key: String,
+    pub created_at: DateTimeWithTimeZone,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::bots::Entity",
+        from = "Column::BotId",
+        to = "super::bots::Column::Id",
+        on_update = "NoAction",
+        on_delete = "Cascade"
+    )]
+    Bots,
+}
+
+impl Related<super::bots::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Bots.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

jupiter/callisto/src/bot_tokens.rs

@@ -0,0 +1,37 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "bot_tokens")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: i64,
+    pub bot_id: i64,
+    pub token_hash: String,
+    pub token_name: String,
+    pub expires_at: Option<DateTimeWithTimeZone>,
+    pub created_at: DateTimeWithTimeZone,
+    pub revoked: bool,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(
+        belongs_to = "super::bots::Entity",
+        from = "Column::BotId",
+        to = "super::bots::Column::Id",
+        on_update = "Cascade",
+        on_delete = "Cascade"
+    )]
+    Bots,
+}
+
+impl Related<super::bots::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::Bots.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

jupiter/callisto/src/bots.rs

@@ -0,0 +1,50 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.19
+
+use sea_orm::entity::prelude::*;
+use serde::{Deserialize, Serialize};
+
+use super::sea_orm_active_enums::{BotStatusEnum, PermissionScopeEnum};
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq, Serialize, Deserialize)]
+#[sea_orm(table_name = "bots")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: i64,
+    pub name: String,
+    pub organization_id: Option<i64>,
+    pub creator_user_id: i64,
+    pub permission_scope: PermissionScopeEnum,
+    pub status: BotStatusEnum,
+    pub created_at: DateTimeWithTimeZone,
+    pub updated_at: DateTimeWithTimeZone,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {
+    #[sea_orm(has_many = "super::bot_installations::Entity")]
+    BotInstallations,
+    #[sea_orm(has_many = "super::bot_keys::Entity")]
+    BotKeys,
+    #[sea_orm(has_many = "super::bot_tokens::Entity")]
+    BotTokens,
+}
+
+impl Related<super::bot_installations::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::BotInstallations.def()
+    }
+}
+
+impl Related<super::bot_keys::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::BotKeys.def()
+    }
+}
+
+impl Related<super::bot_tokens::Entity> for Entity {
+    fn to() -> RelationDef {
+        Relation::BotTokens.def()
+    }
+}
+
+impl ActiveModelBehavior for ActiveModel {}

jupiter/callisto/src/entity_ext/bot_keys.rs

@@ -0,0 +1,17 @@
+use chrono::Utc;
+
+use crate::{bot_keys, entity_ext::generate_id};
+
+impl bot_keys::Model {
+    pub fn new(bot_id: i64, private_key: String, public_key: String) -> Self {
+        let now = Utc::now().into();
+
+        Self {
+            id: generate_id(),
+            bot_id,
+            private_key,
+            public_key,
+            created_at: now,
+        }
+    }
+}

jupiter/callisto/src/entity_ext/bots.rs

@@ -0,0 +1,30 @@
+use chrono::Utc;
+
+use crate::{
+    bots,
+    entity_ext::generate_id,
+    sea_orm_active_enums::{BotStatusEnum, PermissionScopeEnum},
+};
+
+impl bots::Model {
+    pub fn new(
+        name: String,
+        organization_id: Option<i64>,
+        creator_user_id: i64,
+        permission_scope: PermissionScopeEnum,
+        status: BotStatusEnum,
+    ) -> Self {
+        let now = Utc::now().into();
+
+        Self {
+            id: generate_id(),
+            name,
+            organization_id,
+            creator_user_id,
+            permission_scope,
+            status,
+            created_at: now,
+            updated_at: now,
+        }
+    }
+}

jupiter/callisto/src/entity_ext/mod.rs

@@ -1,3 +1,5 @@
+pub mod bot_keys;
+pub mod bots;
 pub mod buck_session;
 pub mod buck_session_file;
 pub mod check_result;