085d41b121
This PR adds a new `billing_subscriptions` table to the database, as well as some accompanying models/queries. In this table we store a minimal amount of data from Stripe: - The Stripe customer ID - The Stripe subscription ID - The status of the Stripe subscription This should be enough for interactions with the Stripe API (e.g., to [create a customer portal session](https://docs.stripe.com/api/customer_portal/sessions/create)), as well as determine whether a subscription is active (based on the `status`). Release Notes: - N/A
86 lines
2.2 KiB
Rust
86 lines
2.2 KiB
Rust
use crate::db::UserId;
|
|
use sea_orm::entity::prelude::*;
|
|
use serde::Serialize;
|
|
|
|
/// A user model.
|
|
#[derive(Clone, Debug, Default, PartialEq, Eq, DeriveEntityModel, Serialize)]
|
|
#[sea_orm(table_name = "users")]
|
|
pub struct Model {
|
|
#[sea_orm(primary_key)]
|
|
pub id: UserId,
|
|
pub github_login: String,
|
|
pub github_user_id: Option<i32>,
|
|
pub email_address: Option<String>,
|
|
pub admin: bool,
|
|
pub invite_code: Option<String>,
|
|
pub invite_count: i32,
|
|
pub inviter_id: Option<UserId>,
|
|
pub connected_once: bool,
|
|
pub metrics_id: Uuid,
|
|
pub created_at: DateTime,
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
pub enum Relation {
|
|
#[sea_orm(has_many = "super::access_token::Entity")]
|
|
AccessToken,
|
|
#[sea_orm(has_many = "super::billing_subscription::Entity")]
|
|
BillingSubscription,
|
|
#[sea_orm(has_one = "super::room_participant::Entity")]
|
|
RoomParticipant,
|
|
#[sea_orm(has_many = "super::project::Entity")]
|
|
HostedProjects,
|
|
#[sea_orm(has_many = "super::channel_member::Entity")]
|
|
ChannelMemberships,
|
|
#[sea_orm(has_many = "super::user_feature::Entity")]
|
|
UserFeatures,
|
|
#[sea_orm(has_one = "super::contributor::Entity")]
|
|
Contributor,
|
|
}
|
|
|
|
impl Related<super::access_token::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::AccessToken.def()
|
|
}
|
|
}
|
|
|
|
impl Related<super::room_participant::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::RoomParticipant.def()
|
|
}
|
|
}
|
|
|
|
impl Related<super::project::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::HostedProjects.def()
|
|
}
|
|
}
|
|
|
|
impl Related<super::channel_member::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::ChannelMemberships.def()
|
|
}
|
|
}
|
|
|
|
impl Related<super::user_feature::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::UserFeatures.def()
|
|
}
|
|
}
|
|
|
|
impl ActiveModelBehavior for ActiveModel {}
|
|
|
|
pub struct UserFlags;
|
|
|
|
impl Linked for UserFlags {
|
|
type FromEntity = Entity;
|
|
|
|
type ToEntity = super::feature_flag::Entity;
|
|
|
|
fn link(&self) -> Vec<RelationDef> {
|
|
vec![
|
|
super::user_feature::Relation::User.def().rev(),
|
|
super::user_feature::Relation::Flag.def(),
|
|
]
|
|
}
|
|
}
|