mirror of
https://github.com/openappsec/openappsec.git
synced 2025-06-28 16:41:02 +03:00
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
# Copyright 2019-present, GraphQL Foundation
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
from c import field_prototype, return_type, struct_name
|
|
from casing import title
|
|
from license import C_LICENSE_COMMENT
|
|
|
|
class Printer(object):
|
|
'''Printer for the implementation of the pure C interface to the AST.
|
|
'''
|
|
|
|
def __init__(self):
|
|
self._current_type = None
|
|
|
|
def start_file(self):
|
|
print(C_LICENSE_COMMENT + '''/** @generated */
|
|
|
|
#include "GraphQLAst.h"
|
|
#include "../Ast.h"
|
|
|
|
using namespace facebook::graphql::ast; // NOLINT
|
|
''')
|
|
|
|
def end_file(self):
|
|
pass
|
|
|
|
def start_type(self, name):
|
|
self._current_type = name
|
|
|
|
def field(self, type, name, nullable, plural):
|
|
print(field_prototype(self._current_type, type, name, nullable, plural) + ' {')
|
|
print(' const auto *realNode = reinterpret_cast<const %s *>(node);' % self._current_type)
|
|
title_name = title(name)
|
|
call_get = 'realNode->get%s()' % title_name
|
|
if plural:
|
|
if nullable:
|
|
print(' return %s ? %s->size() : 0;' % (call_get, call_get))
|
|
else:
|
|
print(' return %s.size();' % call_get)
|
|
else:
|
|
if type in ['string', 'OperationKind', 'boolean']:
|
|
print(' return %s;' % call_get)
|
|
else:
|
|
fmt = ' return reinterpret_cast<const struct %s *>(%s%s);'
|
|
print(fmt % (struct_name(type), '' if nullable else '&', call_get))
|
|
|
|
print('}')
|
|
|
|
def end_type(self, name):
|
|
pass
|
|
|
|
def start_union(self, name):
|
|
pass
|
|
|
|
def union_option(self, option):
|
|
pass
|
|
|
|
def end_union(self, name):
|
|
pass
|