path: root/juick-www/src/main/java/com/juick/www/controllers/User.java

/*
 * Copyright (C) 2008-2017, Juick
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.juick.www.controllers;

import com.juick.server.util.HttpForbiddenException;
import com.juick.server.util.HttpNotFoundException;
import com.juick.server.util.UserUtils;
import com.juick.service.MessagesService;
import com.juick.service.TagService;
import com.juick.service.UserService;
import com.juick.www.Utils;
import org.apache.commons.codec.CharEncoding;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.text.StringEscapeUtils;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.CookieValue;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
import org.springframework.web.util.UriComponents;
import ru.sape.Sape;

import javax.inject.Inject;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.List;
import java.util.stream.Collectors;

/**
 *
 * @author Ugnich Anton
 */
@Controller
public class User {
    @Inject
    private UserService userService;
    @Inject
    private TagService tagService;
    @Inject
    private MessagesService messagesService;
    @Inject
    private Sape sape;

    void fillUserModel(ModelMap model, com.juick.User user, com.juick.User visitor) {
        model.addAttribute("isSubscribed", userService.isSubscribed(visitor.getUid(), user.getUid()));
        model.addAttribute("isInBL", userService.isInBL(visitor.getUid(), user.getUid()));
        model.addAttribute("isInBLAny", userService.isInBLAny(user.getUid(), visitor.getUid()));
        model.addAttribute("statsIRead", userService.getStatsIRead(user.getUid()));
        model.addAttribute("statsMyReaders", userService.getStatsMyReaders(user.getUid()));
        model.addAttribute("statsMyBL", userService.getUserBLUsers(user.getUid()).size());
        model.addAttribute("statsMessages", userService.getStatsMessages(user.getUid()));
        model.addAttribute("statsReplies", userService.getStatsReplies(user.getUid()));
        model.addAttribute("iread", userService.getUserReadLeastPopular(user.getUid(), 8));
        model.addAttribute("tagStats", tagService.getUserTagStats(user.getUid())
                .stream().sorted((e1, e2) -> Integer.compare(e2.getUsageCount(), e1.getUsageCount())).limit(20).map(t -> t.getTag().getName()).collect(Collectors.toList()));
    }

    @GetMapping("/{uname}/")
    protected String doGetBlog(
            @RequestParam(required = false, name = "show") String paramShow,
            @RequestParam(required = false, name = "tag") String paramTagStr,
            @RequestParam(required = false, name = "search") String paramSearch,
            @PathVariable String uname,
            @RequestParam(required = false, defaultValue = "0") Integer before,
            @CookieValue(name = "sape_cookie", required = false, defaultValue = StringUtils.EMPTY) String sapeCookie,
            ModelMap model) throws IOException {
        com.juick.User user = userService.getUserByName(uname);
        com.juick.User visitor = UserUtils.getCurrentUser();
        if (user.isBanned()) {
            throw new HttpNotFoundException();
        }

        List<Integer> mids;

        com.juick.Tag paramTag = null;
        if (paramTagStr != null) {
            if (paramTagStr.length() < 64) {
                paramTag = tagService.getTag(paramTagStr, false);
            }
            if (paramTag == null) {
                throw new HttpNotFoundException();
            } else if (!paramTag.getName().equals(paramTagStr)) {
                String url = user.getName() + "/?tag=" + URLEncoder.encode(paramTag.getName(), CharEncoding.UTF_8);
                return "redirect:/" + url;
            }
        }
        if (paramSearch != null && paramSearch.length() > 64) {
            paramSearch = null;
        }

        int privacy = 0;
        if (visitor.getUid() > 0) {
            if (user.getUid() == visitor.getUid() || visitor.getUid() == 1) {
                privacy = -3;
            } else if (userService.isInWL(user.getUid(), visitor.getUid())) {
                privacy = -2;
            }
        }

        String title;
        if (paramShow == null) {
            if (paramTag != null) {
                title = "Блог " + user.getName() + ": *" + StringEscapeUtils.escapeHtml4(paramTag.getName());
                mids = messagesService.getUserTag(user.getUid(), paramTag.TID, privacy, before);
            } else if (paramSearch != null) {
                title = "Блог " + user.getName() + ": " + StringEscapeUtils.escapeHtml4(paramSearch);
                mids = messagesService.getUserSearch(user.getUid(), Utils.encodeSphinx(paramSearch), privacy, before);
            } else {
                title = "Блог " + user.getName();
                mids = messagesService.getUserBlog(user.getUid(), privacy, before);
            }
        } else if (paramShow.equals("recomm")) {
            title = "Рекомендации " + user.getName();
            mids = messagesService.getUserRecommendations(user.getUid(), before);
        } else if (paramShow.equals("photos")) {
            title = "Фотографии " + user.getName();
            mids = messagesService.getUserPhotos(user.getUid(), privacy, before);
        } else {
            throw new HttpNotFoundException();
        }

        String head = "<link rel=\"alternate\" type=\"application/rss+xml\" title=\"@" +
                user.getName() + "\" href=\"//rss.juick.com/" + user.getName() + "/blog\"/>";
        if (paramTag != null && tagService.getTagNoIndex(paramTag.TID)) {
            head += "<meta name=\"robots\" content=\"noindex,nofollow\"/>";
        } else if (before > 0 || paramShow != null) {
            head += "<meta name=\"robots\" content=\"noindex\"/>";
        }
        model.addAttribute("title", title);
        model.addAttribute("headers", head);
        model.addAttribute("visitor", visitor);
        model.addAttribute("user", user);
        model.addAttribute("noindex", paramShow == null && before == 0);
        fillUserModel(model, user, visitor);
        model.addAttribute("paramTag", paramTag);
        List<com.juick.Message> msgs = messagesService.getMessages(mids);

        if (visitor.getUid() != 0) {
            List<Integer> blUIDs = userService.checkBL(visitor.getUid(),
                    msgs.stream().map(m -> m.getUser().getUid()).collect(Collectors.toList()));
            msgs.forEach(m -> m.ReadOnly |= blUIDs.contains(m.getUser().getUid()));
        }
        model.addAttribute("msgs", msgs);
        model.addAttribute("headers", head);
        model.addAttribute("showAdv",
                paramShow == null && before == 0 && paramSearch == null && visitor.getUid() == 0);
        if (mids.size() >= 20) {
            String nextpage = "?before=" + mids.get(mids.size() - 1);
            if (paramShow != null) {
                nextpage += "&amp;show=" + paramShow;
            }
            if (paramSearch != null) {
                nextpage += "&amp;search=" + URLEncoder.encode(paramSearch, CharEncoding.UTF_8);
            }
            if (paramTag != null) {
                nextpage += "&amp;tag=" + URLEncoder.encode(paramTag.getName(), CharEncoding.UTF_8);
            }
            model.addAttribute("nextpage", nextpage);
        }
        UriComponents builder = ServletUriComponentsBuilder.fromCurrentRequestUri().build();
        String queryString = builder.getQuery();
        String requestURI = builder.toUri().getPath();
        if (sape != null && visitor.getUid() == 0 && queryString == null) {
            String links = sape.getPageLinks(requestURI, sapeCookie).render();
            model.addAttribute("links", links);
        }
        model.addAttribute("isModerator", visitor.getUid() == 3694);
        return "views/blog";
    }

