Use cmd args to get request url

1 files changed, 26 insertions, 2 deletions

diff --git a/main.go b/main.go
index cf5b69e..cae68ab 100644
--- a/main.go
+++ b/main.go
@@ -2,17 +2,41 @@ package main
 
 import (
 	"crypto/tls"
+	"flag"
+	"fmt"
 	"log"
+	"os"
+	"strconv"
 )
 
-const url = "gemini.circumlunar.space:1965"
-const request = "gemini://gemini.circumlunar.space/\r\n"
+var port = 1965
+
+func usage() {
+	fmt.Printf("Usage: %s [options] url\n", os.Args[0])
+	flag.PrintDefaults()
+}
 
 func main() {
+	flag.Usage = usage
+
+	flag.IntVar(&port, "p", port, "server port number")
+
+	flag.Parse()
+	if flag.NArg() != 1 {
+		fmt.Printf("Got %d arguments, expected 1\n", flag.NArg())
+		flag.Usage()
+		os.Exit(1)
+	}
+
+	url := flag.Arg(0) + ":" + strconv.Itoa(port)
+	request := "gemini://" + flag.Arg(0) + "/\r\n"
+
 	conf := tls.Config{
 		InsecureSkipVerify: true,
 	}
 
+	log.Println("connecting to: " + url)
+
 	conn, err := tls.Dial("tcp", url, &conf)
 
 	if err != nil {