374 lines
12 KiB
Python
374 lines
12 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
import json
|
||
import traceback
|
||
import pretty_errors
|
||
|
||
from json import load, dump
|
||
from os import listdir, getenv
|
||
from os.path import isfile, join, getsize
|
||
|
||
import discord
|
||
from discord.utils import get
|
||
from discord.ext import commands, tasks
|
||
from discord import app_commands
|
||
|
||
from random import randint, shuffle, choice, seed
|
||
from time import time
|
||
from inspect import _empty
|
||
import motor.motor_asyncio
|
||
from loguru import logger
|
||
from discord.ext.commands import HybridCommand
|
||
from os.path import splitext
|
||
|
||
import re
|
||
import os
|
||
import ast
|
||
import nekos
|
||
import typing
|
||
import asyncio
|
||
import requests
|
||
import subprocess
|
||
import Levenshtein
|
||
import DiscordUtils
|
||
from utils.emojis import *
|
||
from utils.checks import *
|
||
from cogs.music import *
|
||
TOKEN = getenv('NATSUKO_TOKEN')
|
||
my_id = 710089471835504672
|
||
|
||
|
||
db = motor.motor_asyncio.AsyncIOMotorClient('localhost', 27017).Koteika
|
||
|
||
if os.path.exists('prefixes.json'):
|
||
with open('prefixes.json', 'r') as f:
|
||
prefs = load(f)
|
||
|
||
if not os.path.exists('tmp'):
|
||
os.mkdir('tmp')
|
||
|
||
else:
|
||
prefs = {"default": "*"}
|
||
|
||
|
||
def NFIF(path):
|
||
return len([f for f in os.listdir(path)
|
||
if os.path.isfile(os.path.join(path, f))])
|
||
|
||
|
||
# def func_check(v): # TODO: Выявить ошибку и использовать
|
||
# def premeow(*args, **kwargs):
|
||
# async def meow(ctx):
|
||
# try:
|
||
# await ctx.message.add_reaction(loading)
|
||
# await func(*args, **kwargs)
|
||
# await ctx.message.add_reaction(check_mark)
|
||
# except Exception as e:
|
||
# print(repr(e))
|
||
# await ctx.message.add_reaction(XX)
|
||
#
|
||
# await asyncio.sleep(3)
|
||
# await ctx.message.delete()
|
||
# return meow
|
||
# return premeow
|
||
|
||
|
||
def insert_returns(body):
|
||
if isinstance(body[-1], ast.Expr):
|
||
body[-1] = ast.Return(body[-1].value)
|
||
ast.fix_missing_locations(body[-1])
|
||
|
||
if isinstance(body[-1], ast.If):
|
||
insert_returns(body[-1].body)
|
||
insert_returns(body[-1].orelse)
|
||
|
||
if isinstance(body[-1], ast.With):
|
||
insert_returns(body[-1].body)
|
||
|
||
def region_to_str(_): return "RU" # TODO: Заменить на нормальный код -_-
|
||
|
||
async def add_user_to_DB(member):
|
||
# if member is not discord.Member: return
|
||
if not await db.members.find_one({"id": member.id}):
|
||
await db.members.insert_one({"id": member.id,
|
||
"name": member.name,
|
||
"access_level": "gray",
|
||
"language": region_to_str("Meow"), # TODO: Ну, очевидно
|
||
"money": 0,
|
||
"exp": 0,
|
||
"level": 0,
|
||
"history": {
|
||
"hour": {},
|
||
"day": {},
|
||
},
|
||
"max_level": 0,
|
||
"last_mess_time": 0,
|
||
"stocks": {},
|
||
"guild_stat": {}
|
||
})
|
||
|
||
def translate(text, region):
|
||
return text
|
||
|
||
async def prefixes(bot, message):
|
||
pattern = re.compile(r'/(?:^|[^а-яёa-z]|[,.])(?:(?:ко(?:т(?:(?:е(?:йк|ньк))|(?:ик)|(?:ан)|(?:[её]нок)|(?:я['
|
||
r'рт]))?|(?:шак)))(?:[иаыуэя]|(?:ов)|(ом))?)(?:[^а-яёa-z]|$)/gm', flags=re.IGNORECASE)
|
||
match = pattern.search(message.content)
|
||
|
||
def check():
|
||
return type(message.channel) == discord.channel.DMChannel or str(message.guild.id) not in prefs.keys()
|
||
|
||
return commands.when_mentioned_or(prefs[str(message.guild.id)] if not check() else prefs["default"],
|
||
match[0])(bot, message) if match else \
|
||
commands.when_mentioned_or(prefs[str(message.guild.id)] if not check() else prefs["default"])(bot, message)
|
||
|
||
intents=discord.Intents.all()
|
||
bot = commands.Bot(command_prefix=prefixes, help_command=None, intents=intents)
|
||
|
||
|
||
bot.add_check(is_not_black)
|
||
|
||
bot.owner_id = 459823895256498186
|
||
|
||
|
||
@bot.event
|
||
async def on_ready():
|
||
bot.voice_counter = {}
|
||
|
||
await bot.load_extension("cogs.settings.server")
|
||
|
||
for cog in os.listdir('./cogs'):
|
||
if cog.endswith('.py'):
|
||
logger.info(f"Загрузка {cog}...")
|
||
await bot.load_extension(f'cogs.{splitext(cog)[0]}')
|
||
|
||
await db.members.update_one({"id": 459823895256498186}, {"$set": {"access_level": "secret"}})
|
||
|
||
if os.path.isfile('reboot'):
|
||
with open('reboot', 'r') as f:
|
||
try:
|
||
data = f.readline()
|
||
data = data.split()
|
||
mes = await bot.get_channel(int(data[0])).fetch_message(int(data[1]))
|
||
await mes.clear_reaction(loading)
|
||
await mes.add_reaction(check_mark)
|
||
|
||
await asyncio.sleep(3)
|
||
await mes.delete()
|
||
except: pass
|
||
|
||
await bot.get_channel(773270764522045460) \
|
||
.send(embed=discord.Embed(title="Бот перезагружен",
|
||
color=discord.Color.red()))
|
||
os.remove('reboot')
|
||
else:
|
||
await bot.get_channel(773270764522045460) \
|
||
.send(embed=discord.Embed(title="Бот запущен",
|
||
color=discord.Color.red()))
|
||
|
||
logger.info('Я живой, блин!')
|
||
|
||
|
||
async def on_message_handler(message):
|
||
channel = message.channel
|
||
user = message.author
|
||
|
||
global last_channel
|
||
last_channel = channel
|
||
|
||
# Обработка команд (пре-подготовка)
|
||
if user.name != 'Котейка':
|
||
await add_user_to_DB(user)
|
||
|
||
|
||
@bot.event
|
||
async def on_message(message):
|
||
await bot.process_commands(message)
|
||
await on_message_handler(message)
|
||
|
||
CSF = discord.Object(id=822157545622863902)
|
||
|
||
|
||
@bot.command()
|
||
@commands.has_guild_permissions(manage_messages=True)
|
||
async def clear(ctx, count: int):
|
||
count += 1
|
||
|
||
while count > 100:
|
||
await ctx.channel.purge(limit=100)
|
||
count -= 100
|
||
await ctx.channel.purge(limit=count)
|
||
|
||
|
||
|
||
@bot.command(name='eval')
|
||
@commands.check(is_secret)
|
||
async def eval_fn(ctx, timeout: typing.Optional[int] = 10, *, cmd):
|
||
fn_name = "_eval_expr"
|
||
|
||
cmd = cmd.strip("` ")
|
||
|
||
# add a layer of indentation
|
||
cmd = "\n".join(f" {i}" for i in cmd.splitlines())
|
||
|
||
# wrap in async def body
|
||
body = f"async def {fn_name}():\n{cmd}"
|
||
|
||
parsed = ast.parse(body)
|
||
body = parsed.body[0].body
|
||
|
||
insert_returns(body)
|
||
|
||
env = {
|
||
'bot': ctx.bot,
|
||
'discord': discord,
|
||
'commands': commands,
|
||
'ctx': ctx,
|
||
'__import__': __import__,
|
||
'on_message_handler': on_message_handler,
|
||
'get': get
|
||
}
|
||
exec(compile(parsed, filename="<ast>", mode="exec"), env)
|
||
|
||
try:
|
||
await eval(f"{fn_name}()", env)
|
||
except Exception as e:
|
||
await ctx.message.reply(repr(e))
|
||
|
||
|
||
@bot.command()
|
||
@commands.is_owner()
|
||
async def change_level(ctx, user: typing.Union[discord.Member, int], level):
|
||
if 'int' in str(type(user)):
|
||
id_ = user
|
||
else:
|
||
id_ = user.id
|
||
|
||
if not level in ('secret', 'white', 'gray', 'black'):
|
||
await ctx.message.add_reaction(XX)
|
||
raise TypeError
|
||
await db.members.update_one({"id": id_}, {"$set": {"access_level": level}})
|
||
await ctx.message.add_reaction(check_mark)
|
||
|
||
await asyncio.sleep(3)
|
||
await ctx.message.delete()
|
||
|
||
|
||
@bot.command(aliases=["reload"], brief="Перезагружает бота")
|
||
@commands.check(is_secret)
|
||
async def reboot(ctx):
|
||
if os.name == 'posix':
|
||
await ctx.message.add_reaction(loading)
|
||
with open('reboot', 'w') as f:
|
||
f.write(f"{ctx.message.channel.id} {ctx.message.id}")
|
||
os.system('pm2 restart Koteika')
|
||
|
||
|
||
@bot.command(brief="Управление консолью")
|
||
@commands.check(is_secret)
|
||
async def cmd(ctx, *, command):
|
||
try:
|
||
proc = await asyncio.create_subprocess_shell(command,
|
||
shell=True,
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.PIPE,
|
||
stdin=subprocess.PIPE)
|
||
proc_data = await proc.communicate()
|
||
e = discord.Embed(title="Скрипт завершился без ошибок" if proc_data[1] == b'' else "Ошибка",
|
||
color=discord.Color.green() if proc_data[1] == b'' else discord.Color.red())
|
||
e.add_field(name="STDOUT", value=f"```{proc_data[0][-994:].decode('utf-8')}```") \
|
||
if proc_data[0] != b'' else None
|
||
e.add_field(name="STDERR", value=f"```{proc_data[1][-994:].decode('utf-8')}```") \
|
||
if proc_data[1] != b'' else None
|
||
|
||
await ctx.message.reply(embed=e)
|
||
except Exception as e:
|
||
logger.error(repr(e))
|
||
|
||
|
||
@bot.command(aliases=["vmute", "v_mute", "v_m", "vm"], brief="Выключает микрфоны всех участников ГК")
|
||
@commands.has_guild_permissions(administrator=True)
|
||
async def voice_mute(ctx):
|
||
channel = ctx.message.author.voice.channel
|
||
|
||
if channel is not None:
|
||
for i in channel.members:
|
||
await i.edit(mute=True)
|
||
await ctx.message.add_reaction(check_mark)
|
||
|
||
else:
|
||
await channel.send("Ты не находишься в ГК")
|
||
await ctx.message.add_reaction(XX)
|
||
|
||
await asyncio.sleep(3)
|
||
await ctx.message.delete()
|
||
|
||
|
||
@bot.command(aliases=["vunmute", "v_unmute", "v_unm", "vu"], brief="Влючает микрфоны всех участников ГК")
|
||
@commands.has_guild_permissions(administrator=True)
|
||
async def voice_unmute(ctx):
|
||
channel = ctx.message.author.voice.channel
|
||
|
||
if channel is not None:
|
||
for i in channel.members:
|
||
await i.edit(mute=False)
|
||
await ctx.message.add_reaction(check_mark)
|
||
|
||
else:
|
||
await channel.send("Ты не находишься в ГК")
|
||
await ctx.message.add_reaction(XX)
|
||
|
||
await asyncio.sleep(3)
|
||
await ctx.message.delete()
|
||
|
||
|
||
@bot.command(brief="Выводит права участника сервера (по умолчанию - кота)")
|
||
async def perms(ctx, user: typing.Union[discord.Member, int] = None):
|
||
if user is None:
|
||
user = ctx.author.id
|
||
elif type(user) == discord.Member:
|
||
user = user.id
|
||
|
||
p = {}
|
||
out = ''
|
||
for i in ['create_instant_invite', 'kick_members', 'ban_members', 'administrator', 'manage_channels',
|
||
'manage_guild', 'add_reactions', 'view_audit_log', 'priority_speaker', 'stream', 'read_messages',
|
||
'send_messages', 'send_tts_messages', 'manage_messages', 'embed_links', 'attach_files',
|
||
'read_message_history', 'mention_everyone', 'external_emojis', 'view_guild_insights', 'connect', 'speak',
|
||
'mute_members', 'deafen_members', 'move_members', 'use_voice_activation', 'change_nickname',
|
||
'manage_nicknames', 'manage_roles', 'manage_webhooks', 'manage_emojis']:
|
||
p[i] = False
|
||
|
||
perms = ctx.guild.get_member(user).guild_permissions
|
||
for i in iter(perms):
|
||
p[i[0]] = i[1]
|
||
|
||
for i in p.keys():
|
||
if p[i]:
|
||
out += '+ '
|
||
else:
|
||
out += '- '
|
||
out += i + '\n'
|
||
|
||
await ctx.send('```diff\n' + out + '```')
|
||
|
||
|
||
@bot.command()
|
||
@commands.has_guild_permissions(administrator=True)
|
||
async def change_prefix(ctx, prefix):
|
||
prefs[str(ctx.guild.id)] = prefix
|
||
|
||
with open('prefixes.json', 'w') as f:
|
||
dump(prefs, f)
|
||
|
||
|
||
async def check(itr: discord.Interaction):
|
||
if not itr.channel.permissions_for(itr.guild.me).send_messages:
|
||
await itr.response.send_message("Commands are not allowed here", ephemeral=True)
|
||
return False
|
||
return True
|
||
|
||
bot.tree.interaction_check = check
|
||
|
||
bot.run(TOKEN)
|