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 {
return err
}
tx.Commit(ctx)
return nil
}
@@ -54,7 +55,10 @@ var cli Cli
func main() {
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())
if err := ctx.Run(); err != nil {
println(err.Error())

View File

@@ -1,6 +1,7 @@
package main
import (
"flag"
"fmt"
"github.com/gin-gonic/gin"
"github.com/thequux/qddns/common"
@@ -12,15 +13,7 @@ import (
"strings"
)
func main() {
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) {
func Update(c *gin.Context) {
authHdr := strings.Fields(c.GetHeader("Authorization"))
domain := c.Param("domain")
@@ -87,5 +80,31 @@ func main() {
return
}
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 {
Status string
Code string `json:"omitempty"`
Message string `json:"omitempty"`
Code string //`json:"omitempty"`
Message string //`json:"omitempty"`
}