327 lines
12 KiB
Python
327 lines
12 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
||
|
import subprocess
|
||
|
import sqlite3
|
||
|
import time
|
||
|
import os
|
||
|
import random
|
||
|
import socket
|
||
|
from urllib.parse import urlparse
|
||
|
import datetime
|
||
|
import simplematrixbotlib as botlib
|
||
|
import requests
|
||
|
from pytube import YouTube
|
||
|
import logging
|
||
|
import inspect
|
||
|
|
||
|
|
||
|
MATRIX_URL = "https://matrix.org"
|
||
|
MATRIX_USER = "@hcbot:matrix.org"
|
||
|
MATRIX_PASS = "W4iy$nt^qp~&$K~$w@@jEhnT"
|
||
|
|
||
|
creds = botlib.Creds(MATRIX_URL, MATRIX_USER, MATRIX_PASS)
|
||
|
bot = botlib.Bot(creds)
|
||
|
|
||
|
PREFIX = '!'
|
||
|
|
||
|
# Setup logging
|
||
|
logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
|
||
|
|
||
|
# Database setup
|
||
|
conn = sqlite3.connect('karma.db')
|
||
|
c = conn.cursor()
|
||
|
c.execute('''CREATE TABLE IF NOT EXISTS karma
|
||
|
(username TEXT PRIMARY KEY, points INTEGER)''')
|
||
|
conn.commit()
|
||
|
conn.close()
|
||
|
|
||
|
async def ping_proxy(proxy):
|
||
|
"""
|
||
|
Ping a proxy server to check its availability.
|
||
|
|
||
|
Args:
|
||
|
proxy (str): The proxy server address in the format 'ip:port'.
|
||
|
|
||
|
Returns:
|
||
|
float or str: The ping time in milliseconds if the proxy is up, or 'N/A' if it's down.
|
||
|
"""
|
||
|
ip, port = proxy.split(':')
|
||
|
start_time = time.time()
|
||
|
try:
|
||
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
sock.settimeout(5)
|
||
|
result = sock.connect_ex((ip, int(port)))
|
||
|
if result == 0:
|
||
|
return round((time.time() - start_time) * 1000, 2)
|
||
|
except Exception as e:
|
||
|
logging.error(f"Error pinging proxy {proxy}: {e}")
|
||
|
return "N/A"
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def commands(room, message):
|
||
|
"""
|
||
|
Command to display the list of available commands and their descriptions.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot, PREFIX)
|
||
|
if match.is_not_from_this_bot() and match.prefix() and match.command("commands"):
|
||
|
logging.info("Fetching commands documentation")
|
||
|
commands_message = """
|
||
|
**Available Commands:**
|
||
|
|
||
|
**!fortune**
|
||
|
Returns a random fortune message.
|
||
|
|
||
|
**!date**
|
||
|
Displays the current date and time.
|
||
|
|
||
|
**!proxy**
|
||
|
Retrieves and tests random SOCKS5 and HTTP proxies.
|
||
|
|
||
|
**!isup <domain/ip>**
|
||
|
Checks if the specified domain or IP address is reachable.
|
||
|
|
||
|
**!karma <user>**
|
||
|
Retrieves the karma points for the specified user.
|
||
|
|
||
|
**!karma <user> up**
|
||
|
Increases the karma points for the specified user by 1.
|
||
|
|
||
|
**!karma <user> down**
|
||
|
Decreases the karma points for the specified user by 1.
|
||
|
"""
|
||
|
await bot.api.send_markdown_message(room.room_id, commands_message)
|
||
|
logging.info("Sent commands documentation to the room")
|
||
|
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def fortune(room, message):
|
||
|
"""
|
||
|
Command to get a random fortune.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot, PREFIX)
|
||
|
if match.is_not_from_this_bot() and match.prefix() and match.command("fortune"):
|
||
|
logging.info("Received !fortune command")
|
||
|
fortune_output = subprocess.run(['/usr/games/fortune'], capture_output=True).stdout.decode('UTF-8')
|
||
|
await bot.api.send_text_message(room.room_id, fortune_output)
|
||
|
logging.info("Sent fortune to the room")
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def date(room, message):
|
||
|
"""
|
||
|
Command to get the current date and time.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot, PREFIX)
|
||
|
if match.is_not_from_this_bot() and match.prefix() and match.command("date"):
|
||
|
logging.info("Fetching current date and time")
|
||
|
current_datetime = datetime.datetime.now()
|
||
|
day_of_week = current_datetime.strftime("%A")
|
||
|
month = current_datetime.strftime("%B")
|
||
|
year = current_datetime.strftime("%Y")
|
||
|
time = current_datetime.strftime("%I:%M:%S %p")
|
||
|
date_message = f"Today is **{day_of_week}** of **{month}** in **{year}**. The time is **{time}**"
|
||
|
await bot.api.send_markdown_message(room.room_id, date_message)
|
||
|
logging.info("Sent current date and time to the room")
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def proxy(room, message):
|
||
|
"""
|
||
|
Command to get random SOCKS5 and HTTP proxies.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot, PREFIX)
|
||
|
if match.is_not_from_this_bot() and match.prefix() and match.command("proxy"):
|
||
|
logging.info("Received !proxy command")
|
||
|
socks5_proxies = requests.get('https://raw.githubusercontent.com/TheSpeedX/SOCKS-List/master/socks5.txt').text.splitlines()
|
||
|
http_proxies = requests.get('https://raw.githubusercontent.com/TheSpeedX/SOCKS-List/master/http.txt').text.splitlines()
|
||
|
|
||
|
random_socks5_proxy = random.choice(socks5_proxies)
|
||
|
random_http_proxy = random.choice(http_proxies)
|
||
|
|
||
|
socks5_ping = await ping_proxy(random_socks5_proxy)
|
||
|
http_ping = await ping_proxy(random_http_proxy)
|
||
|
|
||
|
await bot.api.send_text_message(room.room_id, f"Random SOCKS5 Proxy: {random_socks5_proxy} - Ping time: {socks5_ping} ms")
|
||
|
await bot.api.send_text_message(room.room_id, f"Random HTTP Proxy: {random_http_proxy} - Ping time: {http_ping} ms")
|
||
|
logging.info("Sent proxies to the room")
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def isup(room, message):
|
||
|
"""
|
||
|
Command to check if a website or server is up.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot, PREFIX)
|
||
|
if match.is_not_from_this_bot() and match.prefix() and match.command("isup"):
|
||
|
logging.info("Received !isup command")
|
||
|
args = match.args()
|
||
|
if len(args) != 1:
|
||
|
await bot.api.send_text_message(room.room_id, "Usage: !isup <ipv4/ipv6/domain>")
|
||
|
logging.info("Sent usage message to the room")
|
||
|
return
|
||
|
|
||
|
target = args[0]
|
||
|
try:
|
||
|
response = os.system(f"ping -c 1 {target}")
|
||
|
if response == 0:
|
||
|
await bot.api.send_text_message(room.room_id, f"{target} is up")
|
||
|
else:
|
||
|
await bot.api.send_text_message(room.room_id, f"{target} is down")
|
||
|
logging.info(f"Sent status of {target} to the room")
|
||
|
except Exception as e:
|
||
|
await bot.api.send_text_message(room.room_id, f"Error: {e}")
|
||
|
logging.error(f"Error occurred while checking {target}: {e}")
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def karma(room, message):
|
||
|
"""
|
||
|
Command to manage karma points for users.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot, PREFIX)
|
||
|
if match.is_not_from_this_bot() and match.prefix() and match.command("karma"):
|
||
|
logging.info("Received !karma command")
|
||
|
args = match.args()
|
||
|
sender = str(message.sender)
|
||
|
|
||
|
if len(args) == 0:
|
||
|
# Query sender's own karma
|
||
|
conn = sqlite3.connect('karma.db')
|
||
|
c = conn.cursor()
|
||
|
c.execute('''CREATE TABLE IF NOT EXISTS karma
|
||
|
(username TEXT PRIMARY KEY, points INTEGER)''')
|
||
|
c.execute('''INSERT OR IGNORE INTO karma (username, points) VALUES (?, ?)''', (sender, 0))
|
||
|
c.execute('''SELECT points FROM karma WHERE username = ?''', (sender,))
|
||
|
points = c.fetchone()[0]
|
||
|
conn.close()
|
||
|
await bot.api.send_text_message(room.room_id, f"{sender}'s karma points: {points}")
|
||
|
logging.info(f"Sent {sender}'s karma points ({points}) to the room")
|
||
|
elif len(args) == 1:
|
||
|
username = args[0]
|
||
|
|
||
|
if username == sender:
|
||
|
await bot.api.send_text_message(room.room_id, "You cannot modify your own karma.")
|
||
|
logging.info("Sent self-modification warning message to the room")
|
||
|
return
|
||
|
|
||
|
conn = sqlite3.connect('karma.db')
|
||
|
c = conn.cursor()
|
||
|
c.execute('''CREATE TABLE IF NOT EXISTS karma
|
||
|
(username TEXT PRIMARY KEY, points INTEGER)''')
|
||
|
c.execute('''INSERT OR IGNORE INTO karma (username, points) VALUES (?, ?)''', (username, 0))
|
||
|
c.execute('''SELECT points FROM karma WHERE username = ?''', (username,))
|
||
|
points = c.fetchone()[0]
|
||
|
conn.close()
|
||
|
await bot.api.send_text_message(room.room_id, f"{username}'s karma points: {points}")
|
||
|
logging.info(f"Sent {username}'s karma points ({points}) to the room")
|
||
|
elif len(args) == 2:
|
||
|
username, action = args
|
||
|
if action not in ['up', 'down']:
|
||
|
await bot.api.send_text_message(room.room_id, "Invalid action. Use 'up' or 'down'.")
|
||
|
logging.info("Sent invalid action message to the room")
|
||
|
return
|
||
|
|
||
|
if username == sender:
|
||
|
await bot.api.send_text_message(room.room_id, "You cannot modify your own karma.")
|
||
|
logging.info("Sent self-modification warning message to the room")
|
||
|
return
|
||
|
|
||
|
conn = sqlite3.connect('karma.db')
|
||
|
c = conn.cursor()
|
||
|
c.execute('''CREATE TABLE IF NOT EXISTS karma
|
||
|
(username TEXT PRIMARY KEY, points INTEGER)''')
|
||
|
|
||
|
if action == 'up':
|
||
|
c.execute('''UPDATE karma SET points = points + 1 WHERE username = ?''', (username,))
|
||
|
else:
|
||
|
c.execute('''UPDATE karma SET points = points - 1 WHERE username = ?''', (username,))
|
||
|
|
||
|
conn.commit()
|
||
|
c.execute('''SELECT points FROM karma WHERE username = ?''', (username,))
|
||
|
points = c.fetchone()[0]
|
||
|
conn.close()
|
||
|
|
||
|
await bot.api.send_text_message(room.room_id, f"{username}'s karma points: {points}")
|
||
|
logging.info(f"Sent {username}'s karma points ({points}) to the room")
|
||
|
else:
|
||
|
await bot.api.send_text_message(room.room_id, "Usage: !karma [username] [up/down]")
|
||
|
logging.info("Sent usage message to the room")
|
||
|
|
||
|
|
||
|
import re
|
||
|
|
||
|
@bot.listener.on_message_event
|
||
|
async def youtube(room, message):
|
||
|
"""
|
||
|
Command to fetch YouTube video information from links.
|
||
|
|
||
|
Args:
|
||
|
room (Room): The Matrix room where the command was invoked.
|
||
|
message (RoomMessage): The message object containing the command.
|
||
|
|
||
|
Returns:
|
||
|
None
|
||
|
"""
|
||
|
match = botlib.MessageMatch(room, message, bot)
|
||
|
if match.is_not_from_this_bot() and re.search(r'youtube\.com/watch\?v=', message.body):
|
||
|
logging.info("YouTube link detected")
|
||
|
video_id_match = re.search(r'youtube\.com/watch\?v=([^\s]+)', message.body)
|
||
|
if video_id_match:
|
||
|
video_id = video_id_match.group(1)
|
||
|
youtube_url = f"https://www.youtube.com/watch?v={video_id}"
|
||
|
logging.info(f"Fetching information for YouTube video: {youtube_url}")
|
||
|
try:
|
||
|
video = YouTube(youtube_url)
|
||
|
title = video.title
|
||
|
description = video.description
|
||
|
length = video.length
|
||
|
views = video.views
|
||
|
author = video.author
|
||
|
info_message = f"""Title: {title} | Length: {length} seconds | Views: {views} | Description: {description}"""
|
||
|
await bot.api.send_markdown_message(room.room_id, info_message)
|
||
|
logging.info("Sent YouTube video information to the room")
|
||
|
except Exception as e:
|
||
|
logging.error(f"Error fetching YouTube video information: {str(e)}")
|
||
|
await bot.api.send_text_message(room.room_id, "Error fetching YouTube video information.")
|
||
|
|
||
|
|
||
|
bot.run()
|