# Software Name: MOON

# Version: 5.4

# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
# SPDX-License-Identifier: Apache-2.0

# This software is distributed under the 'Apache License 2.0',
# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
# or see the "LICENSE" file for more details.


from uuid import uuid4
import logging
from moon_utilities.security_functions import enforce
from moon_engine.api.orchestration.managers import Managers

logger = logging.getLogger("moon.engine.api.orchestration.pipeline")


class PipelineManager(Managers):

    def __init__(self, connector=None):
        self.driver = connector.driver
        Managers.PipelineManager = self

    @enforce(("read", "write"), "pipelines")
    def update_pipeline(self, moon_user_id, pipeline_id, data):
        return self.driver.update_pipeline(pipeline_id=pipeline_id, data=data)

    @enforce("write", "pipelines")
    def delete_pipeline(self, moon_user_id, pipeline_id):
        return self.driver.delete_pipeline(pipeline_id=pipeline_id)

    @enforce("write", "pipelines")
    def add_pipeline(self, moon_user_id, pipeline_id=None, data=None):
        if not pipeline_id:
            pipeline_id = uuid4().hex
        if data is None:
            data = {}
        if "plugins" not in data:
            data["plugins"] = ["moon_engine.plugins.authz"]
        return self.driver.add_pipeline(pipeline_id=pipeline_id, data=data)

    @enforce("read", "pipelines")
    def get_pipelines(self, moon_user_id, pipeline_id=None):
        return self.driver.get_pipelines(pipeline_id=pipeline_id)

    @enforce("read", "pipelines")
    def get_pipeline_api_key(self, moon_user_id, pipeline_id):
        return self.driver.get_pipeline_api_key(pipeline_id=pipeline_id)