# Licensed to Elasticsearch B.V. under one or more contributor
# license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright
# ownership. Elasticsearch B.V. licenses this file to you under
# the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#	http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

import fnmatch
import re

class MountSearchableSnapshotRunner:
    async def __call__(self, es, params):
        repository_name = params["repository"]
        snapshot_name = params["snapshot"]
        index_pattern = params.get("index_pattern", "*")
        rename_pattern = params.get("rename_pattern", "(.*)")
        rename_replacement = params.get("rename_replacement", "\\1")
        query_params = params.get("query_params")
        snapshots = await es.snapshot.get(repository_name, snapshot_name)

        # ES main branch
        if "responses" in snapshots:
            available_snapshots = snapshots["responses"][0]["snapshots"]
        else:
            available_snapshots = snapshots["snapshots"]

        for snapshot in available_snapshots:
            for index in snapshot["indices"]:
                if fnmatch.fnmatch(index, index_pattern):
                    body={"index": index}
                    renamed_index = re.sub(rename_pattern, rename_replacement, index)
                    if (renamed_index != index):
                        body={"index": index, "renamed_index": renamed_index}

                    await es.transport.perform_request(method="POST",
                                                    url=f"/_snapshot/{repository_name}/{snapshot_name}/_mount",
                                                    body=body,
                                                    params=query_params
                                                    )