Blog/Comment/comment.go

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/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) // хранение айдишника коммента по юзеру
}

func GetByPostId(postId string) []Comment {
	ctx := storage.GetContext()
	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 storage.Get(ctx, commentId).(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))
}