feat(webserver): add license check

TabbyML · Feb 19, 2024 · 4c34340 · 4c34340
1 parent 9e22233
commit 4c34340
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/ee/tabby-webserver/src/schema/mod.rs b/ee/tabby-webserver/src/schema/mod.rs
@@ -11,7 +11,6 @@ use auth::{
     validate_jwt, AuthenticationService, Invitation, RefreshTokenError, RefreshTokenResponse,
     RegisterError, RegisterResponse, TokenAuthError, TokenAuthResponse, User, VerifyTokenResponse,
 };
-use futures::TryFutureExt;
 use job::{JobRun, JobService};
 use juniper::{
     graphql_object, graphql_value, EmptySubscription, FieldError, FieldResult, GraphQLObject,
@@ -72,8 +71,11 @@ pub enum CoreError {
     #[error("{0}")]
     Forbidden(&'static str),
 
-    #[error("Invalid ID Error")]
-    InvalidIDError,
+    #[error("Your license is not valid for this feature")]
+    InvalidLicense,
+
+    #[error("Invalid ID")]
+    InvalidID,
 
     #[error("Invalid input parameters")]
     InvalidInput(#[from] ValidationErrors),

diff --git a/ee/tabby-webserver/src/service/dao.rs b/ee/tabby-webserver/src/service/dao.rs
@@ -143,7 +143,7 @@ impl AsRowid for juniper::ID {
             .decode(self)
             .first()
             .map(|i| *i as i32)
-            .ok_or(CoreError::InvalidIDError)
+            .ok_or(CoreError::InvalidID)
     }
 }