#
# 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",
            "type": "text",
            "body": "text",
            "url": "text",
            "created_at": "date",
        }
        display = {
            "title_field": "title",
            "url_field": "url",
            "detail_fields": [
                {"field_name": "title", "label": "Title"},
                {"field_name": "type", "label": "Type"},
                {"field_name": "body", "label": "Content"},
                {"field_name": "created_at", "label": "Created At"},
            ],
            "color": "#000000",
        }
        self.workplace_search_custom_client.create_content_source(
            schema, display, self.args.name, is_searchable=True
        )