htmgo/examples/chat/partials/index.go
2024-10-01 12:42:01 -05:00

74 lines
1.7 KiB
Go

package partials
import (
"chat/chat"
"chat/components"
"github.com/maddalax/htmgo/framework/h"
"net/http"
"time"
)
func CreateOrJoinRoom(ctx *h.RequestContext) *h.Partial {
locator := ctx.ServiceLocator()
service := chat.NewService(locator)
chatRoomId := ctx.Request.FormValue("join-chat-room")
username := ctx.Request.FormValue("username")
if username == "" {
return h.SwapPartial(ctx, components.FormError("Username is required"))
}
if len(username) > 15 {
return h.SwapPartial(ctx, components.FormError("Username is too long"))
}
user, err := service.CreateUser(username)
if err != nil {
return h.SwapPartial(ctx, components.FormError("Failed to create user"))
}
var redirect = func(path string) *h.Partial {
cookie := &http.Cookie{
Name: "session_id",
Value: user.SessionID,
Path: "/",
Expires: time.Now().Add(24 * 30 * time.Hour),
}
return h.SwapManyPartialWithHeaders(
ctx,
h.NewHeaders(
"Set-Cookie", cookie.String(),
"HX-Redirect", path,
),
h.Fragment(),
)
}
if chatRoomId != "" {
room, _ := service.GetRoom(chatRoomId)
if room == nil {
return h.SwapPartial(ctx, components.FormError("Room not found"))
} else {
return redirect("/chat/" + chatRoomId)
}
}
chatRoomName := ctx.Request.FormValue("new-chat-room")
if len(chatRoomName) > 20 {
return h.SwapPartial(ctx, components.FormError("Chat room name is too long"))
}
if chatRoomName != "" {
room, _ := service.CreateRoom(chatRoomName)
if room == nil {
return h.SwapPartial(ctx, components.FormError("Failed to create room"))
} else {
return redirect("/chat/" + room.ID)
}
}
return h.SwapPartial(ctx, components.FormError("Create a new room or join an existing one"))
}