Merge pull request #21 from indigotech/feature/union

Feature - Union Type

Author: felipesabino

README.md

@@ -13,6 +13,7 @@ Apart from the decorators listed on the original documentation, we have added si
 - @Query: It can be used multiple times on the same file. This way we make it possible to break queries into different folders.
 - @Mutation: It can be used multiple times on the same file. This way we make it possible to break queries into different folders.
 - @UseContainer: Sets the IoC container to be used in order to instantiate the decorated clas.
+- @Uniontype: It can be used to create `GraphQLUnionType` objects.
 
 #### GraphQL Decorator Examples
 

src/decorator.ts

@@ -8,6 +8,9 @@ import { OrderByTypeFactory } from './order-by.type-factory';
 import { PageInfo } from './page-info.type';
 import { PaginationResponse } from './pagination.type';
 
+export * from './decorator/';
+export * from './metadata/options';
+
 export const GQ_QUERY_KEY = 'gq_query';
 export const GQ_MUTATION_KEY = 'gq_mutation';
 export const GQ_SUBSCRIPTION_KEY = 'gq_subscription';
@@ -17,8 +20,6 @@ export const GQ_OBJECT_METADATA_KEY = 'gq_object_type';
 export const GQ_ENUM_METADATA_KEY = 'gq_enum_type';
 export const GQ_DESCRIPTION_KEY = 'gq_description';
 
-
-
 export interface TypeMetadata {
     name?: string;
     description?: string;

src/decorator/index.ts

@@ -0,0 +1 @@
+export * from './union-type.decorator';

src/decorator/union-type.decorator.ts

@@ -0,0 +1,17 @@
+import { MetadataStorage } from '../metadata-storage';
+import { UnionOption } from '../metadata';
+
+/**
+ * Union Type. ref: http://graphql.org/learn/schema/#union-types
+ * @param option Options for a Union Type
+ */
+export function UnionType<T>(option: UnionOption<T>) {
+  return function (target: any) {
+      MetadataStorage.addUnionMetadata({
+          name: target.name,
+          types: option.types,
+          resolver: option.resolver,
+          description: option.description,
+      });
+  } as Function;
+}

src/field_type_factory.ts

@@ -11,13 +11,17 @@ import {
     RootMetadata,
     TypeMetadata,
 } from './decorator';
+import { MetadataStorage } from './metadata-storage';
+
 import { SchemaFactoryError, SchemaFactoryErrorType } from './schema_factory';
 
 import { IoCContainer } from './ioc-container';
 import { OrderByTypeFactory } from './order-by.type-factory';
 import { PaginationType } from './pagination.type';
 import { enumTypeFactory } from './enum.type-factory';
 import { objectTypeFactory } from './object_type_factory';
+import { unionTypeFactory } from './type-factory';
+
 
 export interface ResolverHolder {
     fn: Function;
@@ -43,12 +47,15 @@ function convertType(typeFn: Function, metadata: TypeMetadata, isInput: boolean,
         } else if (typeFn === Boolean) {
             returnType = graphql.GraphQLBoolean;
         } else if (typeFn && typeFn.prototype && Reflect.hasMetadata(GQ_OBJECT_METADATA_KEY, typeFn.prototype)) {
-            // recursively call objectFactory
-            returnType = objectTypeFactory(typeFn, isInput);
+          // recursively call objectFactory
+          returnType = objectTypeFactory(typeFn, isInput);
         }
-    } else {
+      } else {
         returnType = metadata.explicitType;
-        if (returnType && returnType.prototype && Reflect.hasMetadata(GQ_OBJECT_METADATA_KEY, returnType.prototype)) {
+
+        if (returnType && returnType.prototype && MetadataStorage.containsUnionMetadata(returnType.name)) {
+            returnType = unionTypeFactory(returnType.name, isInput);
+        } else if (returnType && returnType.prototype && Reflect.hasMetadata(GQ_OBJECT_METADATA_KEY, returnType.prototype)) {
             // recursively call objectFactory
             returnType = objectTypeFactory(returnType, isInput);
         } else if (returnType && returnType.prototype && Reflect.hasMetadata(GQ_ENUM_METADATA_KEY, returnType.prototype)) {

src/metadata-storage/index.ts

@@ -0,0 +1 @@
+export * from './metadata-storage';

src/metadata-storage/metadata-storage.ts

@@ -0,0 +1,18 @@
+import { UnionTypeMetadata } from '../metadata/types/union.metadata';
+
+const unionMetadata: UnionTypeMetadata[] = [];
+
+const MetadataStorage = {
+
+    addUnionMetadata: function(metadata: UnionTypeMetadata) {
+        unionMetadata.push(metadata);
+    },
+    getUnionMetadata: function(): UnionTypeMetadata[] {
+        return unionMetadata;
+    },
+    containsUnionMetadata: function(name: string) {
+      return unionMetadata.some(metadata => metadata.name === name);
+    },
+};
+
+export { MetadataStorage };

src/metadata/index.ts

@@ -0,0 +1,2 @@
+export * from './options';
+export * from './types';

src/metadata/options/index.ts

@@ -0,0 +1 @@
+export * from './union.option';

src/metadata/options/union.option.ts

@@ -0,0 +1,19 @@
+/**
+ * Arguments for a Union type on graphql schema
+ */
+export interface UnionOption<T> {
+  /**
+   * (Optional) Description
+   */
+  description?: string;
+
+  /**
+   * Concrete object types
+   */
+  types: any[];
+
+  /**
+   * Resolver function to inform schema what type should be returned based on the value provided
+   */
+  resolver: (obj: T, context: any, info: any) => Promise<string> | string | null;
+}