package service;

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

import dao.Dao; // 儲(chǔ)存相關(guān)配置文件
import dao.UserInfoDao; // 用于將用戶信息存盤
import entity.UserInfo; // 用戶信息實(shí)體類，其中的兩個(gè)類成員是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); // 從相關(guān)文件中讀取用戶信息
		if (userInfoSet == null) { // 若尚無(wú)用戶注冊(cè)，則new HashSet<String>()，避免NullPointerException
			userInfoSet = new HashSet<String>();
		}
		if (userInfoSet.contains(getUserInfoHashCode(userInfo))) { // 判斷是否含有相關(guān)用戶信息
			tips = "登錄成功！";
		} else {
			tips = "登錄失??！請(qǐng)檢查用戶名或密碼";
		}
		return tips;
	}

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

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

	/**
	 * 以指定算法algorithm加密用戶名和密碼，以指定基數(shù)radix長(zhǎng)度保存
	 * 
	 * @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)建一個(gè)MessageDigest實(shí)例
			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); // 返回一個(gè)指定進(jìn)制基數(shù)為radix的字符串
		} catch (Exception e) {
			e.printStackTrace();
			return ""; // 若異常則返回空字符串
		}
	}
}