qbit/tsns
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add a response type and pass back errors

Browse Source
This commit is contained in:
Aaron Bieber 2024-08-23 08:51:31 -06:00
parent 259060aa74
commit 94a5f9c8da
No known key found for this signature in database
3 changed files with 32 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
main.go
View File

@ -77,40 +77,46 @@ func main() {
mux := http.NewServeMux() mux := http.NewServeMux()
mux.HandleFunc("GET /", func(w http.ResponseWriter, r *http.Request) { mux.HandleFunc("GET /", func(w http.ResponseWriter, r *http.Request) {
httpLog(r) httpLog(r)
if err := records.templ.ExecuteTemplate(w, "index", records); err != nil { resp := &Response{}
resp.Entries = records.Entries
if err := records.templ.ExecuteTemplate(w, "index", resp); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError) http.Error(w, err.Error(), http.StatusInternalServerError)
} }
}) })
mux.HandleFunc("POST /records", func(w http.ResponseWriter, r *http.Request) { mux.HandleFunc("POST /records", func(w http.ResponseWriter, r *http.Request) {
httpLog(r) httpLog(r)
resp := &Response{}
ipStr := r.FormValue("ip") ipStr := r.FormValue("ip")
ip := net.ParseIP(ipStr) ip := net.ParseIP(ipStr)
rec := Record{ if ip != nil {
Name: r.FormValue("name"), rec := Record{
IP: ip, Name: r.FormValue("name"),
IP: ip,
}
records.Entries = append(records.Entries, rec)
records.Save(*dataDir)
} else {
resp.Error = fmt.Errorf("invalid IP: %q", ipStr)
} }
records.Entries = append(records.Entries, rec) resp.Entries = records.Entries
records.Save(*dataDir)
if err := records.templ.ExecuteTemplate(w, "table", records); err != nil { if err := records.templ.ExecuteTemplate(w, "table", resp); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError) http.Error(w, err.Error(), http.StatusInternalServerError)
} }
}) })
mux.HandleFunc("DELETE /records/{name}", func(w http.ResponseWriter, r *http.Request) { mux.HandleFunc("DELETE /records/{name}", func(w http.ResponseWriter, r *http.Request) {
httpLog(r) httpLog(r)
resp := Response{}
records.Delete(r.PathValue("name")) records.Delete(r.PathValue("name"))
records.Save(*dataDir) records.Save(*dataDir)
if err := records.templ.ExecuteTemplate(w, "table", records); err != nil { resp.Entries = records.Entries
if err := records.templ.ExecuteTemplate(w, "table", resp); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError) http.Error(w, err.Error(), http.StatusInternalServerError)
} }
}) })
mux.HandleFunc("PUT /records/{name}", func(w http.ResponseWriter, r *http.Request) {
httpLog(r)
})
httpServer := &http.Server{ httpServer := &http.Server{
Handler: mux, Handler: mux,

5
records.go
View File

@ -12,6 +12,11 @@ import (
"github.com/miekg/dns" "github.com/miekg/dns"
) )
type Response struct {
Error error `json:"error"`
Entries []Record `json:"entries"`
}
type Record struct { type Record struct {
Name string `json:"name"` Name string `json:"name"`
IP net.IP `json:"ip"` IP net.IP `json:"ip"`

9
templates/templates.html
View File

@ -23,7 +23,6 @@
</style> </style>
</head> </head>
<body> <body>
<h1>tsns</h1> <h1>tsns</h1>
<hr/> <hr/>
@ -48,7 +47,15 @@
{{ template "row" . }} {{ template "row" . }}
{{ end }} {{ end }}
</tbody> </tbody>
<tfoot>
<tr>
<td>
<div id="error" >{{ .Error }}</div>
</td>
</tr>
</tfoot>
</table> </table>
{{ end }} {{ end }}
{{ block "row" . }} {{ block "row" . }}