@RequestMapping(value = "/tweet/comment", method = RequestMethod.POST)
public void comment(@RequestBody Message message) {
Tweet tweet = tweetRepo.findOne(message.getTid());
tweet.setCommentCount(tweet.getCommentCount() + 1);
tweetRepo.saveAndFlush(tweet);
// 保存评论
UserTweet userTweet = new UserTweet();
userTweet.setUid(message.getUid());
userTweet.setTid(message.getTid());
userTweet.setUserTweetType(UserTweet.UserTweetType.COMMENT);
userTweetRepo.saveAndFlush(userTweet);
message.setMessageType(Message.MessageType.COMMENT);
message.setTime(Utils.getTime());
message.setDate(System.currentTimeMillis());
Utils.message(messageRepo.saveAndFlush(message));
}
@RequestMapping(value = "/tweet/star", method = RequestMethod.POST)
public void star(
@RequestParam("uid") long uid,
@RequestParam("tid") long tid,
@RequestParam("star") boolean star) {
Tweet tweet = tweetRepo.findOne(tid);
if (star) {
if (userTweetRepo.findByUidAndTidAndUserTweetType(uid, tid, UserTweet.UserTweetType.STAR)
== null) {
UserTweet userTweet = new UserTweet();
userTweet.setUid(uid);
userTweet.setTid(tid);
userTweet.setUserTweetType(UserTweet.UserTweetType.STAR);
userTweetRepo.saveAndFlush(userTweet);
User user = userRepo.findOne(uid);
Message message = new Message();
message.setUid(uid);
message.setUid2(tweet.getUid());
message.setTid(tid);
message.setMessageType(Message.MessageType.STAR);
message.setTitle(user.getName());
message.setContent(user.getName());
message.setIcon(user.getAvatar());
message.setTime(Utils.getTime());
message.setDate(System.currentTimeMillis());
Utils.message(messageRepo.saveAndFlush(message));
} else {
return;
}
} else {
userTweetRepo.deleteByUidAndTidAndUserTweetType(uid, tid, UserTweet.UserTweetType.STAR);
}
tweet.setStarCount(star ? tweet.getStarCount() + 1 : tweet.getStarCount() - 1);
tweetRepo.saveAndFlush(tweet);
}
@RequestMapping(value = "/tweet", method = RequestMethod.GET)
public Tweet get(@RequestParam("uid") long uid, @RequestParam("tid") long tid) {
Tweet tweet = tweetRepo.findOne(tid);
tweet.setStar(
userTweetRepo.findByUidAndTidAndUserTweetType(
uid, tweet.getId(), UserTweet.UserTweetType.STAR)
!= null);
TweetTweet tweetTweet =
tweetTweetRepo.findByTidAndTweetTweetType(tweet.getId(), TweetTweet.TweetTweetType.REPEAT);
if (tweetTweet != null) {
Tweet originTweet = tweetRepo.findOne(tweetTweet.getTid2());
originTweet.setStar(
userTweetRepo.findByUidAndTidAndUserTweetType(
uid, originTweet.getId(), UserTweet.UserTweetType.STAR)
!= null);
tweet.setOriginTweet(originTweet);
}
return tweet;
}
// 新建Tweet
@RequestMapping(value = "/tweet", method = RequestMethod.POST)
public Tweet add(
@RequestParam("prompt") String prompt,
@RequestParam("originTid") long originTid,
@RequestParam("tweet") MultipartFile tweetFile,
@RequestParam("image") MultipartFile[] images)
throws IOException {
// 保持userTweet
String tweetJSON = new String(tweetFile.getBytes());
Tweet tweet = new Gson().fromJson(URLDecoder.decode(tweetJSON, "UTF8"), Tweet.class);
long uid = tweet.getUid();
User user = userRepo.findOne(uid);
if (user.getUserType() == User.UserType.SPECIAL && originTid > 0) { // 超级用户
Tweet t = tweetRepo.findOne(originTid);
tweet = new Tweet();
tweet.setUid(uid);
tweet.setName(user.getName());
tweet.setIcon(user.getAvatar());
tweet.setTweetType(Tweet.TweetType.SPECIAL);
tweet.setTweetContentType(tweet.getTweetContentType());
tweet.setTitle(t.getTitle());
tweet.setSummary(t.getSummary());
tweet.setImages(t.getImages());
tweet.setContent(t.getContent());
tweet.setLat(t.getLat());
tweet.setLon(t.getLon());
tweet.setLocation(t.location);
tweet.setTime(Utils.getTime());
tweet.setDate(System.currentTimeMillis());
originTid = 0;
prompt = null;
} else {
String content = tweet.getContent();
tweet.setSummary(content);
content = content.replace(" ", " ").replace("\n", "<br>");
Pattern pattern = Pattern.compile("@[^\\p{P}|\\p{S}|\\p{Z}|\\p{M}]*");
String mContent = content;
Matcher matcher = pattern.matcher(mContent);
while (matcher.find()) {
int start = matcher.start();
int end = matcher.end();
String p = mContent.substring(start, end);
content =
content.replace(
p, Constants.TWEET_ADD_PROMPT_HTML.replace(Constants.TWEET_HTML_PROMPT_TAG, p));
}
List<String> imageUrls = new ArrayList<>();
StringBuffer imageContent = new StringBuffer();
if (images != null) {
for (MultipartFile image : images) {
String imageUrl = Utils.image(image);
if (!StringUtils.isEmpty(imageUrl)) {
imageUrls.add(imageUrl);
imageContent.append(
Constants.TWEET_ADD_IMAGE_HTML.replace(Constants.TWEET_HTML_IMAGE_TAG, imageUrl));
}
}
tweet.setImages(String.join(";", imageUrls));
}
tweet.setContent(
Utils.content(
Constants.TWEET_ADD_HTML
.replace(Constants.TWEET_HTML_CONTENT_TAG, content)
.replace(Constants.TWEET_HTML_IMAGES_TAG, imageContent)));
tweet.setTime(Utils.getTime());
tweet.setDate(System.currentTimeMillis());
}
tweet = tweetRepo.saveAndFlush(tweet);
UserTweet userTweet = new UserTweet();
userTweet.setUid(uid);
userTweet.setTid(tweet.getId());
userTweet.setUserTweetType(UserTweet.UserTweetType.ADD);
userTweetRepo.saveAndFlush(userTweet);
// 保存@ 并且发送message
if (!StringUtils.isEmpty(prompt)) {
String[] uids = prompt.split(";");
for (String id : uids) {
// 保存@
UserTweet ut = new UserTweet();
ut.setUid(Long.valueOf(id));
ut.setTid(tweet.getId());
ut.setUserTweetType(UserTweet.UserTweetType.PROMPT);
userTweetRepo.saveAndFlush(ut);
// 保存@ message
Message message = new Message();
message.setUid(uid);
message.setUid2(Long.valueOf(id));
message.setTid(tweet.getId());
message.setMessageType(Message.MessageType.PROMPT);
message.setTitle(user.getName());
message.setContent(user.getName());
message.setIcon(user.getAvatar());
message.setTime(Utils.getTime());
message.setDate(System.currentTimeMillis());
Utils.message(messageRepo.saveAndFlush(message));
}
}
// 查找转发的tweet
if (originTid > 0) {
Tweet originTweet = tweetRepo.findOne(originTid);
tweet.setOriginTweet(originTweet);
TweetTweet tweetTweet = new TweetTweet();
tweetTweet.setTid(tweet.getId());
tweetTweet.setTid2(originTid);
tweetTweet.setTweetTweetType(TweetTweet.TweetTweetType.REPEAT);
tweetTweetRepo.saveAndFlush(tweetTweet);
originTweet.setRepeatCount(originTweet.getRepeatCount() + 1);
tweetRepo.saveAndFlush(originTweet);
}
return tweet;
}
@RequestMapping(value = "/tweet/list", method = RequestMethod.GET)
public List<Tweet> gets(
@RequestParam("getsType") GetsType getsType,
@RequestParam("uid") long uid,
@RequestParam("keyword") String keyword,
@RequestParam("pageIndex") int pageIndex,
@RequestParam("pageSize") int pageSize) {
PageRequest pageRequest = new PageRequest(pageIndex, pageSize, Sort.Direction.DESC, "date");
Page<Tweet> tweets = null;
if (getsType != GetsType.VIDEO) {
List<Long> uids = new ArrayList<>();
if (uid > 0 && (getsType == GetsType.HOME || getsType == GetsType.USER)) {
uids.add(uid);
}
if (getsType != GetsType.USER) {
List<User> users = new ArrayList<>();
if (uid > 0) {
users.addAll(
userRepo.findAll(
userUserRepo.findUid2sByUidAndUserUserType(uid, UserUser.UserUserType.FOCUS)));
} else {
users.add(userRepo.findByName(Constants.SPECIAL_BREAK));
users.add(userRepo.findByName(Constants.NEWS_YINGCHAO));
users.add(userRepo.findByName(Constants.NEWS_XIJIA));
users.add(userRepo.findByName(Constants.NEWS_DEJIA));
users.add(userRepo.findByName(Constants.NEWS_YIJIA));
users.add(userRepo.findByName(Constants.NEWS_FAJIA));
users.add(userRepo.findByName(Constants.NEWS_ZHONGCHAO));
users.add(userRepo.findByName(Constants.NEWS_OUGUAN));
users.add(userRepo.findByName(Constants.NEWS_HUABIAN));
users.add(userRepo.findByName(Constants.NEWS_VIDEO));
}
for (User user : users) {
if (getsType == GetsType.HOME
&& (user.getUserType() == User.UserType.NORMAL
|| user.getUserType() == User.UserType.SPECIAL)) {
uids.add(user.getId());
} else if (getsType == GetsType.TEAM && user.getUserType() == User.UserType.TEAM) {
uids.add(user.getId());
} else if (getsType == GetsType.NEWS && user.getUserType() == User.UserType.NEWS) {
uids.add(user.getId());
}
}
}
if (getsType != GetsType.SEARCH) {
if (uids.size() > 0) {
List<Long> tids =
userTweetRepo.findTidsByUidsAndUserTweetType(uids, UserTweet.UserTweetType.ADD);
tweets = tweetRepo.findByIdIn(tids, pageRequest);
}
} else if (!StringUtils.isEmpty(keyword)) {
keyword = "%" + keyword + "%";
tweets =
tweetRepo
.findByNameLikeIgnoreCaseOrTitleLikeIgnoreCaseOrSourceLikeIgnoreCaseOrSummaryLikeIgnoreCase(
keyword, keyword, keyword, keyword, pageRequest);
}
} else {
tweets = tweetRepo.findByTweetContentType(Tweet.TweetContentType.VIDEO, pageRequest);
}
if (tweets != null) {
for (Tweet tweet : tweets) {
tweet.setStar(
userTweetRepo.findByUidAndTidAndUserTweetType(
uid, tweet.getId(), UserTweet.UserTweetType.STAR)
!= null);
TweetTweet tweetTweet =
tweetTweetRepo.findByTidAndTweetTweetType(
tweet.getId(), TweetTweet.TweetTweetType.REPEAT);
if (tweetTweet != null) {
Tweet originTweet = tweetRepo.findOne(tweetTweet.getTid2());
originTweet.setStar(
userTweetRepo.findByUidAndTidAndUserTweetType(
uid, originTweet.getId(), UserTweet.UserTweetType.STAR)
!= null);
tweet.setOriginTweet(originTweet);
}
}
return tweets.getContent();
} else {
return new ArrayList<>();
}
}