import { StudioProjectState } from '../types';

export default function setTrackArmed(trackId: string, arm: boolean) {
  return (state: StudioProjectState) => {
    return {
      ...state,
      tracks: state.tracks.map((track) =>
        track.id === trackId
          ? { ...track, arm }
          : track.arm && arm === true
            ? { ...track, arm: false }
            : track
      ),
    };
  };
}