import { AbstractResolver } from '@ulangi/resolver';
import * as Joi from 'joi';

import { WiktionaryWordClass } from '../interfaces/WiktionaryWordClass';
import { WiktionaryDefinitionResolver } from './WiktionaryDefinitionResolver';

export class WiktionaryWordClassResolver extends AbstractResolver<
  WiktionaryWordClass
> {
  private definitionResolver = new WiktionaryDefinitionResolver();

  protected rules = {
    kind: Joi.string().valid('wordClass'),
    wordClass: Joi.string(),
    headword: Joi.string().optional(),
    children: Joi.array().items(this.definitionResolver.getRules()),
    categories: Joi.array().items(Joi.string()),
  };
}