juick-api/src/main/java/com/juick/api/controllers/Subscriptions.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

package com.juick.api.controllers;

import com.juick.Message;
import com.juick.User;
import com.juick.api.util.HttpBadRequestException;
import com.juick.api.util.HttpForbiddenException;
import com.juick.service.MessagesService;
import com.juick.service.SubscriptionService;
import com.juick.service.UserService;
import com.juick.util.UserUtils;
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.io.IOException;
import java.security.Principal;
import java.util.List;

/**
 * Created by vitalyster on 24.10.2016.
 */
@Controller
@ResponseBody
public class Subscriptions {
    @Inject
    UserService userService;
    @Inject
    SubscriptionService subscriptionService;
    @Inject
    MessagesService messagesService;

    @RequestMapping(value = "/subscriptions", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
    public List<User> doGet(
            Principal principal,
            @RequestParam(defaultValue = "0") int mid,
            @RequestParam(defaultValue = "0") int uid) throws IOException {
        String name = UserUtils.getUsername(principal, null);
        User visitor = userService.getUserByName(name);
        if ((visitor.getUid() == 0) && !(visitor.getName().equals("juick"))) {
            throw new HttpForbiddenException();
        }
        if (uid > 0) {
            return subscriptionService.getSubscribedUsers(uid, mid);
        } else {
            // thread
            Message msg = messagesService.getMessage(mid);
            if (msg != null) {
                return subscriptionService.getUsersSubscribedToComments(mid, msg.getUser().getUid());
            }
        }
        throw new HttpBadRequestException();
    }
}