from flask_wtf import FlaskForm
from werkzeug.utils import secure_filename
from wtforms import (BooleanField, FileField, StringField, SubmitField,
                     ValidationError, IntegerField, SelectField)
from wtforms.validators import DataRequired, Length, NumberRange


class AddCorpusFileForm(FlaskForm):
    address = StringField('Adress', validators=[Length(0, 255)])
    author = StringField('Author', validators=[DataRequired(), Length(1, 255)])
    booktitle = StringField('Booktitle', validators=[Length(0, 255)])
    chapter = StringField('Chapter', validators=[Length(0, 255)])
    editor = StringField('Editor', validators=[Length(0, 255)])
    file = FileField('File', validators=[DataRequired()])
    institution = StringField('Institution', validators=[Length(0, 255)])
    journal = StringField('Journal', validators=[Length(0, 255)])
    pages = StringField('Pages', validators=[Length(0, 255)])
    publisher = StringField('Publisher', validators=[Length(0, 255)])
    publishing_year = IntegerField('Publishing year',
                                   validators=[DataRequired()])
    school = StringField('School', validators=[Length(0, 255)])
    submit = SubmitField()
    title = StringField('Title', validators=[DataRequired(), Length(1, 255)])

    def __init__(self, corpus, *args, **kwargs):
        super(AddCorpusFileForm, self).__init__(*args, **kwargs)
        self.corpus = corpus

    def validate_file(self, field):
        if not field.data.filename.lower().endswith('.vrt'):
            raise ValidationError('File does not have an approved extension: '
                                  '.vrt')
        field.data.filename = secure_filename(field.data.filename)
        for corpus_file in self.corpus.files:
            if field.data.filename == corpus_file.filename:
                raise ValidationError('File already registered to corpus.')


class EditCorpusFileForm(FlaskForm):
    address = StringField('Adress', validators=[Length(0, 255)])
    author = StringField('Author', validators=[DataRequired(), Length(1, 255)])
    booktitle = StringField('Booktitle', validators=[Length(0, 255)])
    chapter = StringField('Chapter', validators=[Length(0, 255)])
    editor = StringField('Editor', validators=[Length(0, 255)])
    institution = StringField('Institution', validators=[Length(0, 255)])
    journal = StringField('Journal', validators=[Length(0, 255)])
    pages = StringField('Pages', validators=[Length(0, 255)])
    publisher = StringField('Publisher', validators=[Length(0, 255)])
    publishing_year = IntegerField('Publishing year',
                                   validators=[DataRequired()])
    school = StringField('School', validators=[Length(0, 255)])
    submit = SubmitField()
    title = StringField('Title', validators=[DataRequired(), Length(1, 255)])


class AddCorpusForm(FlaskForm):
    description = StringField('Description',
                              validators=[DataRequired(), Length(1, 255)])
    submit = SubmitField()
    title = StringField('Title', validators=[DataRequired(), Length(1, 32)])


class QueryForm(FlaskForm):
    query = StringField('Query',
                        validators=[DataRequired(), Length(1, 1024)])
    submit = SubmitField('Send query')


class DisplayOptionsForm(FlaskForm):
    expert_mode = BooleanField('Expert mode')
    result_context = SelectField('Result context',
                                 choices=[('', 'Choose your option'),
                                          ('10', '10'),
                                          ('20', '20'),
                                          ('30', '30'),
                                          ('40', '40'),
                                          ('50', '50')])
    results_per_page = SelectField('Results per page',
                                   choices=[('', 'Choose your option'),
                                            ('10', '10'),
                                            ('20', '20'),
                                            ('30', '30'),
                                            ('40', '40'),
                                            ('50', '50')])


class InspectDisplayOptionsForm(FlaskForm):
    expert_mode_inspect = BooleanField('Expert mode')
    highlight_sentences = BooleanField('Split sentences')
    context_sentences = IntegerField('Context sentences',
                                     validators=[NumberRange(min=0, max=10)],
                                     default=3)


class QueryDownloadForm(FlaskForm):
    file_type = SelectField('File type',
                            choices=[('', 'Choose file type'),
                                     ('csv', 'csv'),
                                     ('json', 'json'),
                                     ('excel', 'excel'),
                                     ('html', 'html-table')],
                            validators=[DataRequired()])