update with code snippet

1 files changed, 42 insertions, 0 deletions

diff --git a/helpers.go b/helpers.go
new file mode 100644
index 0000000..8f63de0
--- /dev/null
+++ b/helpers.go
@@ -0,0 +1,42 @@
+package chatpb
+
+import (
+	"time"
+
+	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
+)
+
+func (c *Chats) AddGeminiComment(s string) *Chat {
+	chat := new(Chat)
+
+	chat.From = Who_GEMINI
+	chat.Content = s
+	chat.Ctime = timestamppb.New(time.Now())
+
+	c.Append(chat)
+
+	return chat
+}
+
+func (c *Chats) AddUserComment(s string) *Chat {
+	chat := new(Chat)
+
+	chat.From = Who_USER
+	chat.Content = s
+
+	c.Append(chat)
+
+	return chat
+}
+
+func UnmarshalChats(data []byte) (*Chats, error) {
+	c := new(Chats)
+	err := c.Unmarshal(data)
+	return c, err
+}
+
+func UnmarshalChatsTEXT(data []byte) (*Chats, error) {
+	c := new(Chats)
+	err := c.UnmarshalTEXT(data)
+	return c, err
+}