174 lines
4.0 KiB
Python
174 lines
4.0 KiB
Python
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
import pprint
|
|
|
|
from stone.ir import (
|
|
Boolean,
|
|
Bytes,
|
|
Float32,
|
|
Float64,
|
|
Int32,
|
|
Int64,
|
|
List,
|
|
String,
|
|
Timestamp,
|
|
UInt32,
|
|
UInt64,
|
|
Void,
|
|
is_boolean_type,
|
|
is_list_type,
|
|
is_numeric_type,
|
|
is_string_type,
|
|
is_tag_ref,
|
|
is_user_defined_type,
|
|
unwrap_nullable,
|
|
)
|
|
from .helpers import split_words
|
|
|
|
# This file defines *stylistic* choices for Swift
|
|
# (ie, that class names are UpperCamelCase and that variables are lowerCamelCase)
|
|
|
|
|
|
_type_table = {
|
|
Boolean: 'Bool',
|
|
Bytes: 'Data',
|
|
Float32: 'Float',
|
|
Float64: 'Double',
|
|
Int32: 'Int32',
|
|
Int64: 'Int64',
|
|
List: 'Array',
|
|
String: 'String',
|
|
Timestamp: 'Date',
|
|
UInt32: 'UInt32',
|
|
UInt64: 'UInt64',
|
|
Void: 'Void',
|
|
}
|
|
|
|
_reserved_words = {
|
|
'description',
|
|
'bool',
|
|
'nsdata'
|
|
'float',
|
|
'double',
|
|
'int32',
|
|
'int64',
|
|
'list',
|
|
'string',
|
|
'timestamp',
|
|
'uint32',
|
|
'uint64',
|
|
'void',
|
|
'associatedtype',
|
|
'class',
|
|
'deinit',
|
|
'enum',
|
|
'extension',
|
|
'func',
|
|
'import',
|
|
'init',
|
|
'inout',
|
|
'internal',
|
|
'let',
|
|
'operator',
|
|
'private',
|
|
'protocol',
|
|
'public',
|
|
'static',
|
|
'struct',
|
|
'subscript',
|
|
'typealias',
|
|
'var',
|
|
'default',
|
|
}
|
|
|
|
|
|
def fmt_obj(o):
|
|
assert not isinstance(o, dict), "Only use for base type literals"
|
|
if o is True:
|
|
return 'true'
|
|
if o is False:
|
|
return 'false'
|
|
if o is None:
|
|
return 'nil'
|
|
if o == u'':
|
|
return '""'
|
|
return pprint.pformat(o, width=1)
|
|
|
|
|
|
def _format_camelcase(name, lower_first=True):
|
|
words = [word.capitalize() for word in split_words(name)]
|
|
if lower_first:
|
|
words[0] = words[0].lower()
|
|
ret = ''.join(words)
|
|
if ret.lower() in _reserved_words:
|
|
ret += '_'
|
|
return ret
|
|
|
|
|
|
def fmt_class(name):
|
|
return _format_camelcase(name, lower_first=False)
|
|
|
|
|
|
def fmt_func(name, version):
|
|
if version > 1:
|
|
name = '{}_v{}'.format(name, version)
|
|
name = _format_camelcase(name)
|
|
return name
|
|
|
|
|
|
def fmt_type(data_type):
|
|
data_type, nullable = unwrap_nullable(data_type)
|
|
|
|
if is_user_defined_type(data_type):
|
|
result = '{}.{}'.format(fmt_class(data_type.namespace.name),
|
|
fmt_class(data_type.name))
|
|
else:
|
|
result = _type_table.get(data_type.__class__, fmt_class(data_type.name))
|
|
|
|
if is_list_type(data_type):
|
|
result = result + '<{}>'.format(fmt_type(data_type.data_type))
|
|
|
|
return result if not nullable else result + '?'
|
|
|
|
|
|
def fmt_var(name):
|
|
return _format_camelcase(name)
|
|
|
|
|
|
def fmt_default_value(namespace, field):
|
|
if is_tag_ref(field.default):
|
|
return '{}.{}Serializer().serialize(.{})'.format(
|
|
fmt_class(namespace.name),
|
|
fmt_class(field.default.union_data_type.name),
|
|
fmt_var(field.default.tag_name))
|
|
elif is_list_type(field.data_type):
|
|
return '.array({})'.format(field.default)
|
|
elif is_numeric_type(field.data_type):
|
|
return '.number({})'.format(field.default)
|
|
elif is_string_type(field.data_type):
|
|
return '.str({})'.format(fmt_obj(field.default))
|
|
elif is_boolean_type(field.data_type):
|
|
if field.default:
|
|
bool_str = '1'
|
|
else:
|
|
bool_str = '0'
|
|
return '.number({})'.format(bool_str)
|
|
else:
|
|
raise TypeError('Can\'t handle default value type %r' %
|
|
type(field.data_type))
|
|
|
|
def check_route_name_conflict(namespace):
|
|
"""
|
|
Check name conflicts among generated route definitions. Raise a runtime exception when a
|
|
conflict is encountered.
|
|
"""
|
|
|
|
route_by_name = {}
|
|
for route in namespace.routes:
|
|
route_name = fmt_func(route.name, route.version)
|
|
if route_name in route_by_name:
|
|
other_route = route_by_name[route_name]
|
|
raise RuntimeError(
|
|
'There is a name conflict between {!r} and {!r}'.format(other_route, route))
|
|
route_by_name[route_name] = route
|