Add: support feedback

This commit is contained in:
2021-12-13 23:18:46 +08:00
parent 22f7ea8476
commit 0c9048072f
9 changed files with 190 additions and 15 deletions

View File

@@ -76,6 +76,7 @@ func NewAPI(config Config) (*API, error) {
apiMux.HandleFunc("/get_file_stream", api.HandleGetFileStream)
apiMux.HandleFunc("/get_ffmpeg_config_list", api.HandleGetFfmpegConfigs)
apiMux.HandleFunc("/feedback", api.HandleFeedback)
apiMux.HandleFunc("/get_feedbacks", api.HandleGetFeedbacks)
apiMux.HandleFunc("/get_file_info", api.HandleGetFileInfo)
apiMux.HandleFunc("/get_file_stream_direct", api.HandleGetFileStreamDirect)
apiMux.HandleFunc("/prepare_file_stream_direct", api.HandlePrepareFileStreamDirect)

View File

@@ -4,12 +4,13 @@ import (
"bytes"
"encoding/json"
"log"
"msw-open-music/pkg/database"
"net/http"
"time"
)
type FeedbackRequest struct {
Feedback string `json:"feedback"`
Content string `json:"content"`
}
func (api *API) HandleFeedback(w http.ResponseWriter, r *http.Request) {
@@ -21,12 +22,12 @@ func (api *API) HandleFeedback(w http.ResponseWriter, r *http.Request) {
}
// check empty feedback
if feedbackRequest.Feedback == "" {
if feedbackRequest.Content == "" {
api.HandleErrorString(w, r, `"feedback" can't be empty`)
return
}
log.Println("[api] Feedback", feedbackRequest.Feedback)
log.Println("[api] Feedback", feedbackRequest.Content)
headerBuff := &bytes.Buffer{}
err = r.Header.Write(headerBuff)
@@ -36,10 +37,45 @@ func (api *API) HandleFeedback(w http.ResponseWriter, r *http.Request) {
}
header := headerBuff.String()
err = api.Db.InsertFeedback(time.Now().Unix(), feedbackRequest.Feedback, header)
userID, err := api.GetUserID(w, r)
if err != nil {
api.HandleError(w, r, err)
return
}
err = api.Db.InsertFeedback(time.Now().Unix(), feedbackRequest.Content, userID, header)
if err != nil {
api.HandleError(w, r, err)
return
}
api.HandleOK(w, r)
}
type GetFeedbacksResponse struct {
Feedbacks []*database.Feedback `json:"feedbacks"`
}
func (api *API) HandleGetFeedbacks(w http.ResponseWriter, r *http.Request) {
// check if admin
err := api.CheckAdmin(w, r)
if err != nil {
api.HandleError(w, r, err)
return
}
feedbacks, err := api.Db.GetFeedbacks()
if err != nil {
api.HandleError(w, r, err)
return
}
resp := &GetFeedbacksResponse{
Feedbacks: feedbacks,
}
err = json.NewEncoder(w).Encode(resp)
if err != nil {
api.HandleError(w, r, err)
return
}
}