BlogArchive.java

/**
 * myJerry | Evenstar
 * Copyright (C) 2010 myJerry Development Team
 * http://www.myjerry.org
 * 
 * The file is licensed under the the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * 
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * 
 */
package org.myjerry.evenstar.model;

import java.util.ArrayList;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import org.myjerry.evenstar.constants.CalendarConstants;
import org.myjerry.evenstar.core.ArchiveKey;
import org.myjerry.evenstar.enums.BlogArchiveFrequency;

public class BlogArchive {
	
	private Long blogID;
	
	private BlogArchiveFrequency frequency;
	
	private List<BlogArchivePost> archives = new ArrayList<BlogArchivePost>();
	
	@SuppressWarnings("unchecked")
	private Map data;
	
	public BlogArchive(Long blogID, BlogArchiveFrequency frequency) {
		this.blogID = blogID;
		this.frequency = frequency;
	}
	
	public void addArchivePost(BlogArchivePost archivePost) {
		this.archives.add(archivePost);
	}
	
	public void addBlogArchivePost(Long postID, String title, String url, Date postedDate) {
		BlogArchivePost post = new BlogArchivePost(postID, title, url, postedDate);
		this.archives.add(post);
	}
	
	public void reIndex() {
		if(this.archives.size() == 0) {
			// nothing to index
			return;
		}
		
		switch(this.frequency) {
			case DAILY:
				this.data = dailyIndex();
				break;
				
			case WEEKLY:
				this.data = weeklyIndex();
				break;
				
			case MONTHLY:
				this.data = monthlyIndex();
				break;
		}
	}
	
	@SuppressWarnings("unchecked")
	private Map monthlyIndex() {
		Map<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>> yearIndex = new TreeMap<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>>();
		
		ArchiveKey key = null;
		
		for(BlogArchivePost post : this.archives) {
			Calendar calendar = Calendar.getInstance();
			calendar.setTime(post.getPostedDate());
			int year = calendar.get(Calendar.YEAR);
			int month = calendar.get(Calendar.MONTH) + 1;
		
			// month, monthIndex
			Map<ArchiveKey, List<BlogArchivePost>> monthIndex = null;
			key = getYearKey(year);
			if(yearIndex.containsKey(key)) {
				monthIndex = yearIndex.get(key);
			} else {
				monthIndex = new TreeMap<ArchiveKey, List<BlogArchivePost>>();
				yearIndex.put(key, monthIndex);
			}
			
			List<BlogArchivePost> list = null;
			key = getMonthKey(year, month);
			if(monthIndex.containsKey(key)) {
				list = monthIndex.get(key);
			} else {
				list = new ArrayList<BlogArchivePost>();
				monthIndex.put(key, list);
			}
			
			list.add(post);
		}
		
		return yearIndex;
	}

	@SuppressWarnings("unchecked")
	private Map weeklyIndex() {
		Map<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>> yearIndex = new TreeMap<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>>();
		
		ArchiveKey key = null;
		
		for(BlogArchivePost post : this.archives) {
			Calendar calendar = Calendar.getInstance();
			calendar.setTime(post.getPostedDate());
			int year = calendar.get(Calendar.YEAR);
			int month = calendar.get(Calendar.MONTH) + 1;
			int day = calendar.get(Calendar.DATE);
			int week = calendar.get(Calendar.WEEK_OF_YEAR);
			
			Map<ArchiveKey, List<BlogArchivePost>> weekIndex = null;
			key = getYearKey(year);
			if(yearIndex.containsKey(key)) {
				weekIndex = yearIndex.get(key);
			} else {
				weekIndex = new TreeMap<ArchiveKey, List<BlogArchivePost>>();
				yearIndex.put(key, weekIndex);
			}
			
			List<BlogArchivePost> list = null;
			key = getWeekKey(year, month, day, week);
			if(weekIndex.containsKey(key)) {
				list = weekIndex.get(key);
			} else {
				list = new ArrayList<BlogArchivePost>();
				weekIndex.put(key, list);
			}
			
			list.add(post);
		}
		
		return yearIndex;
	}

