cmd/server/main.go

@@ -3,43 +3,37 @@ package main
 import (
 	"fmt"
 	"log"
-	"net/http"
+	"net"
 	"time"
+
+	"git.pbiernat.dev/golang/ngkrok/internal/ngkrok"
+	vhost "github.com/inconshreveable/go-vhost"
 )
 
 func main() {
-	// addr, host, port := "::", "hop.pbiernat.dev", "18181"
+	addr, host, port := "::", "hop.pbiernat.dev", "18181"
 
-	// lst, err := net.Listen("tcp", net.JoinHostPort(addr, port))
-	// if err != nil {
-	// 	log.Fatal(err)
-	// }
-
-	// vmux, err := vhost.NewHTTPMuxer(lst, 3*time.Second)
-	// if err != nil {
-	// 	log.Fatal(err)
-	// }
-	// defer vmux.Close()
-
-	// srv := ngkrok.NewServer(addr, host, port, vmux)
-	// go srv.Serve()
-
-	// log.Printf("NgKrok server [%s] ready!\n", host)
-
-	// for {
-	// 	conn, err := vmux.NextError()
-	// 	fmt.Println("vmux error:", err)
-	// 	if conn != nil {
-	// 		conn.Close()
-	// 	}
-	// }
-
-	port := ":18181"
-
-	reverseProxy := http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
-		fmt.Printf("[reverse proxy server] received request at: %s\n", time.Now())
-	})
-
-	fmt.Printf("Server started at %s\n", port)
-	log.Fatal(http.ListenAndServe(port, reverseProxy))
+	lst, err := net.Listen("tcp", net.JoinHostPort(addr, port))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	vmux, err := vhost.NewHTTPMuxer(lst, 3*time.Second)
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer vmux.Close()
+
+	srv := ngkrok.NewServer(addr, host, port, vmux)
+	go srv.Serve()
+
+	log.Printf("NgKrok server [%s] ready!\n", host)
+
+	for {
+		conn, err := vmux.NextError()
+		fmt.Println("vmux error:", err)
+		if conn != nil {
+			conn.Close()
+		}
+	}
 }