package kr.wisestone.owl.web.controller;

import com.google.gson.Gson;
import kr.wisestone.owl.domain.User;
import kr.wisestone.owl.util.CommonUtil;
import kr.wisestone.owl.util.MapUtil;
import kr.wisestone.owl.util.SecurityUtils;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.messaging.handler.annotation.Header;
import org.springframework.messaging.handler.annotation.MessageMapping;
import org.springframework.messaging.simp.stomp.StompHeaderAccessor;
import org.springframework.stereotype.Controller;

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

@Controller
public class WebSocketSessionController {

    private static final Logger log = LoggerFactory.getLogger(WebSocketSessionController.class);

    @Autowired
    private StringRedisTemplate stringRedisTemplate;


    @MessageMapping("/session/alive")
    public void active(@Header("simpSessionAttributes") Map<String, Object> sessionAttributes, StompHeaderAccessor stompHeaderAccessor) {
        User user = SecurityUtils.getCurrentUserFromWebSocket(sessionAttributes);

        if (user != null) {
            //  클러스터링 환경에서 접속중인 사용자 정보를 저장한다.
            this.updateActiveUser(user);
        }
    }

    //  클러스터링 환경에서 접속중인 사용자 정보를 저장한다.
    private void updateActiveUser(User user) {
        //  전체 사용자 정보
        Map<String, Object> totalActiveUserMap = new HashMap<>();
        Gson gson = new Gson();
        //  살아있다고 요청한 사용자 정보
        Map<String, Object> currentActiveUserMap = new HashMap<>();
        currentActiveUserMap.put("id", user.getId());
        currentActiveUserMap.put("account", CommonUtil.decryptAES128(user.getAccount()));
        currentActiveUserMap.put("name", user.getName());
        currentActiveUserMap.put("sessionActiveTime", String.valueOf(System.currentTimeMillis()));
        currentActiveUserMap.put("lastWorkspaceId", user.getLastWorkspaceId());

        //  테스트용 키 삭제
        //this.stringRedisTemplate.delete(Lists.newArrayList("activeUsers"));

        String activeUsers = this.stringRedisTemplate.opsForValue().get("activeUsers");

        //  redis 에서 로그인 세션 정보를 가져와서 업데이트 한다.
        if (!StringUtils.isEmpty(activeUsers)) {
            totalActiveUserMap = (Map<String, Object>)gson.fromJson(activeUsers, Object.class);
        }

        totalActiveUserMap.put(MapUtil.getString(currentActiveUserMap, "id"), currentActiveUserMap);
        String content = gson.toJson(totalActiveUserMap);

        this.stringRedisTemplate.opsForValue().set("activeUsers", content);
    }
}