122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
package api
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"gopkg.in/oauth2.v3"
|
|
"net/http"
|
|
"openmediacenter/apiGo/api/oauth"
|
|
"openmediacenter/apiGo/database/settings"
|
|
)
|
|
|
|
const APIPREFIX = "/api"
|
|
|
|
const (
|
|
VideoNode = iota
|
|
TagNode = iota
|
|
SettingsNode = iota
|
|
ActorNode = iota
|
|
TVShowNode = iota
|
|
PhotoNode = iota
|
|
)
|
|
|
|
type HandlerInfo struct {
|
|
ID string
|
|
Token string
|
|
Data map[string]interface{}
|
|
}
|
|
|
|
type actionStruct struct {
|
|
Action string
|
|
}
|
|
|
|
type Handler struct {
|
|
action string
|
|
handler func(info *HandlerInfo) []byte
|
|
apiNode int
|
|
}
|
|
|
|
var handlers = make(map[string]Handler)
|
|
|
|
func AddHandler(action string, apiNode int, h func(info *HandlerInfo) []byte) {
|
|
// append new handler to the handlers
|
|
handlers[fmt.Sprintf("%s/%d", action, apiNode)] = Handler{action, h, apiNode}
|
|
}
|
|
|
|
func ServerInit() {
|
|
http.Handle(APIPREFIX+"/video", oauth.ValidateToken(handlefunc, VideoNode))
|
|
http.Handle(APIPREFIX+"/tags", oauth.ValidateToken(handlefunc, TagNode))
|
|
http.Handle(APIPREFIX+"/settings", oauth.ValidateToken(handlefunc, SettingsNode))
|
|
http.Handle(APIPREFIX+"/actor", oauth.ValidateToken(handlefunc, ActorNode))
|
|
|
|
// add tvshow endpoint only if tvshows enabled
|
|
if settings.TVShowsEnabled() {
|
|
http.Handle(APIPREFIX+"/tvshow", oauth.ValidateToken(handlefunc, TVShowNode))
|
|
}
|
|
|
|
http.Handle(APIPREFIX+"/photos", oauth.ValidateToken(handlefunc, PhotoNode))
|
|
|
|
// initialize oauth service and add corresponding auth routes
|
|
oauth.InitOAuth()
|
|
}
|
|
|
|
func handleAPICall(action string, requestBody string, apiNode int, info *HandlerInfo) []byte {
|
|
handler, ok := handlers[fmt.Sprintf("%s/%d", action, apiNode)]
|
|
if !ok {
|
|
// handler doesn't exist!
|
|
fmt.Printf("no handler found for Action: %d/%s\n", apiNode, action)
|
|
return nil
|
|
}
|
|
|
|
// check if info even exists
|
|
if info == nil {
|
|
info = &HandlerInfo{}
|
|
}
|
|
|
|
// parse the arguments
|
|
var args map[string]interface{}
|
|
err := json.Unmarshal([]byte(requestBody), &args)
|
|
|
|
if err != nil {
|
|
fmt.Printf("failed to decode arguments of action %s :: %s\n", action, requestBody)
|
|
} else {
|
|
// check if map has an action
|
|
if _, ok := args["action"]; ok {
|
|
delete(args, "action")
|
|
}
|
|
|
|
info.Data = args
|
|
}
|
|
|
|
// call the handler
|
|
return handler.handler(info)
|
|
}
|
|
|
|
func handlefunc(rw http.ResponseWriter, req *http.Request, node int, tokenInfo *oauth2.TokenInfo) {
|
|
// only allow post requests
|
|
if req.Method != "POST" {
|
|
return
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
buf.ReadFrom(req.Body)
|
|
body := buf.String()
|
|
|
|
var t actionStruct
|
|
err := json.Unmarshal([]byte(body), &t)
|
|
if err != nil {
|
|
fmt.Println("failed to read action from request! :: " + body)
|
|
}
|
|
|
|
// load userid from received token object
|
|
id := (*tokenInfo).GetClientID()
|
|
|
|
userinfo := &HandlerInfo{
|
|
ID: id,
|
|
Token: (*tokenInfo).GetCode(),
|
|
}
|
|
|
|
rw.Write(handleAPICall(t.Action, body, node, userinfo))
|
|
}
|