410 lines
10 KiB
Python
410 lines
10 KiB
Python
import requests
|
|
import websockets
|
|
import asyncio
|
|
import json
|
|
import re
|
|
import httpx
|
|
import traceback
|
|
import os
|
|
import aiohttp
|
|
import random
|
|
from bs4 import BeautifulSoup
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
import aiofiles
|
|
|
|
HERE = Path(__file__).parent
|
|
username = os.environ.get("GLUEBOT_USERNAME")
|
|
password = os.environ.get("GLUEBOT_PASSWORD")
|
|
|
|
headers = {
|
|
"User-Agent": "gluebot",
|
|
"Origin": "https://deek.chat",
|
|
"DNT": "1",
|
|
}
|
|
|
|
url = "https://deek.chat"
|
|
ws_url = "wss://deek.chat/ws"
|
|
prefix = ","
|
|
token = None
|
|
session = None
|
|
delay = 3
|
|
|
|
gifmaker_common = [
|
|
"/usr/bin/gifmaker",
|
|
"--width", 350,
|
|
"--output", "/tmp/gifmaker",
|
|
"--nogrow",
|
|
]
|
|
|
|
|
|
def get_time():
|
|
return datetime.now().timestamp()
|
|
|
|
|
|
def remove_file(path):
|
|
try:
|
|
path.unlink()
|
|
except Exception as e:
|
|
print(f"(Remove) Error: {e}")
|
|
traceback.print_exc()
|
|
|
|
|
|
def get_extension(path):
|
|
return Path(path).suffix.lower().lstrip(".")
|
|
|
|
|
|
def clean_lines(s):
|
|
cleaned = s
|
|
cleaned = re.sub(r" *(\n+|\\n+) *", "\n", cleaned)
|
|
cleaned = re.sub(r" +", " ", cleaned)
|
|
return cleaned.strip()
|
|
|
|
|
|
def random_int(min_val, max_val):
|
|
return random.randint(min_val, max_val)
|
|
|
|
|
|
def get_path(name):
|
|
return str(Path(HERE, name))
|
|
|
|
|
|
def extract_range(string):
|
|
pattern = r"(?:(?P<number1>-?\d+)(?:\s*(.+?)\s*(?P<number2>-?\d+))?)?"
|
|
match = re.search(pattern, string)
|
|
num1 = None
|
|
num2 = None
|
|
|
|
if match["number1"]:
|
|
num1 = int(match["number1"])
|
|
|
|
if match["number2"]:
|
|
num2 = int(match["number2"])
|
|
|
|
return [num1, num2]
|
|
|
|
|
|
def clean_list(lst):
|
|
return list(filter(lambda x: x != "", lst))
|
|
|
|
|
|
def string_to_number(input_string):
|
|
hash_value = hash(input_string)
|
|
absolute_hash = abs(hash_value)
|
|
scaled_number = absolute_hash % 1000
|
|
return scaled_number
|
|
|
|
|
|
def join_command(command):
|
|
return " ".join(f"\"{arg}\"" for arg in command)
|
|
|
|
|
|
def gifmaker_command(args):
|
|
command = gifmaker_common.copy()
|
|
command.extend(args)
|
|
return join_command(command)
|
|
|
|
|
|
cmd_date = get_time()
|
|
|
|
|
|
def update_time():
|
|
global cmd_date
|
|
cmd_date = get_time()
|
|
|
|
|
|
def blocked():
|
|
return (get_time() - cmd_date) < delay
|
|
|
|
|
|
def auth():
|
|
global token, session, headers
|
|
|
|
if not username or not password:
|
|
print("Missing environment variables")
|
|
exit(1)
|
|
|
|
data = {"name": username, "password": password, "submit": "log+in"}
|
|
res = requests.post(url + "/login/submit", headers=headers, data=data, allow_redirects=False)
|
|
token = re.search("(?:api_token)=[^;]+", res.headers.get("Set-Cookie")).group(0)
|
|
session = re.search("(?:session_id)=[^;]+", res.headers.get("Set-Cookie")).group(0)
|
|
headers["Cookie"] = token + "; " + session
|
|
|
|
|
|
async def run():
|
|
async with websockets.connect(ws_url, extra_headers=headers) as ws:
|
|
try:
|
|
while True:
|
|
message = await ws.recv()
|
|
await on_message(ws, message)
|
|
except KeyboardInterrupt:
|
|
exit(0)
|
|
except websockets.exceptions.ConnectionClosedOK:
|
|
print("WebSocket connection closed")
|
|
except Exception as e:
|
|
print("(WebSocket) Error:", e)
|
|
traceback.print_exc()
|
|
|
|
|
|
async def on_message(ws, message):
|
|
if blocked():
|
|
return
|
|
|
|
try:
|
|
data = json.loads(message)
|
|
except:
|
|
return
|
|
|
|
if data["type"] == "message":
|
|
if data["data"]["name"] == username:
|
|
return
|
|
|
|
text = data["data"]["text"].strip()
|
|
|
|
if not text.startswith(prefix):
|
|
return
|
|
|
|
room_id = data["roomId"]
|
|
words = text.lstrip(prefix).split(" ")
|
|
cmd = words[0]
|
|
args = words[1:]
|
|
|
|
if cmd == "ping":
|
|
update_time()
|
|
await send_message(ws, "Pong!", room_id)
|
|
|
|
elif cmd == "help":
|
|
update_time()
|
|
await send_message(ws, f"Commands: describe | wins | numbers | date | bird | shitpost", room_id)
|
|
|
|
elif cmd == "describe":
|
|
if len(args) >= 1:
|
|
update_time()
|
|
arg = " ".join(clean_list(args))
|
|
await gif_describe(arg, room_id)
|
|
|
|
elif cmd == "wins" or cmd == "win":
|
|
if len(args) >= 1:
|
|
update_time()
|
|
arg = " ".join(clean_list(args))
|
|
await gif_wins(arg, room_id)
|
|
|
|
elif cmd == "numbers" or cmd == "number" or cmd == "nums" or cmd == "num":
|
|
update_time()
|
|
|
|
if len(args) > 0:
|
|
arg = " ".join(clean_list(args))
|
|
else:
|
|
arg = None
|
|
|
|
await gif_numbers(arg, room_id)
|
|
|
|
elif cmd == "date" or cmd == "data" or cmd == "time" or cmd == "datetime":
|
|
update_time()
|
|
await gif_date(room_id)
|
|
|
|
elif cmd == "bird" or cmd == "birds" or cmd == "birb" or cmd == "birbs" or cmd == "brb":
|
|
update_time()
|
|
await random_bird(ws, room_id)
|
|
|
|
elif cmd == "post" or cmd == "shitpost" or cmd == "4chan" or cmd == "anon" or cmd == "shit":
|
|
update_time()
|
|
await random_post(ws, room_id)
|
|
|
|
|
|
async def random_bird(ws, room_id):
|
|
birdfile = get_path("data/aves.txt")
|
|
|
|
async with aiofiles.open(birdfile, mode="r", encoding="utf-8") as file:
|
|
birds = await file.readlines()
|
|
bird = random.choice(birds).strip()
|
|
await send_message(ws, f".i \"{bird}\" bird", room_id)
|
|
|
|
|
|
async def gif_describe(who, room_id):
|
|
command = gifmaker_command([
|
|
"--input", get_path("describe.jpg"),
|
|
"--words", f"{who} is\\n[Random] [x5]",
|
|
"--filter", "anyhue2",
|
|
"--opacity", 0.8,
|
|
"--fontsize", 66,
|
|
"--delay", 700,
|
|
"--padding", 50,
|
|
"--fontcolor", "light2",
|
|
"--bgcolor", "black",
|
|
])
|
|
|
|
await run_gifmaker(command, room_id)
|
|
|
|
|
|
async def gif_wins(who, room_id):
|
|
command = gifmaker_command([
|
|
"--input", get_path("wins.gif"),
|
|
"--words", f"{who} wins a ; [repeat] ; [RANDOM] ; [repeat]",
|
|
"--bgcolor", "0,0,0",
|
|
"--bottom", 20,
|
|
"--filter", "anyhue2",
|
|
"--framelist", "11,11,33,33",
|
|
"--fontsize", 42,
|
|
])
|
|
|
|
await run_gifmaker(command, room_id)
|
|
|
|
|
|
async def gif_numbers(arg, room_id):
|
|
num = -1
|
|
|
|
if arg:
|
|
nums = extract_range(arg)
|
|
|
|
if nums[0] is not None:
|
|
if nums[1] is not None:
|
|
if nums[0] < nums[1]:
|
|
num = random_int(nums[0], nums[1])
|
|
else:
|
|
return
|
|
else:
|
|
num = random_int(0, nums[0])
|
|
|
|
if num == -1:
|
|
num = string_to_number(arg)
|
|
|
|
if num == -1:
|
|
num = random_int(0, 999)
|
|
|
|
command = gifmaker_command([
|
|
"--input", get_path("numbers.pnf"),
|
|
"--top", 20,
|
|
"--words", num,
|
|
"--fontcolor", "0,0,0",
|
|
"--fontsize", 66,
|
|
"--format", "jpg",
|
|
])
|
|
|
|
await run_gifmaker(command, room_id)
|
|
|
|
|
|
async def gif_date(room_id):
|
|
command = gifmaker_command([
|
|
"--input", get_path("time.jpg"),
|
|
"--words", "Date: [date %A %d] ; [repeat] ; Time: [date %I:%M %p] ; [repeat]",
|
|
"--filter", "anyhue2",
|
|
"--bottom", 20,
|
|
"--bgcolor", "0,0,0",
|
|
"--fontsize", 80,
|
|
])
|
|
|
|
await run_gifmaker(command, room_id)
|
|
|
|
|
|
async def random_post(ws, room_id):
|
|
boards = ["g", "an", "ck", "lit", "x", "tv", "v", "fit", "k", "o"]
|
|
board = random.choice(boards)
|
|
|
|
try:
|
|
threads_url = f"https://a.4cdn.org/{board}/threads.json"
|
|
async_client = httpx.AsyncClient()
|
|
threads_response = await async_client.get(threads_url)
|
|
threads_response.raise_for_status()
|
|
threads_json = threads_response.json()
|
|
threads = threads_json[0]["threads"]
|
|
|
|
# Select a random thread
|
|
id = threads[random_int(0, len(threads) - 1)]["no"]
|
|
thread_url = f"https://a.4cdn.org/{board}/thread/{id}.json"
|
|
|
|
# Fetch the selected thread
|
|
thread_response = await async_client.get(thread_url)
|
|
thread_response.raise_for_status()
|
|
thread_json = thread_response.json()
|
|
posts = thread_json["posts"]
|
|
|
|
# Select a random post
|
|
post = posts[random_int(0, len(posts) - 1)]
|
|
html = post.get("com", "")
|
|
|
|
if not html:
|
|
return
|
|
|
|
# Parse HTML using BeautifulSoup
|
|
soup = BeautifulSoup(html, "html.parser")
|
|
|
|
# Remove elements with class "quotelink"
|
|
for elem in soup.select(".quotelink"):
|
|
elem.decompose()
|
|
|
|
# Replace <br> with newline
|
|
for br in soup.find_all("br"):
|
|
br.replace_with("\n")
|
|
|
|
# Get text content and limit to 500 characters
|
|
text = soup.get_text(separator="\n")[:500].strip()
|
|
text = clean_lines(text)
|
|
|
|
if not text:
|
|
return
|
|
|
|
await send_message(ws, text, room_id)
|
|
|
|
except Exception as err:
|
|
print(f"Error: {err}")
|
|
|
|
|
|
async def run_gifmaker(command, room_id):
|
|
process = await asyncio.create_subprocess_shell(
|
|
command,
|
|
stdout=asyncio.subprocess.PIPE,
|
|
stderr=asyncio.subprocess.PIPE,
|
|
shell=True,
|
|
)
|
|
|
|
stdout, stderr = await process.communicate()
|
|
|
|
if process.returncode != 0:
|
|
print(f"(Process) Error: {stderr.decode()}")
|
|
return
|
|
|
|
await upload(Path(stdout.decode().strip()), room_id)
|
|
|
|
|
|
async def upload(path, room_id):
|
|
if (not path.exists()) or (not path.is_file()):
|
|
return
|
|
|
|
cookies = {
|
|
"session_id": session.split("=")[1],
|
|
"api_token": token.split("=")[1],
|
|
}
|
|
|
|
ext = get_extension(path)
|
|
ext = "jpeg" if ext == "jpg" else ext
|
|
url = "https://deek.chat/message/send/" + str(room_id)
|
|
|
|
data = aiohttp.FormData()
|
|
data.add_field(name="files[]", value=open(path, "rb"), filename=path.name, content_type=f"image/{ext}")
|
|
|
|
try:
|
|
async with aiohttp.ClientSession(cookies=cookies) as sess:
|
|
async with sess.post(url, data=data, headers={}) as response:
|
|
await response.text()
|
|
except Exception as e:
|
|
print(f"(Upload) Error: {e}")
|
|
traceback.print_exc()
|
|
|
|
remove_file(path)
|
|
|
|
|
|
async def send_message(ws, text, room_id):
|
|
await ws.send(json.dumps({"type": "message", "data": text, "roomId": room_id}))
|
|
|
|
while True:
|
|
try:
|
|
auth()
|
|
print("Authenticated")
|
|
asyncio.run(run())
|
|
except KeyboardInterrupt:
|
|
break
|
|
except Exception as e:
|
|
print("(Main) Error:", e)
|
|
traceback.print_exc()
|