# 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