Athou_commafeed/src/main/java/com/commafeed/backend/StartupBean.java

package com.commafeed.backend;

import java.util.Arrays;
import java.util.Calendar;

import javax.annotation.PostConstruct;
import javax.ejb.Singleton;
import javax.ejb.Startup;
import javax.inject.Inject;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.commafeed.backend.dao.FeedCategoryService;
import com.commafeed.backend.dao.FeedService;
import com.commafeed.backend.dao.FeedSubscriptionService;
import com.commafeed.backend.dao.UserService;
import com.commafeed.backend.model.Feed;
import com.commafeed.backend.model.FeedCategory;
import com.commafeed.backend.model.FeedSubscription;
import com.commafeed.backend.model.User;
import com.commafeed.backend.model.UserRole.Role;
import com.commafeed.backend.security.PasswordEncryptionService;

@Startup
@Singleton
public class StartupBean {

	private static Logger log = LoggerFactory.getLogger(StartupBean.class);
	public static final String ADMIN_NAME = "admin";

	@Inject
	FeedService feedService;

	@Inject
	FeedCategoryService feedCategoryService;

	@Inject
	FeedSubscriptionService feedSubscriptionService;

	@Inject
	UserService userService;

	@Inject
	PasswordEncryptionService encryptionService;

	private long startupTime;

	@PostConstruct
	private void init() {
		startupTime = Calendar.getInstance().getTimeInMillis();
		if (userService.getCount() == 0) {
			log.info("Populating database with default values");

			User user = userService.register(ADMIN_NAME, "admin",
					Arrays.asList(Role.ADMIN, Role.USER));
			userService.register("test", "test", Arrays.asList(Role.USER));

			Feed dilbert = new Feed(
					"http://feed.dilbert.com/dilbert/daily_strip");
			feedService.save(dilbert);

			Feed engadget = new Feed("http://www.engadget.com/rss.xml");
			feedService.save(engadget);

			Feed frandroid = new Feed("http://feeds.feedburner.com/frandroid");
			feedService.save(frandroid);

			FeedCategory newsCategory = new FeedCategory();
			newsCategory.setName("News");
			newsCategory.setUser(user);
			feedCategoryService.save(newsCategory);

			FeedCategory comicsCategory = new FeedCategory();
			comicsCategory.setName("Comics");
			comicsCategory.setUser(user);
			comicsCategory.setParent(newsCategory);
			feedCategoryService.save(comicsCategory);

			FeedCategory techCategory = new FeedCategory();
			techCategory.setName("Tech");
			techCategory.setUser(user);
			techCategory.setParent(newsCategory);
			feedCategoryService.save(techCategory);

			FeedSubscription sub = new FeedSubscription();
			sub.setCategory(comicsCategory);
			sub.setFeed(dilbert);
			sub.setTitle("Dilbert - Strips");
			sub.setUser(user);
			feedSubscriptionService.save(sub);

			FeedSubscription sub2 = new FeedSubscription();
			sub2.setCategory(techCategory);
			sub2.setFeed(engadget);
			sub2.setTitle("Engadget");
			sub2.setUser(user);
			feedSubscriptionService.save(sub2);

			FeedSubscription sub3 = new FeedSubscription();
			sub3.setFeed(frandroid);
			sub3.setTitle("Frandroid");
			sub3.setUser(user);
			feedSubscriptionService.save(sub3);

		}

	}

	public long getStartupTime() {
		return startupTime;
	}

}