Added more examples and updated README.md

Author: MrIceman

README.md

@@ -44,7 +44,11 @@ You don't need to download any dependencies such as plantUML or Graphviz, which
 
 ```
 Result PNG file: 
-![image description](./examples/user_starts_chatting.png)
+![Sequence Diagram generated based on above code](./examples/user_starts_chatting.png)
+
+Some more examples:
+![example_a](./examples/on_new_message_for_user_b_not_read.png)
+![example_a](./examples/on_new_message_for_user_b.png)
 
 ## Warning - AI not welcome here
 Because of my own personal philosophy regarding technology and AI, all the code in this repository that was written by me - I wrote 100% on my own. There is and will be no usage of Github Co-Pilot or any other AI tool. I became a software developer because of my passion for our craft - Software Engineering. I build this tool because I enjoy programming. Every single line of code you'll read in this repo, that was written by me, is produced first in my mind and then manifested into reality through my hands. I encourage any contributor to follow the same principle, though I can't and don't want to put any restrictions on this.

examples/main.go

@@ -3,19 +3,77 @@ package main
 import "github.com/mriceman/go-uml/sequence"
 
 func main() {
+	userStartsChatting()
+	userBIsNotPartOfChat()
+	UserBIsPartOfChat()
+
+}
+
+func UserBIsPartOfChat() {
+	d := sequence.NewDiagram("on_new_message_for_user_b")
+
+	usrA := "User A"
+	usrB := "User B"
+	backend := "Backend"
+	db := "DynamoDB"
+
+	d.SetTitle("User A sends a message, User B is connected to the chat and receives it")
+
+	d.AddParticipants(usrA, usrB, backend, db)
+
+	d.AddDirectionalEdge(usrA, backend, "WS sendmessage/userB_id payload: hi")
+	d.AddDirectionalEdge(backend, db, "get connection ID from userB")
+	d.AddDirectionalEdge(backend, db, "store message")
+	d.AddDirectionalEdge(backend, usrA, "new_message from: userA payload: hi")
+	d.AddDirectionalEdge(backend, db, "set all messages read for usrA")
+	d.Render()
+}
+
+func userBIsNotPartOfChat() {
+	d := sequence.NewDiagram("on_new_message_for_user_b_not_read")
+
+	usrA := "User A"
+	usrB := "User B"
+	backend := "Backend"
+	db := "DynamoDB"
+
+	d.SetTitle("User A sends a message, User B is not connected to the chat")
+
+	d.AddParticipants(usrA, usrB, backend, db)
+
+	d.AddDirectionalEdge(usrA, backend, "WS sendmessage/userB_id payload: hi")
+	d.AddDirectionalEdge(backend, db, "get connection ID from userB")
+	d.AddDirectionalEdge(backend, db, "store message")
+	d.AddDirectionalEdge(backend, db, "increment_unread_count_for_user_b")
+	d.AddDirectionalEdge(backend, backend, "create_push_for_user_b")
+	d.AddDirectionalEdge(usrB, usrB, "receives a push notification")
+	d.AddDirectionalEdge(usrB, usrB, "navigates to chats")
+	d.AddDirectionalEdge(usrB, backend, "GET /chat")
+	d.AddDirectionalEdge(backend, db, "Query * chats for userB")
+	d.AddDirectionalEdge(backend, usrB, "return chats for userB")
+	d.AddDirectionalEdge(
+		usrB,
+		usrB,
+		"renders list of chats + unread count")
+	d.Render()
+}
+
+func userStartsChatting() {
 	d := sequence.NewDiagram("user_starts_chatting")
 
 	client := "Client"
 	backend := "Backend"
 	db := "DynamoDB"
 
 	d.SetTitle("User initiates chat with another User")
-	d.AddParticipant(client)
-	d.AddParticipant(backend)
-	d.AddParticipant(db)
+
+	d.AddParticipants(client)
+	d.AddParticipants(backend)
+	d.AddParticipants(db)
 
 	d.AddDirectionalEdge(client, backend, "PUT /chat/user/<TO_ID>")
 	d.AddDirectionalEdge(backend, db, "checks or create inbox for user")
+	d.AddDirectionalEdge(backend, db, "set all unread messages to read if existing")
 	d.AddDirectionalEdge(backend, client, "return chat meta data + chat secret")
 	d.AddDirectionalEdge(client, backend, "WS /joinchat/<CHAT_ID> payload: secret")
 	d.AddDirectionalEdge(backend, db, "compare inbox secret")