Blog/Post/post_contract.go

package Post

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 Post struct {
	postName string
	header   string
	text     string
	login    string
	likes    int
	dislikes int
	id       string
}

const (
	lastIndex = "user_post_index"
	id        = "current_post_id"
)

func NewPost(login string, text string, postName string) {
	ctx := storage.GetContext()
	updatePostId()
	id := storage.Get(ctx, id).(int)
	post_id := "post_" + std.Itoa10(id)

	post := Post{
		text:     text,
		login:    login,
		dislikes: 0,
		likes:    0,
		postName: postName,
		id:       post_id,
	}

	storage.Put(ctx, post.id, std.Serialize(post))

	updatePostIndex(login)
	lastPostIndex := storage.Get(ctx, login+lastIndex).(string)

	storage.Put(ctx, login+"_p_"+lastPostIndex, post.id)
}

func GetAllPosts() []Post {
	posts := make([]Post, 0)
	ctx := storage.GetReadOnlyContext()
	it := storage.Find(ctx, "post", storage.ValuesOnly|storage.DeserializeValues)
	for iterator.Next(it) {
		post := iterator.Value(it).(Post)
		posts = append(posts, post)
	}
	return posts
}

func GetPost(postId string) Post {

	ctx := storage.GetReadOnlyContext()
	data := storage.Get(ctx, postId)
	if data == nil {
		panic("post not found")
	}

	return std.Deserialize(data.([]byte)).(Post)
}

func GetAllPostsByUser(login string) []Post {
	ctx := storage.GetReadOnlyContext()
	var postsByUser []Post
	i := 0
	n := getPostIndex(login)
	for i < n {
		post := storage.Get(ctx, login+"_p_"+std.Itoa10(i))
		i++
		postsByUser = append(postsByUser, std.Deserialize(post.([]byte)).(Post))
	}
	return postsByUser
}

func RatePost(isLike bool, postId string) {
	ctx := storage.GetContext()
	post := GetPost(postId)
	if isLike {
		post.likes++
	} else {
		post.dislikes++
	}
	storage.Put(ctx, post.id, std.Serialize(post))
}

func getPostIndex(userId string) int {
	ctx := storage.GetContext()
	index := storage.Get(ctx, userId+lastIndex)
	if index == nil {
		index = 0
	}
	return index.(int)
}

func updatePostIndex(userId string) {
	ctx := storage.GetContext()
	index := getPostIndex(userId)
	storage.Put(ctx, userId+lastIndex, index+1)
}

func getPostId() int {
	ctx := storage.GetContext()
	current_id := storage.Get(ctx, id)
	if current_id == nil {
		current_id = 0
	}
	return current_id.(int)
}

func updatePostId() {
	ctx := storage.GetContext()
	cur_id := getPostId()
	storage.Put(ctx, id, cur_id+1)
}