package com.juick.api.controllers; import com.juick.User; import com.juick.api.util.HttpForbiddenException; import com.juick.api.util.HttpNotFoundException; import com.juick.service.UserService; import com.juick.util.UserUtils; import com.juick.util.WebUtils; import org.springframework.http.MediaType; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.ResponseBody; import javax.inject.Inject; import java.security.Principal; import java.util.ArrayList; import java.util.List; /** * @author ugnich */ @Controller @ResponseBody public class Users { @Inject UserService userService; @RequestMapping(value = "/users", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE) public List doGetUsers( @RequestParam(value = "uname", required = false) List unames, @RequestParam(value = "jid", required = false) List jids) { List users = new ArrayList<>(); if (unames != null) { unames.removeIf(WebUtils::isNotUserName); if (!unames.isEmpty() && unames.size() < 20) users.addAll(userService.getUsersByName(unames)); } if (jids != null) { jids.removeIf(WebUtils::isNotJid); if (!jids.isEmpty() && jids.size() < 20) users.addAll(userService.getUsersByJID(jids)); } if (!users.isEmpty()) return users; throw new HttpNotFoundException(); } @RequestMapping(value = "/users/read", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE) public List doGetUserRead( Principal principal, @RequestParam String uname) { String name = UserUtils.getUsername(principal, null); User visitor = userService.getUserByName(name); int vuid = visitor.getUid(); if (vuid == 0) { throw new HttpForbiddenException(); } int uid = 0; if (uname == null) { uid = vuid; } else { if (UserUtils.checkUserNameValid(uname)) { com.juick.User u = userService.getUserByName(uname); if (u != null && u.getUid() > 0) { uid = u.getUid(); } } } if (uid > 0) { List uids = userService.getUserRead(uid); if (uids.size() > 0) { List users = userService.getUsersByID(uids); if (users.size() > 0) { return users; } } } throw new HttpNotFoundException(); } @RequestMapping(value = "/users/readers", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE) public List doGetUserReaders( Principal principal, @RequestParam String uname) { String name = UserUtils.getUsername(principal, null); User visitor = userService.getUserByName(name); int vuid = visitor.getUid(); if (vuid == 0) { throw new HttpForbiddenException(); } int uid = 0; if (uname == null) { uid = vuid; } else { if (UserUtils.checkUserNameValid(uname)) { com.juick.User u = userService.getUserByName(uname); if (u != null && u.getUid() > 0) { uid = u.getUid(); } } } if (uid > 0) { return userService.getUserReaders(uid); } throw new HttpNotFoundException(); } }