zhinengdahui/src/main/java/com/izouma/awesomeadmin/service/impl/UserInfoServiceImpl.java

package com.izouma.awesomeadmin.service.impl;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.izouma.awesomeadmin.constant.AppConstant;
import com.izouma.awesomeadmin.dao.DepartInfoMapper;
import com.izouma.awesomeadmin.dao.SysAppTokenMapper;
import com.izouma.awesomeadmin.dao.SysRoleMapper;
import com.izouma.awesomeadmin.dao.UserInfoMapper;
import com.izouma.awesomeadmin.dto.Page;
import com.izouma.awesomeadmin.model.UserInfo;
import com.izouma.awesomeadmin.service.OSSFileService;
import com.izouma.awesomeadmin.service.UserInfoService;
import com.izouma.awesomeadmin.shiro.AppToken;
import com.izouma.awesomeadmin.util.MD5Util;
import com.izouma.awesomeadmin.util.PropertiesFileLoader;
import com.izouma.awesomeadmin.util.WeixinUtil;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwt;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.security.Keys;
import io.rong.RongCloud;
import io.rong.models.SMSVerifyCodeResult;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import org.json.JSONException;
import org.json.JSONObject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.crypto.SecretKey;
import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * user_info service接口实现类
 * Tue Apr 17 10:32:49 CST 2018  Suo Chen Cheng
 */
@Service
public class UserInfoServiceImpl implements UserInfoService {

    private static Logger logger = Logger.getLogger(UserInfoServiceImpl.class);
    private RongCloud rongCloud = RongCloud.getInstance(PropertiesFileLoader.getProperties("rongyunappkey"), PropertiesFileLoader.getProperties("rongyunappsecret"));

    @Autowired
    private UserInfoMapper userInfoMapper;
    @Autowired
    private SysRoleMapper sysRoleMapper;
    @Autowired
    private DepartInfoMapper departInfoMapper;
    @Autowired
    private SysAppTokenMapper sysAppTokenMapper;
    @Autowired
    private OSSFileService ossFileService;

    @Override
    public List<UserInfo> getUserInfoList(UserInfo record) {

        logger.info("getUserInfoList");
        try {

            return userInfoMapper.queryAllUserInfo(record);
        } catch (Exception e) {
            logger.error("getUserInfoList", e);
        }

        return null;
    }

    @Override
    public List<UserInfo> getUserInfoByPage(Page page, UserInfo record) {

        logger.info("getUserInfoByPage");
        try {

            Map<String, Object> parameter = new HashMap<String, Object>();
            parameter.put("record", record);
            parameter.put(AppConstant.PAGE, page);

            return userInfoMapper.queryUserInfosByPage(parameter);
        } catch (Exception e) {
            logger.error("getUserInfoByPage", e);
        }

        return null;
    }

    @Override
    public UserInfo getUserInfoById(String id) {

        logger.info("getUserInfoById");
        try {

            return userInfoMapper.selectByPrimaryKey(Integer.valueOf(id));
        } catch (Exception e) {
            logger.error("getUserInfoById", e);
        }

        return null;
    }

    @Override
    public UserInfo getUserInfo(UserInfo record) {

        logger.info("getUserInfo");
        try {

            return userInfoMapper.queryUserInfo(record);
        } catch (Exception e) {
            logger.error("getUserInfo", e);
        }

        return null;
    }

    @Override
    public boolean createUserInfo(UserInfo record) {

        logger.info("createUserInfo");
        try {
            if (StringUtils.isNotEmpty(record.getPassword())) {
                record.setPassword(MD5Util.getMD5(record.getPassword()));
            }
            int updates = userInfoMapper.insertSelective(record);

            if (updateUserRolesAndDeparts(record, updates)) return true;
        } catch (Exception e) {
            logger.error("createUserInfo", e);
        }

        return false;
    }

