feat: add message rule types

onesoft-sudo · Aug 22, 2023 · b6be033 · b6be033
1 parent 4963231
commit b6be033
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/src/types/GuildConfigSchema.ts b/src/types/GuildConfigSchema.ts
@@ -19,6 +19,7 @@
 
 import { z } from "zod";
 import { isSnowflake } from "../utils/utils";
+import { MessageRuleSchema } from "./MessageRuleSchema";
 
 const zSnowflake = z.custom<string>(data => {
     return typeof data === "string" && isSnowflake(data);
@@ -255,6 +256,13 @@ export const GuildConfigSchema = z.object({
             disabled_channels: z.array(zSnowflake).default([]),
             blocked_hashes: z.record(z.string(), z.string().nullable()).default({})
         })
+        .optional(),
+    message_rules: z
+        .object({
+            enabled: z.boolean().default(false),
+            rules: z.array(MessageRuleSchema).default([]),
+            global_disabled_channels: z.array(zSnowflake).default([])
+        })
         .optional()
 });
 

diff --git a/src/types/MessageRuleSchema.ts b/src/types/MessageRuleSchema.ts
@@ -0,0 +1,41 @@
+import { z } from "zod";
+
+export const MessageRuleAction = z.enum(["delete", "verbal_warn", "warn", "mute", "clear"]);
+
+const hasStringArrayData = {
+    data: z.array(z.string()).default([])
+};
+
+const Common = {
+    actions: z.array(MessageRuleAction).default([]),
+    verbal_warning_reason: z.string().optional(),
+    warning_reason: z.string().optional(),
+    mute_reason: z.string().optional(),
+    common_reason: z.string().optional(),
+    mute_duration: z.number().int().default(-1)
+};
+
+export const BlockedDomainRule = z
+    .object({
+        type: z.literal("blocked_domain"),
+        scan_links_only: z.boolean().default(false)
+    })
+    .extend(Common)
+    .extend(hasStringArrayData);
+
+export const BlockedMimeTypeRule = z
+    .object({
+        type: z.literal("blocked_mime_type")
+    })
+    .extend(Common)
+    .extend(hasStringArrayData);
+
+export const BlockedFileExtensionRule = z
+    .object({
+        type: z.literal("blocked_file_extension")
+    })
+    .extend(Common)
+    .extend(hasStringArrayData);
+
+export const MessageRuleSchema = z.union([BlockedDomainRule, BlockedMimeTypeRule, BlockedFileExtensionRule]);
+export type MessageRuleType = z.infer<typeof MessageRuleSchema>;