acme-dns/api.go

77 lines
2.4 KiB
Go
Raw Normal View History

2016-11-11 21:48:00 +07:00
package main
import (
"encoding/json"
2016-11-11 21:48:00 +07:00
"fmt"
"net/http"
2017-01-30 17:19:22 +07:00
"github.com/julienschmidt/httprouter"
log "github.com/sirupsen/logrus"
2016-11-11 21:48:00 +07:00
)
// RegResponse is a struct for registration response JSON
type RegResponse struct {
Username string `json:"username"`
Password string `json:"password"`
Fulldomain string `json:"fulldomain"`
Subdomain string `json:"subdomain"`
Allowfrom []string `json:"allowfrom"`
2016-11-13 19:50:44 +07:00
}
func webRegisterPost(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
2016-11-17 00:15:36 +07:00
var regStatus int
var reg []byte
2016-12-03 15:31:15 +07:00
aTXT := ACMETxt{}
json.NewDecoder(r.Body).Decode(&aTXT)
// Create new user
2016-12-03 15:31:15 +07:00
nu, err := DB.Register(aTXT.AllowFrom)
2016-11-11 21:48:00 +07:00
if err != nil {
errstr := fmt.Sprintf("%v", err)
reg = jsonError(errstr)
regStatus = http.StatusInternalServerError
2016-11-26 20:42:35 +07:00
log.WithFields(log.Fields{"error": err.Error()}).Debug("Error in registration")
2016-11-11 21:48:00 +07:00
} else {
2016-11-26 20:42:35 +07:00
log.WithFields(log.Fields{"user": nu.Username.String()}).Debug("Created new user")
regStruct := RegResponse{nu.Username.String(), nu.Password, nu.Subdomain + "." + Config.General.Domain, nu.Subdomain, nu.AllowFrom.ValidEntries()}
regStatus = http.StatusCreated
reg, err = json.Marshal(regStruct)
if err != nil {
regStatus = http.StatusInternalServerError
reg = jsonError("json_error")
log.WithFields(log.Fields{"error": "json"}).Debug("Could not marshal JSON")
}
2016-11-11 21:48:00 +07:00
}
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(regStatus)
w.Write(reg)
2016-11-11 21:48:00 +07:00
}
func webUpdatePost(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
var updStatus int
var upd []byte
// Get user
a, ok := r.Context().Value(ACMETxtKey).(ACMETxt)
if !ok {
log.WithFields(log.Fields{"error": "context"}).Error("Context error")
}
2016-11-23 22:11:31 +07:00
if validSubdomain(a.Subdomain) && validTXT(a.Value) {
2016-11-13 19:50:44 +07:00
err := DB.Update(a)
if err != nil {
2016-11-26 20:42:35 +07:00
log.WithFields(log.Fields{"error": err.Error()}).Debug("Error while trying to update record")
updStatus = http.StatusInternalServerError
upd = jsonError("db_error")
} else {
log.WithFields(log.Fields{"subdomain": a.Subdomain, "txt": a.Value}).Debug("TXT updated")
updStatus = http.StatusOK
upd = []byte("{\"txt\": \"" + a.Value + "\"}")
2016-11-11 21:48:00 +07:00
}
2016-11-13 19:50:44 +07:00
} else {
log.WithFields(log.Fields{"error": "subdomain", "subdomain": a.Subdomain, "txt": a.Value}).Debug("Bad update data")
updStatus = http.StatusBadRequest
upd = jsonError("bad_subdomain")
2016-11-11 21:48:00 +07:00
}
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(updStatus)
w.Write(upd)
2016-11-11 21:48:00 +07:00
}