80c519a66f
Modifies the Telegram bot's 'Add User' functionality. After adding a user, the bot now attempts to retrieve and display the Normal-SUB subscription link and its QR code if available. If Normal-SUB is not found, it falls back to showing the direct Hysteria2 IPv4 URI and QR code. The direct URI is also provided as a fallback if Normal-SUB is shown. Usernames are quoted in CLI calls for robustness. Improved input validation for username, traffic, and expiration days.
151 lines
7.5 KiB
Python
151 lines
7.5 KiB
Python
import qrcode
|
|
import io
|
|
import json
|
|
from telebot import types
|
|
from utils.command import *
|
|
from utils.common import create_main_markup
|
|
|
|
|
|
def create_cancel_markup(back_step=None):
|
|
markup = types.ReplyKeyboardMarkup(resize_keyboard=True, one_time_keyboard=True)
|
|
if back_step:
|
|
markup.row(types.KeyboardButton("⬅️ Back"))
|
|
markup.row(types.KeyboardButton("❌ Cancel"))
|
|
return markup
|
|
|
|
@bot.message_handler(func=lambda message: is_admin(message.from_user.id) and message.text == 'Add User')
|
|
def add_user(message):
|
|
msg = bot.reply_to(message, "Enter username:", reply_markup=create_cancel_markup())
|
|
bot.register_next_step_handler(msg, process_add_user_step1)
|
|
|
|
def process_add_user_step1(message):
|
|
if message.text == "❌ Cancel":
|
|
bot.reply_to(message, "Process canceled.", reply_markup=create_main_markup())
|
|
return
|
|
|
|
username = message.text.strip()
|
|
if not username:
|
|
bot.reply_to(message, "Username cannot be empty. Please enter a valid username:", reply_markup=create_cancel_markup())
|
|
bot.register_next_step_handler(message, process_add_user_step1)
|
|
return
|
|
|
|
if '\n' in username or len(username) > 50:
|
|
bot.reply_to(message, "Invalid username format. Please use a shorter username without newlines.", reply_markup=create_cancel_markup())
|
|
bot.register_next_step_handler(message, process_add_user_step1)
|
|
return
|
|
|
|
command = f"python3 {CLI_PATH} list-users"
|
|
result = run_cli_command(command)
|
|
|
|
try:
|
|
users_data = json.loads(result)
|
|
existing_users = {user_key.lower() for user_key in users_data.keys()}
|
|
if username.lower() in existing_users:
|
|
bot.reply_to(message, f"Username '{username}' already exists. Please choose a different username:", reply_markup=create_cancel_markup())
|
|
bot.register_next_step_handler(message, process_add_user_step1)
|
|
return
|
|
except json.JSONDecodeError:
|
|
if "No such file or directory" in result or result.strip() == "" or "Could not find users" in result.lower():
|
|
pass
|
|
else:
|
|
bot.reply_to(message, "Error checking existing users. Please try again.", reply_markup=create_main_markup())
|
|
return
|
|
|
|
msg = bot.reply_to(message, "Enter traffic limit (GB):", reply_markup=create_cancel_markup(back_step=process_add_user_step1))
|
|
bot.register_next_step_handler(msg, process_add_user_step2, username)
|
|
|
|
def process_add_user_step2(message, username):
|
|
if message.text == "❌ Cancel":
|
|
bot.reply_to(message, "Process canceled.", reply_markup=create_main_markup())
|
|
return
|
|
if message.text == "⬅️ Back":
|
|
msg = bot.reply_to(message, "Enter username:", reply_markup=create_cancel_markup())
|
|
bot.register_next_step_handler(msg, process_add_user_step1)
|
|
return
|
|
|
|
try:
|
|
traffic_limit = int(message.text.strip())
|
|
if traffic_limit < 0:
|
|
bot.reply_to(message, "Traffic limit cannot be negative. Please enter a valid number (GB):", reply_markup=create_cancel_markup(back_step=process_add_user_step1))
|
|
bot.register_next_step_handler(message, process_add_user_step2, username)
|
|
return
|
|
msg = bot.reply_to(message, "Enter expiration days:", reply_markup=create_cancel_markup(back_step=process_add_user_step2))
|
|
bot.register_next_step_handler(msg, process_add_user_step3, username, traffic_limit)
|
|
except ValueError:
|
|
bot.reply_to(message, "Invalid traffic limit. Please enter a number (GB):", reply_markup=create_cancel_markup(back_step=process_add_user_step1))
|
|
bot.register_next_step_handler(message, process_add_user_step2, username)
|
|
|
|
def process_add_user_step3(message, username, traffic_limit):
|
|
if message.text == "❌ Cancel":
|
|
bot.reply_to(message, "Process canceled.", reply_markup=create_main_markup())
|
|
return
|
|
if message.text == "⬅️ Back":
|
|
msg = bot.reply_to(message, "Enter traffic limit (GB):", reply_markup=create_cancel_markup(back_step=process_add_user_step1))
|
|
bot.register_next_step_handler(msg, process_add_user_step2, username)
|
|
return
|
|
|
|
try:
|
|
expiration_days = int(message.text.strip())
|
|
if expiration_days < 0:
|
|
bot.reply_to(message, "Expiration days cannot be negative. Please enter a valid number:", reply_markup=create_cancel_markup(back_step=process_add_user_step2))
|
|
bot.register_next_step_handler(message, process_add_user_step3, username, traffic_limit)
|
|
return
|
|
|
|
add_user_command = f"python3 {CLI_PATH} add-user -u \"{username}\" -t {traffic_limit} -e {expiration_days}"
|
|
add_user_feedback = run_cli_command(add_user_command).strip()
|
|
|
|
bot.send_chat_action(message.chat.id, 'typing')
|
|
|
|
lower_username = username.lower()
|
|
uri_info_command = f"python3 {CLI_PATH} show-user-uri -u \"{lower_username}\" -ip 4 -n"
|
|
uri_info_output = run_cli_command(uri_info_command)
|
|
|
|
direct_uri = None
|
|
normal_sub_link = None
|
|
|
|
if "IPv4:" in uri_info_output:
|
|
try:
|
|
parts_after_ipv4 = uri_info_output.split("IPv4:\n", 1)[1]
|
|
potential_direct_uri = parts_after_ipv4.split('\n', 1)[0].strip()
|
|
if potential_direct_uri.startswith("hy2://"):
|
|
direct_uri = potential_direct_uri
|
|
except (IndexError, AttributeError):
|
|
pass
|
|
|
|
if "Normal-SUB Sublink:" in uri_info_output:
|
|
try:
|
|
parts_after_sublink_label = uri_info_output.split("Normal-SUB Sublink:\n", 1)[1]
|
|
potential_sub_link = parts_after_sublink_label.split('\n', 1)[0].strip()
|
|
if potential_sub_link.startswith("http://") or potential_sub_link.startswith("https://"):
|
|
normal_sub_link = potential_sub_link
|
|
except (IndexError, AttributeError):
|
|
pass
|
|
|
|
caption_text = f"{add_user_feedback}\n"
|
|
link_to_generate_qr_for = None
|
|
link_type_for_caption = ""
|
|
|
|
if normal_sub_link:
|
|
link_to_generate_qr_for = normal_sub_link
|
|
link_type_for_caption = "Normal Subscription Link"
|
|
caption_text += f"\n{link_type_for_caption} for `{username}`:\n`{normal_sub_link}`"
|
|
elif direct_uri:
|
|
link_to_generate_qr_for = direct_uri
|
|
link_type_for_caption = "Hysteria2 IPv4 URI"
|
|
caption_text += f"\n{link_type_for_caption} for `{username}`:\n`{direct_uri}`"
|
|
|
|
if link_to_generate_qr_for:
|
|
qr_img = qrcode.make(link_to_generate_qr_for)
|
|
bio = io.BytesIO()
|
|
qr_img.save(bio, 'PNG')
|
|
bio.seek(0)
|
|
bot.send_photo(message.chat.id, photo=bio, caption=caption_text, parse_mode="Markdown", reply_markup=create_main_markup())
|
|
else:
|
|
caption_text += "\nCould not retrieve specific Hysteria2 URI or Subscription link details."
|
|
bot.send_message(message.chat.id, caption_text, parse_mode="Markdown", reply_markup=create_main_markup())
|
|
|
|
except ValueError:
|
|
bot.reply_to(message, "Invalid expiration days. Please enter a number:", reply_markup=create_cancel_markup(back_step=process_add_user_step2))
|
|
bot.register_next_step_handler(message, process_add_user_step3, username, traffic_limit)
|
|
except Exception as e:
|
|
bot.reply_to(message, f"An unexpected error occurred: {str(e)}", reply_markup=create_main_markup()) |