Fix nix builds: go dependencies are awful

main.go

@@ -1,22 +1,21 @@
 package main
 
 import (
-	"flag"
 	"fmt"
 	"log"
 	"net"
 	"os"
 
-	"github.com/goburrow/netforward"
+	"ben.soroos.net/tsnet-proxy/netforward"
 	"tailscale.com/tsnet"
 )
 
 var (
-	host_proto = flag.Arg(0)
-	hostname   = flag.Arg(1)
-	host_port  = flag.Arg(2)
-	dst_proto  = flag.Arg(3)
-	dst_addr   = flag.Arg(4)
+	host_proto = os.Args[1]
+	hostname   = os.Args[2]
+	host_port  = os.Args[3]
+	dst_proto  = os.Args[4]
+	dst_addr   = os.Args[5]
 )
 
 type Dialer struct{}
@@ -26,15 +25,20 @@ func (dialer Dialer) Dial() (net.Conn, error) {
 }
 
 func main() {
-	flag.Parse()
 	if !(host_proto != "" && hostname != "" && host_port != "" && dst_proto != "" && dst_addr != "") {
 		fmt.Println("Usage: tsnet-proxy PROTO HOSTNAME PORT DST-PROTO DST")
 		fmt.Println("Where PROTO is one of tcp, udp, unix")
+		fmt.Println("Got: ", host_proto, hostname, host_port, dst_proto, dst_addr)
 		os.Exit(1)
 	}
 
+	dir, valid := os.LookupEnv("RUNTIME_DIRECTORY")
+	if !valid {
+		log.Fatal("RUNTIME_DIRECTORY must be set")
+	}
 	s := &tsnet.Server{
 		Hostname: hostname,
+		Dir: dir,
 	}
 	defer s.Close()
 
@@ -43,8 +47,6 @@ func main() {
 		log.Fatal(err)
 	}
 	defer ln.Close()
-	err = netforward.Forward(Dialer{}, ln)
-	if err != nil {
-		log.Fatal(err)
-	}
+
+	netforward.Forward(Dialer {}, ln)
 }