99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package ctrlsubsonic
|
|
|
|
import (
|
|
"encoding/json"
|
|
"encoding/xml"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
|
|
"senan.xyz/g/gonic/server/ctrlbase"
|
|
"senan.xyz/g/gonic/server/ctrlsubsonic/spec"
|
|
"senan.xyz/g/gonic/server/parsing"
|
|
)
|
|
|
|
type Controller struct {
|
|
*ctrlbase.Controller
|
|
}
|
|
|
|
func New(base *ctrlbase.Controller) *Controller {
|
|
return &Controller{
|
|
Controller: base,
|
|
}
|
|
}
|
|
|
|
type metaResponse struct {
|
|
XMLName xml.Name `xml:"subsonic-response" json:"-"`
|
|
*spec.Response `json:"subsonic-response"`
|
|
}
|
|
|
|
type errWriter struct {
|
|
w io.Writer
|
|
err error
|
|
}
|
|
|
|
func (ew *errWriter) write(buf []byte) {
|
|
if ew.err != nil {
|
|
return
|
|
}
|
|
_, ew.err = ew.w.Write(buf)
|
|
}
|
|
|
|
func writeResp(w http.ResponseWriter, r *http.Request, resp *spec.Response) {
|
|
res := metaResponse{Response: resp}
|
|
ew := &errWriter{w: w}
|
|
switch parsing.GetStrParam(r, "f") {
|
|
case "json":
|
|
w.Header().Set("Content-Type", "application/json")
|
|
data, err := json.Marshal(res)
|
|
if err != nil {
|
|
log.Printf("could not marshall to json: %v\n", err)
|
|
return
|
|
}
|
|
ew.write(data)
|
|
case "jsonp":
|
|
w.Header().Set("Content-Type", "application/javascript")
|
|
data, err := json.Marshal(res)
|
|
if err != nil {
|
|
log.Printf("could not marshall to json: %v\n", err)
|
|
return
|
|
}
|
|
ew.write([]byte(parsing.GetStrParamOr(r, "callback", "cb")))
|
|
ew.write([]byte("("))
|
|
ew.write(data)
|
|
ew.write([]byte(");"))
|
|
default:
|
|
w.Header().Set("Content-Type", "application/xml")
|
|
data, err := xml.MarshalIndent(res, "", " ")
|
|
if err != nil {
|
|
log.Printf("could not marshall to xml: %v\n", err)
|
|
return
|
|
}
|
|
ew.write(data)
|
|
}
|
|
if ew.err != nil {
|
|
log.Printf("error writing to response: %v\n", ew.err)
|
|
}
|
|
}
|
|
|
|
type subsonicHandler func(r *http.Request) *spec.Response
|
|
|
|
func (c *Controller) H(h subsonicHandler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// TODO: write a non 200 if has err
|
|
response := h(r)
|
|
writeResp(w, r, response)
|
|
})
|
|
}
|
|
|
|
type subsonicHandlerRaw func(w http.ResponseWriter, r *http.Request) *spec.Response
|
|
|
|
func (c *Controller) HR(h subsonicHandlerRaw) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// TODO: write a non 200 if has err
|
|
// TODO: ensure no mixed return/writer
|
|
response := h(w, r)
|
|
writeResp(w, r, response)
|
|
})
|
|
}
|