Athou_commafeed/src/main/java/com/commafeed/CommaFeedApplication.java

package com.commafeed;

import io.dropwizard.Application;
import io.dropwizard.assets.AssetsBundle;
import io.dropwizard.auth.CachingAuthenticator;
import io.dropwizard.auth.basic.BasicAuthProvider;
import io.dropwizard.auth.basic.BasicCredentials;
import io.dropwizard.db.DataSourceFactory;
import io.dropwizard.hibernate.HibernateBundle;
import io.dropwizard.migrations.MigrationsBundle;
import io.dropwizard.setup.Bootstrap;
import io.dropwizard.setup.Environment;

import java.util.Date;

import org.hibernate.SessionFactory;

import com.codahale.metrics.MetricRegistry;
import com.commafeed.backend.HttpGetter;
import com.commafeed.backend.cache.CacheService;
import com.commafeed.backend.cache.NoopCacheService;
import com.commafeed.backend.dao.FeedCategoryDAO;
import com.commafeed.backend.dao.FeedDAO;
import com.commafeed.backend.dao.FeedEntryContentDAO;
import com.commafeed.backend.dao.FeedEntryDAO;
import com.commafeed.backend.dao.FeedEntryStatusDAO;
import com.commafeed.backend.dao.FeedEntryTagDAO;
import com.commafeed.backend.dao.FeedSubscriptionDAO;
import com.commafeed.backend.dao.UserDAO;
import com.commafeed.backend.dao.UserRoleDAO;
import com.commafeed.backend.dao.UserSettingsDAO;
import com.commafeed.backend.feed.FaviconFetcher;
import com.commafeed.backend.feed.FeedFetcher;
import com.commafeed.backend.feed.FeedParser;
import com.commafeed.backend.feed.FeedQueues;
import com.commafeed.backend.feed.FeedRefreshTaskGiver;
import com.commafeed.backend.feed.FeedRefreshUpdater;
import com.commafeed.backend.feed.FeedRefreshWorker;
import com.commafeed.backend.model.AbstractModel;
import com.commafeed.backend.model.Feed;
import com.commafeed.backend.model.FeedCategory;
import com.commafeed.backend.model.FeedEntry;
import com.commafeed.backend.model.FeedEntryContent;
import com.commafeed.backend.model.FeedEntryStatus;
import com.commafeed.backend.model.FeedEntryTag;
import com.commafeed.backend.model.FeedSubscription;
import com.commafeed.backend.model.User;
import com.commafeed.backend.model.UserRole;
import com.commafeed.backend.model.UserSettings;
import com.commafeed.backend.opml.OPMLExporter;
import com.commafeed.backend.opml.OPMLImporter;
import com.commafeed.backend.service.ApplicationPropertiesService;
import com.commafeed.backend.service.DatabaseCleaningService;
import com.commafeed.backend.service.FeedEntryContentService;
import com.commafeed.backend.service.FeedEntryService;
import com.commafeed.backend.service.FeedEntryTagService;
import com.commafeed.backend.service.FeedService;
import com.commafeed.backend.service.FeedSubscriptionService;
import com.commafeed.backend.service.FeedUpdateService;
import com.commafeed.backend.service.PasswordEncryptionService;
import com.commafeed.backend.service.PubSubService;
import com.commafeed.backend.service.StartupService;
import com.commafeed.backend.service.UserService;
import com.commafeed.frontend.resource.AdminREST;
import com.commafeed.frontend.resource.CategoryREST;
import com.commafeed.frontend.resource.EntryREST;
import com.commafeed.frontend.resource.FeedREST;
import com.commafeed.frontend.resource.PubSubHubbubCallbackREST;
import com.commafeed.frontend.resource.ServerREST;
import com.commafeed.frontend.resource.UserREST;

public class CommaFeedApplication extends Application<CommaFeedConfiguration> {

	public static final String USERNAME_ADMIN = "admin";
	public static final String USERNAME_DEMO = "demo";

	public static final Date STARTUP_TIME = new Date();

	private HibernateBundle<CommaFeedConfiguration> hibernateBundle;
	private MigrationsBundle<CommaFeedConfiguration> migrationsBundle;

	@Override
	public void initialize(Bootstrap<CommaFeedConfiguration> bootstrap) {
		hibernateBundle = new HibernateBundle<CommaFeedConfiguration>(AbstractModel.class, Feed.class, FeedCategory.class, FeedEntry.class,
				FeedEntryContent.class, FeedEntryStatus.class, FeedEntryTag.class, FeedSubscription.class, User.class, UserRole.class,
				UserSettings.class) {
			@Override
			public DataSourceFactory getDataSourceFactory(CommaFeedConfiguration configuration) {
				return configuration.getDatabase();
			}
		};
		bootstrap.addBundle(hibernateBundle);

		migrationsBundle = new MigrationsBundle<CommaFeedConfiguration>() {
			@Override
			public DataSourceFactory getDataSourceFactory(CommaFeedConfiguration configuration) {
				return configuration.getDatabase();
			}
		};
		bootstrap.addBundle(migrationsBundle);

		bootstrap.addBundle(new AssetsBundle("/assets/", "/", "index.html"));
	}

