WatchedTaskController.java

package com.qprogramming.tasq.task.watched;

import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.MessageSource;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort.Direction;
import org.springframework.data.web.PageableDefault;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
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 com.qprogramming.tasq.account.Roles;
import com.qprogramming.tasq.support.ResultData;
import com.qprogramming.tasq.support.Utils;
import com.qprogramming.tasq.task.Task;
import com.qprogramming.tasq.task.TaskService;

@Controller
public class WatchedTaskController {

    private WatchedTaskService watchSrv;
    private MessageSource msg;
    private TaskService taskSrv;

    @Autowired
    public WatchedTaskController(WatchedTaskService watchSrv,
                                 MessageSource msg, TaskService taskSrv) {
        this.watchSrv = watchSrv;
        this.msg = msg;
        this.taskSrv = taskSrv;
    }

    @RequestMapping(value = "/task/watch", method = RequestMethod.POST)
    @ResponseBody
    public ResultData watch(@RequestParam(value = "id") String id) {
        ResultData result = new ResultData();
        // check if not admin or user
        if (!Roles.isUser()) {
            String role = msg.getMessage(Utils.getCurrentAccount().getRole()
                    .getCode(), null, Utils.getCurrentLocale());
            String message = msg.getMessage("role.error.auth",
                    new Object[]{role}, Utils.getCurrentLocale());
            result = new ResultData(ResultData.Code.ERROR, message);
        } else {
            Task task = taskSrv.findById(id);
            WatchedTask watched = watchSrv.getByTask(id);
            if (watched != null
                    && watched.getWatchers()
                    .contains(Utils.getCurrentAccount())) {
                watched = watchSrv.stopWatching(task);
                result.code = ResultData.Code.OK;
                result.message = msg.getMessage("task.watch.stoped",
                        new Object[]{id}, Utils.getCurrentLocale());
            } else {
                watched = watchSrv.startWatching(task);
                result.code = ResultData.Code.OK;
                result.message = msg.getMessage("task.watch.started",
                        new Object[]{id}, Utils.getCurrentLocale());
            }
        }
        return result;
    }

    /**
     * Get list of all tasks watched by currently logged user
     *
     * @return
     */
    @RequestMapping(value = "/watching", method = RequestMethod.GET)
    public String watching() {
        return "user/watching";
    }

    @RequestMapping(value = "/listWatches", method = RequestMethod.GET)
    public @ResponseBody
    Page<DisplayWatch> getWatches(
            @RequestParam(required = false) String term,
            @PageableDefault(size = 25, page = 0, sort = "id", direction = Direction.DESC) Pageable p) {
        Page<WatchedTask> page = watchSrv.findByWatcher(
                Utils.getCurrentAccount(), p);
        List<DisplayWatch> result = page.getContent().stream().map(DisplayWatch::new).collect(Collectors.toList());
        return new PageImpl<>(result, p, page.getTotalElements());
    }

}