from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length

from app import models as m, db
from app.logger import log


class BaseBookForm(FlaskForm):
    label = StringField("Label", [DataRequired(), Length(6, 256)])


class CreateBookForm(BaseBookForm):
    submit = SubmitField("Add new book")


class EditBookForm(BaseBookForm):
    book_id = StringField("User ID", [DataRequired()])
    submit = SubmitField("Edit book")

    def validate_book_id(self, field):
        book_id = field.data
        book: m.Book = db.session.get(m.Book, book_id)
        if not book or book.is_deleted:
            log(log.WARNING, "Book with id [%s] not found", book_id)
            raise ValidationError("Book not found")

    def validate_label(self, field):
        label = field.data
        book_id = self.book_id.data

        existing_book: m.Book = m.Book.query.filter_by(
            is_deleted=False,
            label=label,
        ).first()
        if existing_book:
            log(
                log.WARNING, "Book with label [%s] already exists: [%s]", label, book_id
            )
            raise ValidationError("Book label must be unique!")