52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"os/signal"
|
||
|
|
"strings"
|
||
|
|
"net/http"
|
||
|
|
"bytes"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
|
||
|
|
args := os.Args[1:]
|
||
|
|
tunnelSpecParts := strings.Split(args[0], ":")
|
||
|
|
tunnelId := args[0]
|
||
|
|
host := tunnelSpecParts[0]
|
||
|
|
port := tunnelSpecParts[1]
|
||
|
|
|
||
|
|
client := &http.Client{}
|
||
|
|
|
||
|
|
caddyAddRouteStr := fmt.Sprintf("{\"@id\":\"%s\",\"match\":[{\"host\":[\"%s\"]}],\"handle\":[{\"handler\":\"reverse_proxy\",\"upstreams\":[{\"dial\":\":%s\"}]}]}", tunnelId, host, port);
|
||
|
|
|
||
|
|
resp, err := http.Post("http://127.0.0.1:2019/config/apps/http/servers/sirtunnel/routes", "application/json", bytes.NewBuffer([]byte(caddyAddRouteStr)))
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
fmt.Println("Tunnel creation failed")
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
fmt.Println("Tunnel created successfully")
|
||
|
|
|
||
|
|
// wait for CTRL-C
|
||
|
|
c := make(chan os.Signal, 1)
|
||
|
|
signal.Notify(c, os.Interrupt)
|
||
|
|
<-c
|
||
|
|
|
||
|
|
fmt.Println("Cleaning up tunnel")
|
||
|
|
|
||
|
|
req, err := http.NewRequest("DELETE", fmt.Sprintf("http://127.0.0.1:2019/id/%s", tunnelId), nil)
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
req.Header.Add("Content-Type", "application/json")
|
||
|
|
_, err = client.Do(req)
|
||
|
|
|
||
|
|
fmt.Println("Exiting")
|
||
|
|
}
|