/[sudobot]/trunk/src/api/controllers/ConfigController.ts
ViewVC logotype

Diff of /trunk/src/api/controllers/ConfigController.ts

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 349 by rakin, Mon Jul 29 17:29:43 2024 UTC revision 370 by rakin, Mon Jul 29 17:29:50 2024 UTC
# Line 1  Line 1 
1    import { dot, object } from "dot-object";
2    import { body } from "express-validator";
3    import { z as zod, ZodSchema } from "zod";
4    import KeyValuePair from "../../types/KeyValuePair";
5  import Controller from "../Controller";  import Controller from "../Controller";
6  import RequireAuth from "../middleware/RequireAuth";  import RequireAuth from "../middleware/RequireAuth";
7    import ValidatorError from "../middleware/ValidatorError";
8  import Request from "../Request";  import Request from "../Request";
9    import merge from 'ts-deepmerge';
10    
11  export default class ConfigController extends Controller {  export default class ConfigController extends Controller {
12      globalMiddleware(): Function[] {      globalMiddleware(): Function[] {
13          return [RequireAuth];          return [RequireAuth, ValidatorError];
14        }
15    
16        middleware(): KeyValuePair<Function[]> {
17            return {
18                update: [
19                    body(["config"]).isObject()
20                ]
21            };
22        }
23    
24        private zodSchema() {
25            const snowflake = zod.string().regex(/\d+/, { message: "The given value is not a Snowflake" });
26    
27            const schema = zod.object({
28                "prefix": zod.string().optional(),
29                "debug": zod.boolean().optional(),
30                "mute_role": snowflake.optional(),
31                "gen_role": snowflake.optional(),
32                "logging_channel": snowflake.optional(),
33                "logging_channel_join_leave": snowflake.optional(),
34                "mod_role": snowflake.optional(),
35                "announcement_channel": snowflake.optional(),
36                "admin": snowflake.optional(),
37                "lockall": zod.array(zod.string()).optional(),
38                "warn_notallowed": zod.boolean().optional(),
39                "role_commands": zod.record(
40                    snowflake,
41                    zod.array(zod.string().min(1))
42                ).optional(),
43                "autoclear": zod.object({
44                    "enabled": zod.boolean().optional(),
45                    "channels": zod.array(snowflake).optional()
46                }).optional(),
47                "verification": zod.object({
48                    "enabled": zod.boolean().optional(),
49                    "role": snowflake.optional()
50                }).optional(),
51                "welcomer": zod.object({
52                    "enabled": zod.boolean().optional(),
53                    "channel": snowflake.optional(),
54                    "message": zod.string().min(1).or(zod.null()).optional(),
55                    "randomize": zod.boolean().optional()
56                }).optional(),
57                "cooldown": zod.object({
58                    "enabled": zod.boolean().optional(),
59                    "global": zod.any().optional(),
60                    "cmds": zod.object({}).optional()
61                }).optional(),
62                "starboard": zod.object({
63                    "enabled": zod.boolean().optional(),
64                    "reactions": zod.number().int().optional(),
65                    "channel": snowflake.optional()
66                }).optional(),
67                "autorole": zod.object({
68                    "enabled": zod.boolean().optional(),
69                    "roles": zod.array(snowflake).optional()
70                }).optional(),
71                "spam_filter": zod.object({
72                    "enabled": zod.boolean().optional(),
73                    "limit": zod.number().int().optional(),
74                    "time": zod.number().optional(),
75                    "diff": zod.number().optional(),
76                    "exclude": zod.array(snowflake).optional(),
77                    "samelimit": zod.number().int().optional(),
78                    "unmute_in": zod.number().optional()
79                }).optional(),
80                "raid": zod.object({
81                    "enabled": zod.boolean().optional(),
82                    "max_joins": zod.number().int().optional(),
83                    "time": zod.number().optional(),
84                    "channels": zod.array(snowflake).optional(),
85                    "exclude": zod.boolean().optional()
86                }).optional(),
87                "global_commands": zod.array(zod.string()).optional(),
88                "filters": zod.object({
89                    "ignore_staff": zod.boolean().optional(),
90                    "chars_repeated": zod.number().int().optional(),
91                    "words_repeated": zod.number().int().optional(),
92                    "words": zod.array(zod.string()).optional(),
93                    "tokens": zod.array(zod.string()).optional(),
94                    "invite_message": zod.string().optional(),
95                    "words_excluded": zod.array(snowflake).optional(),
96                    "domain_excluded": zod.array(snowflake).optional(),
97                    "invite_excluded": zod.array(snowflake).optional(),
98                    "words_enabled": zod.boolean().optional(),
99                    "invite_enabled": zod.boolean().optional(),
100                    "domain_enabled": zod.boolean().optional(),
101                    "regex": zod.boolean().optional(),
102                    "file_mimes_excluded": zod.array(zod.string()).optional(),
103                    "file_types_excluded": zod.array(zod.string()).optional(),
104                    "domains": zod.array(zod.string()).optional(),
105                    "regex_patterns": zod.array(zod.string()).optional(),
106                    "rickrolls_enabled": zod.boolean().optional(),
107                    "pings": zod.number().int().optional()
108                }).optional()
109            });
110    
111            return schema;
112      }      }
113    
114      public async index(request: Request) {      public async index(request: Request) {
# Line 22  export default class ConfigController ex Line 126  export default class ConfigController ex
126      }      }
127    
128      public async update(request: Request) {      public async update(request: Request) {
129                    const { id } = request.params;
130            const { config } = request.body;
131    
132            console.log(config);        
133    
134            try {
135                const currentConfigDotObject = dot(this.client.config.props[id]);
136                let newConfigDotObject = {...currentConfigDotObject};
137        
138                console.log("Input: ", config);
139    
140                const result = this.zodSchema().safeParse(object({...config}));
141    
142                if (!result?.success) {
143                    return this.response({ error: "The data schema does not match.", error_type: 'validation', errors: result.error.errors }, 422);
144                }
145    
146                for (const configKey in config) {
147                    const regexMatched = /(.+)\[\d+\]/g.test(configKey);
148                    
149                    if (typeof currentConfigDotObject[configKey] === 'undefined' && !regexMatched) {
150                        return this.response({ error: `The key '${configKey}' is not allowed` }, 422);
151                    }
152    
153                    if (!regexMatched && config[configKey] !== null && config[configKey] !== null && typeof config[configKey] !== typeof currentConfigDotObject[configKey]) {
154                        console.log(typeof config[configKey], typeof currentConfigDotObject[configKey]);    
155                        
156                        if (typeof currentConfigDotObject[configKey] === 'number' && typeof config[configKey] === 'string') {
157                            const int = parseInt(config[configKey]);
158    
159                            if (int !== NaN) {
160                                newConfigDotObject[configKey] = int;
161                                console.log("Updating: ", configKey, config[configKey], newConfigDotObject[configKey]);
162                                continue;
163                            }
164                        }
165                        
166                        return this.response({ error: `The key '${configKey}' has incompatible value type '${config[configKey] === null ? 'null' : typeof config[configKey]}'` }, 422);
167                    }
168    
169                    console.log("Updating: ", configKey, config[configKey], newConfigDotObject[configKey]);
170                }
171    
172    
173                newConfigDotObject = merge.withOptions({ mergeArrays: false }, object({...newConfigDotObject}), object({...config}));
174                console.log("Output: ", newConfigDotObject);
175    
176                this.client.config.props[id] = newConfigDotObject;
177                this.client.config.write();
178    
179                return { message: "Configuration updated", previous: currentConfigDotObject, new: dot(this.client.config.props[id]) };
180            }
181            catch (e) {
182                console.log(e);
183                return this.response({ error: 'Internal Server Error', error_type: 'internal' }, 500);
184            }
185      }      }
186  }  }
Legend:
Removed from v.349  
changed lines
  Added in v.370
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26