feat(http): add server and auth middleware
This commit is contained in:
@@ -2,10 +2,13 @@ package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
"tomatentum.net/svg-templater/internal/database"
|
||||
svgtemplater "tomatentum.net/svg-templater/internal/svg-templater"
|
||||
"tomatentum.net/svg-templater/pkg/auth"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -18,10 +21,11 @@ func main() {
|
||||
log.Fatal("Failed opening DB:\n", err)
|
||||
return
|
||||
}
|
||||
defer database.Close()
|
||||
database.InitDB()
|
||||
prepareCommandLine()
|
||||
prepareHTTP()
|
||||
handleCommandline()
|
||||
defer database.Close()
|
||||
}
|
||||
|
||||
func prepareCommandLine() {
|
||||
@@ -29,6 +33,12 @@ func prepareCommandLine() {
|
||||
flag.BoolVar(&deleteTokenFlag, "tokendel", false, "svg-templater --tokendel <name> : Delete token with name")
|
||||
}
|
||||
|
||||
func prepareHTTP() {
|
||||
http.HandleFunc("/", auth.AuthMiddleware(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
fmt.Fprintln(w, "You are authorized!")
|
||||
})))
|
||||
}
|
||||
|
||||
func handleCommandline() {
|
||||
flag.Parse()
|
||||
if generateTokenFlag {
|
||||
@@ -36,7 +46,7 @@ func handleCommandline() {
|
||||
} else if deleteTokenFlag {
|
||||
svgtemplater.DeleteTokenCommand()
|
||||
} else {
|
||||
log.Println("No commend provided!")
|
||||
flag.PrintDefaults()
|
||||
log.Println("Starting http server on :3000")
|
||||
http.ListenAndServe(":3000", nil)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user