package main

import (
	"bytes"
	"context"
	"encoding/hex"
	"flag"
	"fmt"
	"github.com/wiretrustee/wiretrustee/browser/conn"
	"github.com/wiretrustee/wiretrustee/signal/client"
	"golang.zx2c4.com/wireguard/device"
	"golang.zx2c4.com/wireguard/tun/netstack"
	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
	"io"
	"log"
	"net"
	"net/http"
	"time"
)

//my private key qJi7zSrgdokeoXE27fbca2hvMlgg1NQIW6KbrTJhhmc=
//remote private key KLuBc6tM/NRV1071bfPiNUxZmMhGBCXfxoDg+A+J7ns=
func main() {

	keyFlag := flag.String("key", "", "a Wireguard private key")
	remoteKeyFlag := flag.String("remote-key", "", "a Wireguard remote peer public key")
	signalEndpoint := flag.String("signal-endpoint", "ws://apitest.wiretrustee.com:80/signal", "a Signal service Websocket endpoint")
	cl := flag.Bool("client", false, "indicates whether the program is a client")

	flag.Parse()

	key, err := wgtypes.ParseKey(*keyFlag)
	if err != nil {
		panic(err)
	}

	log.Printf("my public key: %s", key.PublicKey().String())

	remoteKey, err := wgtypes.ParseKey(*remoteKeyFlag)

	if err != nil {
		panic(err)
	}

	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
	defer cancel()

	signal, err := client.NewWebsocketClient(ctx, *signalEndpoint, key)

	time.Sleep(5 * time.Second)

	tun, tnet, err := netstack.CreateNetTUN(
		[]net.IP{net.ParseIP("10.100.0.2")},
		[]net.IP{net.ParseIP("8.8.8.8")},
		1420)

	b := conn.NewWebRTCBind("chann-1", signal, key.PublicKey().String(), remoteKey.String())
	dev := device.NewDevice(tun, b, device.NewLogger(device.LogLevelVerbose, ""))
	err = dev.IpcSet(fmt.Sprintf("private_key=%s\npublic_key=%s\npersistent_keepalive_interval=100\nendpoint=webrtc://datachannel\nallowed_ip=0.0.0.0/0",
		hex.EncodeToString(key[:]),
		hex.EncodeToString(remoteKey[:]),
	))

	dev.Up()

	if err != nil {
		panic(err)
	}

	client := http.Client{
		Transport: &http.Transport{
			DialContext: tnet.DialContext,
		},
	}
	time.Sleep(2 * time.Second)

	if *cl {

		req, _ := http.NewRequest("POST", "https://httpbin.org/ip", bytes.NewBufferString("ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd"))
		//req.Header.Set("js.fetch:mode", "no-cors")
		resp, err := client.Do(req)
		if err != nil {
			log.Panic(err)
		}
		body, err := io.ReadAll(resp.Body)
		if err != nil {
			log.Panic(err)
		}
		log.Printf(string(body))
		log.Printf(resp.Status)

	} else {
		listener, err := tnet.ListenTCP(&net.TCPAddr{Port: 80})
		if err != nil {
			log.Panicln(err)
		}
		http.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
			log.Printf("> %s - %s - %s", request.RemoteAddr, request.URL.String(), request.UserAgent())
			io.WriteString(writer, "Hello from userspace TCP!")
		})
		err = http.Serve(listener, nil)
		if err != nil {
			log.Panicln(err)
		}
	}

	select {}

}