-
Notifications
You must be signed in to change notification settings - Fork 29
/
action_http_api.go
81 lines (69 loc) · 1.82 KB
/
action_http_api.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
package iplibrary
import (
"bytes"
"encoding/json"
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/firewallconfigs"
teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
"github.com/iwind/TeaGo/maps"
"net/http"
"time"
)
var httpAPIClient = &http.Client{
Timeout: 5 * time.Second,
}
type HTTPAPIAction struct {
BaseAction
config *firewallconfigs.FirewallActionHTTPAPIConfig
}
func NewHTTPAPIAction() *HTTPAPIAction {
return &HTTPAPIAction{}
}
func (this *HTTPAPIAction) Init(config *firewallconfigs.FirewallActionConfig) error {
this.config = &firewallconfigs.FirewallActionHTTPAPIConfig{}
err := this.convertParams(config.Params, this.config)
if err != nil {
return err
}
if len(this.config.URL) == 0 {
return NewFataError("'url' should not be empty")
}
return nil
}
func (this *HTTPAPIAction) AddItem(listType IPListType, item *pb.IPItem) error {
return this.runAction("addItem", listType, item)
}
func (this *HTTPAPIAction) DeleteItem(listType IPListType, item *pb.IPItem) error {
return this.runAction("deleteItem", listType, item)
}
func (this *HTTPAPIAction) runAction(action string, listType IPListType, item *pb.IPItem) error {
if item == nil {
return nil
}
// TODO 增加节点ID等信息
m := maps.Map{
"action": action,
"listType": listType,
"item": maps.Map{
"type": item.Type,
"ipFrom": item.IpFrom,
"ipTo": item.IpTo,
"expiredAt": item.ExpiredAt,
},
}
mJSON, err := json.Marshal(m)
if err != nil {
return err
}
req, err := http.NewRequest(http.MethodPost, this.config.URL, bytes.NewReader(mJSON))
if err != nil {
return err
}
req.Header.Set("User-Agent", teaconst.GlobalProductName+"-Node/"+teaconst.Version)
resp, err := httpAPIClient.Do(req)
if err != nil {
return err
}
_ = resp.Body.Close()
return nil
}