import { connect } from "react-redux";
import { addUnit } from "../../../../../actions/topology/machine";
import UnitAddComponent from "../../../../../components/app/sidebars/topology/machine/UnitAddComponent";

const mapStateToProps = (state, ownProps) => {
  return {
    units: Object.values(state.objects[ownProps.unitType])
  };
};

const mapDispatchToProps = (dispatch, ownProps) => {
  return {
    onAdd: id => dispatch(addUnit(ownProps.unitType, id))
  };
};

const UnitAddContainer = connect(mapStateToProps, mapDispatchToProps)(
  UnitAddComponent
);

export default UnitAddContainer;