improve: 使用标准项目结构

2023-03-01 18:58:11 +08:00
parent cb11426bd6
commit 3c63e9ddc3
29 changed files with 178 additions and 183 deletions
--- a/pkg/srun/api.go
+++ b/pkg/srun/api.go
@@ -0,0 +1,121 @@
+package srun
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/Mmx233/tool"
+	log "github.com/sirupsen/logrus"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+)
+
+type Api struct {
+	inited  bool
+	BaseUrl string
+	Client  *http.Client
+	Header  http.Header
+}
+
+func (a *Api) Init(https bool, domain string, client *http.Client, header http.Header) {
+	if a.inited {
+		return
+	}
+
+	a.BaseUrl = "http"
+	if https {
+		a.BaseUrl += "s"
+	}
+	a.BaseUrl = a.BaseUrl + "://" + domain + "/"
+
+	a.Client = client
+
+	a.inited = true
+}
+
+func (a *Api) request(path string, query map[string]interface{}) (map[string]interface{}, error) {
+	log.Debugln("HTTP GET ", a.BaseUrl+path)
+	timestamp := fmt.Sprint(time.Now().UnixNano())
+	callback := "jQuery" + timestamp
+	if query == nil {
+		query = make(map[string]interface{}, 2)
+	}
+	query["callback"] = callback
+	query["_"] = timestamp
+	httpTool := tool.NewHttpTool(a.Client)
+	req, e := httpTool.GenReq("GET", &tool.DoHttpReq{
+		Url:   a.BaseUrl + path,
+		Query: query,
+	})
+	if e != nil {
+		log.Debugln(e)
+		return nil, e
+	}
+
+	for k, v := range a.Header {
+		req.Header[k] = v
+	}
+
+	resp, e := httpTool.Client.Do(req)
+	if e != nil {
+		log.Debugln(e)
+		return nil, e
+	}
+	defer resp.Body.Close()
+
+	data, e := io.ReadAll(resp.Body)
+	if e != nil {
+		log.Debugln(e)
+		return nil, e
+	}
+	res := string(data)
+
+	log.Debugln(res)
+	res = strings.TrimPrefix(res, callback+"(")
+	res = strings.TrimSuffix(res, ")")
+
+	var r map[string]interface{}
+	return r, json.Unmarshal([]byte(res), &r)
+}
+
+func (a *Api) GetUserInfo() (map[string]interface{}, error) {
+	return a.request("cgi-bin/rad_user_info", nil)
+}
+
+func (a *Api) Login(
+	Username,
+	Password,
+	AcID,
+	Ip,
+	Info,
+	ChkSum,
+	N,
+	Type string,
+) (map[string]interface{}, error) {
+	return a.request(
+		"cgi-bin/srun_portal",
+		map[string]interface{}{
+			"action":       "login",
+			"username":     Username,
+			"password":     Password,
+			"ac_id":        AcID,
+			"ip":           Ip,
+			"info":         Info,
+			"chksum":       ChkSum,
+			"n":            N,
+			"type":         Type,
+			"os":           "Windows 10",
+			"name":         "windows",
+			"double_stack": 0,
+		})
+}
+
+func (a *Api) GetChallenge(username, ip string) (map[string]interface{}, error) {
+	return a.request(
+		"cgi-bin/get_challenge",
+		map[string]interface{}{
+			"username": username,
+			"ip":       ip,
+		})
+}