package service;

import java.math.BigInteger;
import java.security.MessageDigest;
import java.util.HashSet;
import java.util.Set;

import dao.Dao; // 儲存相關配置文件
import dao.UserInfoDao; // 用于將用戶信息存盤
import entity.UserInfo; // 用戶信息實體類，其中的兩個類成員是userName和userPassword，即用戶名和密碼

public class UserInfoService extends Dao {
	private String userInfoPath; // 用戶信息保存的文件路徑
	private String userNamePath; // 用戶名保存的文件路徑
	private UserInfoDao dao = new UserInfoDao();

	public UserInfoService() {
		super();
		userInfoPath = super.getResource().getString("userInfo");
		userNamePath = super.getResource().getString("userName");
	}

	/**
	 * 用戶登錄。若用戶信息存在，則登錄成功；若用戶信息不存在，則登錄失敗
	 * 
	 * @param userInfo
	 * @return 提示信息
	 */
	public String userSignIn(UserInfo userInfo) {
		Set<String> userInfoSet = null;
		String tips;
		userInfoSet = dao.readInfo(userInfoPath); // 從相關文件中讀取用戶信息
		if (userInfoSet == null) { // 若尚無用戶注冊，則new HashSet<String>()，避免NullPointerException
			userInfoSet = new HashSet<String>();
		}
		if (userInfoSet.contains(getUserInfoHashCode(userInfo))) { // 判斷是否含有相關用戶信息
			tips = "登錄成功！";
		} else {
			tips = "登錄失敗！請檢查用戶名或密碼";
		}
		return tips;
	}

	/**
	 * 用戶注冊。若用戶名不存在，則注冊成功；若用戶名存在，則注冊失敗
	 * 
	 * @param userInfo
	 * @return 提示信息
	 */
	public String userSignUp(UserInfo userInfo) {
		Set<String> userInfoSet = null;
		Set<String> userNameSet = null;
		String tips;
		userInfoSet = dao.readInfo(userInfoPath); // 從相關文件中讀取用戶信息
		userNameSet = dao.readInfo(userNamePath); // 從相關文件中讀取用戶名
		if (userInfoSet == null) { // 若尚無用戶注冊，則new HashSet<String>()，避免NullPointerException
			userInfoSet = new HashSet<String>();
		}
		if (userNameSet == null) { // 若尚無用戶注冊，則new HashSet<String>()，避免NullPointerException
			userNameSet = new HashSet<String>();
		}
		if (userNameSet.add(userInfo.getUserName())) { // 判斷用戶名是否已注冊
			userInfoSet.add(getUserInfoHashCode(userInfo)); // 若用戶名未注冊，則將用戶信息添加至Set中
			dao.saveInfo(userInfoSet, userInfoPath); // 保存用戶信息到相關文件
			dao.saveInfo(userNameSet, userNamePath); // 保存用戶名到相關文件
			tips = "注冊成功！";
		} else {
			tips = "注冊失?。∮脩粢汛嬖?;
		}
		return tips;
	}

	/**
	 * 以預設算法SHA-1加密用戶名和密碼，以預設基數(shù)36位保存
	 * 
	 * @param userInfo
	 * @return 加密后的用戶信息
	 */
	public String getUserInfoHashCode(UserInfo userInfo) {
		return getUserInfoHashCode(userInfo, "SHA-1", 36); // 用SHA-1算法生成用戶信息密鑰，進制為36進制
	}

	/**
	 * 以指定算法algorithm加密用戶名和密碼，以指定基數(shù)radix長度保存
	 * 
	 * @param userInfo
	 * @param algorithm
	 * @param radix
	 * @return 加密后的用戶信息
	 */
	public String getUserInfoHashCode(UserInfo userInfo, String algorithm, int radix) {
		try {
			MessageDigest md = MessageDigest.getInstance(algorithm); // 用指定算法algorithm創(chuàng)建一個MessageDigest實例
			md.update((userInfo.getUserName() + userInfo.getUserPassword()).getBytes("UTF-8")); // 將用戶名和密碼合并，調(diào)用update()輸入數(shù)據(jù)
			byte[] res = md.digest(); // 將摘要存放在byte[]中
			return new BigInteger(1, res).toString(radix); // 返回一個指定進制基數(shù)為radix的字符串
		} catch (Exception e) {
			e.printStackTrace();
			return ""; // 若異常則返回空字符串
		}
	}
}