import logging
from telebot import types
from database.models import Courses

logger = logging.getLogger(__name__)


def register(bot):
    @bot.message_handler(func=lambda message: message.text == "Show Courses")
    def get_students(message):
        try:
            data = Courses.getAllCourses()
            if data:
                markup = types.InlineKeyboardMarkup(row_width=2)
                for row in data:
                    btn = types.InlineKeyboardButton(
                        f"{row[1][:26] + "..."} | id#{row[0]}", callback_data=f"course_{row[0]}")
                    markup.add(btn)

                # add button for creating a new course
                markup.add(types.InlineKeyboardButton(
                    "➕ Create New Course", callback_data="create_course"))

                bot.send_message(
                    message.chat.id, "Here is the data:", reply_markup=markup)
            else:
                markup = types.InlineKeyboardMarkup(row_width=1)
                markup.add(types.InlineKeyboardButton(
                    "➕ Create New Course", callback_data="create_course"))
                bot.reply_to(message, "No data found.", reply_markup=markup)
        except Exception as e:
            logger.error(f"Error in (Show Courses) handler: {e}")
            bot.reply_to(message, "Sorry, an error occurred.")