from flask_restful import fields  # type: ignore

from libs.helper import TimestampField

intention_base_fields = {
    "id": fields.String,
    "name": fields.String,
}

intention_fields = {
    "id": fields.String,
    "name": fields.String,
    "type_id": fields.String,
    "type_name": fields.String,
}

intention_type_fields = {
    "id": fields.String,
    "name": fields.String,
}

intention_type_page_fields = {
    "id": fields.String,
    "name": fields.String,
    "intention_count": fields.Integer,
}

intention_type_detail_fields = {
    "id": fields.String,
    "name": fields.String,
    "intentions": fields.List(fields.Nested(intention_base_fields)),
}

intention_keyword_fields = {
    "id": fields.String,
    "name": fields.String,
    "intention_id": fields.String,
}

intention_keyword_detail_fields = {
    "id": fields.String,
    "name": fields.String,
    "intention": fields.Nested(intention_fields)
}

intention_corpus_fields = {
    "id": fields.String,
    "question": fields.String,
    "question_config": fields.String,
    "intention_id": fields.String,
    "created_by": fields.String,
    "created_at": TimestampField,
    "updated_by": fields.String,
    "updated_at": TimestampField,
}

intention_corpus_similarity_question_fields = {
    "id": fields.String,
    "question": fields.String,
    "question_config": fields.String,
    "corpus_id": fields.String,
}

intention_page_fields = {
    "id": fields.String,
    "name": fields.String,
    "type_id": fields.String,
    "type_name": fields.String,
    "corpus_count": fields.Integer,
    "keywords_count": fields.Integer,
    "created_by": fields.String,
    "created_at": TimestampField,
}

intention_corpus_detail_fields = {
    "id": fields.String,
    "question": fields.String,
    "question_config": fields.String,
    "intention": fields.Nested(intention_page_fields),
    "similarity_questions": fields.List(fields.Nested(intention_corpus_similarity_question_fields)),
    "created_by": fields.String,
    "created_at": TimestampField,
}

intention_detail_fields = {
    "id": fields.String,
    "name": fields.String,
    "type": fields.Nested(intention_type_fields),
    "corpus": fields.List(fields.Nested(intention_corpus_fields)),
    "keywords": fields.List(fields.Nested(intention_keyword_fields)),
    "created_by": fields.String,
    "created_at": TimestampField,
    "updated_by": fields.String,
    "updated_at": TimestampField,
}

intention_train_file_fields = {
    "id": fields.String,
    "name": fields.String,
    "version": fields.String,
    "type": fields.String,
    "data_source_type": fields.String,
    "data_source_info": fields.String,
    "created_by": fields.String,
    "created_at": TimestampField,
}

intention_train_file_binding_fields = {
    "id": fields.String,
    "file_id": fields.String,
    "task_id": fields.String,
}

intention_train_task_fields = {
    "id": fields.String,
    "name": fields.String,
    "status": fields.String,
    "dataset_info": fields.Nested(intention_train_file_fields, allow_null=True),
    "model_info": fields.Nested(intention_train_file_fields, allow_null=True),
    "created_by": fields.String,
    "created_at": TimestampField,
}