graphbrainz/src/extensions/mediawiki/index.js

import schema from './schema'
import resolvers from './resolvers'
import createLoader from './loader'
import MediaWikiClient from './client'
import { ONE_DAY } from '../../util'

export default {
  name: 'MediaWiki',
  description:
    'Retrieve information from MediaWiki image pages, like the actual image ' +
    'file URL and EXIF metadata.',
  extendContext (context, { mediaWiki = {} } = {}) {
    const client = new MediaWikiClient(mediaWiki)
    const cacheSize = parseInt(
      process.env.MEDIAWIKI_CACHE_SIZE || process.env.GRAPHBRAINZ_CACHE_SIZE || 8192,
      10
    )
    const cacheTTL = parseInt(
      process.env.MEDIAWIKI_CACHE_TTL || process.env.GRAPHBRAINZ_CACHE_TTL || ONE_DAY,
      10
    )
    return {
      ...context,
      loaders: {
        ...context.loaders,
        mediaWiki: createLoader({ client, cacheSize, cacheTTL })
      }
    }
  },
  extendSchema: {
    schemas: [schema],
    resolvers
  }
}