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

Add version flag

Browse Source
This commit is contained in:
Aaron Bieber 2021-12-10 15:50:07 -07:00
parent 0ea90c5ddd
commit 14bdf17e49
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.go
View File

@ -85,6 +85,8 @@ var (
tlsCert string tlsCert string
tlsKey string tlsKey string
users map[string]string users map[string]string
version bool
build string
) )
var pledges = "stdio wpath rpath cpath tty inet dns unveil" var pledges = "stdio wpath rpath cpath tty inet dns unveil"
@ -103,6 +105,7 @@ func init() {
flag.StringVar(&passPath, "htpass", fmt.Sprintf("%s/.htpasswd", dir), "Path to .htpasswd file..") flag.StringVar(&passPath, "htpass", fmt.Sprintf("%s/.htpasswd", dir), "Path to .htpasswd file..")
flag.StringVar(&auth, "auth", "basic", "Enable HTTP Basic Authentication.") flag.StringVar(&auth, "auth", "basic", "Enable HTTP Basic Authentication.")
flag.BoolVar(&genHtpass, "gen", false, "Generate a .htpasswd file or add a new entry to an existing file.") flag.BoolVar(&genHtpass, "gen", false, "Generate a .htpasswd file or add a new entry to an existing file.")
flag.BoolVar(&version, "v", false, "Show version and exit.")
flag.Parse() flag.Parse()
// These are OpenBSD specific protections used to prevent unnecessary file access. // These are OpenBSD specific protections used to prevent unnecessary file access.
@ -177,7 +180,10 @@ func prompt(prompt string, secure bool) (string, error) {
} }
func main() { func main() {
if version {
fmt.Println(build)
os.Exit(0)
}
if genHtpass { if genHtpass {
user, err := prompt("Username: ", false) user, err := prompt("Username: ", false)
if err != nil { if err != nil {