Export complexity directive

Author: dimatill

src/estimators/directive/README.md

@@ -63,3 +63,19 @@ input Filter {
 ```
 
 The multipliers can be combined. Configured multipliers that don't have a value or `NULL` are ignored. 
+
+Code first approach can use the `createComplexityDirective` function to generate directive definition:
+
+```typescript
+import {createComplexityDirective} from 'graphql-query-complexity';
+
+const schema = new GraphQLSchema({
+  directives: [
+    createComplexityDirective({
+      // Optionally change the name of the directive here... Default value is `complexity`
+      name: 'complexity'
+    })
+  ]
+  // ... other config
+});
+```

src/estimators/directive/__tests__/directiveEstimator-test.ts

@@ -3,7 +3,10 @@
  */
 
 import {
+  GraphQLSchema,
   parse,
+  print,
+  printSchema,
   TypeInfo,
   visit,
   visitWithTypeInfo,
@@ -14,7 +17,7 @@ import {expect} from 'chai';
 import schema from './fixtures/schema';
 
 import ComplexityVisitor from '../../../QueryComplexity';
-import directiveEstimator from '../index';
+import directiveEstimator, {createComplexityDirective} from '../index';
 import { CompatibleValidationContext } from '../../../__tests__/fixtures/CompatibleValidationContext';
 
 describe('directiveEstimator analysis', () => {
@@ -224,7 +227,7 @@ describe('directiveEstimator analysis', () => {
     expect(visitor.complexity).to.equal(10);
   });
 
-  it('ignores fields without compexity directive', () => {
+  it('ignores fields without complexity directive', () => {
     const ast = parse(`
       query {
         noDirective
@@ -245,4 +248,28 @@ describe('directiveEstimator analysis', () => {
       'No complexity could be calculated for field Query.noDirective',
     );
   });
+
+  it('should create complexity directive that can be used to generate directive definition', () => {
+    const complexityDirective = createComplexityDirective();
+    const codeFirstSchema = new GraphQLSchema({
+      directives: [complexityDirective]
+    });
+    
+    const printedCodeFirstSchema = printSchema(codeFirstSchema).trim();
+    const printedComplexityDirective = print(schema.getDirective('complexity').astNode).trim();
+
+    expect(printedCodeFirstSchema).to.equal(printedComplexityDirective);
+  });
+
+  it('should create complexity directive with configured name', () => {
+    const complexityDirective = createComplexityDirective({name: 'cost'});
+    const codeFirstSchema = new GraphQLSchema({
+      directives: [complexityDirective]
+    });
+
+    const printedCodeFirstSchema = printSchema(codeFirstSchema).trim();
+    const printedComplexityDirective = print(schema.getDirective('cost').astNode).trim();
+
+    expect(printedCodeFirstSchema).to.equal(printedComplexityDirective);
+  });
 });

src/estimators/directive/__tests__/fixtures/schema.ts

@@ -7,12 +7,16 @@ import {
 } from 'graphql';
 
 export default buildSchema(`
+"""Define a relation between the field and other nodes"""
 directive @cost(
+  """The complexity value for the field"""
   value: Int!,
   multipliers: [String!]
 ) on FIELD_DEFINITION
 
+"""Define a relation between the field and other nodes"""
 directive @complexity(
+  """The complexity value for the field"""
   value: Int!,
   multipliers: [String!]
 ) on FIELD_DEFINITION

src/estimators/directive/index.ts

@@ -4,12 +4,21 @@ import { GraphQLDirective } from 'graphql/type/directives';
 import { DirectiveLocation } from 'graphql/language/directiveLocation';
 import get from 'lodash.get';
 
-export function complexityDirective(name: string = 'complexity') {
+export type ComplexityDirectiveOptions = {
+  name?: string
+}
+
+export function createComplexityDirective(options: ComplexityDirectiveOptions = {}) {
+  const mergedOptions = {
+    name: 'complexity',
+    ...(options || {})
+  };
+
   return new GraphQLDirective({
-    name,
+    name: mergedOptions.name,
     description: 'Define a relation between the field and other nodes',
     locations: [
-      DirectiveLocation.FIELD,
+      DirectiveLocation.FIELD_DEFINITION,
     ],
     args: {
       value: {
@@ -23,8 +32,8 @@ export function complexityDirective(name: string = 'complexity') {
   });
 }
 
-export default function (options: { name?: string } = {}): ComplexityEstimator {
-  const directive = complexityDirective(options.name);
+export default function (options: ComplexityDirectiveOptions = {}): ComplexityEstimator {
+  const directive = createComplexityDirective(options);
 
   return (args: ComplexityEstimatorArgs) => {
     // Ignore if astNode is undefined

src/estimators/index.ts

@@ -1,3 +1,3 @@
 export {default as simpleEstimator} from './simple';
-export {default as directiveEstimator, complexityDirective} from './directive';
+export {default as directiveEstimator, ComplexityDirectiveOptions, createComplexityDirective} from './directive';
 export {default as fieldExtensionsEstimator} from './fieldExtensions';