#
# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
# or more contributor license agreements. Licensed under the Elastic License 2.0;
# you may not use this file except in compliance with the Elastic License 2.0.
#
"""This module allows to create Content Source in Elastic Enterprise Search.

    It can be used to create a Content Source that will be used to upload the
    data to Elastic Enterprise Search instance.

    Otherwise, it's possible to use Content Source that was pre-created
    in Elastic Enterprise Search
"""

from .base_command import BaseCommand


class BootstrapCommand(BaseCommand):
    """This class defines a method to create a content source.
    """
    def execute(self):
        """This function attempts to create a Content Source.

        It will use data from configuration file to determine
        which instance of Elastic Enterprise Search will be used
        to create a Content Source."""
        schema = {
            "title": "text",
            "body": "text",
            "url": "text",
            "created_at": "date",
            "name": "text",
            "description": "text",
            "type": "text",
            "size": "text",
        }
        display = {
            "title_field": "title",
            "description_field": "description",
            "url_field": "url",
            "detail_fields": [
                {"field_name": 'created_at', "label": 'Created At'},
                {"field_name": 'type', "label": 'Type'},
                {"field_name": 'size', "label": 'Size (in bytes)'},
                {"field_name": 'description', "label": 'Description'},
                {"field_name": 'body', "label": 'Content'},
            ],
            "color": "#000000",
        }
        self.workplace_search_custom_client.create_content_source(schema, display, self.args.name, True)