var assign = require('object-assign');

var RestaurantConstants = require('../constants/RestaurantConstants');
var UserConstants = require('../constants/UserConstants');

module.exports = function(state = {}, action) {
	switch (action.type) {
		case RestaurantConstants.RESTAURANTS_FETCHED:
			var restaurants = {};
			for (var i = 0; i < action.restaurants.length; i++) {
				var restaurant = action.restaurants[i];
				restaurants[restaurant.RestaurantId] = restaurant;
			}
			return restaurants;
		case RestaurantConstants.RESTAURANT_CREATED:
			var restaurant = action.restaurant;
			var restaurants = assign({}, state, {
				[restaurant.RestaurantId]: restaurant
			});
			return restaurants;
		case UserConstants.USER_LOGGEDOUT:
			return {};
		default:
			return state;
	}
};