ACL, channel_charing

main.py

@@ -19,22 +19,46 @@ logging.getLogger("httpx").setLevel(logging.WARNING)
 logger = logging.getLogger(__name__)
 
 import asyncio
-
+from telegram.ext import CommandHandler
+from sqlalchemy import select
+from datetime import datetime
+from db import AsyncSessionLocal
+from models import ChannelAccess
+from handlers.permissions import get_or_create_admin, token_hash
 
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    session = AsyncSessionLocal()
-    user_id = update.effective_user.id if update.effective_user else None
-    result = await session.execute(Admin.__table__.select().where(Admin.tg_id == user_id))
-    admin = result.first() if user_id else None
-    if not admin and user_id:
-        await session.execute(Admin.__table__.insert().values(tg_id=user_id))
-        await session.commit()
-        if update.message:
-            await update.message.reply_text('Вы зарегистрированы как админ.')
-    else:
-        if update.message:
-            await update.message.reply_text('Вы уже зарегистрированы.')
-    await session.close()
+    args = (context.args or [])
+    if args and args[0].startswith("sch_"):
+        # формат: sch_<invite_id>_<token>
+        try:
+            _, sid, token = args[0].split("_", 2)
+            invite_id = int(sid)
+        except Exception:
+            await update.message.reply_text("Неверная ссылка приглашения.")
+            return
+
+        async with AsyncSessionLocal() as session:
+            me = await get_or_create_admin(session, update.effective_user.id)
+            res = await session.execute(select(ChannelAccess).where(ChannelAccess.id == invite_id))
+            acc = res.scalar_one_or_none()
+            if not acc or acc.status != "pending":
+                await update.message.reply_text("Приглашение не найдено или уже активировано/отозвано.")
+                return
+            if acc.expires_at and acc.expires_at < datetime.utcnow():
+                await update.message.reply_text("Срок действия приглашения истёк.")
+                return
+            if token_hash(token) != acc.token_hash:
+                await update.message.reply_text("Неверный токен приглашения.")
+                return
+
+            acc.invited_admin_id = me.id
+            acc.accepted_at = datetime.utcnow()
+            acc.status = "active"
+            await session.commit()
+
+        await update.message.reply_text("Доступ к каналу успешно активирован. Можно постить через /new_post.")
+        return
+
 
 async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
     help_text = (
@@ -73,6 +97,8 @@ from handlers.group_buttons import group_buttons_conv
 from handlers.channel_buttons import channel_buttons_conv
 from handlers.edit_button import edit_button
 from handlers.del_button import del_button
+from handlers.share_channel import share_channel_conv
+
 
 
 
@@ -93,6 +119,7 @@ def main():
     application.add_handler(CommandHandler('edit_button', edit_button))
     application.add_handler(CommandHandler('del_button', del_button))
     application.add_handler(admin_panel_conv)
+    application.add_handler(share_channel_conv)
     import sys
     import asyncio
     if sys.platform.startswith('win'):