    private boolean updateUserRolesAndDeparts(UserInfo record, int updates) {
        if (updates > 0) {
            if (record.getDepartId() != null) {
                departInfoMapper.clearUserDeparts(record.getId());
                departInfoMapper.setUserDeparts(record.getId(), Arrays.asList(record.getDepartId().split(",")));
            }
            if (record.getRoleId() != null) {
                sysRoleMapper.clearUserRoles(record.getId());
                sysRoleMapper.setUserRoles(record.getId(), Arrays.asList(record.getRoleId().split(",")));
            }
            return true;
        }
        return false;
    }

    @Override
    public boolean deleteUserInfo(String id) {

        logger.info("deleteUserInfo");
        try {

            int updates = userInfoMapper.delete(id);
            departInfoMapper.clearUserDeparts(Integer.valueOf(id));
            sysRoleMapper.clearUserRoles(Integer.valueOf(id));

            if (updates > 0) {
                return true;
            }
        } catch (Exception e) {
            logger.error("deleteUserInfo", e);
        }

        return false;
    }

    @Override
    public boolean updateUserInfo(UserInfo record) {

        logger.info("updateUserInfo");
        try {

            int updates = userInfoMapper.updateByPrimaryKeySelective(record);

            if (updateUserRolesAndDeparts(record, updates)) return true;
        } catch (Exception e) {
            logger.error("updateUserInfo", e);
        }

        return false;
    }

    @Override
    public boolean updatePassword(UserInfo record) {

        logger.info("updatePassword");
        try {
            if (StringUtils.isNotEmpty(record.getPassword())) {
                record.setPassword(MD5Util.getMD5(record.getPassword()));
            }

            int updates = userInfoMapper.updatePassword(record);

            if (updates > 0) {
                return true;
            }
        } catch (Exception e) {
            logger.error("updatePassword", e);
        }

        return false;
    }

    @Override
    public UserInfo login(String username, String password) {
        logger.info("login");
        try {
            Map<String, Object> map = new HashMap<>();
            map.put("username", username);
            map.put("password", MD5Util.getMD5(password));
            UserInfo result = userInfoMapper.login(map);
            return result;
        } catch (Exception e) {
            logger.error("login", e);
        }
        return null;
    }

    @Override
    public UserInfo loginSms(String phone, String code, String sessionId) throws LoginException {
        logger.info("loginSms");
        SMSVerifyCodeResult sMSVerifyCodeResult;
        try {
            sMSVerifyCodeResult = rongCloud.sms.verifyCode(sessionId, code);
        } catch (Exception e) {
            e.printStackTrace();
            throw new LoginException("验证码错误");
        }
        if (200 == sMSVerifyCodeResult.getCode()) {
            Boolean success = sMSVerifyCodeResult.getSuccess();
            if (success) {
                UserInfo userInfo = new UserInfo();
                userInfo.setPhone(phone);
                userInfo = getUserInfo(userInfo);
                if (userInfo == null) {
                    userInfo = new UserInfo();
                    userInfo.setPhone(phone);
                    userInfo.setUsername(phone);
                    userInfo.setNickname(phone);
                    userInfo.setIcon("https://microball.oss-cn-hangzhou.aliyuncs.com/huanbaojia/icon_morentouxiang.png");
                    if (!createUserInfo(userInfo)) {
                        throw new LoginException("登录失败");
                    }
                }
                return userInfo;
            }
        }
        throw new LoginException("验证码错误");
    }

    @Override
    public UserInfo loginAppToken(String token) {
        logger.info("loginAppToken");
        UserInfo userInfo = null;
        try {
            AppToken appToken = sysAppTokenMapper.getToken(token);
            if (appToken != null) {
                SecretKey key = Keys.hmacShaKeyFor(Base64.getDecoder().decode(PropertiesFileLoader.getProperties("jwtsecret").getBytes()));
                Jwt jwt = Jwts.parser()
                        .setSigningKey(key)
                        .parse(token);
                Claims claims = (Claims) jwt.getBody();
                if (claims.getExpiration() != null) {
                    if (claims.getExpiration().before(new Date())) {
                        return null;
                    }
                }
                userInfo = getUserInfoById(claims.getSubject());
            }
        } catch (Exception e) {
            logger.error("loginAppToken", e);
        }
        return userInfo;
    }

