DbInit.java

package com.floreantpos.util;

import java.util.Arrays;
import java.util.Date;
import java.util.HashSet;

import com.floreantpos.model.MenuCategory;
import com.floreantpos.model.MenuGroup;
import com.floreantpos.model.MenuItem;
import com.floreantpos.model.Restaurant;
import com.floreantpos.model.Shift;
import com.floreantpos.model.Tax;
import com.floreantpos.model.User;
import com.floreantpos.model.UserPermission;
import com.floreantpos.model.UserType;
import com.floreantpos.model.dao.MenuCategoryDAO;
import com.floreantpos.model.dao.MenuGroupDAO;
import com.floreantpos.model.dao.MenuItemDAO;
import com.floreantpos.model.dao.RestaurantDAO;
import com.floreantpos.model.dao.ShiftDAO;
import com.floreantpos.model.dao.TaxDAO;
import com.floreantpos.model.dao.UserDAO;
import com.floreantpos.model.dao.UserTypeDAO;
import com.floreantpos.model.dao._RootDAO;

public class DbInit {
	public static void main(String[] args) throws Exception {
		_RootDAO.dbCleanerInitialize();
		
		Restaurant restaurant = new Restaurant();
		restaurant.setId(Integer.valueOf(1));
		restaurant.setName("Floreant");
		RestaurantDAO.getInstance().saveOrUpdate(restaurant);
		
		Tax tax = new Tax();
		tax.setName("US");
		tax.setRate(Double.valueOf(6));
		TaxDAO.getInstance().saveOrUpdate(tax);
		
		Shift shift = new Shift();
		shift.setName("General");
		Date shiftStartTime = ShiftUtil.buildShiftStartTime(0, 0, 0, 11, 59, 1);
		Date shiftEndTime = ShiftUtil.buildShiftEndTime(0, 0, 0, 11, 59, 1);
		
		shift.setStartTime(shiftStartTime);
		shift.setEndTime(shiftEndTime);
		long length = Math.abs(shiftStartTime.getTime() - shiftEndTime.getTime());

        shift.setShiftLength(Long.valueOf(length));
		ShiftDAO.getInstance().saveOrUpdate(shift);
		
		UserType type = new UserType();
		type.setName("Administrator");
		type.setPermissions(new HashSet<UserPermission>(Arrays.asList(UserPermission.permissions)));
		UserTypeDAO.getInstance().saveOrUpdate(type);
		
		User u = new User();
		u.setUserId(123);
		u.setSsn("123");
		u.setPassword("123");
		u.setFirstName("Test");
		u.setLastName("User");
		u.setNewUserType(type);
		
		UserDAO dao = new UserDAO();
		dao.saveOrUpdate(u);
		
		MenuCategory category = new MenuCategory();
		category.setName("BEVERAGE");
		category.setBeverage(Boolean.TRUE);
		category.setVisible(Boolean.TRUE);
		MenuCategoryDAO.getInstance().saveOrUpdate(category);
		
		MenuCategory category2 = new MenuCategory();
		category2.setName("BREAKFAST");
		category2.setBeverage(Boolean.FALSE);
		category2.setVisible(Boolean.TRUE);
		MenuCategoryDAO.getInstance().saveOrUpdate(category2);
		
		MenuGroup group1 = new MenuGroup();
		group1.setParent(category);
		group1.setName("COLD BEVERAGE");
		group1.setVisible(Boolean.TRUE);
		MenuGroupDAO.getInstance().save(group1);
		
		MenuGroup group2 = new MenuGroup();
		group2.setParent(category2);
		group2.setName("FAVOURITE");
		group2.setVisible(Boolean.TRUE);
		MenuGroupDAO.getInstance().save(group2);

		MenuItem item1 = new MenuItem();
		item1.setParent(group1);
		item1.setName("Coke");
		item1.setPrice(Double.valueOf(2.0));
		item1.setTax(tax);
		item1.setVisible(Boolean.TRUE);
		MenuItemDAO.getInstance().save(item1);
		
		MenuItem item2 = new MenuItem();
		item2.setParent(group2);
		item2.setName("Egg");
		item2.setPrice(Double.valueOf(2.0));
		item2.setTax(tax);
		item2.setVisible(Boolean.TRUE);
		MenuItemDAO.getInstance().save(item2);
		
		
		System.exit(0);
	}
}