package xyz.aiinirii.postalk.bean;

import lombok.Data;
import lombok.EqualsAndHashCode;

import java.util.Date;

@Data
public class User {

    private int id;
    private String username;
    private String password;
    private String sex;
    private int age;
    private String phoneNumber;
    private String email;
    private Date registrationDate;
}

@Data
public abstract class Text {

    private int id;
    private String content;
    private Date time;

    private User user;
}

@EqualsAndHashCode(callSuper = true)
@Data
public class Comment extends Text{

    private Text text;
}

@EqualsAndHashCode(callSuper = true)
@Data
public class Post extends Text{
}

@Data
public class Like {

    private User user;
    private Text text;
}

@Data
public class Friend {

    private User user1;
    private User user2;
    private Data sinceTime;
}

@Data
public class Conversation {

    private int id;
    private User toUser;
    private User fromUser;
    private int conversationId;
    private Date time;
    private String content;
    private boolean read;
}

package xyz.aiinirii.postalk.mapper;

import org.apache.ibatis.annotations.*;
import org.springframework.stereotype.Repository;
import xyz.aiinirii.postalk.bean.User;

import java.util.List;

/**
 * @author AIINIRII
 */
@Mapper
@Repository
public interface UserMapper {

    @Select("select * from user")
    List<User> findAllUser();

    @Select("select * from user where id=#{id}")
    User findUserById(Integer id);

    @Select("delete from user where id=#{id}")
    int deleteUserById(Integer id);

    @Options(useGeneratedKeys = true, keyProperty = "id")
    @Insert("insert into user(username, password, sex, age, phone_number, email, registration_date) values (#{username}, #{password}, #{sex}, #{age}, #{phoneNumber}, #{email}, #{registrationDate})")
    void insertUser(User user);

    @Insert("update user set username=#{username}, password=#{password}, sex=#{sex}, age=#{age}, phone_number=#{phoneNumber}, email=#{email}, registration_date=#{registration_date} where id=#{id}")
    int updateUser(User user);
}

package xyz.aiinirii.postalk.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.servlet.ModelAndView;
import xyz.aiinirii.postalk.bean.User;
import xyz.aiinirii.postalk.service.UserService;

import java.util.LinkedList;
import java.util.List;

/**
 * @author AIINIRII
 */
@Controller
public class UserController {

    UserService userService = new UserService();

    @Autowired
    public void setUserService(UserService userService) {
        this.userService = userService;
    }

    @PostMapping("/user/login")
    public ModelAndView checkUser(User user, ModelAndView modelAndView){
        int res = userService.checkUser(user);
        modelAndView.addObject("checkResult", res);
        if (res == 0) {
            modelAndView.setViewName("/post/index");
        } else {
            modelAndView.setViewName("/index");
        }
        return modelAndView;
    }

    @GetMapping("/user/{id}")
    public ModelAndView findUserById(@PathVariable("id") Integer id, ModelAndView modelAndView) {
        User user = userService.findUserById(id);
        modelAndView.setViewName("user/list");
        List<User> users = new LinkedList<>();
        users.add(user);
        modelAndView.addObject("users", users);
        return modelAndView;
    }

    @GetMapping("/user")
    public String toRegisterPage(){
        return "user/update";
    }

    @PostMapping("/user")
    public ModelAndView registerUser(User user, ModelAndView modelAndView) {

        int res = userService.registerUser(user);
        if (res == 0) {
            // success insert, go to login page
            modelAndView.setViewName("/index");
        } else if (res == 1){
            // the username is used, go back to the register page
            modelAndView.addObject("userR", user);
            modelAndView.setViewName("/user/update");
        }
        return modelAndView;
    }
}

<!DOCTYPE html>
<html lang="en" xmlns:th="http://www.thymeleaf.org">
<head>
    <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
    <meta name="description" content="">
    <meta name="author" content="">
    <title>Signin Template for Bootstrap</title>
    <!-- Bootstrap core CSS -->
    <link href="/asserts/css/bootstrap.min.css" rel="stylesheet">
    <!-- Custom styles for this template -->
    <link href="/asserts/css/signin.css" rel="stylesheet">
</head>

<body class="text-center">
<div>
<form class="form-signin" method="post" th:action="@{/user/login}">
    <img class="mb-4" src="/asserts/img/bootstrap-solid.svg" alt="" width="72" height="72">
    <h1 class="h3 mb-3 font-weight-normal">Please sign in</h1>
    <p style="color: #b21f2d" th:if="${checkResult == 1}">wrong password</p>
    <p style="color: #b21f2d" th:if="${checkResult == 2}">no such user</p>
    <label class="sr-only">Username</label>
    <label>
        <input name="username" type="text" class="form-control" placeholder="Username" required="" autofocus=""/>
    </label>
    <label class="sr-only">Password</label>
    <label>
        <input name="password" type="password" class="form-control" placeholder="Password" required=""/>
    </label>
    <div class="checkbox mb-3">
        <label>
            <input type="checkbox" value="remember-me"/> Remember me
        </label>
    </div>
    <button class="btn btn-lg btn-primary btn-block" type="submit">Sign in</button>
