lux-framework/src/packages/database/model/utils/get-columns.js

Build scalable, Node.js-powered REST APIs with almost no code.

// @flow
import pick from '../../../../utils/pick';
import entries from '../../../../utils/entries';
import type Model from '../index';

/**
 * @private
 */
export default function getColumns(record: Model, only?: Array<string>) {
  let { constructor: { attributes: columns } } = record;

  if (only) {
    columns = pick(columns, ...only);
  }

  return entries(columns).reduce((obj, [key, { columnName }]) => ({
    ...obj,
    [columnName]: Reflect.get(record, key)
  }), {});
}