package com.it.controller;


import com.alibaba.fastjson.JSONObject;
import com.it.constant.ResultKey;
import com.it.exception.AuthException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.ExceptionHandler;

import javax.annotation.Resource;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.nio.charset.Charset;

@Controller
public class DefaultController {
  protected static Logger LOGGER = LoggerFactory.getLogger(DefaultController.class);

  protected HttpHeaders httpHeaders = new HttpHeaders();

  @Resource
  protected ServletContext servletContext;

  @Resource
  protected HttpServletRequest httpServletRequest;

  @Resource
  protected HttpServletResponse httpServletResponse;

  public DefaultController() {
    MediaType mediaType = new MediaType("text", "html", Charset.forName("UTF-8"));
    httpHeaders.setContentType(mediaType);
  }

  @ExceptionHandler(value = {Exception.class})
  public ResponseEntity<String> exceptionHandler(Exception ex) {
    JSONObject result = new JSONObject();

    if (ex instanceof AuthException) {
      LOGGER.debug("error message = {}, data = {}", ex.getMessage(), ((AuthException) ex).getData());
    } else {
      ex.printStackTrace();
    }

//    if (ex instanceof InvalidSessionException) {
//      result.put("invalidSession", true);
//    }

    result.put(ResultKey.success.name(), false);
    result.put("message", ex.getMessage());

    return new ResponseEntity<String>(result.toJSONString(), httpHeaders, HttpStatus.OK);
  }

  /**
   * 简化操作, 生成字符串为:
   * { "success":true, "data": obj }
   */
  protected String successDataToJSONString() {
    JSONObject result = new JSONObject();
    result.put(ResultKey.success.name(), true);
    return result.toJSONString();
  }

  /**
   * 简化操作, 生成字符串为:
   * { "success":true, "data": obj }
   */
  protected String successDataToJSONString(Object obj) {
    JSONObject result = new JSONObject();
    result.put(ResultKey.success.name(), true);
    result.put(ResultKey.data.name(), obj);
    return result.toJSONString();
  }
}