    @GetMapping("/{uname}/tags")
    protected String doGetTags(@PathVariable String uname, ModelMap model) throws IOException {
        com.juick.User user = userService.getUserByName(uname);
        com.juick.User visitor = UserUtils.getCurrentUser();
        if (visitor.isBanned()) {
            throw new HttpNotFoundException();
        }

        model.addAttribute("title", "Теги " + user.getName());
        model.addAttribute("headers", "<meta name=\"robots\" content=\"noindex,nofollow\"/>");
        model.addAttribute("visitor", visitor);
        model.addAttribute("user", user);
        fillUserModel(model, user, visitor);
        model.addAttribute("tags", tagService.getUserTagStats(user.getUid()).stream()
                .sorted((e1, e2) -> Integer.compare(e2.getUsageCount(), e1.getUsageCount())).map(t -> t.getTag().getName()).collect(Collectors.toList()));

        return "views/blog_tags";
    }

    @GetMapping("/{uname}/friends")
    protected String doGetFriends(@PathVariable String uname, ModelMap model) throws IOException {
        com.juick.User user = userService.getUserByName(uname);
        com.juick.User visitor = UserUtils.getCurrentUser();
        if (visitor.isBanned()) {
            throw new HttpNotFoundException();
        }
        model.addAttribute("title", "Подписки " + user.getName());
        model.addAttribute("headers", "<meta name=\"robots\" content=\"noindex\"/>");
        model.addAttribute("visitor", visitor);
        model.addAttribute("user", user);
        fillUserModel(model, user, visitor);
        model.addAttribute("users", userService.getUserFriends(user.getUid()));

        return "views/users";
    }

    @GetMapping("/{uname}/readers")
    protected String doGetReaders(@PathVariable String uname, ModelMap model) throws IOException {
        com.juick.User user = userService.getUserByName(uname);
        com.juick.User visitor = UserUtils.getCurrentUser();
        if (visitor.isBanned()) {
            throw new HttpForbiddenException();
        }
        model.addAttribute("title", "Читатели " + user.getName());
        model.addAttribute("headers", "<meta name=\"robots\" content=\"noindex\"/>");
        model.addAttribute("visitor", visitor);
        model.addAttribute("user", user);
        fillUserModel(model, user, visitor);
        model.addAttribute("users", userService.getUserReaders(user.getUid()));

        return "views/users";
    }

    @GetMapping("/{uname}/bl")
    protected String doGetBL(@PathVariable String uname, ModelMap model) throws IOException {
        com.juick.User user = userService.getUserByName(uname);
        com.juick.User visitor = UserUtils.getCurrentUser();
        if (visitor.isBanned() || visitor.getUid() != user.getUid()) {
            throw new HttpForbiddenException();
        }
        model.addAttribute("title", "Черный список " + user.getName());
        model.addAttribute("headers", "<meta name=\"robots\" content=\"noindex\"/>");
        model.addAttribute("visitor", visitor);
        model.addAttribute("user", user);
        fillUserModel(model, user, visitor);
        model.addAttribute("users", userService.getUserBLUsers(user.getUid()));

        return "views/users";
    }
}