-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathendpoint_update_handler.py
387 lines (349 loc) · 22.9 KB
/
endpoint_update_handler.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
import asyncio
import json
import logging
import pathlib
import re
import traceback
import fastapi
import astroidapi.sending_handler as sending_handler
import astroidapi.surrealdb_handler as surrealdb_handler
import astroidapi.queue_processor as queue_processor
from Bot import config
import astroidapi.health_check as health_check
class UpdateHandler:
@classmethod
async def update_endpoint(
cls,
endpoint: int,
token: str = None,
webhook_discord: str = None,
webhook_guilded: str = None,
webhook_revolt: str = None,
webhook_nerimity: str = None,
selfuse: bool = None,
log_discord: str = None,
log_guilded: str = None,
log_revolt: str = None,
log_nerimity: str = None,
channel_discord: str = None,
channel_guilded: str = None,
channel_revolt: str = None,
channel_nerimity: str = None,
blacklist: str = None,
trigger: bool = None,
sender_channel: str = None,
sender: str = None,
message_author_name: str = None,
message_author_avatar: str = None,
allowed_ids: str = None,
message_reply: bool = False,
message_reply_message: str = None,
message_reply_author: str = None,
message_author_id: str = None,
message_content: str = None,
message_attachments: str = None,
message_embed: str = None,
beta: bool = False,
index: int = None,
only_check: bool = False,
):
try:
data_token = json.load(open(f"{pathlib.Path(__file__).parent.parent.resolve()}/tokens.json", "r"))[f"{endpoint}"]
if token is not None:
if token == config.MASTER_TOKEN or token == data_token:
endpoint_data = await surrealdb_handler.get_endpoint(endpoint, __file__)
if endpoint_data:
if only_check:
if endpoint_data["meta"]["read"]["discord"] and endpoint_data["meta"]["read"]["guilded"] and endpoint_data["meta"]["read"]["revolt"]:
return fastapi.responses.JSONResponse(status_code=200, content={"message": "All endpoints are read."})
else:
return fastapi.responses.JSONResponse(status_code=200, content={"message": "Not all endpoints are read."})
if beta is True:
endpoint_data["config"]["isbeta"] = True
if webhook_discord:
if webhook_discord in endpoint_data["config"]["webhooks"]["discord"]:
pass
elif index is not None:
endpoint_data["config"]["webhooks"]["discord"][index] = webhook_discord
else:
endpoint_data["config"]["webhooks"]["discord"].append(webhook_discord)
if webhook_guilded:
if webhook_guilded in endpoint_data["config"]["webhooks"]["guilded"]:
pass
elif index is not None:
endpoint_data["config"]["webhooks"]["guilded"][index] = webhook_guilded
else:
endpoint_data["config"]["webhooks"]["guilded"].append(webhook_guilded)
if webhook_revolt:
if webhook_revolt in endpoint_data["config"]["webhooks"]["revolt"]:
pass
elif index is not None:
endpoint_data["config"]["webhooks"]["revolt"][index] = webhook_revolt
else:
endpoint_data["config"]["webhooks"]["revolt"].append(webhook_revolt)
if webhook_nerimity:
if webhook_nerimity in endpoint_data["config"]["webhooks"]["nerimity"]:
pass
elif index is not None:
endpoint_data["config"]["webhooks"]["nerimity"][index] = webhook_nerimity
else:
endpoint_data["config"]["webhooks"]["nerimity"].append(webhook_nerimity)
if selfuse is True or selfuse is False:
endpoint_data["config"]["self-user"] = selfuse
if log_discord:
endpoint_data["config"]["logs"]["discord"] = log_discord
if log_guilded:
endpoint_data["config"]["logs"]["guilded"] = log_guilded
if log_revolt:
endpoint_data["config"]["logs"]["revolt"] = log_revolt
if log_nerimity:
endpoint_data["config"]["logs"]["nerimity"] = log_nerimity
if channel_discord:
if channel_discord in endpoint_data["config"]["channels"]["discord"]:
pass
elif index is not None:
endpoint_data["config"]["channels"]["discord"][index] = channel_discord
else:
endpoint_data["config"]["channels"]["discord"].append(channel_discord)
if channel_guilded:
if channel_guilded in endpoint_data["config"]["channels"]["guilded"]:
pass
elif index is not None:
endpoint_data["config"]["channels"]["guilded"][index] = channel_guilded
else:
endpoint_data["config"]["channels"]["guilded"].append(channel_guilded)
if channel_revolt:
if channel_revolt in endpoint_data["config"]["channels"]["revolt"]:
pass
elif index is not None:
endpoint_data["config"]["channels"]["revolt"][index] = channel_revolt
else:
endpoint_data["config"]["channels"]["revolt"].append(channel_revolt)
if channel_nerimity:
if channel_nerimity in endpoint_data["config"]["channels"]["nerimity"]:
pass
elif index is not None:
endpoint_data["config"]["channels"]["nerimity"][index] = channel_nerimity
else:
endpoint_data["config"]["channels"]["nerimity"].append(channel_nerimity)
if blacklist:
if "," in blacklist:
for val in blacklist.split(","):
if val.lower() not in [x.lower() for x in endpoint_data["config"]["blacklist"]]:
endpoint_data["config"]["blacklist"].append(val.lower())
else:
if blacklist.lower() not in endpoint_data["config"]["blacklist"]:
if index is not None:
endpoint_data["config"]["blacklist"][index] = blacklist.lower()
else:
endpoint_data["config"]["blacklist"].append(blacklist.lower())
if trigger:
endpoint_data["meta"]["trigger"] = trigger
if sender_channel:
endpoint_data["meta"]["sender-channel"] = sender_channel
if sender:
if sender in ["discord", "guilded", "revolt", "nerimity"]:
endpoint_data["meta"]["sender"] = sender
endpoint_data["meta"]["read"][sender] = True
else:
return fastapi.responses.JSONResponse(status_code=400, content={"message": "Invalid sender."})
if message_author_name:
endpoint_data["meta"]["message"]["author"]["name"] = message_author_name
if message_author_avatar:
endpoint_data["meta"]["message"]["author"]["avatar"] = message_author_avatar
if allowed_ids:
if "," in allowed_ids:
for val in allowed_ids.split(","):
if val not in endpoint_data["config"]["allowed-ids"]:
endpoint_data["config"]["allowed-ids"].append(val)
else:
if allowed_ids not in endpoint_data["config"]["allowed-ids"]:
if index is not None:
endpoint_data["config"]["allowed-ids"][index] = allowed_ids
else:
endpoint_data["config"]["allowed-ids"].append(allowed_ids)
if message_reply:
try:
endpoint_data["meta"]["message"]["isReply"] = message_reply
except:
await health_check.HealthCheck.EndpointCheck.repair_structure(endpoint)
endpoint_data["meta"]["message"]["isReply"] = message_reply
if message_reply_message:
try:
endpoint_data["meta"]["message"]["reply"]["message"] = message_reply_message
except:
await health_check.HealthCheck.EndpointCheck.repair_structure(endpoint)
endpoint_data["meta"]["message"]["reply"]["message"] = message_reply_message
if message_reply_author:
try:
endpoint_data["meta"]["message"]["reply"]["author"] = message_reply_author
except:
await health_check.HealthCheck.EndpointCheck.repair_structure(endpoint)
endpoint_data["meta"]["message"]["reply"]["author"] = message_reply_author
if message_author_id:
endpoint_data["meta"]["message"]["author"]["id"] = message_author_id
if message_content:
if sender == "discord":
if "http" in message_content or "https" in message_content:
urls = re.findall(r'(https?://\S+)', message_content)
for url in urls:
image_markdown = f""
message_content = message_content.replace(url, image_markdown) + message_content.split(url)[1]
endpoint_data["meta"]["message"]["content"] = message_content
else:
endpoint_data["meta"]["message"]["content"] = message_content
if message_attachments:
if "," in message_attachments:
for val in message_attachments.split(","):
if val.lower() not in [x.lower() for x in endpoint_data["meta"]["message"]["attachments"]]:
if sender == "discord":
endpoint_data["meta"]["message"]["attachments"].append(f"")
else:
endpoint_data["meta"]["message"]["attachments"].append(val)
else:
endpoint_data["meta"]["message"]["attachments"] = [message_attachments]
if message_embed:
embed_object = json.loads(message_embed.replace("'", '"'))
endpoint_data["meta"]["message"]["embed"] = embed_object
if len(endpoint_data["config"]["channels"]["discord"]) == 0:
endpoint_data["meta"]["read"]["discord"] = True
if len(endpoint_data["config"]["channels"]["guilded"]) == 0:
endpoint_data["meta"]["read"]["guilded"] = True
if len(endpoint_data["config"]["channels"]["revolt"]) == 0:
endpoint_data["meta"]["read"]["revolt"] = True
if len(endpoint_data["config"]["channels"]["nerimity"]) == 0:
endpoint_data["meta"]["read"]["nerimity"] = True
updated_json = await surrealdb_handler.update(endpoint, endpoint_data)
try:
updated_json = updated_json[0]["result"][0]
updated_json.pop("id")
except:
pass
finally:
if not updated_json["config"]["self-user"] is True:
if updated_json["meta"]["trigger"]:
asyncio.create_task(queue_processor.QueueProcessor.handleUpdatedEndpointData(endpoint, updated_json))
waiting_secs = 0
max_secs = 10
while True:
check_json = await surrealdb_handler.get_endpoint(endpoint, __file__)
if (check_json["meta"]["read"]["discord"] == True
and check_json["meta"]["read"]["guilded"] == True
and check_json["meta"]["read"]["revolt"] == True
and check_json["meta"]["read"]["nerimity"] == True):
check_json["meta"]["message"]["content"] = None
check_json["meta"]["message"]["attachments"].clear()
check_json["meta"]["message"]["author"]["avatar"] = None
check_json["meta"]["message"]["author"]["name"] = None
check_json["meta"]["message"]["author"]["id"] = None
check_json["meta"]["trigger"] = False
check_json["meta"]["sender"] = None
check_json["meta"]["sender-channel"] = None
try:
check_json["meta"]["message"]["isReply"] = False
check_json["meta"]["message"]["reply"]["message"] = None
check_json["meta"]["message"]["reply"]["author"] = None
except:
await health_check.HealthCheck.EndpointCheck.repair_structure(endpoint)
check_json["meta"]["message"]["isReply"] = False
check_json["meta"]["message"]["reply"]["message"] = None
check_json["meta"]["message"]["reply"]["author"] = None
check_json["meta"]["read"]["discord"] = False
check_json["meta"]["read"]["guilded"] = False
check_json["meta"]["read"]["revolt"] = False
check_json["meta"]["read"]["nerimity"] = False
await surrealdb_handler.update(endpoint, check_json)
print("Everything is read. Cleared.")
break
await asyncio.sleep(1)
waiting_secs += 1
if waiting_secs >= max_secs:
check_json["meta"]["message"]["content"] = None
check_json["meta"]["message"]["attachments"].clear()
check_json["meta"]["message"]["author"]["avatar"] = None
check_json["meta"]["message"]["author"]["name"] = None
check_json["meta"]["message"]["author"]["id"] = None
check_json["meta"]["trigger"] = False
check_json["meta"]["sender"] = None
check_json["meta"]["sender-channel"] = None
try:
check_json["meta"]["message"]["isReply"] = False
check_json["meta"]["message"]["reply"]["message"] = None
check_json["meta"]["message"]["reply"]["author"] = None
except:
await health_check.HealthCheck.EndpointCheck.repair_structure(endpoint)
check_json["meta"]["message"]["isReply"] = False
check_json["meta"]["message"]["reply"]["message"] = None
check_json["meta"]["message"]["reply"]["author"] = None
check_json["meta"]["read"]["discord"] = False
check_json["meta"]["read"]["guilded"] = False
check_json["meta"]["read"]["revolt"] = False
check_json["meta"]["read"]["nerimity"] = False
await surrealdb_handler.update(endpoint, check_json)
print("Not everything is read. Cleared anyways.")
break
else:
return fastapi.responses.JSONResponse(
status_code=200,
content={"message": "This endpoint activated self-usage."},
)
return fastapi.responses.JSONResponse(status_code=200, content=endpoint_data)
else:
return fastapi.responses.JSONResponse(status_code=404, content={"message": "This endpoint does not exist."})
else:
return fastapi.responses.JSONResponse(status_code=401, content={"message": "The provided token is invalid."})
else:
return fastapi.responses.JSONResponse(status_code=401, content={"message": "You must provide a token."})
except IndexError:
waiting_secs = 0
max_secs = 10
while True:
check_json = await surrealdb_handler.get_endpoint(endpoint, __file__)
if check_json["meta"]["trigger"] is False and check_json["meta"]["message"]["content"] is None:
return fastapi.responses.JSONResponse(status_code=200, content=check_json)
if (check_json["meta"]["read"]["discord"] == True
and check_json["meta"]["read"]["guilded"] == True
and check_json["meta"]["read"]["revolt"] == True
and check_json["meta"]["read"]["nerimity"] == True and check_json["config"]["self-user"] is False):
check_json["meta"]["message"]["content"] = None
check_json["meta"]["message"]["attachments"].clear()
check_json["meta"]["message"]["author"]["avatar"] = None
check_json["meta"]["message"]["author"]["name"] = None
check_json["meta"]["message"]["author"]["id"] = None
check_json["meta"]["trigger"] = False
check_json["meta"]["sender"] = None
check_json["meta"]["sender-channel"] = None
check_json["meta"]["read"]["discord"] = False
check_json["meta"]["read"]["guilded"] = False
check_json["meta"]["read"]["revolt"] = False
check_json["meta"]["read"]["nerimity"] = False
await surrealdb_handler.update(endpoint, check_json)
print("Everything is read. Cleared.")
break
await asyncio.sleep(1)
waiting_secs += 1
if waiting_secs >= max_secs:
check_json["meta"]["message"]["content"] = None
check_json["meta"]["message"]["attachments"].clear()
check_json["meta"]["message"]["author"]["avatar"] = None
check_json["meta"]["message"]["author"]["name"] = None
check_json["meta"]["message"]["author"]["id"] = None
check_json["meta"]["trigger"] = False
check_json["meta"]["sender"] = None
check_json["meta"]["sender-channel"] = None
check_json["meta"]["read"]["discord"] = False
check_json["meta"]["read"]["guilded"] = False
check_json["meta"]["read"]["revolt"] = False
check_json["meta"]["read"]["nerimity"] = False
await surrealdb_handler.update(endpoint, check_json)
print("Not everything is read. Cleared anyways.")
break
elif check_json["config"]["self-user"] is True:
return fastapi.responses.JSONResponse(
status_code=200,
content={"message": "This endpoint activated self-usage."},
)
except Exception as e:
logging.error("An error occurred: %s", e)
logging.error(traceback.format_exc())
return fastapi.responses.JSONResponse(status_code=500, content={"message": f"An error occurred: {e}"})