ICIJ/datashare-client

import { isInteger, max, min, uniq } from 'lodash'
import moment from 'moment'

import FilterDate from './FilterDate'

export default class FilterDateRange extends FilterDate {
  constructor({ interval = 'year', ...options }) {
    super(options)
    this.component = 'FilterDateRange'
    this.interval = interval
  }

  itemLabel(item) {
    if (isInteger(item.key)) {
      const timestamp = item.key + new Date().getTimezoneOffset() * 60 * 1000
      const locale = localStorage.getItem('locale')
      return moment(timestamp).locale(locale).format('L')
    } else {
      return item.key
    }
  }

  queryBuilder(body, param, func) {
    return body.query('bool', (sub) => {
      const gte = new Date(min(param.values))
      const lte = new Date(max(param.values))
      sub[func]('range', this.key, { gte, lte })
      return sub
    })
  }

  body(body, { interval = this.interval } = {}) {
    return body
      .query('match', 'type', 'Document')
      .agg('date_histogram', this.key, FilterDateRange.getIntervalAgg(interval), this.key)
  }

  static getIntervalAgg(interval = this.interval) {
    return { ...FilterDate.getIntervalOption(interval), min_doc_count: 1 }
  }

  static getIntervalOption(interval = this.interval) {
    switch (interval) {
      case 'day':
        return { interval: '1d', format: 'yyyy-MM-dd', missing: '1970-01-01' }
      case 'month':
        return { interval: '1M', format: 'yyyy-MM', missing: '1970-01' }
      default:
        return { interval: '1y', format: 'yyyy', missing: '1970' }
    }
  }

  get values() {
    return uniq(super.values.map((v) => parseInt(v)))
  }
}