# Copyright 2016 Casey Jaymes

# This file is part of Expatriate.
#
# Expatriate is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Expatriate is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with Expatriate.  If not, see <http://www.gnu.org/licenses/>.

import logging

from ..decorators import *
from .AnnotatedType import AnnotatedType
from .NCNameType import NCNameType
from .RestrictionType import RestrictionType
from .UnionElement import UnionElement

logger = logging.getLogger(__name__)

@element(local_name='restriction', list='tags', cls=RestrictionType, min=0)
@element(local_name='list', list='tags', cls=('expatriate.model.xs.ListElement', 'ListElement'), min=0)
@element(local_name='union', list='tags', cls=UnionElement, min=0)
@attribute(local_name='final', enum=['#all', 'list', 'union', 'restriction'])
@attribute(local_name='name', type=NCNameType)
@attribute(local_name='*', )
class SimpleTypeType(AnnotatedType):
    def stub(self, path, schema):
        class_name = ''.join(cap_first(w) for w in self.name.split('_'))
        if not class_name.endswith('Type'):
            class_name = class_name + 'Type'
        super().stub(path, schema, class_name)
