const { Component } = require('react')
const connect = require('react-redux').connect
const PropTypes = require('prop-types')
const { withRouter } = require('react-router-dom')
const { compose } = require('recompose')
const t = require('../i18n-helper').getMessage

class I18nProvider extends Component {
  tOrDefault = (key, defaultValue, ...args) => {
    const { localeMessages: { current, en } = {} } = this.props
    return t(current, key, ...args) || t(en, key, ...args) || defaultValue
  }

  getChildContext () {
    const { localeMessages } = this.props
    const { current, en } = localeMessages
    return {
      t (key, ...args) {
        return t(current, key, ...args) || t(en, key, ...args) || `[${key}]`
      },
      tOrDefault: this.tOrDefault,
      tOrKey (key, ...args) {
        return this.tOrDefault(key, key, ...args)
      },
    }
  }

  render () {
    return this.props.children
  }
}

I18nProvider.propTypes = {
  localeMessages: PropTypes.object,
  children: PropTypes.object,
}

I18nProvider.childContextTypes = {
  t: PropTypes.func,
  tOrDefault: PropTypes.func,
  tOrKey: PropTypes.func,
}

const mapStateToProps = state => {
  const { localeMessages } = state
  return {
    localeMessages,
  }
}

module.exports = compose(
  withRouter,
  connect(mapStateToProps)
)(I18nProvider)