langateam/graphiql.io

'use strict'

const Model = require('trails/model')
const faker = require('faker')
const _ = require('lodash')

/**
 * @module Company
 * @description TODO document Model
 */
module.exports = class Company extends Model {

  static config () {
    return {
      type: 'graphql'
    }
  }

  static schema (graphql) {
    return graphql`

      # Company type
      type Company {
        companyName: String
        catchPhrase: String
        suffixes: String
        bs: String
      }

      type CompanyQuery {
        findOne: Company
        findAll (count: Int!): [Company]
      }
    `
  }

  static resolver (app) {
    return {
      findOne () {
        return new Company(app)
      },
      findAll ({ count }) {
        return _.range(0, count).map(i => new Company(app))
      }
    }
  }

  get companyName () {
    return faker.company.companyName()
  }

  get catchPhrase () {
    return faker.company.catchPhrase()
  }

  get suffixes () {
    return faker.company.suffixes()
  }

  bs () {
    return faker.company.bs()
  }
}