/*
 * This file is part of Nuts Framework.
 * Copyright(C) 2009-2012 Nuts Develop Team.
 *
 * Nuts Framework is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License any later version.
 * 
 * Nuts Framework is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Nuts Framework. If not, see <http://www.gnu.org/licenses/>.
 */
package nuts.core.servlet;

import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

import javax.servlet.http.HttpServletRequest;

import nuts.core.lang.StringUtils;
import nuts.core.net.HttpHeaderDefine;



/**
 */
public class UserAgent {

	private String userAgent;
	private Map<String, String> browsers;

	public static final String CHROME = "Chrome";
	public static final String FIREFOX = "Firefox";
	public static final String MSIE = "MSIE";
	public static final String NETSCAPE = "Netscape";
	public static final String OPERA = "Opera";
	public static final String WEBKIT = "WebKit";
	public static final String SAFARI = "Safari";
	public static final String IPHONE = "iPhone";
	public static final String IPAD = "iPad";
	public static final String IPOD = "iPod";
	public static final String ANDROID = "Android";
	
	public static final String GOOGLEBOT = "Googlebot";
	public static final String BINGBOT = "bingbot";

	public static final String MOBILE = "mobile";
	public static final String ROBOT = "robot";

	/**
	 * @param request request
	 */
	public UserAgent(HttpServletRequest request) {
		this(request.getHeader(HttpHeaderDefine.USER_AGENT));
	}

	/**
	 * @param userAgent user agent
	 */
	public UserAgent(String userAgent) {
		super();
		this.userAgent = userAgent;
		init();
	}

	private void init() {
		browsers = new HashMap<String, String>();
		
		if (StringUtils.isEmpty(userAgent)) {
			return;
		}
		
		parse(CHROME);
		parse(FIREFOX);
		parse(MSIE);
		parse(NETSCAPE);
		parse(OPERA);
		parse(WEBKIT);
		parse(SAFARI);
		parse(IPHONE);
		parse(IPAD);
		parse(IPOD);
		parse(ANDROID);
		
		parse(GOOGLEBOT);
		parse(BINGBOT);
		
		if (isGooglebot() || isBingbot()) {
			browsers.put(ROBOT, "");
		}
		if (isAndroid() || isIphone() || isIpad() || isIpod()) {
			browsers.put(MOBILE, "");
		}
	}
	
	private void parse(String client) {
		int i = userAgent.indexOf(client);
		if (i >= 0) {
			for (i += client.length(); i < userAgent.length(); i++) {
				char c = userAgent.charAt(i);
				if (Character.isDigit(c)) {
					break;
				}
			}
			
			int j = i;
			for (; j < userAgent.length(); j++) {
				char c = userAgent.charAt(j);
				if (!Character.isDigit(c) && c != '.') {
					break;
				}
			}
			
			String ver = "";
			if (i < userAgent.length()) {
				ver = userAgent.substring(i, j);
			}
			
			browsers.put(client, ver);
		}
	}
	
	/**
	 * @return true if user agent is Chrome
	 */
	public boolean isChrome() {
		return browsers.containsKey(CHROME);
	}

	/**
	 * @return true if user agent is MSIE
	 */
	public boolean isMsie() {
		return browsers.containsKey(MSIE);
	}

	/**
	 * @return true if user agent is Netscape
	 */
	public boolean isNetscape() {
		return browsers.containsKey(NETSCAPE);
	}

	/**
	 * @return true if user agent is Firefox
	 */
	public boolean isFirefox() {
		return browsers.containsKey(FIREFOX);
	}

	/**
	 * @return true if user agent is Safari
	 */
	public boolean isSafari() {
		return browsers.containsKey(SAFARI);
	}

	/**
	 * @return true if user agent is Opera
	 */
	public boolean isOpera() {
		return browsers.containsKey(OPERA);
	}

	/**
	 * @return true if user agent is Mobile
	 */
	public boolean isMobile() {
		return browsers.containsKey(MOBILE);
	}

	/**
	 * @return true if user agent is Robot
	 */
	public boolean isRobot() {
		return browsers.containsKey(ROBOT);
	}

	/**
	 * @return true if user agent is Googlebot
	 */
	public boolean isGooglebot() {
		return browsers.containsKey(GOOGLEBOT);
	}

	/**
	 * @return true if user agent is Bingbot
	 */
	public boolean isBingbot() {
		return browsers.containsKey(BINGBOT);
	}

	/**
	 * @return true if user agent is Android
	 */
	public boolean isAndroid() {
		return browsers.containsKey(ANDROID);
	}

	/**
	 * @return true if user agent is iPad
	 */
	public boolean isIpad() {
		return browsers.containsKey(IPAD);
	}

	/**
	 * @return true if user agent is iPhone
	 */
	public boolean isIphone() {
		return browsers.containsKey(IPHONE);
	}

	/**
	 * @return true if user agent is iPod
	 */
	public boolean isIpod() {
		return browsers.containsKey(IPOD);
	}

	/**
	 * @return true if user agent is ios
	 */
	public boolean isIos() {
		return browsers.containsKey("ios");
	}

	private Integer parseMajorVersion(String ver) {
		if (ver != null) {
			int i = ver.indexOf('.');
			if (i > 0) {
				int v = Integer.parseInt(ver.substring(0, i));
				return v;
			}
		}
		return null;
	}

	/**
	 * @param client client 
	 * @return major version of the client
	 */
	public Integer getMajorVersion(String client) {
		return parseMajorVersion(browsers.get(client));
	}

	/**
	 * @return simple string
	 */
	public String toSimpleString() {
		StringBuilder sb = new StringBuilder();
		for (Entry<String, String> en : browsers.entrySet()) {
			String b = en.getKey().toLowerCase();
			sb.append(b).append(' ');
			Integer v = parseMajorVersion(en.getValue());
			if (v != null) {
				sb.append(b).append(v).append(' ');
			}
		}
		return sb.toString();
	}
	
	/**
	 * @return user agent string
	 */
	@Override
	public String toString() {
		return userAgent;
	}
}
