package handlers import ( "encoding/json" "net/http" "github.com/owncast/owncast/config" "github.com/owncast/owncast/core/chat" "github.com/owncast/owncast/core/data" "github.com/owncast/owncast/models" "github.com/owncast/owncast/persistence/userrepository" "github.com/owncast/owncast/utils" "github.com/owncast/owncast/webserver/handlers/generated" "github.com/owncast/owncast/webserver/router/middleware" webutils "github.com/owncast/owncast/webserver/utils" log "github.com/sirupsen/logrus" ) // ExternalGetChatMessages gets all of the chat messages. func ExternalGetChatMessages(integration models.ExternalAPIUser, w http.ResponseWriter, r *http.Request) { middleware.EnableCors(w) getChatMessages(w, r) } // GetChatMessages gets all of the chat messages. func GetChatMessages(u models.User, w http.ResponseWriter, r *http.Request) { middleware.EnableCors(w) getChatMessages(w, r) } func getChatMessages(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "application/json") switch r.Method { case http.MethodGet: messages := chat.GetChatHistory() if err := json.NewEncoder(w).Encode(messages); err != nil { log.Debugln(err) } default: w.WriteHeader(http.StatusNotImplemented) if err := json.NewEncoder(w).Encode(webutils.J{"error": "method not implemented (PRs are accepted)"}); err != nil { webutils.InternalErrorHandler(w, err) } } } // RegisterAnonymousChatUser will register a new user. func RegisterAnonymousChatUser(w http.ResponseWriter, r *http.Request) { middleware.EnableCors(w) userRepository := userrepository.Get() if r.Method == http.MethodOptions { // All OPTIONS requests should have a wildcard CORS header. w.Header().Set("Access-Control-Allow-Origin", "*") w.WriteHeader(http.StatusNoContent) return } if r.Method != http.MethodPost { // nolint:goconst webutils.WriteSimpleResponse(w, false, r.Method+" not supported") return } type registerAnonymousUserResponse struct { ID string `json:"id"` AccessToken string `json:"accessToken"` DisplayName string `json:"displayName"` } decoder := json.NewDecoder(r.Body) var request generated.RegisterAnonymousChatUserJSONBody // registerAnonymousUserRequest if err := decoder.Decode(&request); err != nil { //nolint // this is fine. register a new user anyway. } proposedNewDisplayName := r.Header.Get("X-Forwarded-User") if proposedNewDisplayName == "" && request.DisplayName != nil { proposedNewDisplayName = *request.DisplayName } if proposedNewDisplayName == "" { proposedNewDisplayName = generateDisplayName() } proposedNewDisplayName = utils.MakeSafeStringOfLength(proposedNewDisplayName, config.MaxChatDisplayNameLength) newUser, accessToken, err := userRepository.CreateAnonymousUser(proposedNewDisplayName) if err != nil { webutils.WriteSimpleResponse(w, false, err.Error()) return } response := registerAnonymousUserResponse{ ID: newUser.ID, AccessToken: accessToken, DisplayName: newUser.DisplayName, } w.Header().Set("Content-Type", "application/json") middleware.DisableCache(w) webutils.WriteResponse(w, response) } func generateDisplayName() string { suggestedUsernamesList := data.GetSuggestedUsernamesList() minSuggestedUsernamePoolLength := 10 if len(suggestedUsernamesList) >= minSuggestedUsernamePoolLength { index := utils.RandomIndex(len(suggestedUsernamesList)) return suggestedUsernamesList[index] } else { return utils.GeneratePhrase() } }