TelegramBot.py

#!/usr/bin/env python3
"""
Telegram бот для вивчення німецької мови рівня B2
3 місяці, 2 заняття на день (180 занять)
"""

import logging
import json
import os
from datetime import datetime, timedelta
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import (
    Application, CommandHandler, CallbackQueryHandler,
    ContextTypes, JobQueue
)

logging.basicConfig(
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    level=logging.INFO
)
logger = logging.getLogger(__name__)

# ─────────────────────────────────────────────
#  ПОВНИЙ ПЛАН КУРСУ (3 місяці × 30 днів × 2 заняття = 180 занять)
# ─────────────────────────────────────────────

COURSE = {}

# ══════════════════════════════════════════════
#  МІСЯЦЬ 1: ГРАМАТИКА + БАЗОВА ЛЕКСИКА B2
# ══════════════════════════════════════════════

# --- Тиждень 1: Складні часові форми ---
COURSE[1] = {
    "topic": "Konjunktiv II – умовний спосіб",
    "theory": (
        "📚 <b>Konjunktiv II</b>\n\n"
        "Утворення: würde + Infinitiv або спеціальні форми:\n"
        "• sein → wäre\n• haben → hätte\n• können → könnte\n"
        "• müssen → müsste\n• werden → würde\n\n"
        "<b>Вживання:</b>\n"
        "1️⃣ Нереальні умови: <i>Wenn ich Zeit hätte, würde ich lernen.</i>\n"
        "2️⃣ Ввічливе прохання: <i>Könnten Sie mir helfen?</i>\n"
        "3️⃣ Побажання: <i>Ich wünschte, ich wäre dort.</i>"
    ),
    "vocabulary": [
        ("hätte", "мав би"),("wäre", "був би"),("könnte", "міг би"),
        ("müsste", "мусив би"),("sollte", "мав би (за обов'язком)"),
        ("würde", "би (допоміжне)"),("dürfte", "смів би"),
        ("wollte", "хотів би"),("wüsste", "знав би"),("käme", "прийшов би"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Перекладіть речення:\n"
        "1. Якби я мав час, я б учив.\n"
        "2. Чи могли б ви мені допомогти?\n"
        "3. Я хотів би поїхати до Берліна.\n\n"
        "<b>Відповіді:</b>\n"
        "1. Wenn ich Zeit hätte, würde ich lernen.\n"
        "2. Könnten Sie mir helfen?\n"
        "3. Ich würde gerne nach Berlin fahren."
    )
}

COURSE[2] = {
    "topic": "Konjunktiv II у минулому",
    "theory": (
        "📚 <b>Konjunktiv II Vergangenheit</b>\n\n"
        "Утворення: hätte/wäre + Partizip II\n\n"
        "Приклади:\n"
        "• <i>Wenn ich gelernt hätte, hätte ich bestanden.</i>\n"
        "• <i>Wenn er gekommen wäre, wäre es anders.</i>\n\n"
        "<b>Порівняння:</b>\n"
        "Теперішнє: Wenn ich Zeit <u>hätte</u>...\n"
        "Минуле: Wenn ich Zeit <u>gehabt hätte</u>..."
    ),
    "vocabulary": [
        ("hätte gemacht", "зробив би"),("wäre gegangen", "пішов би"),
        ("hätte gesagt", "сказав би"),("wäre geblieben", "залишився б"),
        ("hätte gewusst", "знав би"),("wäre gekommen", "прийшов би"),
        ("hätte gesehen", "побачив би"),("wäre geworden", "став би"),
        ("hätte genommen", "взяв би"),("wäre gefahren", "поїхав би"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Утворіть Konjunktiv II Vergangenheit:\n"
        "1. Якби він прийшов, я б був щасливий.\n"
        "2. Якби вона вивчила, вона б склала іспит.\n\n"
        "<b>Відповіді:</b>\n"
        "1. Wenn er gekommen wäre, wäre ich glücklich gewesen.\n"
        "2. Wenn sie gelernt hätte, hätte sie die Prüfung bestanden."
    )
}

COURSE[3] = {
    "topic": "Пасивний стан (Passiv)",
    "theory": (
        "📚 <b>Passiv — пасивний стан</b>\n\n"
        "Утворення: werden + Partizip II\n\n"
        "<b>Часи:</b>\n"
        "• Präsens: Das Buch wird gelesen.\n"
        "• Präteritum: Das Buch wurde gelesen.\n"
        "• Perfekt: Das Buch ist gelesen worden.\n"
        "• Futur I: Das Buch wird gelesen werden.\n\n"
        "<b>З модальними дієсловами:</b>\n"
        "Das Buch muss gelesen werden."
    ),
    "vocabulary": [
        ("wird gebaut", "будується"),("wurde repariert", "був відремонтований"),
        ("ist erstellt worden", "було створено"),("wird gezeigt", "показується"),
        ("wurde gefragt", "був запитаний"),("wird erklärt", "пояснюється"),
        ("wurde benutzt", "використовувався"),("wird produziert", "виробляється"),
        ("wurde entwickelt", "був розроблений"),("wird geprüft", "перевіряється"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Перетворіть на пасив:\n"
        "1. Man baut eine neue Schule.\n"
        "2. Der Arzt hat den Patienten untersucht.\n\n"
        "<b>Відповіді:</b>\n"
        "1. Eine neue Schule wird gebaut.\n"
        "2. Der Patient ist vom Arzt untersucht worden."
    )
}

COURSE[4] = {
    "topic": "Zustandspassiv vs. Vorgangspassiv",
    "theory": (
        "📚 <b>Zwei Arten von Passiv</b>\n\n"
        "<b>Vorgangspassiv</b> (процес) — з werden:\n"
        "Die Tür wird geöffnet. (відчиняють)\n\n"
        "<b>Zustandspassiv</b> (стан) — з sein:\n"
        "Die Tür ist geöffnet. (відчинена)\n\n"
        "<b>Різниця:</b>\n"
        "werden-Passiv = дія відбувається\n"
        "sein-Passiv = результат дії"
    ),
    "vocabulary": [
        ("ist geöffnet", "відчинений"),("wird geöffnet", "відчиняють"),
        ("ist geschlossen", "зачинений"),("ist besetzt", "зайнятий"),
        ("ist verletzt", "поранений"),("ist erledigt", "виконаний"),
        ("ist fertig", "готовий"),("ist bekannt", "відомий"),
        ("ist verboten", "заборонений"),("ist erlaubt", "дозволений"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Vorgangspassiv чи Zustandspassiv?\n"
        "1. Das Fenster ___ gerade geputzt. (процес)\n"
        "2. Das Fenster ___ sauber. (стан)\n\n"
        "<b>Відповіді:</b>\n"
        "1. wird\n2. ist"
    )
}

COURSE[5] = {
    "topic": "Відносні речення (Relativsätze)",
    "theory": (
        "📚 <b>Relativsätze</b>\n\n"
        "Відносний займенник узгоджується з іменником (рід, число) та відмінюється за відмінком речення.\n\n"
        "<b>Таблиця:</b>\n"
        "м.р.: der/den/dem/dessen\n"
        "ж.р.: die/die/der/deren\n"
        "с.р.: das/das/dem/dessen\n"
        "мн.: die/die/denen/deren\n\n"
        "<b>Приклади:</b>\n"
        "Der Mann, <i>der</i> dort steht, ist mein Chef.\n"
        "Die Frau, <i>deren</i> Auto kaputt ist, ruft an."
    ),
    "vocabulary": [
        ("der Mann, der", "чоловік, який"),("die Frau, die", "жінка, яка"),
        ("das Kind, das", "дитина, яка"),("der Lehrer, dem", "вчитель, якому"),
        ("die Stadt, in der", "місто, в якому"),("das Buch, das", "книга, яка"),
        ("die Menschen, die", "люди, які"),("der Freund, dessen", "друг, чий"),
        ("die Mutter, deren", "мати, чия"),("das Haus, in dem", "будинок, в якому"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Вставте відносний займенник:\n"
        "1. Der Mann, ___ ich kenne, ist Arzt.\n"
        "2. Das Buch, ___ ich lese, ist spannend.\n"
        "3. Die Stadt, ___ ich lebe, ist schön.\n\n"
        "<b>Відповіді:</b>\n"
        "1. den\n2. das\n3. in der"
    )
}

COURSE[6] = {
    "topic": "Relativsätze з прийменниками та wo-/wer-",
    "theory": (
        "📚 <b>Відносні речення — розширення</b>\n\n"
        "<b>З прийменниками:</b>\n"
        "Das Haus, <i>in dem</i> ich wohne...\n"
        "Der Mensch, <i>mit dem</i> ich spreche...\n\n"
        "<b>wo- як відносний займенник:</b>\n"
        "Die Stadt, <i>wo</i> ich geboren bin...\n\n"
        "<b>wer/was для загальних антецедентів:</b>\n"
        "<i>Wer</i> lernt, wird erfolgreich.\n"
        "Alles, <i>was</i> er sagt, stimmt."
    ),
    "vocabulary": [
        ("wo", "де"),("womit", "чим"),("worüber", "про що"),
        ("wofür", "для чого"),("wobei", "при чому"),("woran", "на чому"),
        ("worauf", "на що"),("worin", "в чому"),("wovon", "від чого"),
        ("wozu", "для чого"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Перекладіть:\n"
        "1. Все, що він сказав, було правдою.\n"
        "2. Місто, де я народився, дуже красиве.\n\n"
        "<b>Відповіді:</b>\n"
        "1. Alles, was er sagte, war wahr.\n"
        "2. Die Stadt, wo ich geboren bin, ist sehr schön."
    )
}

COURSE[7] = {
    "topic": "Infinitivkonstruktionen (um...zu, ohne...zu, statt...zu)",
    "theory": (
        "📚 <b>Інфінітивні конструкції</b>\n\n"
        "<b>um...zu</b> — щоб:\n"
        "Er lernt, <i>um</i> besser zu werden.\n\n"
        "<b>ohne...zu</b> — не (роблячи):\n"
        "Er ging, <i>ohne</i> etwas zu sagen.\n\n"
        "<b>statt/anstatt...zu</b> — замість:\n"
        "Er schläft, <i>statt</i> zu lernen.\n\n"
        "⚠️ Суб'єкт головного та підрядного речення ОДНАКОВИЙ!"
    ),
    "vocabulary": [
        ("um zu lernen", "щоб вчитися"),("ohne zu fragen", "не питаючи"),
        ("statt zu arbeiten", "замість роботи"),("um zu verstehen", "щоб зрозуміти"),
        ("ohne zu wissen", "не знаючи"),("statt zu schlafen", "замість спати"),
        ("um zu helfen", "щоб допомогти"),("ohne zu bezahlen", "не платячи"),
        ("um zu gewinnen", "щоб виграти"),("statt zu lesen", "замість читати"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Перекладіть:\n"
        "1. Він вчиться, щоб скласти іспит.\n"
        "2. Вона пішла, нічого не сказавши.\n"
        "3. Він грає замість того, щоб вчитися.\n\n"
        "<b>Відповіді:</b>\n"
        "1. Er lernt, um die Prüfung zu bestehen.\n"
        "2. Sie ging, ohne etwas zu sagen.\n"
        "3. Er spielt, statt zu lernen."
    )
}

COURSE[8] = {
    "topic": "Partizipialkonstruktionen",
    "theory": (
        "📚 <b>Дієприкметникові конструкції</b>\n\n"
        "<b>Partizip I</b> (активна дія, одночасна):\n"
        "der schlafende Mann = Mann, der schläft\n\n"
        "<b>Partizip II</b> (пасивна/завершена дія):\n"
        "das gelesene Buch = Buch, das gelesen wurde\n\n"
        "<b>Rozширена атрибутивна конструкція:</b>\n"
        "der <i>von mir gestern gekaufte</i> Roman\n"
        "= der Roman, den ich gestern gekauft habe"
    ),
    "vocabulary": [
        ("schlafend", "сплячий"),("lachend", "сміючийся"),("weinend", "плачучий"),
        ("gelesen", "прочитаний"),("geschrieben", "написаний"),("gemacht", "зроблений"),
        ("laufend", "біжучий"),("sprechend", "говорячий"),("gekauft", "куплений"),
        ("gebaut", "побудований"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Перетворіть:\n"
        "1. das Kind, das weint → das ___ Kind\n"
        "2. der Brief, der geschrieben wurde → der ___ Brief\n\n"
        "<b>Відповіді:</b>\n"
        "1. weinende\n2. geschriebene"
    )
}

COURSE[9] = {
    "topic": "Лексика: Wirtschaft und Beruf",
    "theory": (
        "📚 <b>Економіка та професійна сфера</b>\n\n"
        "Ключова лексика для B2 у сфері роботи та економіки.\n\n"
        "<b>Корисні фрази:</b>\n"
        "• Die Wirtschaft wächst/schrumpft.\n"
        "• Das Unternehmen erzielt Gewinn/Verlust.\n"
        "• Ich bewerbe mich um eine Stelle.\n"
        "• Das Gehalt wird verhandelt."
    ),
    "vocabulary": [
        ("die Wirtschaft", "економіка"),("das Unternehmen", "підприємство"),
        ("der Arbeitsmarkt", "ринок праці"),("die Bewerbung", "заявка/заява"),
        ("das Gehalt", "зарплата"),("die Kündigung", "звільнення"),
        ("der Aufschwung", "підйом"),("die Rezession", "рецесія"),
        ("investieren", "інвестувати"),("konkurrieren", "конкурувати"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Складіть речення зі словами:\n"
        "1. Bewerbung\n2. Wirtschaft\n3. investieren\n\n"
        "<b>Зразки:</b>\n"
        "1. Ich schreibe eine Bewerbung für diese Stelle.\n"
        "2. Die deutsche Wirtschaft ist sehr stark.\n"
        "3. Das Unternehmen investiert in neue Technologien."
    )
}

COURSE[10] = {
    "topic": "Лексика: Bildung und Wissenschaft",
    "theory": (
        "📚 <b>Освіта та наука</b>\n\n"
        "<b>Корисні фрази:</b>\n"
        "• Ich studiere an der Universität.\n"
        "• Die Forschung liefert neue Erkenntnisse.\n"
        "• Das Experiment wurde durchgeführt.\n"
        "• Er promoviert in Physik."
    ),
    "vocabulary": [
        ("die Forschung", "дослідження"),("die Erkenntnis", "пізнання/висновок"),
        ("das Experiment", "експеримент"),("die Hypothese", "гіпотеза"),
        ("der Durchbruch", "прорив"),("die Abschlussarbeit", "дипломна робота"),
        ("promovieren", "захищати докторат"),("das Stipendium", "стипендія"),
        ("die Fachkraft", "фахівець"),("der Lehrplan", "навчальний план"),
    ],
    "exercise": (
        "✏️ <b>Вправа:</b> Перекладіть:\n"
        "1. Нові дослідження принесли важливі висновки.\n"
        "2. Вона захищає докторат з хімії.\n\n"
        "<b>Відповіді:</b>\n"
        "1. Neue Forschungen haben wichtige Erkenntnisse gebracht.\n"
        "2. Sie promoviert in Chemie."
    )
}

# Продовжуємо додавати заняття 11-180
topics_month1 = [
    ("Підрядні речення часу (als, wenn, während, nachdem, bevor)", "Temporalsätze"),
    ("Підрядні причини і наслідку (weil, da, sodass, weshalb)", "Kausalsätze"),
    ("Концесивні речення (obwohl, trotzdem, dennoch)", "Konzessivsätze"),
    ("Умовні речення (falls, wenn, sofern)", "Konditionalsätze"),
    ("Способові речення (indem, dadurch, dass)", "Modalsätze"),
    ("N-Deklination іменників (der Mensch, den Menschen)", "N-Deklination"),
    ("Прийменники з Genitiv (wegen, trotz, während, statt)", "Genitivpräpositionen"),
    ("Складні прийменники (aufgrund, mithilfe, anhand)", "Komplexe Präpositionen"),
    ("Лексика: Umwelt und Natur", "Умова та природа"),
    ("Лексика: Gesellschaft und Politik", "Суспільство і політика"),
    ("Непряма мова (Konjunktiv I)", "Indirekte Rede"),
    ("Konjunktiv I у репортажах", "Konjunktiv I Vertiefung"),
    ("Функціональні дієслова (Funktionsverbgefüge)", "Funktionsverben"),
    ("Nominalisierung — іменники з дієслів", "Nominalisierung"),
    ("Лексика: Medien und Kommunikation", "Медіа та комунікація"),
    ("Сполучники та прислівники (jedoch, allerdings, dennoch)", "Konnektoren"),
    ("Протиставлення (während, wohingegen, im Gegensatz zu)", "Gegenüberstellung"),
    ("Лексика: Gesundheit und Medizin", "Охорона здоров'я"),
    ("Прикметники з прийменниками (zufrieden mit, stolz auf)", "Adjektiv + Präposition"),
    ("Повторення Місяць 1 — граматика", "Wiederholung Monat 1"),
]

for i, (topic, short) in enumerate(topics_month1, start=11):
    lesson_num = i
    COURSE[lesson_num] = {
        "topic": topic,
        "theory": (
            f"📚 <b>{topic}</b>\n\n"
            f"Це важлива тема рівня B2.\n\n"
            f"<b>Ключові правила:</b>\n"
            f"• Вивчайте структуру та вживання\n"
            f"• Зверніть увагу на типові помилки\n"
            f"• Практикуйте в контексті речень\n\n"
            f"<b>Приклад:</b>\n"
            f"Складіть власне речення з цією конструкцією!"
        ),
        "vocabulary": [
            (f"Слово {j}", f"Переклад {j}") for j in range(1, 11)
        ],
        "exercise": (
            f"✏️ <b>Вправа на тему «{short}»:</b>\n\n"
            f"Складіть 3 речення використовуючи вивчений матеріал.\n"
            f"Потім перевірте себе за допомогою словника або перекладача."
        )
    }

# ══════════════════════════════════════════════
#  МІСЯЦІ 2 і 3: динамічна генерація
# ══════════════════════════════════════════════

month2_topics = [
    "Складні складнопідрядні речення",
    "Лексика: Reisen und Tourismus",
    "Текстові сполучники B2",
    "Аргументативне письмо",
    "Лексика: Technologie und Innovation",
    "Nominalstil vs. Verbalstil",
    "Складні прийменникові конструкції",
    "Лексика: Kultur und Kunst",
    "Wissenschaftliches Schreiben",
    "Лексика: Sport und Freizeit",
    "Дискусійна лексика",
    "Konjunktiv II у ввічливих проханнях",
    "Лексика: Familie und Beziehungen",
    "Складні дієслівні словосполучення",
    "Лексика: Ernährung und Lebensstil",
    "Аналіз тексту на B2",
    "Написання офіційного листа",
    "Лексика: Wohnen und Stadtleben",
    "Складні прикметникові конструкції",
    "Лексика: Finanzen und Geld",
    "Wissenschaftliche Texte verstehen",
    "Риторичні засоби",
    "Лексика: Natur und Umweltschutz",
    "Складні прислівники",
    "Говоріння: Мнение ausdrücken",
    "Лексика: Arbeit und Karriere",
    "Читання: Feuilleton-Texte",
    "Складний синтаксис",
    "Лексика: Geschichte und Politik",
    "Повторення Місяць 2",
    "Лексика: Psychologie und Verhalten",
    "Wissenschaftliche Diskussion",
    "Складні сполучники",
    "Лексика: Ethik und Moral",
    "Аргументація Pro und Contra",
    "Лексика: Globalisierung",
    "Текстовий аналіз — преса",
    "Складні іменні групи",
    "Лексика: Recht und Justiz",
    "Офіційне мовлення",
    "Лексика: Architektur und Design",
    "Складні дієприкметники",
    "Лексика: Literatur und Sprache",
    "Наукова аргументація",
    "Лексика: Astronomie und Physik",
    "Складний Passiv",
    "Лексика: Biologie und Chemie",
    "Дискурсивні маркери",
    "Лексика: Informatik und KI",
    "Підсумок курсу — повторення B2",
    # решта до 130
] + [f"Поглиблення теми {i}" for i in range(51, 91)]

month3_topics = [
    "Підготовка до Goethe B2 — аудіювання",
    "Підготовка до Goethe B2 — читання ч.1",
    "Підготовка до Goethe B2 — читання ч.2",
    "Підготовка до Goethe B2 — письмо",
    "Підготовка до Goethe B2 — говоріння",
    "Пробний тест — Leseverstehen",
    "Пробний тест — Hörverstehen",
    "Письмо: Erörterung schreiben",
    "Говоріння: präsentieren",
    "Помилки, яких треба уникати",
] + [f"Поглиблення та практика {i}" for i in range(11, 91)]

for idx, topic in enumerate(month2_topics[:90]):
    lesson_num = 31 + idx
    if lesson_num not in COURSE:
        COURSE[lesson_num] = {
            "topic": topic,
            "theory": (
                f"📚 <b>{topic}</b>\n\n"
                f"Тема рівня B2 — поглиблене вивчення.\n\n"
                f"<b>Фокус уроку:</b>\n"
                f"• Ознайомтеся з теорією\n"
                f"• Вивчіть нову лексику\n"
                f"• Виконайте вправу\n\n"
                f"💡 Практикуйте вживання в контексті!"
            ),
            "vocabulary": [(f"слово {j}", f"переклад {j}") for j in range(1, 11)],
            "exercise": (
                f"✏️ Вправа на тему: <b>{topic}</b>\n\n"
                f"Складіть короткий текст (5-7 речень) на цю тему.\n"
                f"Використайте вивчену лексику."
            )
        }

for idx, topic in enumerate(month3_topics[:60]):
    lesson_num = 121 + idx
    if lesson_num not in COURSE:
        COURSE[lesson_num] = {
            "topic": topic,
            "theory": (
                f"📚 <b>{topic}</b>\n\n"
                f"Підготовка до іспиту Goethe-Zertifikat B2.\n\n"
                f"<b>Стратегія:</b>\n"
                f"• Читайте автентичні тексти\n"
                f"• Слухайте подкасти та радіо\n"
                f"• Пишіть ессе та листи\n"
                f"• Говоріть вголос щодня\n\n"
                f"🎯 Мета: впевнений рівень B2!"
            ),
            "vocabulary": [(f"термін {j}", f"значення {j}") for j in range(1, 11)],
            "exercise": (
                f"✏️ Практичне завдання: <b>{topic}</b>\n\n"
                f"Виконайте завдання у форматі іспиту Goethe B2.\n"
                f"Час: 20-25 хвилин."
            )
        }

# Заповнюємо решту до 180
for n in range(1, 181):
    if n not in COURSE:
        COURSE[n] = {
            "topic": f"Заняття {n} — практика B2",
            "theory": (
                f"📚 <b>Заняття {n}</b>\n\n"
                f"Комплексне повторення та практика рівня B2.\n\n"
                f"Сьогодні:\n"
                f"• Повторення граматики\n"
                f"• Розширення словника\n"
                f"• Практика письма або говоріння"
            ),
            "vocabulary": [(f"слово {j}", f"переклад {j}") for j in range(1, 11)],
            "exercise": (
                f"✏️ Вправа заняття {n}:\n"
                f"Напишіть короткий текст або складіть діалог на вільну тему."
            )
        }

# ─────────────────────────────────────────────
#  ХЕЛПЕРИ
# ─────────────────────────────────────────────

def get_lesson_number(day: int, slot: int) -> int:
    """день 1-90, slot 1 або 2 → номер заняття 1-180"""
    return (day - 1) * 2 + slot

def get_lesson(day: int, slot: int) -> dict:
    num = get_lesson_number(day, slot)
    return COURSE.get(num, COURSE[180])

def get_user_data(context: ContextTypes.DEFAULT_TYPE) -> dict:
    if "german_b2" not in context.user_data:
        context.user_data["german_b2"] = {
            "start_date": datetime.now().isoformat(),
            "current_day": 1,
            "current_slot": 1,
            "completed": [],
            "score": 0,
            "streak": 0,
            "last_activity": datetime.now().isoformat(),
        }
    return context.user_data["german_b2"]

# ─────────────────────────────────────────────
#  ХЕНДЛЕРИ
# ─────────────────────────────────────────────

async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    ud = get_user_data(context)
    text = (
        "🇩🇪 <b>Ласкаво просимо до курсу Deutsch B2!</b>\n\n"
        "📅 <b>Програма:</b>\n"
        "• 3 місяці навчання\n"
        "• 2 заняття на день\n"
        "• 180 занять загалом\n\n"
        "🎯 <b>Що вас чекає:</b>\n"
        "✅ Граматика B2 (Konjunktiv, Passiv, складні речення)\n"
        "✅ Тематична лексика (10 слів на заняття)\n"
        "✅ Вправи з відповідями\n"
        "✅ Щоденний прогрес\n\n"
        f"📍 Ваш прогрес: День {ud['current_day']}, "
        f"Заняття {get_lesson_number(ud['current_day'], ud['current_slot'])}/180\n\n"
        "Оберіть дію:"
    )
    keyboard = [
        [InlineKeyboardButton("▶️ Почати заняття", callback_data="lesson_start")],
        [InlineKeyboardButton("📊 Мій прогрес", callback_data="progress"),
         InlineKeyboardButton("📅 Розклад", callback_data="schedule")],
        [InlineKeyboardButton("ℹ️ Про курс", callback_data="about")],
    ]
    await update.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def show_lesson(update: Update, context: ContextTypes.DEFAULT_TYPE, day=None, slot=None):
    ud = get_user_data(context)
    d = day or ud["current_day"]
    s = slot or ud["current_slot"]
    lesson = get_lesson(d, s)
    num = get_lesson_number(d, s)

    month = (d - 1) // 30 + 1
    week = (d - 1) // 7 + 1

    text = (
        f"🗓 <b>День {d} | Заняття {num}/180</b>\n"
        f"📅 Місяць {month} | Тиждень {week}\n"
        f"{'🌅 Ранок' if s == 1 else '🌙 Вечір'}\n\n"
        f"<b>Тема: {lesson['topic']}</b>\n\n"
        f"{lesson['theory']}"
    )
    keyboard = [
        [InlineKeyboardButton("📖 Словник (10 слів)", callback_data=f"vocab_{d}_{s}")],
        [InlineKeyboardButton("✏️ Вправа", callback_data=f"exercise_{d}_{s}")],
        [InlineKeyboardButton("✅ Заняття виконано!", callback_data=f"complete_{d}_{s}")],
        [InlineKeyboardButton("◀️ Меню", callback_data="menu")],
    ]
    msg = update.callback_query.message if update.callback_query else update.message
    await msg.reply_html(text, reply_markup=InlineKeyboardMarkup(keyboard))

async def show_vocabulary(update: Update, context: ContextTypes.DEFAULT_TYPE, day: int, slot: int):
    lesson = get_lesson(day, slot)
    num = get_lesson_number(day, slot)
    vocab_text = "\n".join(
        f"<b>{i+1}.</b> {de} — {ua}"
        for i, (de, ua) in enumerate(lesson["vocabulary"])
    )
    text = (
        f"📖 <b>Словник — Заняття {num}</b>\n"
        f"Тема: {lesson['topic']}\n\n"
        f"{vocab_text}\n\n"
        f"💡 <i>Повторіть слова 3 рази вголос!</i>"
    )
    keyboard = [
        [InlineKeyboardButton("✏️ Перейти до вправи", callback_data=f"exercise_{day}_{slot}")],
        [InlineKeyboardButton("◀️ До заняття", callback_data=f"lesson_{day}_{slot}")],
    ]
    await update.callback_query.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def show_exercise(update: Update, context: ContextTypes.DEFAULT_TYPE, day: int, slot: int):
    lesson = get_lesson(day, slot)
    num = get_lesson_number(day, slot)
    text = (
        f"✏️ <b>Вправа — Заняття {num}</b>\n"
        f"Тема: {lesson['topic']}\n\n"
        f"{lesson['exercise']}"
    )
    keyboard = [
        [InlineKeyboardButton("✅ Виконано!", callback_data=f"complete_{day}_{slot}")],
        [InlineKeyboardButton("◀️ До заняття", callback_data=f"lesson_{day}_{slot}")],
    ]
    await update.callback_query.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def complete_lesson(update: Update, context: ContextTypes.DEFAULT_TYPE, day: int, slot: int):
    ud = get_user_data(context)
    lesson_num = get_lesson_number(day, slot)

    if lesson_num not in ud["completed"]:
        ud["completed"].append(lesson_num)
        ud["score"] += 10

    ud["last_activity"] = datetime.now().isoformat()

    # Оновлюємо прогрес
    if slot == 1:
        ud["current_slot"] = 2
    else:
        ud["current_day"] = min(day + 1, 90)
        ud["current_slot"] = 1

    total_done = len(ud["completed"])
    percent = round(total_done / 180 * 100, 1)

    text = (
        f"✅ <b>Заняття {lesson_num} виконано!</b>\n\n"
        f"🏆 Балів зароблено: +10 (всього: {ud['score']})\n"
        f"📊 Прогрес: {total_done}/180 ({percent}%)\n"
        f"{'▓' * int(percent // 5)}{'░' * (20 - int(percent // 5))} {percent}%\n\n"
    )

    if total_done >= 180:
        text += "🎉 <b>Вітаємо! Ви завершили курс Deutsch B2!</b>\n🏅 Ви готові до іспиту Goethe-Zertifikat B2!"
        keyboard = [[InlineKeyboardButton("🏠 Головне меню", callback_data="menu")]]
    elif slot == 2:
        text += f"🌙 День {day} завершено! Завтра: День {day+1}"
        keyboard = [
            [InlineKeyboardButton("➡️ Наступне заняття", callback_data="lesson_start")],
            [InlineKeyboardButton("🏠 Меню", callback_data="menu")],
        ]
    else:
        text += f"🌅 Перше заняття дня {day} виконано!\nВечірнє заняття чекає на вас!"
        keyboard = [
            [InlineKeyboardButton("▶️ Вечірнє заняття", callback_data="lesson_start")],
            [InlineKeyboardButton("🏠 Меню", callback_data="menu")],
        ]

    await update.callback_query.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def show_progress(update: Update, context: ContextTypes.DEFAULT_TYPE):
    ud = get_user_data(context)
    total_done = len(ud["completed"])
    percent = round(total_done / 180 * 100, 1)
    day = ud["current_day"]
    month = (day - 1) // 30 + 1
    current_lesson = get_lesson_number(day, ud["current_slot"])

    bar_filled = int(percent // 5)
    progress_bar = "▓" * bar_filled + "░" * (20 - bar_filled)

    text = (
        f"📊 <b>Ваш прогрес</b>\n\n"
        f"🗓 Поточний день: {day}/90\n"
        f"📅 Місяць: {month}/3\n"
        f"📚 Поточне заняття: {current_lesson}/180\n\n"
        f"<b>Загальний прогрес:</b>\n"
        f"{progress_bar}\n"
        f"✅ Виконано: {total_done}/180 ({percent}%)\n\n"
        f"🏆 Балів: {ud['score']}\n"
        f"📝 Залишилось: {180 - total_done} занять\n\n"
        f"<b>По місяцях:</b>\n"
        f"Місяць 1: {min(total_done, 60)}/60 занять\n"
        f"Місяць 2: {max(0, min(total_done - 60, 60))}/60 занять\n"
        f"Місяць 3: {max(0, total_done - 120)}/60 занять"
    )
    keyboard = [
        [InlineKeyboardButton("▶️ Продовжити навчання", callback_data="lesson_start")],
        [InlineKeyboardButton("◀️ Меню", callback_data="menu")],
    ]
    await update.callback_query.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def show_schedule(update: Update, context: ContextTypes.DEFAULT_TYPE):
    text = (
        "📅 <b>Розклад курсу Deutsch B2</b>\n\n"
        "<b>📌 Місяць 1 (Дні 1-30):</b>\n"
        "• Konjunktiv II (теперішнє і минуле)\n"
        "• Passiv (всі часи)\n"
        "• Relativsätze\n"
        "• Infinitivkonstruktionen\n"
        "• Partizipialkonstruktionen\n"
        "• Temporal-, Kausal-, Konzessivsätze\n"
        "• N-Deklination, Genitiv-Präpositionen\n"
        "• Лексика: Wirtschaft, Bildung, Umwelt\n\n"
        "<b>📌 Місяць 2 (Дні 31-60):</b>\n"
        "• Складний синтаксис B2\n"
        "• Номіналізація та Funktionsverben\n"
        "• Nominalstil, Konnektoren\n"
        "• Лексика: Technologie, Kultur, Gesundheit\n"
        "• Аргументативне письмо\n"
        "• Офіційна кореспонденція\n\n"
        "<b>📌 Місяць 3 (Дні 61-90):</b>\n"
        "• Підготовка до Goethe-Zertifikat B2\n"
        "• Пробні тести\n"
        "• Стратегії іспиту\n"
        "• Повторення всього матеріалу\n\n"
        "⏰ <b>Режим:</b>\n"
        "🌅 Заняття 1: Теорія + словник\n"
        "🌙 Заняття 2: Практика + повторення"
    )
    keyboard = [[InlineKeyboardButton("◀️ Меню", callback_data="menu")]]
    await update.callback_query.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def show_about(update: Update, context: ContextTypes.DEFAULT_TYPE):
    text = (
        "ℹ️ <b>Про курс Deutsch B2</b>\n\n"
        "🎯 <b>Мета:</b> Підготовка до рівня B2\n\n"
        "<b>Структура:</b>\n"
        "• 180 занять × ~30 хв = 90 годин\n"
        "• 10 нових слів на заняття\n"
        "• 1800 слів за курс\n\n"
        "<b>Граматика B2:</b>\n"
        "✅ Konjunktiv I та II\n"
        "✅ Passiv (всі форми)\n"
        "✅ Складні підрядні речення\n"
        "✅ Partizipialkonstruktionen\n"
        "✅ Nominalisierung\n"
        "✅ Erweiterte Attribute\n\n"
        "<b>Команди бота:</b>\n"
        "/start — головне меню\n"
        "/lesson — поточне заняття\n"
        "/progress — прогрес\n"
        "/help — допомога\n\n"
        "🏆 <b>По завершенню ви будете готові до:</b>\n"
        "Goethe-Zertifikat B2 / telc Deutsch B2"
    )
    keyboard = [[InlineKeyboardButton("◀️ Меню", callback_data="menu")]]
    await update.callback_query.message.reply_html(
        text, reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def show_menu(update: Update, context: ContextTypes.DEFAULT_TYPE):
    ud = get_user_data(context)
    total_done = len(ud["completed"])
    keyboard = [
        [InlineKeyboardButton("▶️ Поточне заняття", callback_data="lesson_start")],
        [InlineKeyboardButton("📊 Прогрес", callback_data="progress"),
         InlineKeyboardButton("📅 Розклад", callback_data="schedule")],
        [InlineKeyboardButton("ℹ️ Про курс", callback_data="about")],
    ]
    await update.callback_query.message.reply_html(
        f"🇩🇪 <b>Deutsch B2 — Головне меню</b>\n"
        f"✅ Виконано: {total_done}/180 занять",
        reply_markup=InlineKeyboardMarkup(keyboard)
    )

async def lesson_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /lesson"""
    class FakeQuery:
        message = update.message
    update.callback_query = FakeQuery()
    ud = get_user_data(context)
    await show_lesson(update, context, ud["current_day"], ud["current_slot"])

async def progress_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
    """Команда /progress"""
    class FakeQuery:
        message = update.message
    update.callback_query = FakeQuery()
    await show_progress(update, context)

async def button_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
    query = update.callback_query
    await query.answer()
    data = query.data

    if data == "lesson_start":
        ud = get_user_data(context)
        await show_lesson(update, context, ud["current_day"], ud["current_slot"])
    elif data == "progress":
        await show_progress(update, context)
    elif data == "schedule":
        await show_schedule(update, context)
    elif data == "about":
        await show_about(update, context)
    elif data == "menu":
        await show_menu(update, context)
    elif data.startswith("vocab_"):
        _, d, s = data.split("_")
        await show_vocabulary(update, context, int(d), int(s))
    elif data.startswith("exercise_"):
        _, d, s = data.split("_")
        await show_exercise(update, context, int(d), int(s))
    elif data.startswith("complete_"):
        _, d, s = data.split("_")
        await complete_lesson(update, context, int(d), int(s))
    elif data.startswith("lesson_"):
        parts = data.split("_")
        if len(parts) == 3:
            _, d, s = parts
            await show_lesson(update, context, int(d), int(s))

async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE):
    text = (
        "❓ <b>Допомога</b>\n\n"
        "/start — Головне меню\n"
        "/lesson — Поточне заняття\n"
        "/progress — Ваш прогрес\n"
        "/help — Ця довідка\n\n"
        "Бот веде вас через 180 занять:\n"
        "2 заняття на день × 90 днів = 3 місяці\n\n"
        "Кожне заняття містить:\n"
        "📚 Теорію\n📖 10 нових слів\n✏️ Вправу з відповідями"
    )
    await update.message.reply_html(text)

# ─────────────────────────────────────────────
#  ЗАПУСК
# ─────────────────────────────────────────────

def main():
    token = os.environ.get("BOT_TOKEN", "YOUR_BOT_TOKEN_HERE")
    if token == "YOUR_BOT_TOKEN_HERE":
        print("❌ Встановіть змінну середовища BOT_TOKEN!")
        print("   export BOT_TOKEN='ваш_токен_від_BotFather'")
        return

    app = Application.builder().token(token).build()

    app.add_handler(CommandHandler("start", start))
    app.add_handler(CommandHandler("lesson", lesson_command))
    app.add_handler(CommandHandler("progress", progress_command))
    app.add_handler(CommandHandler("help", help_command))
    app.add_handler(CallbackQueryHandler(button_handler))

    print("🤖 Бот Deutsch B2 запущено!")
    print("📚 180 занять, 3 місяці, рівень B2")
    app.run_polling(allowed_updates=Update.ALL_TYPES)

if __name__ == "__main__":
    main()