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

Change listen addr/port flags

Browse Source
This commit is contained in:
Tom Andrade 2019-04-26 17:46:15 +02:00
parent ded42136c4
commit 18c3271249
Signed by: wolvie
GPG Key ID: 31AAB07872E82669
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@ -82,7 +82,8 @@ const returnPage = `
var domain string var domain string
var redisServer string var redisServer string
var listenAddr string var addr string
var port string
var proto string var proto string
var path string var path string
var src = rand.NewSource(time.Now().UnixNano()) var src = rand.NewSource(time.Now().UnixNano())
@ -138,15 +139,16 @@ func shortner(w http.ResponseWriter, r *http.Request) {
break break
} }
} }
var hostSuf string
if port != "80" && proto == "http" { if port != "80" && proto == "http" {
port = ":" + port + "/" hostSuf = ":" + port + "/"
} else if port != "443" && proto == "https" { } else if port != "443" && proto == "https" {
port = ":" + port + "/" hostSuf = ":" + port + "/"
} else if port == "443" || port == "80" { } else if port == "443" || port == "80" {
port = "/" hostSuf = "/"
} }
set(u.String(), suffix) set(u.String(), suffix)
shortend := proto + "://" + domain + port + path + suffix shortend := proto + "://" + domain + hostSuf + path + suffix
output := fmt.Sprintf(returnPage, shortend, shortend) output := fmt.Sprintf(returnPage, shortend, shortend)
w.Write([]byte(output)) w.Write([]byte(output))
} }
@ -172,7 +174,8 @@ func RandStringBytesMaskImprSrc(n int) string {
func main() { func main() {
flag.StringVar(&domain, "domain", "localhost", "Domain to write to the URLs") flag.StringVar(&domain, "domain", "localhost", "Domain to write to the URLs")
flag.StringVar(&listenAddr, "addr", "localhost:8080", "Address to listen for connections") flag.StringVar(&addr, "addr", "localhost", "Address to listen for connections")
flag.StringVar(&port, "port", "8080", "Port to listen for connections")
flag.StringVar(&path, "path", "", "Path to the base URL (https://localhost/PATH/... remember to append a / at the end") flag.StringVar(&path, "path", "", "Path to the base URL (https://localhost/PATH/... remember to append a / at the end")
flag.StringVar(&proto, "proto", "https", "proto to the base URL (HTTPS://localhost/path/... no real https here just to set the url (for like a proxy offloading https") flag.StringVar(&proto, "proto", "https", "proto to the base URL (HTTPS://localhost/path/... no real https here just to set the url (for like a proxy offloading https")
version := flag.Bool("v", false, "prints current version") version := flag.Bool("v", false, "prints current version")
@ -186,6 +189,7 @@ func main() {
path = path + "/" path = path + "/"
} }
listenAddr := addr + ":" + port
r := mux.NewRouter() r := mux.NewRouter()