Forgot to connect to the DB in the admin tool

This commit is contained in:
2022-08-06 16:47:46 +02:00
parent f3f8d353f1
commit cbf17e7d27
3 changed files with 92 additions and 69 deletions

View File

@@ -36,6 +36,7 @@ func (cmd *AddToken) Run() error {
if err != nil { if err != nil {
return err return err
} }
tx.Commit(ctx)
return nil return nil
} }
@@ -54,7 +55,10 @@ var cli Cli
func main() { func main() {
ctx := kong.Parse(&cli) ctx := kong.Parse(&cli)
//db.Connect(cli.Db) if err := db.Connect(cli.Db); err != nil {
println(err.Error())
os.Exit(0)
}
println(ctx.Command()) println(ctx.Command())
if err := ctx.Run(); err != nil { if err := ctx.Run(); err != nil {
println(err.Error()) println(err.Error())

View File

@@ -1,6 +1,7 @@
package main package main
import ( import (
"flag"
"fmt" "fmt"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/thequux/qddns/common" "github.com/thequux/qddns/common"
@@ -12,15 +13,7 @@ import (
"strings" "strings"
) )
func main() { func Update(c *gin.Context) {
if err := db.Connect(""); err != nil {
fmt.Fprintf(os.Stderr, "Unable to connect to database: %v", err)
os.Exit(1)
}
// Set up server
r := gin.Default()
r.POST("/update/:domain", func(c *gin.Context) {
authHdr := strings.Fields(c.GetHeader("Authorization")) authHdr := strings.Fields(c.GetHeader("Authorization"))
domain := c.Param("domain") domain := c.Param("domain")
@@ -87,5 +80,31 @@ func main() {
return return
} }
tx.Commit(c) tx.Commit(c)
}) }
var listen = flag.String("listen", ":8081", "Address or path on which to listen")
func main() {
flag.Parse()
if err := db.Connect(""); err != nil {
fmt.Fprintf(os.Stderr, "Unable to connect to database: %v", err)
os.Exit(1)
}
// Set up server
r := gin.Default()
r.POST("/update/:domain", Update)
var err error
if _, err = net.ResolveTCPAddr("tcp", *listen); err == nil {
err = r.Run(*listen)
} else {
// Probably a UNIX address
err = r.RunUnix(*listen)
}
if err != nil {
println("Failed to listen: " + err.Error())
os.Exit(1)
}
} }

View File

@@ -2,6 +2,6 @@ package common
type Response struct { type Response struct {
Status string Status string
Code string `json:"omitempty"` Code string //`json:"omitempty"`
Message string `json:"omitempty"` Message string //`json:"omitempty"`
} }