All files local-directive.js

100% Statements 14/14
100% Branches 5/5
100% Functions 4/4
100% Lines 13/13
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38    2x 7x   7x   5x 3x 3x         7x     2x 7x   7x 6x     7x                   2x      
import { visit } from 'graphql'
 
const extractDirectives = (originalQuery, directive) => {
  let hasDirective = false
 
  const query = visit(originalQuery, {
    Directive: ({ name: { value: name } }) => {
      if (name === directive) {
        hasDirective = true
        return null
      }
    }
  })
 
  return { query, hasDirective }
}
 
const createHasLocalDirective = directive => (operation, mutate = true) => {
  const { query, hasDirective } = extractDirectives(operation.query, directive)
 
  if (mutate) {
    operation.query = query
  }
 
  return hasDirective
}
 
/**
 * Checks if a given operation uses @local directive.
 *
 * @param {Object} operation The GraphQL operation.
 * @param {boolean} [mutate] Wheter or not the operation's query should
 *  get it's @local directives extracted.
 */
const hasLocalDirective = createHasLocalDirective('local')
 
export { hasLocalDirective, createHasLocalDirective }