@@ -0,0 +1,93 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"net/http"
|
||||
"strconv"
|
||||
|
||||
"github.com/labstack/echo"
|
||||
"github.com/labstack/echo/middleware"
|
||||
)
|
||||
|
||||
type passwordData struct {
|
||||
Length int
|
||||
Digits int
|
||||
Symbols int
|
||||
NoUpper bool
|
||||
DenyRepeat bool
|
||||
Password string
|
||||
}
|
||||
|
||||
var d, defaults passwordData
|
||||
|
||||
func generatePassword(c echo.Context) error {
|
||||
|
||||
var err error
|
||||
|
||||
d.Length, err = strconv.Atoi(c.FormValue("length"))
|
||||
if err != nil {
|
||||
d.Length = defaults.Length
|
||||
}
|
||||
|
||||
d.Digits, err = strconv.Atoi(c.FormValue("digits"))
|
||||
if err != nil {
|
||||
d.Digits = defaults.Digits
|
||||
}
|
||||
|
||||
d.Symbols, err = strconv.Atoi(c.FormValue("symbols"))
|
||||
if err != nil {
|
||||
d.Symbols = defaults.Symbols
|
||||
}
|
||||
|
||||
if c.FormValue("noupper") == "on" {
|
||||
d.NoUpper = true
|
||||
}
|
||||
|
||||
if c.FormValue("denyrepeat") == "on" {
|
||||
d.DenyRepeat = true
|
||||
}
|
||||
|
||||
// Generate a password
|
||||
d.Password, err = generate(d.Length, d.Digits, d.Symbols, d.NoUpper, !d.DenyRepeat)
|
||||
if err != nil {
|
||||
d.Password = "Error: " + err.Error()
|
||||
}
|
||||
if d.Length == 0 {
|
||||
d.Password = "Error: password can not have zero length"
|
||||
}
|
||||
|
||||
return c.String(http.StatusOK, d.Password)
|
||||
}
|
||||
|
||||
func checkHealth(c echo.Context) error {
|
||||
s := `{"status":"OK"}`
|
||||
return c.String(http.StatusOK, s)
|
||||
}
|
||||
|
||||
func main() {
|
||||
|
||||
// Set defaults
|
||||
portPtr := flag.Int("listen", 8080, "Specify on which port to listen")
|
||||
lengthPtr := flag.Int("length", 16, "Specify the password length")
|
||||
digitsPtr := flag.Int("digits", 2, "Specify the the number of digits in the password")
|
||||
symbolsPtr := flag.Int("symbols", 2, "Specify the the number of symbols in the password")
|
||||
flag.Parse()
|
||||
|
||||
defaults.Length = *lengthPtr
|
||||
defaults.Digits = *digitsPtr
|
||||
defaults.Symbols = *symbolsPtr
|
||||
|
||||
e := echo.New()
|
||||
e.HideBanner = true
|
||||
|
||||
// Middleware
|
||||
e.Use(middleware.Logger())
|
||||
e.Use(middleware.Recover())
|
||||
|
||||
e.GET("/", generatePassword)
|
||||
e.POST("/", generatePassword)
|
||||
|
||||
e.GET("/health", checkHealth)
|
||||
|
||||
e.Logger.Fatal(e.Start(":" + strconv.Itoa(*portPtr)))
|
||||
}
|
||||
Reference in New Issue
Block a user