2021-03-31-TIL

package honux.spring.playground.web;

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;

@RestController
@RequestMapping("/login")
public class UserController {

    //sessionid, userid;
    private Map<String,String> session = new HashMap<>();

    @GetMapping
    public MyResponse welcome() {
        return new MyResponse("ok", "Kite");
    }

    @GetMapping("/check")
    public MyResponse check(HttpServletRequest req) {
        String key = "sid";
        Cookie[] cookies = req.getCookies();
        String uid = "Null";
        for(Cookie c: cookies) {
            if (c.getName().equals(key)) {
                uid = session.getOrDefault(c.getValue(), "Null");
                System.out.println("check: " + c.getValue() + ": " + uid);
            }
        }

        return new MyResponse("ok", "login id: " + uid);
    }

    @PostMapping
    public MyResponse login(String id, HttpServletResponse response) {
        System.out.println(id);
        Random r = new Random();
        String sid = Integer.toString(r.nextInt());
        session.put(sid, id);
        System.out.println("login: " + sid+ ": " + session.get(sid));
        response.addCookie(new Cookie("sid", sid));
        return new MyResponse("ok","sid: " + sid);
    }
}