	@Override
	public void run(CommaFeedConfiguration config, Environment environment) throws Exception {
		MetricRegistry metrics = environment.metrics();
		SessionFactory sessionFactory = hibernateBundle.getSessionFactory();

		// TODO select cache service at runtime from config
		CacheService cacheService = new NoopCacheService();

		FeedCategoryDAO feedCategoryDAO = new FeedCategoryDAO(sessionFactory);
		FeedDAO feedDAO = new FeedDAO(sessionFactory);
		FeedEntryContentDAO feedEntryContentDAO = new FeedEntryContentDAO(sessionFactory);
		FeedEntryDAO feedEntryDAO = new FeedEntryDAO(sessionFactory);
		FeedEntryTagDAO feedEntryTagDAO = new FeedEntryTagDAO(sessionFactory);
		FeedSubscriptionDAO feedSubscriptionDAO = new FeedSubscriptionDAO(sessionFactory);
		UserDAO userDAO = new UserDAO(sessionFactory);
		UserRoleDAO userRoleDAO = new UserRoleDAO(sessionFactory);
		UserSettingsDAO userSettingsDAO = new UserSettingsDAO(sessionFactory);
		FeedEntryStatusDAO feedEntryStatusDAO = new FeedEntryStatusDAO(sessionFactory, feedEntryDAO, feedEntryTagDAO, config);

		FeedQueues queues = new FeedQueues(feedDAO, config, metrics);

		ApplicationPropertiesService applicationPropertiesService = new ApplicationPropertiesService();
		DatabaseCleaningService cleaningService = new DatabaseCleaningService(feedDAO, feedEntryDAO, feedEntryContentDAO,
				feedEntryStatusDAO, feedSubscriptionDAO);
		FeedEntryContentService feedEntryContentService = new FeedEntryContentService(feedEntryContentDAO);
		FeedEntryService feedEntryService = new FeedEntryService(feedSubscriptionDAO, feedEntryDAO, feedEntryStatusDAO, cacheService);
		FeedEntryTagService feedEntryTagService = new FeedEntryTagService(feedEntryDAO, feedEntryTagDAO);
		FeedService feedService = new FeedService(feedDAO);
		FeedSubscriptionService feedSubscriptionService = new FeedSubscriptionService(feedEntryStatusDAO, feedSubscriptionDAO, feedService,
				queues, cacheService, config);
		FeedUpdateService feedUpdateService = new FeedUpdateService(feedEntryDAO, feedEntryContentService);
		PasswordEncryptionService encryptionService = new PasswordEncryptionService();
		PubSubService pubSubService = new PubSubService(config, queues);
		UserService userService = new UserService(feedCategoryDAO, userDAO, userSettingsDAO, feedSubscriptionService, encryptionService,
				config);
		StartupService startupService = new StartupService(sessionFactory, userDAO, userService);

		OPMLImporter opmlImporter = new OPMLImporter(feedCategoryDAO, feedSubscriptionService, cacheService);
		OPMLExporter opmlExporter = new OPMLExporter(feedCategoryDAO, feedSubscriptionDAO);

		HttpGetter httpGetter = new HttpGetter();
		FeedParser feedParser = new FeedParser();
		FaviconFetcher faviconFetcher = new FaviconFetcher(httpGetter);

		FeedFetcher feedFetcher = new FeedFetcher(feedParser, httpGetter);
		FeedRefreshUpdater feedUpdater = new FeedRefreshUpdater(sessionFactory, feedUpdateService, pubSubService, queues, config, metrics,
				feedSubscriptionDAO, cacheService);
		FeedRefreshWorker feedWorker = new FeedRefreshWorker(feedUpdater, feedFetcher, queues, config, metrics);
		FeedRefreshTaskGiver taskGiver = new FeedRefreshTaskGiver(sessionFactory, queues, feedDAO, feedWorker, config, metrics);

		CachingAuthenticator<BasicCredentials, User> cachingAuthenticator = new CachingAuthenticator<BasicCredentials, User>(
				environment.metrics(), new CommaFeedAuthenticator(userService), config.getAuthenticationCachePolicy());
		environment.jersey().register(new BasicAuthProvider<User>(cachingAuthenticator, "CommaFeed"));

		environment.jersey().setUrlPattern("/rest/*");
		environment.jersey()
				.register(new AdminREST(userDAO, userRoleDAO, userService, encryptionService, cleaningService, config, metrics));
		environment.jersey().register(
				new CategoryREST(feedCategoryDAO, feedEntryStatusDAO, feedSubscriptionDAO, feedEntryService, feedSubscriptionService,
						cacheService, config));
		environment.jersey().register(new EntryREST(feedEntryTagDAO, feedEntryService, feedEntryTagService));
		environment.jersey().register(
				new FeedREST(feedSubscriptionDAO, feedCategoryDAO, feedEntryStatusDAO, faviconFetcher, feedFetcher, feedEntryService,
						feedSubscriptionService, queues, opmlImporter, opmlExporter, cacheService, config));
		environment.jersey().register(new PubSubHubbubCallbackREST(feedDAO, feedParser, queues, config, metrics));
		environment.jersey().register(new ServerREST(httpGetter, config, applicationPropertiesService));
		environment.jersey().register(new UserREST(userDAO, userRoleDAO, userSettingsDAO, userService, encryptionService));

		environment.lifecycle().manage(startupService);
		environment.lifecycle().manage(taskGiver);
		environment.lifecycle().manage(feedWorker);
		environment.lifecycle().manage(feedUpdater);
	}

	public static void main(String[] args) throws Exception {
		new CommaFeedApplication().run(args);
	}
}