    @Override
    public List<String> findDepartLeader(String userId) {
        logger.info("findDepartLeader");
        try {
            return userInfoMapper.findDepartLeader(userId);
        } catch (Exception e) {
            logger.error("findDepartLeader", e);
        }
        return null;
    }

    @Override
    public List<String> findUserByRoleName(String roleName) {
        logger.info("findUserByRoleName");
        try {
            return userInfoMapper.findUserByRoleName(roleName);
        } catch (Exception e) {
            logger.error("findUserByRoleName", e);
        }
        return null;
    }

    public class LoginException extends Exception {
        public LoginException(String message) {
            super(message);
        }
    }

    @Override
    public UserInfo loginWechat(String code) {
        try {
            final String APP_ID = PropertiesFileLoader.getProperties("weixinappid");
            final String APP_SECRET = PropertiesFileLoader.getProperties("weixinsecret");

            String accessTokenUrl = "https://api.weixin.qq.com/sns/oauth2/access_token?appid=" + APP_ID + "&secret=" + APP_SECRET + "&code=" + code
                    + "&grant_type=authorization_code";
            JSONObject data = WeixinUtil.loadJSON(accessTokenUrl);
            logger.debug("微信授权获取access_token：\n" + data.toString(4));
            String openId = data.getString("openid");
            String access_token = data.getString("access_token");

            String userDataUrl = "https://api.weixin.qq.com/sns/userinfo?access_token=" + access_token + "&openid=" + openId;
            JSONObject userData = WeixinUtil.loadJSON(userDataUrl);
            logger.debug("微信授权获取用户信息：\n" + userData.toString(4));

            UserInfo userInfo = new UserInfo();
            userInfo.setOpenId(openId);
            userInfo = getUserInfo(userInfo);

            if (userInfo != null) {
                return userInfo;
            }
            userInfo = new UserInfo();
            userInfo.setOpenId(openId);
            String nickname = userData.getString("nickname");
            userInfo.setNickname(nickname);
            userInfo.setUsername(nickname);
            try {
                String country = userData.getString("country");
                userInfo.setCountry(country);
            } catch (JSONException ignored) {
            }
            try {
                String province = userData.getString("province");
                userInfo.setProvince(province);
            } catch (JSONException ignored) {
            }
            try {
                String city = userData.getString("city");
                userInfo.setCity(city);
            } catch (JSONException ignored) {
            }
            int sex = 1;
            try {
                sex = userData.getInt("sex");
            } catch (JSONException ignored) {
            }
            userInfo.setSex(sex == 1 ? "男" : "女");
            String headimgurl = null;
            try {
                headimgurl = userData.getString("headimgurl");
            } catch (JSONException ignored) {
            }
            userInfo.setIcon(saveAvatar(headimgurl));
            if (createUserInfo(userInfo)) {
                return userInfo;
            }
        } catch (Exception e) {
            logger.error("loginWechat", e);
        }
        return null;
    }

    private String saveAvatar(String url) {
        String path;
        HttpURLConnection httpUrl = null;
        URL iconUrl = null;
        try {
            iconUrl = new URL(url);
            httpUrl = (HttpURLConnection) iconUrl.openConnection();
            httpUrl.connect();

            Random random = new Random();
            StringBuilder randomCode = new StringBuilder();
            for (int i = 0; i < 8; i++) {
                randomCode.append(Integer.toString(random.nextInt(36), 36));
            }
            String uploadPath = String.format("images/%s-%s.jpg", new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss").format(new Date()), randomCode);
            path = ossFileService.upload(httpUrl.getInputStream(), uploadPath);
        } catch (IOException e) {
            path = "https://microball.oss-cn-hangzhou.aliyuncs.com/huanbaojia/icon_morentouxiang.png";
            e.printStackTrace();
        } finally {
            if (httpUrl != null) {
                httpUrl.disconnect();
            }
        }
        return path;
    }
}