Files
Ultroid-fork/addons/chatbot.py
Cursor User 7f1ba3cdc3 Add comprehensive Docker deployment with safety features
Features:
- Complete Ubuntu dependency resolution in requirements.txt
- Safe Docker deployment that preserves existing bash startup setups
- Isolated Docker environment (docker-ultroid/) with different ports
- Automatic detection of existing bot configurations
- Session generation scripts for Docker deployment
- Health check and diagnostic tools
- Comprehensive documentation and deployment guides

 Safety:
- Detects existing 'bash startup' method and creates isolated environment
- Uses different ports (8081) to avoid conflicts
- Separate volumes and configs for Docker deployment
- Both bash startup and Docker can run side by side
- No interference with existing bot setups

 Files added/updated:
- requirements.txt (all missing dependencies)
- Docker setup (Dockerfile, docker-compose.yml, .env.sample)
- Deployment scripts (ubuntu_setup.sh, docker-deploy.sh, quick-start.sh)
- Safety scripts (safe-docker-setup.sh with isolation logic)
- Management tools (Makefile, health_check.sh, generate-session.sh)
- Documentation (SAFE_DOCKER_GUIDE.md, DOCKER_DEPLOYMENT.md, etc.)

Ready for production Ubuntu server deployment!
2025-06-18 20:42:10 +02:00

110 lines
3.2 KiB
Python

# Ultroid - UserBot
# Copyright (C) 2021-2025 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
from . import get_help
__doc__ = get_help("help_chatbot")
import asyncio
from pyUltroid.fns.tools import get_chatbot_reply
from telethon.tl.types import Channel
from . import LOGS, eod, get_string, inline_mention, udB, ultroid_cmd, ultroid_bot, events
@ultroid_cmd(pattern="repai")
async def im_lonely_chat_with_me(event):
if event.reply_to:
message = (await event.get_reply_message()).message
else:
try:
message = event.text.split(" ", 1)[1]
except IndexError:
return await eod(event, get_string("tban_1"), time=10)
reply_ = await get_chatbot_reply(message=message)
await event.eor(reply_)
@ultroid_cmd(pattern="addai")
async def add_chatBot(event):
await chat_bot_fn(event, type_="add")
@ultroid_cmd(pattern="remai")
async def rem_chatBot(event):
await chat_bot_fn(event, type_="remov")
@ultroid_cmd(pattern="listai")
async def lister(event):
key = udB.get_key("CHATBOT_USERS") or {}
users = key.get(event.chat_id, [])
if not users:
return await event.eor(get_string("chab_2"), time=5)
msg = "**Total List Of AI Enabled Users In This Chat :**\n\n"
for i in users:
try:
user = await event.client.get_entity(int(i))
user = inline_mention(user)
except BaseException:
user = f"`{i}`"
msg += f"{user}\n"
await event.eor(msg, link_preview=False)
async def chat_bot_fn(event, type_):
if event.reply_to:
user_ = (await event.get_reply_message()).sender
else:
temp = event.text.split(maxsplit=1)
try:
user_ = await event.client.get_entity(await event.client.parse_id(temp[1]))
except BaseException as er:
LOGS.exception(er)
user_ = event.chat if event.is_private else None
if not user_:
return await eod(
event,
get_string("chab_1"),
)
key = udB.get_key("CHATBOT_USERS") or {}
chat = event.chat_id
user = user_.id
if type_ == "add":
if key.get(chat):
if user not in key[chat]:
key[chat].append(user)
else:
key.update({chat: [user]})
elif type_ == "remov":
if key.get(chat):
if user in key[chat]:
key[chat].remove(user)
if chat in key and not key[chat]:
del key[chat]
udB.set_key("CHATBOT_USERS", key)
await event.eor(f"**ChatBot:**\n{type_}ed {inline_mention(user_)}")
@ultroid_bot.on(events.NewMessage(incoming=True))
async def chatBot_replies(e):
sender = await e.get_sender()
if isinstance(sender, Channel):
return
if sender.bot:
return
key = udB.get_key("CHATBOT_USERS") or {}
if e.text and key.get(e.chat_id) and sender.id in key[e.chat_id]:
msg = await get_chatbot_reply(e.message.message)
if msg:
sleep = udB.get_key("CHATBOT_SLEEP") or 1.5
await asyncio.sleep(sleep)
await e.reply(msg)