<?php
namespace GraphQL\Tests\Utils;

use GraphQL\GraphQL;
use GraphQL\Schema;
use GraphQL\Type\Definition\CustomScalarType;
use GraphQL\Type\Definition\InputObjectType;
use GraphQL\Type\Definition\InterfaceType;
use GraphQL\Type\Definition\ObjectType;
use GraphQL\Type\Definition\Type;
use GraphQL\Type\Definition\EnumType;
use GraphQL\Type\Definition\UnionType;
use GraphQL\Utils\SchemaPrinter;

class SchemaPrinterTest extends \PHPUnit_Framework_TestCase
{
    // Describe: Type System Printer

    private function printForTest($schema)
    {
        return "\n" . SchemaPrinter::doPrint($schema);
    }

    private function printSingleFieldSchema($fieldConfig)
    {
        $root = new ObjectType([
            'name' => 'Root',
            'fields' => [
                'singleField' => $fieldConfig
            ]
        ]);
        return $this->printForTest(new Schema(['query' => $root]));
    }

    /**
     * @it Prints String Field
     */
    public function testPrintsStringField()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string()
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField: String
}
');
    }

    /**
     * @it Prints [String] Field
     */
    public function testPrintArrayStringField()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::listOf(Type::string())
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField: [String]
}
');
    }

    /**
     * @it Prints String! Field
     */
    public function testPrintNonNullStringField()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::nonNull(Type::string())
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField: String!
}
');
    }

    /**
     * @it Prints [String]! Field
     */
    public function testPrintNonNullArrayStringField()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::nonNull(Type::listOf(Type::string()))
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField: [String]!
}
');
    }

    /**
     * @it Prints [String!] Field
     */
    public function testPrintArrayNonNullStringField()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::listOf(Type::nonNull(Type::string()))
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField: [String!]
}
');
    }

    /**
     * @it Prints [String!]! Field
     */
    public function testPrintNonNullArrayNonNullStringField()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::nonNull(Type::listOf(Type::nonNull(Type::string())))
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField: [String!]!
}
');
    }

    /**
     * @it Prints Object Field
     */
    public function testPrintObjectField()
    {
        $fooType = new ObjectType([
            'name' => 'Foo',
            'fields' => ['str' => ['type' => Type::string()]]
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => ['foo' => ['type' => $fooType]]
        ]);

        $schema = new Schema(['query' => $root]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Foo {
  str: String
}

type Root {
  foo: Foo
}
');
    }

    /**
     * @it Prints String Field With Int Arg
     */
    public function testPrintsStringFieldWithIntArg()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => ['argOne' => ['type' => Type::int()]]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int): String
}
');
    }

    /**
     * @it Prints String Field With Int Arg With Default
     */
    public function testPrintsStringFieldWithIntArgWithDefault()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => ['argOne' => ['type' => Type::int(), 'defaultValue' => 2]]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int = 2): String
}
');
    }

    /**
     * @it Prints String Field With Int Arg With Default Null
     */
    public function testPrintsStringFieldWithIntArgWithDefaultNull()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => ['argOne' => ['type' => Type::int(), 'defaultValue' => null]]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int = null): String
}
');
    }

    /**
     * @it Prints String Field With Int! Arg
     */
    public function testPrintsStringFieldWithNonNullIntArg()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => ['argOne' => ['type' => Type::nonNull(Type::int())]]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int!): String
}
');
    }

    /**
     * @it Prints String Field With Multiple Args
     */
    public function testPrintsStringFieldWithMultipleArgs()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => [
                'argOne' => ['type' => Type::int()],
                'argTwo' => ['type' => Type::string()]
            ]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int, argTwo: String): String
}
');
    }

    /**
     * @it Prints String Field With Multiple Args, First is Default
     */
    public function testPrintsStringFieldWithMultipleArgsFirstIsDefault()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => [
                'argOne' => ['type' => Type::int(), 'defaultValue' => 1],
                'argTwo' => ['type' => Type::string()],
                'argThree' => ['type' => Type::boolean()]
            ]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int = 1, argTwo: String, argThree: Boolean): String
}
');
    }

    /**
     * @it Prints String Field With Multiple Args, Second is Default
     */
    public function testPrintsStringFieldWithMultipleArgsSecondIsDefault()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => [
                'argOne' => ['type' => Type::int()],
                'argTwo' => ['type' => Type::string(), 'defaultValue' => 'foo'],
                'argThree' => ['type' => Type::boolean()]
            ]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int, argTwo: String = "foo", argThree: Boolean): String
}
');
    }

    /**
     * @it Prints String Field With Multiple Args, Last is Default
     */
    public function testPrintsStringFieldWithMultipleArgsLastIsDefault()
    {
        $output = $this->printSingleFieldSchema([
            'type' => Type::string(),
            'args' => [
                'argOne' => ['type' => Type::int()],
                'argTwo' => ['type' => Type::string()],
                'argThree' => ['type' => Type::boolean(), 'defaultValue' => false]
            ]
        ]);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Root {
  singleField(argOne: Int, argTwo: String, argThree: Boolean = false): String
}
');
    }

    /**
     * @it Print Interface
     */
    public function testPrintInterface()
    {
        $fooType = new InterfaceType([
            'name' => 'Foo',
            'resolveType' => function() { return null; },
            'fields' => ['str' => ['type' => Type::string()]]
        ]);

        $barType = new ObjectType([
            'name' => 'Bar',
            'fields' => ['str' => ['type' => Type::string()]],
            'interfaces' => [$fooType]
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => ['bar' => ['type' => $barType]]
        ]);

        $schema = new Schema([
            'query' => $root,
            'types' => [$barType]
        ]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Bar implements Foo {
  str: String
}

interface Foo {
  str: String
}

type Root {
  bar: Bar
}
');
    }

    /**
     * @it Print Multiple Interface
     */
    public function testPrintMultipleInterface()
    {
        $fooType = new InterfaceType([
            'name' => 'Foo',
            'resolveType' => function() { return null; },
            'fields' => ['str' => ['type' => Type::string()]]
        ]);

        $baazType = new InterfaceType([
            'name' => 'Baaz',
            'resolveType' => function() { return null; },
            'fields' => ['int' => ['type' => Type::int()]]
        ]);

        $barType = new ObjectType([
            'name' => 'Bar',
            'fields' => [
                'str' => ['type' => Type::string()],
                'int' => ['type' => Type::int()]
            ],
            'interfaces' => [$fooType, $baazType]
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => ['bar' => ['type' => $barType]]
        ]);

        $schema = new Schema([
            'query' => $root,
            'types' => [$barType]
        ]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

interface Baaz {
  int: Int
}

type Bar implements Foo, Baaz {
  int: Int
  str: String
}

interface Foo {
  str: String
}

type Root {
  bar: Bar
}
');
    }

    /**
     * @it Print Unions
     */
    public function testPrintUnions()
    {
        $fooType = new ObjectType([
            'name' => 'Foo',
            'fields' => ['bool' => ['type' => Type::boolean()]]
        ]);

        $barType = new ObjectType([
            'name' => 'Bar',
            'fields' => ['str' => ['type' => Type::string()]]
        ]);

        $singleUnion = new UnionType([
            'name' => 'SingleUnion',
            'resolveType' => function() { return null; },
            'types' => [$fooType]
        ]);

        $multipleUnion = new UnionType([
            'name' => 'MultipleUnion',
            'resolveType' => function() { return null; },
            'types' => [$fooType, $barType]
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => [
                'single' => ['type' => $singleUnion],
                'multiple' => ['type' => $multipleUnion]
            ]
        ]);

        $schema = new Schema(['query' => $root]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

type Bar {
  str: String
}

type Foo {
  bool: Boolean
}

union MultipleUnion = Foo | Bar

type Root {
  multiple: MultipleUnion
  single: SingleUnion
}

union SingleUnion = Foo
');
    }

    /**
     * @it Print Input Type
     */
    public function testInputType()
    {
        $inputType = new InputObjectType([
            'name' => 'InputType',
            'fields' => ['int' => ['type' => Type::int()]]
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => [
                'str' => [
                    'type' => Type::string(),
                    'args' => ['argOne' => ['type' => $inputType]]
                ]
            ]
        ]);

        $schema = new Schema(['query' => $root]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

input InputType {
  int: Int
}

type Root {
  str(argOne: InputType): String
}
');
    }

    /**
     * @it Custom Scalar
     */
    public function testCustomScalar()
    {
        $oddType = new CustomScalarType([
            'name' => 'Odd',
            'serialize' => function($value) {
                return $value % 2 === 1 ? $value : null;
            }
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => [
                'odd' => ['type' => $oddType]
            ]
        ]);

        $schema = new Schema(['query' => $root]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

scalar Odd

type Root {
  odd: Odd
}
');
    }

    /**
     * @it Enum
     */
    public function testEnum()
    {
        $RGBType = new EnumType([
            'name' => 'RGB',
            'values' => [
                'RED' => ['value' => 0],
                'GREEN' => ['value' => 1],
                'BLUE' => ['value' => 2]
            ]
        ]);

        $root = new ObjectType([
            'name' => 'Root',
            'fields' => [
                'rgb' => ['type' => $RGBType]
            ]
        ]);

        $schema = new Schema(['query' => $root]);
        $output = $this->printForTest($schema);
        $this->assertEquals($output, '
schema {
  query: Root
}

enum RGB {
  RED
  GREEN
  BLUE
}

type Root {
  rgb: RGB
}
');
    }

    /**
     * @it Print Introspection Schema
     */
    public function testPrintIntrospectionSchema()
    {
        $root = new ObjectType([
            'name' => 'Root',
            'fields' => [
                'onlyField' => ['type' => Type::string()]
            ]
        ]);

        $schema = new Schema(['query' => $root]);
        $output = SchemaPrinter::printIntrosepctionSchema($schema);
        $introspectionSchema = <<<'EOT'
schema {
  query: Root
}

# Directs the executor to include this field or fragment only when the `if` argument is true.
directive @include(
  # Included when true.
  if: Boolean!
) on FIELD | FRAGMENT_SPREAD | INLINE_FRAGMENT

# Directs the executor to skip this field or fragment when the `if` argument is true.
directive @skip(
  # Skipped when true.
  if: Boolean!
) on FIELD | FRAGMENT_SPREAD | INLINE_FRAGMENT

# Marks an element of a GraphQL schema as no longer supported.
directive @deprecated(
  # Explains why this element was deprecated, usually also including a suggestion
  # for how to access supported similar data. Formatted in
  # [Markdown](https://daringfireball.net/projects/markdown/).
  reason: String = "No longer supported"
) on FIELD_DEFINITION | ENUM_VALUE

# A Directive provides a way to describe alternate runtime execution and type validation behavior in a GraphQL document.
#
# In some cases, you need to provide options to alter GraphQL's execution behavior
# in ways field arguments will not suffice, such as conditionally including or
# skipping a field. Directives provide this by describing additional information
# to the executor.
type __Directive {
  args: [__InputValue!]!
  description: String
  locations: [__DirectiveLocation!]!
  name: String!
  onField: Boolean! @deprecated(reason: "Use `locations`.")
  onFragment: Boolean! @deprecated(reason: "Use `locations`.")
  onOperation: Boolean! @deprecated(reason: "Use `locations`.")
}

# A Directive can be adjacent to many parts of the GraphQL language, a
# __DirectiveLocation describes one such possible adjacencies.
enum __DirectiveLocation {
  # Location adjacent to a query operation.
  QUERY

  # Location adjacent to a mutation operation.
  MUTATION

  # Location adjacent to a subscription operation.
  SUBSCRIPTION

  # Location adjacent to a field.
  FIELD

  # Location adjacent to a fragment definition.
  FRAGMENT_DEFINITION

  # Location adjacent to a fragment spread.
  FRAGMENT_SPREAD

  # Location adjacent to an inline fragment.
  INLINE_FRAGMENT

  # Location adjacent to a schema definition.
  SCHEMA

  # Location adjacent to a scalar definition.
  SCALAR

  # Location adjacent to an object type definition.
  OBJECT

  # Location adjacent to a field definition.
  FIELD_DEFINITION

  # Location adjacent to an argument definition.
  ARGUMENT_DEFINITION

  # Location adjacent to an interface definition.
  INTERFACE

  # Location adjacent to a union definition.
  UNION

  # Location adjacent to an enum definition.
  ENUM

  # Location adjacent to an enum value definition.
  ENUM_VALUE

  # Location adjacent to an input object type definition.
  INPUT_OBJECT

  # Location adjacent to an input object field definition.
  INPUT_FIELD_DEFINITION
}

# One possible value for a given Enum. Enum values are unique values, not a
# placeholder for a string or numeric value. However an Enum value is returned in
# a JSON response as a string.
type __EnumValue {
  deprecationReason: String
  description: String
  isDeprecated: Boolean!
  name: String!
}

# Object and Interface types are described by a list of Fields, each of which has
# a name, potentially a list of arguments, and a return type.
type __Field {
  args: [__InputValue!]!
  deprecationReason: String
  description: String
  isDeprecated: Boolean!
  name: String!
  type: __Type!
}

# Arguments provided to Fields or Directives and the input fields of an
# InputObject are represented as Input Values which describe their type and
# optionally a default value.
type __InputValue {
  # A GraphQL-formatted string representing the default value for this input value.
  defaultValue: String
  description: String
  name: String!
  type: __Type!
}

# A GraphQL Schema defines the capabilities of a GraphQL server. It exposes all
# available types and directives on the server, as well as the entry points for
# query, mutation, and subscription operations.
type __Schema {
  # A list of all directives supported by this server.
  directives: [__Directive!]!

  # If this server supports mutation, the type that mutation operations will be rooted at.
  mutationType: __Type

  # The type that query operations will be rooted at.
  queryType: __Type!

  # If this server support subscription, the type that subscription operations will be rooted at.
  subscriptionType: __Type

  # A list of all types supported by this server.
  types: [__Type!]!
}

# The fundamental unit of any GraphQL Schema is the type. There are many kinds of
# types in GraphQL as represented by the `__TypeKind` enum.
#
# Depending on the kind of a type, certain fields describe information about that
# type. Scalar types provide no information beyond a name and description, while
# Enum types provide their values. Object and Interface types provide the fields
# they describe. Abstract types, Union and Interface, provide the Object types
# possible at runtime. List and NonNull types compose other types.
type __Type {
  description: String
  enumValues(includeDeprecated: Boolean = false): [__EnumValue!]
  fields(includeDeprecated: Boolean = false): [__Field!]
  inputFields: [__InputValue!]
  interfaces: [__Type!]
  kind: __TypeKind!
  name: String
  ofType: __Type
  possibleTypes: [__Type!]
}

# An enum describing what kind of type a given `__Type` is.
enum __TypeKind {
  # Indicates this type is a scalar.
  SCALAR

  # Indicates this type is an object. `fields` and `interfaces` are valid fields.
  OBJECT

  # Indicates this type is an interface. `fields` and `possibleTypes` are valid fields.
  INTERFACE

  # Indicates this type is a union. `possibleTypes` is a valid field.
  UNION

  # Indicates this type is an enum. `enumValues` is a valid field.
  ENUM

  # Indicates this type is an input object. `inputFields` is a valid field.
  INPUT_OBJECT

  # Indicates this type is a list. `ofType` is a valid field.
  LIST

  # Indicates this type is a non-null. `ofType` is a valid field.
  NON_NULL
}

EOT;
        $this->assertEquals($output, $introspectionSchema);
    }
}