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;