135 lines
3.8 KiB
Go
Executable file
135 lines
3.8 KiB
Go
Executable file
package Comment
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/iterator"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/native/std"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/runtime"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
)
|
|
|
|
type Comment struct {
|
|
id string
|
|
userLogin string
|
|
postId string
|
|
text string
|
|
likes int
|
|
dislikes int
|
|
}
|
|
|
|
const (
|
|
comment_prefix = "comment_"
|
|
post_prefix = "post_"
|
|
user_comment_count_prefix = "comment_count_"
|
|
)
|
|
|
|
func _deploy(data interface{}, isUpdate bool) {
|
|
ctx := storage.GetContext()
|
|
storage.Put(ctx, "index_comment", 0)
|
|
}
|
|
|
|
func CreateNewComment(userLogin string, postId string, text string) {
|
|
ctx := storage.GetContext()
|
|
|
|
id := storage.Get(ctx, "index_comment").(int)
|
|
id++
|
|
storage.Put(ctx, "index_comment", id)
|
|
|
|
if storage.Get(ctx, user_comment_count_prefix+userLogin) == nil {
|
|
storage.Put(ctx, user_comment_count_prefix+userLogin, 0)
|
|
}
|
|
|
|
commentId := comment_prefix + std.Itoa10(id)
|
|
newComment := Comment{
|
|
id: commentId,
|
|
userLogin: userLogin,
|
|
postId: postId,
|
|
text: text,
|
|
likes: 0,
|
|
dislikes: 0,
|
|
}
|
|
|
|
comments := GetByPostId(postId)
|
|
comments = append(comments, newComment)
|
|
storage.Put(ctx, comment_prefix+postId, std.Serialize(comments)) // добавление комментов в общий список комментов поста
|
|
storage.Put(ctx, commentId, std.Serialize(newComment)) // добавление коммента отдельно по ключу
|
|
|
|
commentCount := storage.Get(ctx, user_comment_count_prefix+userLogin).(int)
|
|
commentCount++
|
|
storage.Put(ctx, user_comment_count_prefix+userLogin, commentCount) // обновление счетчика коммента юзера
|
|
storage.Put(ctx, userLogin+"_c_"+std.Itoa10(commentCount), commentId) // хранение айдишника коммента по юзеру
|
|
|
|
runtime.Log("Comment created: " + commentId + " ----- " + user_comment_count_prefix + userLogin + " ------ " + userLogin + "_c_" + std.Itoa10(commentCount))
|
|
}
|
|
|
|
func GetByPostId(postId string) []Comment {
|
|
ctx := storage.GetContext()
|
|
comments := storage.Get(ctx, comment_prefix+postId)
|
|
if comments == nil {
|
|
storage.Put(ctx, comment_prefix+postId, std.Serialize(make([]Comment, 0)))
|
|
}
|
|
return std.Deserialize(storage.Get(ctx, comment_prefix+postId).([]byte)).([]Comment)
|
|
}
|
|
|
|
func GetByLoginInPost(postId string, login string) []Comment {
|
|
comments := GetByPostId(postId)
|
|
var commentsByAuthor []Comment
|
|
for _, comment := range comments {
|
|
if comment.userLogin == login {
|
|
commentsByAuthor = append(commentsByAuthor, comment)
|
|
}
|
|
}
|
|
|
|
return commentsByAuthor
|
|
}
|
|
|
|
func GetByLogin(login string) []Comment {
|
|
var comments []Comment
|
|
ctx := storage.GetContext()
|
|
it := storage.Find(ctx, login+"_c_", storage.ValuesOnly|storage.DeserializeValues)
|
|
for iterator.Next(it) {
|
|
commentId := iterator.Value(it).(string)
|
|
comments = append(comments, GetComment(commentId))
|
|
}
|
|
|
|
return comments
|
|
}
|
|
|
|
func GetComment(commentId string) Comment {
|
|
ctx := storage.GetContext()
|
|
return std.Deserialize(storage.Get(ctx, commentId).([]byte)).(Comment)
|
|
}
|
|
|
|
func GetCommentInPost(commentId string, postId string) Comment {
|
|
comments := GetByPostId(postId)
|
|
for _, comment := range comments {
|
|
if comment.id == commentId {
|
|
return comment
|
|
}
|
|
}
|
|
|
|
panic("Коммента с таким айдишником нету")
|
|
}
|
|
|
|
func RateComment(isLike bool, postId string, commentId string) {
|
|
comment := GetComment(commentId)
|
|
if isLike {
|
|
comment.likes++
|
|
} else {
|
|
comment.dislikes++
|
|
}
|
|
|
|
UpdateComment(comment, postId)
|
|
}
|
|
|
|
func UpdateComment(comment Comment, postId string) {
|
|
ctx := storage.GetContext()
|
|
comments := GetByPostId(postId)
|
|
for i := 0; i < len(comments); i++ {
|
|
if comments[i].id == comment.id {
|
|
comments[i] = comment
|
|
}
|
|
}
|
|
|
|
storage.Put(ctx, comment_prefix+postId, std.Serialize(comments))
|
|
storage.Put(ctx, comment.id, std.Serialize(comment))
|
|
}
|