feat: allow configuration of read and write timeout (#9)

cmd/main.go

@@ -12,6 +12,7 @@ import (
 	"os/signal"
 	"strings"
 	"syscall"
+	"time"
 
 	"github.com/a-h/gemini"
 )
@@ -73,6 +74,8 @@ func request(args []string) {
 	verboseFlag := cmd.Bool("verbose", false, "Print both headers and body.")
 	headersFlag := cmd.Bool("headers", false, "Print only the headers.")
 	allowBinaryFlag := cmd.Bool("allowBinary", false, "Set to true to enable printing binary to the console.")
+	readTimeoutFlag := cmd.Duration("readTimeout", time.Second*5, "Set the duration, e.g. 1m or 5s (default is 5 seconds).")
+	writeTimeoutFlag := cmd.Duration("writeTimeout", time.Second*5, "Set the duration, e.g. 1m or 5s (default is 5 seconds).")
 	helpFlag := cmd.Bool("help", false, "Print help and exit.")
 	err := cmd.Parse(args)
 	if err != nil || *helpFlag {
@@ -91,6 +94,8 @@ func request(args []string) {
 	}
 
 	client := gemini.NewClient()
+	client.ReadTimeout = *readTimeoutFlag
+	client.WriteTimeout = *writeTimeoutFlag
 	if *insecureFlag {
 		client.Insecure = true
 	}
@@ -164,6 +169,8 @@ func serve(args []string) {
 	domainFlag := cmd.String("domain", "localhost", "The domain to listen on.")
 	pathFlag := cmd.String("path", ".", "Path containing content.")
 	portFlag := cmd.Int("port", 1965, "Address to listen on.")
+	readTimeoutFlag := cmd.Duration("readTimeout", time.Second*5, "Set the duration, e.g. 1m or 5s (default is 5 seconds).")
+	writeTimeoutFlag := cmd.Duration("writeTimeout", time.Second*10, "Set the duration, e.g. 1m or 5s (default is 10 seconds).")
 	helpFlag := cmd.Bool("help", false, "Print help and exit.")
 	err := cmd.Parse(args)
 	if err != nil || *helpFlag {
@@ -189,6 +196,8 @@ func serve(args []string) {
 		*domainFlag: dh,
 	}
 	server := gemini.NewServer(ctx, fmt.Sprintf(":%d", *portFlag), domainToHandler)
+	server.ReadTimeout = *readTimeoutFlag
+	server.WriteTimeout = *writeTimeoutFlag
 	err = server.ListenAndServe()
 	if err != nil {
 		fmt.Printf("error: %v\n", err)