from flask_restful import fields  # type: ignore

from libs.helper import TimestampField

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_count": fields.Integer,
}

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)),
}

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,
}