	@SuppressWarnings("unchecked")
	private Map dailyIndex() {
		// year, yearIndex
		Map<ArchiveKey, Map<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>>> yearIndex 
				= new TreeMap<ArchiveKey, Map<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>>>();
		
		ArchiveKey key = null;
		
		for(BlogArchivePost post : this.archives) {
			Calendar calendar = Calendar.getInstance();
			calendar.setTime(post.getPostedDate());
			int year = calendar.get(Calendar.YEAR);
			int month = calendar.get(Calendar.MONTH) + 1;
			int day = calendar.get(Calendar.DAY_OF_MONTH);
		
			// month, monthIndex
			Map<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>> monthIndex = null;
			key = getYearKey(year);
			if(yearIndex.containsKey(key)) {
				monthIndex = yearIndex.get(key);
			} else {
				monthIndex = new TreeMap<ArchiveKey, Map<ArchiveKey, List<BlogArchivePost>>>();
				yearIndex.put(key, monthIndex);
			}
			
			// day, posts
			Map<ArchiveKey, List<BlogArchivePost>> dayIndex = null;
			key = getMonthKey(year, month);
			if(monthIndex.containsKey(key)) {
				dayIndex = monthIndex.get(key);
			} else {
				dayIndex = new TreeMap<ArchiveKey, List<BlogArchivePost>>();
				monthIndex.put(key, dayIndex);
			}
			
			List<BlogArchivePost> list = null;
			key = getDayKey(year, month, day);
			if(dayIndex.containsKey(key)) {
				list = dayIndex.get(key);
			} else {
				list = new ArrayList<BlogArchivePost>();
				dayIndex.put(key, list);
			}
			
			list.add(post);
		}
		
		return yearIndex;
	}
	
	private static final ArchiveKey getYearKey(int year) {
		ArchiveKey key = new ArchiveKey(year);
		
		Calendar calendar = Calendar.getInstance();
		calendar.set(year, 0, 1, 0, 0, 0);
		Long firstDay = calendar.getTimeInMillis();
		calendar.set(year, 11, 31, 23, 59, 59);
		Long lastDay = calendar.getTimeInMillis();
		
		key.setUrl("/showPosts.html?newer=" + firstDay + "&older=" + lastDay);
		
		return key;
	}
	
	private static final ArchiveKey getMonthKey(int year, int month) {
		ArchiveKey key = new ArchiveKey(month, CalendarConstants.MONTHS[month - 1]);

		Calendar calendar = Calendar.getInstance();
		calendar.set(year, month - 1, 1, 0, 0, 0);
		Long firstDay = calendar.getTimeInMillis();
		int lastDate = calendar.getActualMaximum(Calendar.DATE);
		calendar.set(year, month - 1, lastDate, 23, 59, 59);
		Long lastDay = calendar.getTimeInMillis();
		
		key.setUrl("/showPosts.html?newer=" + firstDay + "&older=" + lastDay);

		return key;
	}
	
	private static final ArchiveKey getWeekKey(int year, int month, int day, int week) {
		ArchiveKey key = new ArchiveKey(week);
		
		Calendar calendar = Calendar.getInstance();
		calendar.set(year, month, day);

		return key;
	}
	
	private static final ArchiveKey getDayKey(int year, int month, int day) {
		ArchiveKey key = new ArchiveKey(day, CalendarConstants.MONTHS[month - 1] + " " + day);

		Calendar calendar = Calendar.getInstance();
		calendar.set(year, month - 1, day, 0, 0, 0);
		Long firstDay = calendar.getTimeInMillis();
		calendar.set(year, month - 1, day, 23, 59, 59);
		Long lastDay = calendar.getTimeInMillis();
		
		key.setUrl("/showPosts.html?newer=" + firstDay + "&older=" + lastDay);

		return key;
	}

	/**
	 * @return the blogID
	 */
	public Long getBlogID() {
		return blogID;
	}

	/**
	 * @return the archives
	 */
	public List<BlogArchivePost> getArchives() {
		return archives;
	}

	/**
	 * @return the frequency
	 */
	public BlogArchiveFrequency getFrequency() {
		return frequency;
	}
	
	@SuppressWarnings("unchecked")
	public Map getData() {
		return this.data;
	}

}