happyDomain/api/source_specs.go

139 lines
2.9 KiB
Go

package api
import (
"bytes"
"errors"
"io"
"net/http"
"reflect"
"strings"
"github.com/julienschmidt/httprouter"
"git.happydns.org/happydns/config"
"git.happydns.org/happydns/sources"
)
func init() {
router.GET("/api/source_specs", apiHandler(getSourceSpecs))
router.GET("/api/source_specs/*ssid", apiHandler(getSourceSpec))
}
type field struct {
Id string `json:"id"`
Label string `json:"label,omitempty"`
Placeholder string `json:"placeholder,omitempty"`
Default string `json:"default,omitempty"`
Choices []string `json:"choices,omitempty"`
Required bool `json:"required,omitempty"`
Secret bool `json:"secret,omitempty"`
Description string `json:"description,omitempty"`
}
func getSourceSpecs(_ *config.Options, p httprouter.Params, body io.Reader) Response {
srcs := sources.GetSources()
ret := map[string]sources.SourceInfos{}
for k, src := range *srcs {
ret[k] = src.Infos
}
return APIResponse{
response: ret,
}
}
func getSourceSpecImg(ssid string) Response {
if cnt, ok := sources.Icons[strings.TrimSuffix(ssid, ".png")]; ok {
return &FileResponse{
contentType: "image/png",
content: bytes.NewBuffer(cnt),
}
} else {
return APIErrorResponse{
status: http.StatusNotFound,
err: errors.New("Icon not found."),
}
}
}
type viewSourceSpec struct {
Fields []field `json:"fields,omitempty"`
Capabilities []string `json:"capabilities,omitempty"`
}
func getSourceSpec(_ *config.Options, p httprouter.Params, body io.Reader) Response {
ssid := string(p.ByName("ssid"))
if len(ssid) > 1 {
ssid = ssid[1:]
}
if strings.HasSuffix(ssid, ".png") {
return getSourceSpecImg(ssid)
}
src, err := sources.FindSource(ssid)
if err != nil {
return APIErrorResponse{
err: err,
}
}
srcType := reflect.Indirect(reflect.ValueOf(src)).Type()
fields := []field{}
for i := 0; i < srcType.NumField(); i += 1 {
jsonTag := srcType.Field(i).Tag.Get("json")
jsonTuples := strings.Split(jsonTag, ",")
f := field{}
if len(jsonTuples) > 0 && len(jsonTuples[0]) > 0 {
f.Id = jsonTuples[0]
} else {
f.Id = srcType.Field(i).Name
}
tag := srcType.Field(i).Tag.Get("happydns")
tuples := strings.Split(tag, ",")
for _, t := range tuples {
kv := strings.SplitN(t, "=", 2)
if len(kv) > 1 {
switch strings.ToLower(kv[0]) {
case "label":
f.Label = kv[1]
case "placeholder":
f.Placeholder = kv[1]
case "default":
f.Default = kv[1]
case "description":
f.Description = kv[1]
case "choices":
f.Choices = strings.Split(kv[1], ";")
}
} else {
switch strings.ToLower(kv[0]) {
case "required":
f.Required = true
case "secret":
f.Secret = true
default:
f.Label = kv[0]
}
}
}
fields = append(fields, f)
}
var caps []string
if _, ok := src.(sources.ListDomainsSource); ok {
caps = append(caps, "ListDomains")
}
return APIResponse{
response: viewSourceSpec{fields, caps},
}
}