# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

"""Schema for gitlab-ci.yaml configuration file.

.. literalinclude:: ../spack/schema/gitlab_ci.py
   :lines: 13-
"""

from llnl.util.lang import union_dicts

image_schema = {
    'oneOf': [
        {
            'type': 'string'
        }, {
            'type': 'object',
            'properties': {
                'name': {'type': 'string'},
                'entrypoint': {
                    'type': 'array',
                    'items': {
                        'type': 'string',
                    },
                },
            },
        },
    ],
}

runner_attributes_schema_items = {
    'image': image_schema,
    'tags': {
        'type': 'array',
        'items': {'type': 'string'}
    },
    'variables': {
        'type': 'object',
        'patternProperties': {
            r'[\w\d\-_\.]+': {
                'type': 'string',
            },
        },
    },
    'before_script': {
        'type': 'array',
        'items': {'type': 'string'}
    },
    'script': {
        'type': 'array',
        'items': {'type': 'string'}
    },
    'after_script': {
        'type': 'array',
        'items': {'type': 'string'}
    },
}

runner_selector_schema = {
    'type': 'object',
    'additionalProperties': False,
    'required': ['tags'],
    'properties': runner_attributes_schema_items,
}


core_shared_properties = union_dicts(
    runner_attributes_schema_items,
    {
        'bootstrap': {
            'type': 'array',
            'items': {
                'anyOf': [
                    {
                        'type': 'string',
                    }, {
                        'type': 'object',
                        'additionalProperties': False,
                        'required': ['name'],
                        'properties': {
                            'name': {
                                'type': 'string',
                            },
                            'compiler-agnostic': {
                                'type': 'boolean',
                                'default': False,
                            },
                        },
                    },
                ],
            },
        },
        'mappings': {
            'type': 'array',
            'items': {
                'type': 'object',
                'additionalProperties': False,
                'required': ['match'],
                'properties': {
                    'match': {
                        'type': 'array',
                        'items': {
                            'type': 'string',
                        },
                    },
                    'runner-attributes': runner_selector_schema,
                },
            },
        },
        'service-job-attributes': runner_selector_schema,
        'rebuild-index': {'type': 'boolean'},
        'broken-specs-url': {'type': 'string'},
    },
)

gitlab_ci_properties = {
    'anyOf': [
        {
            'type': 'object',
            'additionalProperties': False,
            'required': ['mappings'],
            'properties': union_dicts(
                core_shared_properties,
                {
                    'enable-artifacts-buildcache': {
                        'type': 'boolean',
                    },
                },
            ),
        },
        {
            'type': 'object',
            'additionalProperties': False,
            'required': ['mappings'],
            'properties': union_dicts(
                core_shared_properties,
                {
                    'temporary-storage-url-prefix': {
                        'type': 'string',
                    },
                },
            ),
        },
    ]
}

#: Properties for inclusion in other schemas
properties = {
    'gitlab-ci': gitlab_ci_properties,
}

#: Full schema with metadata
schema = {
    '$schema': 'http://json-schema.org/draft-07/schema#',
    'title': 'Spack gitlab-ci configuration file schema',
    'type': 'object',
    'additionalProperties': False,
    'properties': properties,
}
