import {connect} from "react-redux";
import {deleteUnit} from "../../../../actions/topology/machine";
import UnitComponent from "../../../../components/sidebars/topology/machine/UnitComponent";

const mapStateToProps = (state, ownProps) => {
    return {
        unit: state.objects[ownProps.unitType][ownProps.unitId],
        inSimulation: state.currentExperimentId !== -1
    };
};

const mapDispatchToProps = (dispatch, ownProps) => {
    return {
        onDelete: () => dispatch(deleteUnit(ownProps.unitType, ownProps.index)),
    };
};

const UnitContainer = connect(
    mapStateToProps,
    mapDispatchToProps
)(UnitComponent);

export default UnitContainer;