wolvie/short
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix add an extra flag for the URL port

Browse Source
This commit is contained in:
Tom Andrade 2022-05-18 15:11:07 +02:00
parent d8e77cc756
commit f88b908ade
Signed by: wolvie
GPG Key ID: 31AAB07872E82669
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.go
View File

@ -71,8 +71,9 @@ func main() {
version = flag.Bool("v", false, "prints current version") version = flag.Bool("v", false, "prints current version")
listenAddr string listenAddr string
) )
flag.StringVar(&shortie.DumpFile, "dumpfile", "Path to the file to dump the kv db", "urls.json") flag.StringVar(&shortie.DumpFile, "dumpFile", "Path to the file to dump the kv db", "urls.json")
flag.IntVar(&shortie.URLSize, "size", 10, "Define the size of the shortened String") flag.IntVar(&shortie.URLSize, "size", 10, "Define the size of the shortened String")
flag.IntVar(&shortie.Port, "urlPort", 443, "Port to use for building URLs")
flag.Parse() flag.Parse()
@ -81,10 +82,14 @@ func main() {
os.Exit(0) os.Exit(0)
} }
if shortie.Port > 65535 || shortie.Port < 1 {
log.Fatalln("Invalid port number")
}
if *port > 65535 || *port < 1 { if *port > 65535 || *port < 1 {
log.Fatalln("Invalid port number") log.Fatalln("Invalid port number")
} }
shortie.Port = *port
if *path != "" && !strings.HasSuffix(*path, "/") { if *path != "" && !strings.HasSuffix(*path, "/") {
*path = *path + "/" *path = *path + "/"
} }