</form>
<form class="form-signin" action="/user/" method="get">
    <button class="btn btn-lg btn-success btn-block" type="submit">Register</button>
</form>
<p class="mt-5 mb-3 text-muted">© 2019-2020</p>
</div>
</body>

</html>

<!DOCTYPE html>
<html lang="en" xmlns:th="http://www.thymeleaf.org">
<style>
    #wrapper {
        width: 60%;
        margin: 0 auto;
        border: 0;
    }
</style>
<head>
    <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
    <title>Register</title>
    <!-- Bootstrap core CSS -->
    <link href="/asserts/css/bootstrap.min.css" rel="stylesheet">
    <!-- Custom styles for this template -->
    <!--<link href="/asserts/css/signin.css" rel="stylesheet">-->
</head>
<body>
<div id="wrapper">
    <form class="needs-validation" novalidate method="post" action="/user">
        <div class="form-group row">
            <label for="inputUsername" class="col-sm-2 col-form-label">User name*</label>
            <div class="col-sm-10 mb-3">
                <input name="username" type="text" class="form-control" id="inputUsername" th:value="${userR!=null}?${userR.username}" required>
                <div class="valid-feedback">
                    Looks good!
                </div>
                <div class="invalid-feedback">
                    Please enter a valid username.
                </div>
            </div>
        </div>
        <div class="form-group row">
            <label for="inputPassword" class="col-sm-2 col-form-label">Password*</label>
            <div class="col-sm-10">
                <input name="password" type="password" class="form-control" id="inputPassword" required>
                <div class="valid-feedback">
                    Looks good!
                </div>
                <div class="invalid-feedback">
                    Please enter a valid password.
                </div>
            </div>
        </div>
        <fieldset class="form-group">
            <div class="row">
                <legend class="col-form-label col-sm-2 pt-0">Gender*</legend>
                <div class="col-sm-10">
                    <div class="form-check">
                        <input class="form-check-input" type="radio" name="sex" id="gridRadios1" value="M"  th:checked="${userR!=null}?${userR.sex=='M'}" checked>
                        <label class="form-check-label" for="gridRadios1">
                            Male
                        </label>
                    </div>
                    <div class="form-check">
                        <input class="form-check-input" type="radio" name="sex" id="gridRadios2" th:checked="${userR!=null}?${userR.sex=='F'}" value="F">
                        <label class="form-check-label" for="gridRadios2">
                            Female
                        </label>
                    </div>
                </div>
            </div>
        </fieldset>
        <div class="form-group row">
            <label for="inputAge" class="col-sm-2 col-form-label">Age*</label>
            <div class="col-sm-10">
                <input name="age" type="number" class="form-control" id="inputAge" th:value="${userR!=null}?${userR.age}" required>
                <div class="valid-feedback">
                    Looks good!
                </div>
                <div class="invalid-feedback">
                    Please enter a valid age.
                </div>
            </div>
        </div>
        <div class="form-group row">
            <label for="inputPhoneNumber" class="col-sm-2 col-form-label">Phone number</label>
            <div class="col-sm-10">
                <input name="phoneNumber" type="text" class="form-control" th:value="${userR!=null}?${userR.phoneNumber}" id="inputPhoneNumber">
                <div class="valid-feedback">
                    Looks good!
                </div>
                <div class="invalid-feedback">
                    Please enter a valid phone number.
                </div>
            </div>
        </div>
        <div class="form-group row">
            <label for="inputEmail" class="col-sm-2 col-form-label">Email</label>
            <div class="col-sm-10">
                <input name="email" type="email" class="form-control" th:value="${userR!=null}?${userR.email}" id="inputEmail">
                <div class="valid-feedback">
                    Looks good!
                </div>
                <div class="invalid-feedback">
                    Please enter a valid email.
                </div>
            </div>
        </div>
        <div class="form-group row">
            <div class="col-sm-10">
                <button type="submit" class="btn btn-primary">Sign in</button>
            </div>
        </div>
    </form>
</div>
</body>
<script>
    // Example starter JavaScript for disabling form submissions if there are invalid fields
    (function () {
        'use strict';
        window.addEventListener('load', function () {
            // Fetch all the forms we want to apply custom Bootstrap validation styles to
            var forms = document.getElementsByClassName('needs-validation');
            // Loop over them and prevent submission
            var validation = Array.prototype.filter.call(forms, function (form) {
                form.addEventListener('submit', function (event) {
                    if (form.checkValidity() === false) {
                        event.preventDefault();
                        event.stopPropagation();
                    }
                    form.classList.add('was-validated');
                }, false);
            });
        }, false);
    })();
</script>
</html>

<a href="#" class="badge badge-light" th:text="${!post.anonymous?post.user.username:'anonymous'}"></a>

