# Copyright (c) 2024 CloudZero - ALL RIGHTS RESERVED - PROPRIETARY AND CONFIDENTIAL
# Unauthorized copying of this file and/or project, via any medium is strictly prohibited.
# Direct all questions to legal@cloudzero.com

from pydantic import Field

from czpulumi.sumo_logic.models import ConfiguredBaseModel
from czpulumi.sumo_logic.roles import SumoRole


class Team(ConfiguredBaseModel):
    name: str = Field(..., description='The name of the team')
    title: str = Field(..., description='The title of the team')
    role: str = Field(..., description='The role to assign the permission to')


CZ_TEAMS = {
    'cirrus': Team(
        name='cirrus',
        title='Cirrus',
        role=SumoRole.TEAM_ENGINEERING_CIRRUS,
    ),
    'dev_null': Team(
        name='dev_null',
        title='Dev Null',
        role=SumoRole.TEAM_ENGINEERING_DEV_NULL,
    ),
    'drop_table': Team(
        name='drop_table',
        title='Drop Table',
        role=SumoRole.TEAM_ENGINEERING_DROP_TABLE,
    ),
    'not_found': Team(
        name='not_found',
        title='Not Found',
        role=SumoRole.TEAM_ENGINEERING_NOT_FOUND,
    ),
    'stratus': Team(
        name='stratus',
        title='Stratus',
        role=SumoRole.TEAM_ENGINEERING_STRATUS,
    ),
}
