Updated README.md
This commit is contained in:
Hash Borgir
parent
e6b1ffb66b
commit
46c1e60794
326
hcbot.py
326
hcbot.py
@ -1,326 +0,0 @@
|
|||||||
#!/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()
|
|
||||||
Loading…
Reference in New Issue
Block a user