@GetMapping("/friends/")
public ModelAndView toMyFriendPage(ModelAndView modelAndView,
                                   @SessionAttribute("loginUser") User user) {
    List<User> friends = friendService.findAllFriendsByUId(user.getId());
    modelAndView.addObject("friends", friends);
    modelAndView.setViewName("friend/myFriend");
    return modelAndView;
}

@GetMapping("/friend/search/")
public ModelAndView findFriend(ModelAndView modelAndView,
                               @Param("findWay") String findWay,
                               @Param("inputSearch") String inputSearch,
                               @SessionAttribute("loginUser") User user) throws Exception {
    List<User> friends = new LinkedList<>();
    if (findWay.equals("id") && !user.getId().equals(Integer.valueOf(inputSearch))) {
        friends.add(userService.findUserById(Integer.valueOf(inputSearch)));
    } else if (findWay.equals("username") && !user.getUsername().equals(inputSearch)) {
        friends = userService.findUserByUsername(inputSearch);
    } else {
        throw new Exception("Wrong with the findWay");
    }
    modelAndView.setViewName("/friend/result");
    modelAndView.addObject("friends", friends);
    return modelAndView;
}

@PostMapping("/friend/")
private ModelAndView addFriend(ModelAndView modelAndView,
                               Integer id,
                               @SessionAttribute("loginUser") User user) {
    friendService.addFriend(user.getId(), id);
    List<User> friends = friendService.findAllFriendsByUId(user.getId());
    modelAndView.addObject("friends", friends);
    modelAndView.setViewName("friend/myFriend");
    return modelAndView;
}

@DeleteMapping("/friend/{id}")
public ModelAndView deleteFriend(ModelAndView modelAndView,
                                 @PathVariable("id") Integer id,
                                 @SessionAttribute("loginUser") User user) throws Exception {

    if (!friendService.deleteFriendById(user.getId(), id)) {
        throw new Exception("the delete operation failed");
    } else {
        List<User> friends = friendService.findAllFriendsByUId(user.getId());
        modelAndView.addObject("friends", friends);
        modelAndView.setViewName("friend/myFriend");
    }
    return modelAndView;
}

    @Transactional(propagation = Propagation.REQUIRED)
    public void createComment(Integer pid, Comment comment, User user) {
        comment.setUser(user);
        comment.setTime(new Date(System.currentTimeMillis()));
        textMapper.insertText(comment);
        commentMapper.insertComment(pid);
    }

    @Transactional(propagation = Propagation.REQUIRED)
    public boolean deleteCommentById(Integer id, User user) throws Exception {
        if (commentMapper.findCommentById(id).getUser().getId().equals(user.getId())) {
            return commentMapper.deleteCommentById(id) == 1 &&
                    textMapper.deleteTextById(id) == 1;
        } else {
            throw new Exception("the user do not have right to delete the comment");
        }
    }

<div id="commentModule" class="card" style="width: auto">
    <ul class="list-group list-group-flush">
        <li th:each="comment: ${post.comments}" class="list-group-item">
            <div class="row">
                <a href="#" class="badge badge-light card-text"
                   th:text="${!comment.anonymous?comment.user.username:'anonymous'}">Light</a>
                <p class="card-text col-sm-8" th:text="${comment.content}">content</p>
                <form th:action="@{'/comment/' + ${comment.id}}" method="post" th:if="${session.loginUser.id == comment.user.id}">
                    <input type="hidden" name="_method" value="delete"/>
                    <button type="submit" class="btn btn-danger">Delete</button>
                </form>
            </div>
        </li>
    </ul>
</div>

    @Transactional(propagation = Propagation.REQUIRED)
    public boolean deleteCommentById(Integer id, User user) throws Exception {
        if (commentMapper.findCommentById(id).getUser().getId().equals(user.getId())) { // 检查该用户是否有权删除
            return commentMapper.deleteCommentById(id) == 1 &&
                    textMapper.deleteTextById(id) == 1;
        } else {
            throw new Exception("the user do not have right to delete the comment");
        }
    }

<!-- 前端代码调整，调整了按钮出现的逻辑 -->
<form th:action="@{'/comment/' + ${comment.id}}" method="post" th:if="${session.loginUser.id} == ${comment.user.id} or ${session.loginUser.id} == ${comment.text.user.id}">

// 后端service 层删除代码调整
@Transactional(propagation = Propagation.REQUIRED)
public boolean deleteCommentById(Integer id, User user) throws Exception {
    if (commentMapper.findCommentById(id).getUser().getId().equals(user.getId()) || commentMapper.findCommentById(id).getText().getUser().getId().equals(user.getId())) { // 加入了判断该评论的父贴子的用户是否为登录用户的逻辑
        return commentMapper.deleteCommentById(id) == 1 &&
            textMapper.deleteTextById(id) == 1;
    } else {
        throw new Exception("the user do not have right to delete the comment");
    }
}