import { get } from 'lodash';

const addQuery = (querys) => (newQuery = []) => (nameClass) =>
                      [...querys, ...newQuery].map(query => `${query(nameClass)} \n` ).join('');

const addResolver = (target, key, descriptor) => {
  return async function() {
      try {
        const isAllow = get(target[key], 'allow', function() { return true; });
        if (isAllow.bind(target)(...arguments)) {
          return await descriptor.value.bind(target)(...arguments);
        }

        return null;
      } catch (error) {
        throw new Error('Decorators query failed. \n' + error);
      }
    };
}

const query = function(params) {
  return (target, key, descriptor) => {
    const defaultFields = 'id: String, skip: Int, limit: Int, sort: String';
    let newQuery = '';

    switch (typeof params) {
      case 'string':
      newQuery = (nameType) => `${key}(${params}): [${nameType}],`;
      break;
      case 'object':
      const fields = get(params, 'fields', defaultFields);
      newQuery = (nameType) => `${key}(${fields}): [${get(params, 'responseType', nameType)}],`;
      break;
      default:
      newQuery = (nameType) => `${key}(${defaultFields}): [${nameType}],`;
    }

    //target.Query = addQuery(newQuery);

    target.QUERYS = get(target, 'QUERYS', []);
    target.QUERYS.push(newQuery);
    target.Query = addQuery(target.QUERYS);

    target.Resolvers = Object.assign({}, target.Resolvers);
    target.Resolvers.Query = Object.assign({}, target.Resolvers.Query);
    target.Resolvers.Query[key] = addResolver(target, key, descriptor);
  }
};

export default query;

@@ -1208,50 +1208,8 @@
  %7D%0A%0A
-    //target.Query = addQuery(newQuery);%0A%0A
     

remove comment code