-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
136 lines (113 loc) · 2.75 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
package main
import (
"bytes"
"encoding/json"
"flag"
"fmt"
"github.com/gin-gonic/gin"
"io/ioutil"
"net/http"
"os"
"time"
)
type Configuration struct {
ForwardTo map[string][]string `json:"forward_to"`
}
func main() {
listenPtr := flag.String("listen", "", "The port/address to listen on, overrides sock")
sockPtr := flag.String("sock", "SlackProxy.sock", "The socket file to listen on.")
flag.Parse()
if _, err := os.Stat("config.json"); os.IsNotExist(err) {
data, err := json.MarshalIndent(Configuration{ForwardTo:map[string][]string{"key": {"https://discordapp.com/meme", "https://discordapp.com/dank"}}}, "", " ")
if err != nil {
panic(err)
}
if err := ioutil.WriteFile("config.json", data, os.FileMode(0644)); err != nil {
panic(err)
}
fmt.Println("DEFAULT CONFIG SAVED, PLEASE EDIT!")
fmt.Println("DEFAULT CONFIG SAVED, PLEASE EDIT!")
os.Exit(1)
return
}
data, err := ioutil.ReadFile("config.json")
if err != nil {
panic(err)
}
config := &Configuration{}
if err := json.Unmarshal(data, config); err != nil {
panic(err)
}
engine := gin.Default()
for k, v := range config.ForwardTo {
engine.POST(k, CreateHandler(v))
}
if *listenPtr == "" {
socketString := *sockPtr
go func() {
for {
time.Sleep(500 * time.Millisecond)
if _, err := os.Stat(socketString); err == nil {
if err := os.Chmod(socketString, 0770); err != nil {
panic(err)
}
fmt.Println("Chmodded the socket file!")
break
}
}
}()
engine.RunUnix(socketString)
} else {
engine.Run(*listenPtr)
}
}
func CreateHandler(recepients []string) func(*gin.Context) {
return func(ctx *gin.Context) {
client := http.Client{}
errorOcurred := false
var lastResponse *http.Response
body, err := ioutil.ReadAll(ctx.Request.Body)
if err != nil {
ctx.Error(err)
return
}
for _, v := range recepients {
buf := new(bytes.Buffer)
buf.Write(body)
req, err := http.NewRequest("POST", v, buf)
if err != nil {
fmt.Println(err)
continue
}
for headerName, vals := range ctx.Request.Header {
for _, val := range vals {
req.Header.Set(headerName, val)
}
}
resp, err := client.Do(req)
if err != nil {
fmt.Println(err)
continue
}
if resp.StatusCode < 200 || resp.StatusCode > 209 {
errorOcurred = true
lastResponse = resp
} else if !errorOcurred {
lastResponse = resp
}
}
ctx.Status(lastResponse.StatusCode)
for headerName, vals := range lastResponse.Header {
for _, val := range vals {
ctx.Header(headerName, val)
}
}
contentType := lastResponse.Header.Get("Content-type")
data, err := ioutil.ReadAll(lastResponse.Body)
if err != nil {
ctx.Error(err)
return
}
ctx.Data(lastResponse.StatusCode, contentType, data)
}
}