From eadfefe4a5b1d6bf059583b99a3b5f5af50d8cb0 Mon Sep 17 00:00:00 2001
From: NaMe2te <vladyushashishkin81@gmail.com>
Date: Fri, 5 Jan 2024 16:39:52 +0300
Subject: [PATCH] fix: comment.go

---
 comment/comment.go | 94 +++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 84 insertions(+), 10 deletions(-)

diff --git a/comment/comment.go b/comment/comment.go
index 8dc2ccd..a5c996c 100644
--- a/comment/comment.go
+++ b/comment/comment.go
@@ -1,15 +1,89 @@
 package comment
 
-import "github.com/nspcc-dev/neo-go/pkg/interop/runtime"
+import (
+	"github.com/nspcc-dev/neo-go/pkg/interop/native/std"
+	"github.com/nspcc-dev/neo-go/pkg/interop/storage"
+	"github.com/rs/xid"
+)
 
-var notificationName string
-
-// init initializes notificationName before calling any other smart-contract method
-func init() {
-	notificationName = "Hello world!"
+type Comment struct {
+	id       string
+	authorId string
+	postId   string
+	text     string
+	likes    int
+	dislikes int
 }
 
-// RuntimeNotify sends runtime notification with "Hello world!" name
-func RuntimeNotify(args []any) {
-    runtime.Notify(notificationName, args)
-}
\ No newline at end of file
+func CreateNewComment(authorId string, postId string, text string) {
+	ctx := storage.GetContext()
+	newComment := Comment{
+		id:       xid.New().String(),
+		authorId: authorId,
+		postId:   postId,
+		text:     text,
+		likes:    0,
+		dislikes: 0,
+	}
+
+	comments := GetByPostId(postId)
+	comments = append(comments, newComment)
+	storage.Put(ctx, postId+"_comment", std.Serialize(comments))
+	storage.Put(ctx, newComment.id, std.Serialize(newComment))
+}
+
+func GetByPostId(postId string) []Comment {
+	ctx := storage.GetContext()
+	return std.Deserialize(storage.Get(ctx, postId+"_comment").([]byte)).([]Comment)
+}
+
+func GetByAuthorId(postId string, authorId string) []Comment {
+	comments := GetByPostId(postId)
+	var commentsByAuthor []Comment
+	for _, comment := range comments {
+		if comment.authorId == authorId {
+			commentsByAuthor = append(commentsByAuthor, comment)
+		}
+	}
+
+	return commentsByAuthor
+}
+
+func GetComment(commentId string, postId string) Comment {
+	comments := GetByPostId(postId)
+	for _, comment := range comments {
+		if comment.id == commentId {
+			return comment
+		}
+	}
+
+	panic("Коммента с таким айдишником нету v poste nety")
+}
+
+func GetComment(commentId string) Comment {
+	ctx := storage.GetContext()
+	return std.Deserialize(storage.Get(ctx, commentId).([]byte)).(Comment)
+}
+
+func RateComment(isLike bool, postId string, commentId string) {
+	comment := GetComment(commentId, postId)
+	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, postId+"_comment", std.Serialize(comments))
+}