feat(license): add license expiration check and daily cron job

ee/tabby-schema/src/schema/license.rs

@@ -81,6 +81,14 @@
 
         Ok(())
     }
+
+    pub fn expire_in_days(&self) -> Option<i64> {
+        self.expires_at.map(|expires_at| {
+            let now = Utc::now();
+            let duration = expires_at.signed_duration_since(now);
+            duration.num_days()
+        })
+    }
 }
 
 #[async_trait]

ee/tabby-webserver/src/service/background_job/license_check.rs

@@ -0,0 +1,54 @@
+use std::sync::Arc;
+
+use chrono::{DateTime, Utc};
+use serde::{Deserialize, Serialize};
+use tabby_schema::{
+    context::ContextService,
+    license::{LicenseService, LicenseType},
+    notification::{NotificationRecipient, NotificationService},
+};
+
+use super::helper::Job;
+
+#[derive(Debug, Serialize, Deserialize, Clone)]
+pub struct LicenseCheckJob;
+
+impl Job for LicenseCheckJob {
+    const NAME: &'static str = "license_check";
+}
+
+impl LicenseCheckJob {
+    pub async fn cron(
+        _now: DateTime<Utc>,
+        license_service: Arc<dyn LicenseService>,
+        notification_service: Arc<dyn NotificationService>,
+    ) -> tabby_schema::Result<()> {
+        let license = license_service.read().await?;
+        if license.r#type == LicenseType::Community {
+            return Ok(());
+        }
+        if let Some(expire_in_days) = license.expire_in_days() {
+            if expire_in_days < 7 && expire_in_days > 0 {
+                notification_service
+                    .create(
+                        NotificationRecipient::Admin,
+                        &make_expring_message(expire_in_days),
+                    )
+                    .await?;
+            }
+        }
+        Ok(())
+    }
+}
+
+fn make_expring_message(expire_in_days: i64) -> String {
+    format!(
+        r#"Your license will expire in {} days.
+
+To ensure uninterrupted access to Tabby and to continue enjoying all the premium features, please renew your license before it expires.
+
+If you have any questions or need assistance with the renewal process, please don't hesitate to contact our support team at [email protected].
+"#,
+        expire_in_days
+    )
+}

ee/tabby-webserver/src/service/background_job/mod.rs

@@ -2,6 +2,7 @@
 mod git;
 mod helper;
 mod index_garbage_collection;
+mod license_check;
 mod third_party_integration;
 mod web_crawler;
 
@@ -13,6 +14,7 @@
 use helper::{CronStream, Job, JobLogger};
 use index_garbage_collection::IndexGarbageCollection;
 use juniper::ID;
+use license_check::LicenseCheckJob;
 use serde::{Deserialize, Serialize};
 use tabby_common::config::CodeRepository;
 use tabby_db::DbConn;
@@ -21,6 +23,8 @@
     context::ContextService,
     integration::IntegrationService,
     job::JobService,
+    license::LicenseService,
+    notification::NotificationService,
     repository::{GitRepositoryService, RepositoryService, ThirdPartyRepositoryService},
 };
 use third_party_integration::SchedulerGithubGitlabJob;
@@ -64,12 +68,17 @@
     integration_service: Arc<dyn IntegrationService>,
     repository_service: Arc<dyn RepositoryService>,
     context_service: Arc<dyn ContextService>,
+    license_service: Arc<dyn LicenseService>,
+    notification_service: Arc<dyn NotificationService>,
     embedding: Arc<dyn Embedding>,
 ) {
     let mut hourly =
         CronStream::new(Schedule::from_str("@hourly").expect("Invalid cron expression"))
             .into_stream();
 
+    let mut daily = CronStream::new(Schedule::from_str("@daily").expect("Invalid cron expression"))
+        .into_stream();
+
     tokio::spawn(async move {
         loop {
             tokio::select! {
@@ -141,6 +150,11 @@
                     }
 
                 },
+                Some(now) = daily.next() => {
+                    if let Err(err) = LicenseCheckJob::cron(now, license_service.clone(), notification_service.clone()).await {
+                        warn!("License check job failed: {err:?}");
+                    }
+                }
                 else => {
                     warn!("Background job channel closed");
                     break;

ee/tabby-webserver/src/service/mod.rs

@@ -131,6 +131,8 @@
             integration.clone(),
             repository.clone(),
             context.clone(),
+            license.clone(),
+            notification.clone(),
             embedding.clone(),
         )
         .await;