xync-bot 0.0.9__tar.gz → 0.2.9__tar.gz

{xync-bot-0.0.9 → xync_bot-0.2.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xync-bot
-Version: 0.0.9
+Version: 0.2.9
 Summary: Telegram bot with web app for xync net
 Author-email: Artemiev <mixartemev@gmail.com>
 License: MIT
@@ -10,7 +10,6 @@ Keywords: aiogram,xync-net
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: aiogram
-Requires-Dist: antifragility-schema
 Requires-Dist: python-dotenv
-Requires-Dist: tortoise-api-model
-Requires-Dist: python-wireguard
+Requires-Dist: tortoise-api
+Requires-Dist: xync-schema

{xync-bot-0.0.9 → xync_bot-0.2.9}/pyproject.toml

@@ -10,12 +10,11 @@ readme = "README.md"
 license = {text = "MIT"}
 dependencies = [
     "aiogram",
-    "antifragility-schema",
     "python-dotenv",
-    "tortoise-api-model",
-    "python-wireguard"
+    "tortoise-api",
+    "xync-schema",
 ]
-version = "0.0.9"
+version = "0.2.9"
 
 [project.urls]
 Homepage = "https://gitlab.com/xync/back/tg-bot"

{xync-bot-0.0.9 → xync_bot-0.2.9}/xync_bot/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.0.5"
+from aiogram.client.default import DefaultBotProperties
 from aiogram.types import MenuButtonWebApp, WebAppInfo
 import logging
 from os import getenv as env
@@ -10,12 +10,10 @@ from xync_bot.handlers import main, vpn
 
 load_dotenv()
 
-TOKEN = env('TOKEN')
 logging.basicConfig(filemode='a', level=logging.DEBUG)
-bot = Bot(token=TOKEN, parse_mode='Markdown')
+
+bot = Bot(token=env('TOKEN'), default=DefaultBotProperties(parse_mode='Markdown'))
 dp = Dispatcher(bot=bot)
-dp.include_routers(vpn)
-dp.include_routers(main)
 
 
 async def on_startup(wh_url: str, twa_url: str, cn: AsyncpgDBClient, mbt: str = 'Go!'):
@@ -24,16 +22,18 @@ async def on_startup(wh_url: str, twa_url: str, cn: AsyncpgDBClient, mbt: str =
     """ WEBHOOK SETUP """
     webhook_info = await bot.get_webhook_info()
     if webhook_info.url != wh_url:
-        await bot.set_webhook(url=wh_url)
+        await bot.set_webhook(url=wh_url, drop_pending_updates=True)
     """ WEBAPP URL SETUP IN MENU """
     await bot.set_chat_menu_button(menu_button=MenuButtonWebApp(text=mbt, web_app=WebAppInfo(url=twa_url)))
 
 
 async def on_shutdown():
     """ CLOSE BOT SESSION """
-    await bot.delete_webhook()
+    await bot.delete_webhook(drop_pending_updates=True)
     await bot.session.close()
 
 
 dp.startup.register(on_startup)
 dp.shutdown.register(on_shutdown)
+
+dp.include_routers(vpn, main)

xync_bot-0.2.9/xync_bot/handlers/main.py

@@ -0,0 +1,117 @@
+import logging
+from aiogram import Router, F
+from aiogram.filters import CommandStart, CommandObject
+from aiogram.filters.callback_data import CallbackData
+from aiogram.types import User as TgUser, ChatMemberUpdated, Message, InlineKeyboardMarkup, InlineKeyboardButton, CallbackQuery
+from aiogram.utils.deep_linking import create_start_link
+from aiogram.utils.web_app import WebAppUser
+from xync_schema.models import User, UserStatus, Lang
+
+from xync_bot.shared import NavCallbackData
+
+main = Router()
+
+
+class RrCallbackData(CallbackData, prefix="reg_res"):  # registration response
+    to: int
+    res: bool
+
+
+home_btns = InlineKeyboardMarkup(
+    inline_keyboard=[[
+        InlineKeyboardButton(text='Invite', callback_data=NavCallbackData(to='ref_link').pack()),
+        InlineKeyboardButton(text='Get VPN', callback_data=NavCallbackData(to='get_vpn').pack()),
+    ]])
+
+
+@main.message(CommandStart(deep_link=True, deep_link_encoded=True))
+async def start_handler(msg: Message, command: CommandObject):
+    me: TgUser = msg.from_user
+    ref_id: int = command.args.isnumeric() and int(command.args)
+    user = await User.get_or_none(id=me.id, status__gte=UserStatus.RESTRICTED)
+    rm = None
+    if user:
+        rs, rm = f'{me.full_name}, you have registered already😉', home_btns
+    elif not (ref := await User.get_or_none(id=ref_id)):
+        rs = f'No registered user #{ref_id}😬'
+    else:  # new user created
+        user, cr = await user_upsert(me)
+        await user.update_from_dict({'ref': ref}).save()
+        approve_btns = InlineKeyboardMarkup(
+            inline_keyboard=[[
+                InlineKeyboardButton(text='Отклонить', callback_data=RrCallbackData(to=user.id, res=False).pack()),
+                InlineKeyboardButton(text='Одобрить', callback_data=RrCallbackData(to=user.id, res=True).pack())
+            ]])
+        await msg.bot.send_message(ref.id, f'{me.full_name} просит что б Вы взяли за него/ее ответственность', reply_markup=approve_btns)
+        return await msg.answer(f'Please wait for @{ref.username} approving...')
+    return await msg.answer(rs, reply_markup=rm)
+
+
+@main.callback_query(RrCallbackData.filter())
+async def phrases_input_request(cb: CallbackQuery, callback_data: RrCallbackData) -> None:
+    protege = await User[callback_data.to]
+    if callback_data.res:
+        # protege.status = UserStatus.RESTRICTED
+        await protege.save()
+        rs = f'{cb.from_user.full_name}, теперь Вы несете ответветвенность за {protege.username}'
+    else:
+        rs = f'Вы отклонили запрос юзера "{protege.username}" на Вашу протекцию'
+    res = {True: 'одобрил', False: 'отклонил'}
+    txt = f'{cb.from_user.full_name} {res[callback_data.res]} вашу регистрацию'
+    txt, rm = (f'Поздравляем! {txt}💥', home_btns) if callback_data.res else (f'К сожалению {txt}😢', None)
+    await cb.bot.send_message(protege.id, txt, reply_markup=rm)
+    await cb.answer('👌🏼')
+    await cb.message.edit_text(rs)
+
+
+@main.message(CommandStart(deep_link=True))  # attempt to reg by fake link
+async def fraud_handler(msg: Message):
+    logging.info(f'Start: {msg.from_user.id}. Msg: {msg}')
+    # todo: alert to admins! Fraud attempt!
+    await msg.answer('🤔')
+
+
+@main.message(CommandStart())
+async def start_no_ref_handler(msg: Message):
+    me = msg.from_user
+    user, cr = await user_upsert(me)
+    rr = 'сначала вы должны найти поручителя, и перейти по его реферальной ссылке.\nhttps://telegra.ph/XyncNet-02-13'
+    if cr:  # has ref and created now
+        await msg.answer(f'Здравствуйте {me.full_name}, что бы использовать возможности нашей сети, {rr}')
+    elif not user.ref_id:
+        await msg.answer(rr.capitalize())
+    else:
+        await msg.answer(f'{me.full_name}, не балуйтесь, вы и так уже активный участник👌🏼', reply_markup=home_btns)
+
+
+@main.callback_query(NavCallbackData.filter(F.to == 'ref_link'))
+async def ref_link_handler(cbq: CallbackQuery):
+    me = cbq.from_user
+    if not (u := await User.get_or_none(id=me.id, status__gt=UserStatus.RESTRICTED).prefetch_related('ref')):
+        return await cbq.answer(f"{me.full_name}, сначала сами получите одобрение поручителя😉")
+    link = await create_start_link(cbq.bot, str(u.id), encode=True)
+    logging.info(f'Start: {me.id}. Msg: {cbq}')
+    await cbq.message.answer(f"Your referrer is {u.ref_id and u.ref.username}"
+                             f"\nThis is your invite link: {link}"
+                             f"\nGive it to your protege, and approve his request")
+    await cbq.answer('Wait for your protege request..')
+
+
+async def user_upsert(u: TgUser | WebAppUser, status: UserStatus = None) -> (User, bool):
+    pic = (gpp := await u.get_profile_photos(0, 1)).photos and gpp.photos[0][-1].file_unique_id if type(u) is TgUser else (u.photo_url[0] if u.photo_url else None)
+    udf = {'username': u.username, 'first_name': u.first_name, 'last_name': u.last_name, 'status': UserStatus.MEMBER, 'lang': u.language_code and Lang[u.language_code], 'pic': pic}
+    if status:
+        udf.update({'status': status})
+    return await User.update_or_create(udf, id=u.id)
+
+
+@main.my_chat_member()
+async def user_set_status(my_chat_member: ChatMemberUpdated):
+    u: TgUser = my_chat_member.from_user
+    new_status = my_chat_member.new_chat_member.status
+    await user_upsert(u, status=new_status)
+
+
+@main.message()
+async def del_cbq(msg: Message):
+    await msg.delete()

xync_bot-0.2.9/xync_bot/handlers/vpn.py

@@ -0,0 +1,48 @@
+from aiogram import Router, F
+from aiogram.types import BufferedInputFile, CallbackQuery
+from xync_schema.models import User, Vpn, UserStatus
+from os import getenv as env
+from dotenv import load_dotenv
+
+from xync_bot.shared import NavCallbackData
+
+load_dotenv()
+
+vpn = Router()
+
+
+@vpn.callback_query(NavCallbackData.filter(F.to == 'get_vpn'))
+async def get_vpn(cbq: CallbackQuery):
+    me = cbq.from_user
+    if not (u := await User.get_or_none(id=me.id, status__gt=UserStatus.RESTRICTED).prefetch_related('vpn')):
+        return await cbq.answer('Этот сервис только для наших доверенных участников. Что бы зарегистрироваться, '
+                                'вы должны перейти в бот по пригласительный ссылке от одного из действующих участников сети,'
+                                'и дождаться пока он одобрит вашу регистрацию.')
+    if not u.vpn:
+        try:
+            from python_wireguard import Key
+            private, public = Key.key_pair()
+        except OSError:
+            private, public = 'local_test_private_key_'+me.username, 'local_test_public_key_'+me.username
+        v = await Vpn.create(priv=private, pub=public, user=u)
+    else:
+        v = u.vpn
+        private = v.priv
+    txt = f'''[Interface]
+PrivateKey={private}
+Address=10.0.0.{v.id}/24
+DNS=8.8.8.8
+[Peer]
+PublicKey={env('SWGPUB')}
+AllowedIPs=0.0.0.0/0
+Endpoint=vpn.xync.net:51820
+PersistentKeepalive=60
+'''
+    file = BufferedInputFile(txt.encode(), f'XyncVPN_{u.username}.conf')
+    caption = 'Import this config file to Wireguard:\n[iOS](https://apps.apple.com/us/app/wireguard/id1441195209)\n' \
+              '[Android](https://play.google.com/store/apps/details?id=com.wireguard.android)\n' \
+              '[MacOS](https://itunes.apple.com/us/app/wireguard/id1451685025)\n' \
+              '[Windows](https://download.wireguard.com/windows-client/wireguard-installer.exe)\n\n' \
+              '[Instructions](https://graph.org/VPN-02-19)'
+    await cbq.message.answer_document(file, caption=caption)
+    await cbq.answer('Your VPN is ready')

xync_bot-0.2.9/xync_bot/shared.py

@@ -0,0 +1,5 @@
+from aiogram.filters.callback_data import CallbackData
+
+
+class NavCallbackData(CallbackData, prefix="nav"):  # navigate menu
+    to: str

{xync-bot-0.0.9 → xync_bot-0.2.9}/xync_bot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xync-bot
-Version: 0.0.9
+Version: 0.2.9
 Summary: Telegram bot with web app for xync net
 Author-email: Artemiev <mixartemev@gmail.com>
 License: MIT
@@ -10,7 +10,6 @@ Keywords: aiogram,xync-net
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 Requires-Dist: aiogram
-Requires-Dist: antifragility-schema
 Requires-Dist: python-dotenv
-Requires-Dist: tortoise-api-model
-Requires-Dist: python-wireguard
+Requires-Dist: tortoise-api
+Requires-Dist: xync-schema

{xync-bot-0.0.9 → xync_bot-0.2.9}/xync_bot.egg-info/SOURCES.txt

@@ -1,5 +1,6 @@
 pyproject.toml
 xync_bot/__init__.py
+xync_bot/shared.py
 xync_bot.egg-info/PKG-INFO
 xync_bot.egg-info/SOURCES.txt
 xync_bot.egg-info/dependency_links.txt

xync_bot-0.2.9/xync_bot.egg-info/requires.txt

@@ -0,0 +1,4 @@
+aiogram
+python-dotenv
+tortoise-api
+xync-schema

xync-bot-0.0.9/xync_bot/handlers/main.py

@@ -1,73 +0,0 @@
-import logging
-from aiogram import Router
-from aiogram.filters import CommandStart, CommandObject, Command
-from aiogram.types import User as TgUser, ChatMemberUpdated, Message
-from aiogram.utils.deep_linking import create_start_link
-from antifragility_schema.models import Client
-from tortoise_api_model import User
-from tortoise_api_model.enum import UserStatus
-
-main = Router()
-
-
-@main.message(CommandStart(deep_link=True, deep_link_encoded=True))
-async def start_handler(msg: Message, command: CommandObject):
-    me: TgUser = msg.from_user
-    ref_id: int = command.args.isnumeric() and int(command.args)
-    client: Client
-    client, cr = await user_upsert(me)
-    if cr:
-        rs = 'You have registered already😉'
-    elif client.id == ref_id:
-        rs = 'You can not to ref yourself😒'
-    elif not (ref := await Client.get_or_none(id=ref_id)):
-        rs = f'No registered user #{ref_id}😬'
-    else:
-        rs = f'Please wait for {ref.name} approving...'
-        await client.update_from_dict({'ref': ref}).save()
-    return await msg.answer(rs)
-
-
-@main.message(CommandStart(deep_link=True))
-async def fraud_handler(msg: Message):
-    logging.info(f'Start: {msg.from_user.id}. Msg: {msg}')
-    # todo: alert to admins! Fraud attempt!
-    await msg.answer('🤔')
-
-
-@main.message(CommandStart())
-async def start_no_ref_handler(msg: Message):
-    me = msg.from_user
-    # client, cr = await user_upsert(me) # todo: baskdoor for first user
-    logging.info(f'Start: {me.id}. Msg: {msg}')
-    await msg.answer(f'Sorry {me.full_name}, we do not accept only persons who has a guarantor.\n'
-                     'https://telegra.ph/XyncNet-02-13')
-
-
-@main.message(Command('ref_link'))
-async def ref_link_handler(msg: Message):
-    my_id = msg.from_user.id
-    cl = await Client.get(user_id=my_id)
-    link = await create_start_link(msg.bot, str(cl.id), encode=True)
-    logging.info(f'Start: {my_id}. Msg: {msg}')
-    await msg.answer(f"Give it to your protege: {link}")
-
-
-async def user_upsert(u: TgUser, status: UserStatus = None) -> (Client, bool):
-    udf = {'username': u.username, 'password': ''}
-    if status:
-        udf.update({'status': status})
-    user, is_created = await User.update_or_create(udf, id=u.id)
-    return await Client.update_or_create({'name': u.full_name}, user=user)
-
-
-@main.my_chat_member()
-async def user_set_status(my_chat_member: ChatMemberUpdated):
-    u: TgUser = my_chat_member.from_user
-    new_status = UserStatus[my_chat_member.new_chat_member.status]
-    await user_upsert(u, status=new_status)
-
-
-@main.message()
-async def del_msg(msg: Message):
-    await msg.delete()

xync-bot-0.0.9/xync_bot/handlers/vpn.py

@@ -1,38 +0,0 @@
-from aiogram.types import MenuButtonWebApp, WebAppInfo
-import logging
-from os import getenv as env
-from dotenv import load_dotenv
-from aiogram import Bot, Dispatcher
-from tortoise.backends.asyncpg import AsyncpgDBClient
-
-from xync_bot.handlers import main, vpn
-
-load_dotenv()
-
-TOKEN = env('TOKEN')
-logging.basicConfig(filemode='a', level=logging.DEBUG)
-bot = Bot(token=TOKEN, parse_mode='Markdown')
-dp = Dispatcher(bot=bot)
-dp.include_routers(vpn)
-dp.include_routers(main)
-
-
-async def on_startup(wh_url: str, twa_url: str, cn: AsyncpgDBClient, mbt: str = 'Go!'):
-    """ SET DEISPATCHER GLOBAL WORKFLOW DATA FOR DB Connection """
-    dp['dbc'] = cn
-    """ WEBHOOK SETUP """
-    webhook_info = await bot.get_webhook_info()
-    if webhook_info.url != wh_url:
-        await bot.set_webhook(url=wh_url)
-    """ WEBAPP URL SETUP IN MENU """
-    await bot.set_chat_menu_button(menu_button=MenuButtonWebApp(text=mbt, web_app=WebAppInfo(url=twa_url)))
-
-
-async def on_shutdown():
-    """ CLOSE BOT SESSION """
-    await bot.delete_webhook()
-    await bot.session.close()
-
-
-dp.startup.register(on_startup)
-dp.shutdown.register(on_shutdown)

xync-bot-0.0.9/xync_bot.egg-info/requires.txt

@@ -1,5 +0,0 @@
-aiogram
-antifragility-schema
-python-dotenv
-tortoise-api-model
-python-wireguard