Update message schema to handle cloud attachments

2 files changed, 4 insertions, 7 deletions

diff --git a/src/util/schemas/MessageCreateSchema.ts b/src/util/schemas/MessageCreateSchema.ts

index 2538d39e..2bd6ccf5 100644
--- a/src/util/schemas/MessageCreateSchema.ts
+++ b/src/util/schemas/MessageCreateSchema.ts
@@ -16,12 +16,8 @@
 	along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
-import {
-	ActionRowComponent,
-	Embed,
-	PollAnswer,
-	PollMedia,
-} from "@spacebar/util";
+import { ActionRowComponent, Embed, PollAnswer, PollMedia, UploadAttachmentResponse } from "@spacebar/util";
+import { CloudAttachment } from "../entities/CloudAttachment";
 
 export type MessageCreateAttachment = {
 	id: string;
@@ -57,7 +53,7 @@ export interface MessageCreateSchema {
 	TODO: we should create an interface for attachments
 	TODO: OpenWAAO<-->attachment-style metadata conversion
 	**/
-	attachments?: MessageCreateAttachment[];
+	attachments?: (MessageCreateAttachment | UploadAttachmentResponse)[];
 	sticker_ids?: string[];
 	components?: ActionRowComponent[];
 	// TODO: Fix TypeScript errors in src\api\util\handlers\Message.ts once this is enabled
diff --git a/src/util/schemas/responses/UploadAttachmentResponseSchema.ts b/src/util/schemas/responses/UploadAttachmentResponseSchema.ts

index 322137da..eab5f22e 100644
--- a/src/util/schemas/responses/UploadAttachmentResponseSchema.ts
+++ b/src/util/schemas/responses/UploadAttachmentResponseSchema.ts
@@ -25,4 +25,5 @@ export interface UploadAttachmentResponse {
 	id?: string;
 	upload_url: string;
 	upload_filename: string;
+	original_content_type?: string;
 }