add AD 2FA group check to instruct Trueconf module

utils/ad_checker.py

@@ -0,0 +1,73 @@
+# /opt/trueconf_bot/utils/ad_checker.py
+# Проверка членства пользователя в группах Active Directory
+
+
+from ldap3 import Server, Connection, ALL
+
+
+def _get_ad_connection():
+    """Создать соединение с AD."""
+    from config.config import AD_SERVER, AD_USER, AD_PASSWORD, AD_BASE  # noqa: E501
+    server = Server(AD_SERVER, get_info=ALL)
+    return Connection(server, user=AD_USER, password=AD_PASSWORD, auto_bind=True), AD_BASE
+
+
+def get_user_groups(cn: str) -> list[str] | None:
+    """
+    Найти пользователя по CN и вернуть список групп.
+    Возвращает None если пользователь не найден.
+    """
+    conn, ad_base = _get_ad_connection()
+    try:
+        conn.search(
+            search_base=ad_base,
+            search_filter=f"(cn={cn})",
+            attributes=["cn", "sAMAccountName", "memberOf"],
+        )
+        if not conn.entries:
+            return None
+        entry = conn.entries[0]
+        return [str(g) for g in entry.memberOf]
+    except Exception as e:
+        print(f"❌ Ошибка AD при поиске {cn}: {e}")
+        return None
+
+
+def check_group_membership(cn: str, group_pattern: str = "2FA") -> dict:
+    """
+    Проверить членство пользователя в группе.
+
+    Args:
+        cn: CN пользователя (например 'Krivochenko_Denis_Sergeevich')
+        group_pattern: подстрока для поиска в имени группы (по умолчанию '2FA')
+
+    Returns:
+        dict с полями:
+            found: bool — найден ли пользователь
+            cn: str | None — CN пользователя
+            sam: str | None — sAMAccountName
+            in_group: bool — в группе ли
+            groups: list[str] — все группы
+            matching_groups: list[str] — группы с паттерном
+    """
+    groups = get_user_groups(cn)
+    if groups is None:
+        return {
+            "found": False,
+            "cn": None,
+            "sam": None,
+            "in_group": False,
+            "groups": [],
+            "matching_groups": [],
+        }
+
+    matching = [g for g in groups if group_pattern.upper() in g.upper()]
+
+    return {
+        "found": True,
+        "cn": cn,
+        "sam": groups[0].split(",")[0].split("=")[-1],  # грубо из DN
+        "in_group": len(matching) > 0,
+        "groups": groups,
+        "matching_groups": matching,
+    }

utils/texts.py

@@ -586,14 +586,55 @@ def _instruct_main_menu(emojis: dict) -> str:
         f"{emojis['2']} — 📧 Почта\n\n"
         f"<i>{emojis['0']} — В главное меню</i>"
     )
-
 def _instruct_trueconf(emojis: dict) -> str:
+    """Статический текст Trueconf — без проверки группы."""
     return (
         "🖥 <b>Инструкция по Trueconf</b>\n\n"
         "Данная функция находится в разработке.\n\n"
         f"<i>{emojis['9']} — Назад\n{emojis['0']} — В главное меню</i>"
     )
 
+
+def _instruct_trueconf_with_ad(emojis: dict, cn: str) -> str:
+    """
+    Текст Trueconf с проверкой группы 2FA.
+    cn — CN пользователя из AD.
+    """
+    from utils.ad_checker import check_group_membership
+
+    result = check_group_membership(cn, "2FA")
+
+    if not result["found"]:
+        return (
+            f"🖥 <b>Инструкция по Trueconf</b>\n\n"
+            f"Пользователь {cn} не найден в Active Directory.\n\n"
+            f"<i>{emojis['9']} — Назад\n{emojis['0']} — В главное меню</i>"
+        )
+
+    if result["in_group"]:
+        body = (
+            f"🖥 <b>Инструкция по Trueconf</b>\n\n"
+            f"✅ <b>Доступ к Trueconf разрешён.</b>\n\n"
+            f"У вас есть доступ к функциям Trueconf.\n\n"
+            f"Детали:\n"
+            f"• CN: {result['cn']}\n"
+            f"• Группы 2FA: {', '.join(result['matching_groups'])}\n\n"
+            f"<i>{emojis['9']} — Назад\n{emojis['0']} — В главное меню</i>"
+        )
+    else:
+        body = (
+            f"🖥 <b>Инструкция по Trueconf</b>\n\n"
+            f"❌ <b>Доступ к Trueconf запрещён.</b>\n\n"
+            f"У вас нет группы 2FA в Active Directory.\n\n"
+            f"Детали:\n"
+            f"• CN: {result['cn']}\n"
+            f"• Групп 2FA: 0\n\n"
+            f"<i>{emojis['9']} — Назад\n{emojis['0']} — В главное меню</i>"
+        )
+
+    return body
+
+
 def _instruct_email(emojis: dict) -> str:
     return (
         "📧 <b>Инструкция по Почте</b>\n\n"
@@ -605,4 +646,5 @@ def _instruct_email(emojis: dict) -> str:
 
 INSTRUCT_MAIN_MENU_TEXT = _instruct_main_menu(EMOJI_DIGITS)
 INSTRUCT_TRUECONF_TEXT = _instruct_trueconf(EMOJI_DIGITS)
+INSTRUCT_TRUECONF_WITH_AD = _instruct_trueconf_with_ad
 INSTRUCT_EMAIL_TEXT = _instruct_email(EMOJI_DIGITS)