fix(webserver): forbid empty content message update

TabbyML · Nov 8, 2024 · dd29aac · dd29aac
1 parent 8d2439d
commit dd29aac
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/ee/tabby-schema/src/schema/mod.rs b/ee/tabby-schema/src/schema/mod.rs
@@ -1183,6 +1183,8 @@ impl Mutation {
         input: thread::UpdateMessageInput,
     ) -> Result<bool> {
         let user = check_user(ctx).await?;
+        input.validate()?;
+
         let svc = ctx.locator.thread();
         let Some(thread) = svc.get(&input.thread_id).await? else {
             return Err(CoreError::NotFound("Thread not found"));

diff --git a/ee/tabby-schema/src/schema/thread/types.rs b/ee/tabby-schema/src/schema/thread/types.rs
@@ -5,6 +5,7 @@ use tabby_common::api::{
     code::{CodeSearchDocument, CodeSearchHit, CodeSearchScores},
     doc::{DocSearchDocument, DocSearchHit},
 };
+use validator::Validate;
 
 use crate::{juniper::relay::NodeType, Context};
 
@@ -44,11 +45,12 @@ impl NodeType for Message {
     }
 }
 
-#[derive(GraphQLInputObject, Clone)]
+#[derive(GraphQLInputObject, Clone, Validate)]
 #[graphql(context = Context)]
 pub struct UpdateMessageInput {
     pub id: ID,
     pub thread_id: ID,
+    #[validate(length(min = 1, message = "content can not be empty"))]
     pub content: String,
 }