import { entries, toIteratee } from '../_internal/iteratee'

/**
 * 按 iteratee 结果建立元素索引，重复键保留最后一个元素。
 * @function keyBy
 * @memberof module:collection
 * @param {Array|Object} collection 要遍历的集合
 * @param {Function|string|number|Array|Object} [iteratee] 索引函数或简写
 * @returns {Object} 索引结果对象
 * @throws {TypeError} 集合或 iteratee 类型不正确时抛出
 */
function keyBy(collection, iteratee) {
  const fn = toIteratee(iteratee)
  return entries(collection).reduce((result, [key, value]) => {
    result[String(fn(value, key, collection))] = value
    return result
  }, {})
}

export default keyBy