package main

import (
	"bufio"
	"flag"
	"fmt"
	"io"
	"log"
	"net"
	"net/http"
	"net/http/httputil"

	"github.com/progrium/qmux/golang/session"
)

const (
	CLIENT_VERSION      = "v1.0"
	HEADER_CLIENT_TOKEN = "X-Client-App-Token"
	HEADER_USER_AGENT   = "User-Agent"
)

func main() {
	flag.Parse()

	srvPort, srvHost, clientToken := "18181", "hop.pbiernat.dev", "ghd7H8H*&6tg8*tg68"
	targetPort := flag.Arg(0)

	if targetPort == "" {
		log.Fatal("Missing port!")
	}

	conn, err := net.Dial("tcp", net.JoinHostPort(srvHost, srvPort))
	fatal("main net.Dial", err)
	client := httputil.NewClientConn(conn, bufio.NewReader(conn))
	req, err := http.NewRequest("GET", "/register", nil)
	fatal("http.NewRequest", err)
	req.Host = net.JoinHostPort(srvHost, srvPort)
	req.Header.Add(HEADER_CLIENT_TOKEN, clientToken)
	req.Header.Add(HEADER_USER_AGENT, "HopClient/"+CLIENT_VERSION)
	client.Write(req)
	resp, _ := client.Read(req)
	fmt.Printf("http port %s available at:\n", targetPort)
	fmt.Printf("https://%s\n", resp.Header.Get("X-Public-Host"))
	c, _ := client.Hijack()
	sess := session.New(c)
	defer sess.Close()
	for {
		ch, err := sess.Accept()
		fatal("Accept", err)
		conn, err := net.Dial("tcp", "127.0.0.1:"+targetPort)
		fatal("net.Dial", err)
		go join(conn, ch)
	}
}

func join(a io.ReadWriteCloser, b io.ReadWriteCloser) {
	go io.Copy(b, a)
	io.Copy(a, b)
	a.Close()
	b.Close()
}

func fatal(title string, err error) {
	if err != nil {
		log.Fatal(title+": ", err)
	}
}