From 6a34c730715dbc6561f149a4c4f91a6066a3c22b Mon Sep 17 00:00:00 2001
From: Akkariin Meiko <akkariins@gmail.com>
Date: Sun, 8 Sep 2019 18:32:21 +0800
Subject: [PATCH] Update api.go

---
 extend/api/api.go | 38 +++++++++++++++++++++++++++++++++++++-
 1 file changed, 37 insertions(+), 1 deletion(-)

diff --git a/extend/api/api.go b/extend/api/api.go
index b018558..36b9837 100755
--- a/extend/api/api.go
+++ b/extend/api/api.go
@@ -148,7 +148,38 @@ func (s Service) CheckProxy(user string, pMsg *msg.NewProxy, timestamp int64, st
 
 // GetProxyLimit 获取隧道限速信息
 func (s Service) GetProxyLimit(pxyConf *config.BaseProxyConf) (inLimit, outLimit uint64, err error) {
-	return 1 * limit.MB, 1 * limit.MB, nil
+	// 这部分就照之前的搬过去了，能跑就行x
+	values := url.Values{}
+	values.Set("do", "getlimit")
+	values.Set("user", user)
+	values.Set("frpstoken", stk)
+	s.Host.RawQuery = values.Encode()
+	defer func(u *url.URL) {
+		u.RawQuery = ""
+	}(&s.Host)
+	resp, err := http.Get(s.Host.String())
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+
+	er := &ErrHTTPStatus{}
+	if err = json.Unmarshal(body, er); err != nil {
+		return nil, err
+	}
+	if er.Status != 200 {
+		return nil, er
+	}
+
+	r = &ResponseGetLimit{}
+	if err = json.Unmarshal(body, r); err != nil {
+		return nil, err
+	}
+	return r.MaxIn, r.MaxOut, nil
 }
 
 func BoolToString(val bool) (str string) {
@@ -168,6 +199,11 @@ func (e ErrHTTPStatus) Error() string {
 	return fmt.Sprintf("%s", e.Text)
 }
 
+type ResponseGetLimit struct {
+	MaxIn  uint64 `json:"max-in"`
+	MaxOut uint64 `json:"max-out"`
+}
+
 type ResponseCheckToken struct {
 	Success bool   `json:"success"`
 	Message string `json:"message"`