From 21a4ff3b3e781b8d74f7e4200f1c3a1567585dda Mon Sep 17 00:00:00 2001
From: xing <fthvgb1@163.com>
Date: Wed, 4 Oct 2023 21:48:21 +0800
Subject: [PATCH] http tool get

---
 helper/func.go               |  8 +++++
 helper/func_test.go          | 13 +++++++
 helper/httptool/http.go      | 69 ++++++++++++++++++++++++++++++++++++
 helper/httptool/http_test.go | 51 ++++++++++++++++++++++++++
 4 files changed, 141 insertions(+)
 create mode 100644 helper/httptool/http.go
 create mode 100644 helper/httptool/http_test.go

diff --git a/helper/func.go b/helper/func.go
index 801065b..2533388 100644
--- a/helper/func.go
+++ b/helper/func.go
@@ -126,3 +126,11 @@ func FileExist(path string) bool {
 	_, err := os.Stat(path)
 	return err == nil
 }
+
+func GetAnyVal[T any](v any, defaults T) T {
+	vv, ok := v.(T)
+	if !ok {
+		return defaults
+	}
+	return vv
+}
diff --git a/helper/func_test.go b/helper/func_test.go
index eca596f..da45230 100644
--- a/helper/func_test.go
+++ b/helper/func_test.go
@@ -267,3 +267,16 @@ func TestGetValFromContext(t *testing.T) {
 		})
 	}
 }
+
+func TestGetAnyVal(t *testing.T) {
+	t.Run("string", func(t *testing.T) {
+		want := "string"
+		if got := GetAnyVal(any("string"), "s"); !reflect.DeepEqual(got, want) {
+			t.Errorf("GetAnyVal() = %v, want %v", got, want)
+		}
+		want = "s"
+		if got := GetAnyVal(any(1), "s"); !reflect.DeepEqual(got, want) {
+			t.Errorf("GetAnyVal() = %v, want %v", got, want)
+		}
+	})
+}
diff --git a/helper/httptool/http.go b/helper/httptool/http.go
new file mode 100644
index 0000000..467ba09
--- /dev/null
+++ b/helper/httptool/http.go
@@ -0,0 +1,69 @@
+package httptool
+
+import (
+	"io"
+	"net/http"
+	"net/url"
+	"time"
+)
+
+func GetString(u string, q map[string]string, timeout int64, a ...any) (r string, code int, err error) {
+	res, err := Get(u, q, timeout, a...)
+	if res != nil {
+		code = res.StatusCode
+	}
+	if err != nil {
+
+		return "", code, err
+	}
+	defer res.Body.Close()
+	rr, err := io.ReadAll(res.Body)
+	if err != nil {
+		return "", code, err
+	}
+	r = string(rr)
+	return
+}
+
+func Get(u string, q map[string]string, timeout int64, a ...any) (res *http.Response, err error) {
+	parse, err := url.Parse(u)
+	if err != nil {
+		return nil, err
+	}
+	cli := http.Client{
+		Timeout: time.Duration(timeout) * time.Second,
+	}
+	values := parse.Query()
+	for k, v := range q {
+		values.Add(k, v)
+	}
+	parse.RawQuery = values.Encode()
+	req := http.Request{
+		Method: "GET",
+		URL:    parse,
+	}
+	if len(a) > 0 {
+		for _, arg := range a {
+			h, ok := arg.(map[string]string)
+			if ok && len(h) > 0 {
+				for k, v := range h {
+					req.Header.Add(k, v)
+				}
+			}
+			t, ok := arg.(time.Duration)
+			if ok {
+				cli.Timeout = t
+			}
+			checkRedirect, ok := arg.(func(req *http.Request, via []*http.Request) error)
+			if ok {
+				cli.CheckRedirect = checkRedirect
+			}
+			jar, ok := arg.(http.CookieJar)
+			if ok {
+				cli.Jar = jar
+			}
+		}
+	}
+	res, err = cli.Do(&req)
+	return
+}
diff --git a/helper/httptool/http_test.go b/helper/httptool/http_test.go
new file mode 100644
index 0000000..867749f
--- /dev/null
+++ b/helper/httptool/http_test.go
@@ -0,0 +1,51 @@
+package httptool
+
+import (
+	"testing"
+)
+
+func TestGetString(t *testing.T) {
+	type args struct {
+		u       string
+		q       map[string]string
+		timeout int64
+		a       []any
+	}
+	tests := []struct {
+		name     string
+		args     args
+		wantR    string
+		wantCode int
+		wantErr  bool
+	}{
+		{
+			name: "wp.test",
+			args: args{
+				u: "http://wp.test",
+				q: map[string]string{
+					"p":                    "2",
+					"XDEBUG_SESSION_START": "34343",
+				},
+				timeout: 3,
+			},
+			wantR:    `{"XDEBUG_SESSION_START":"34343","p":"2"}`,
+			wantCode: 200,
+			wantErr:  false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotR, gotCode, err := GetString(tt.args.u, tt.args.q, tt.args.timeout, tt.args.a...)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("GetString() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if gotR != tt.wantR {
+				t.Errorf("GetString() gotR = %v, want %v", gotR, tt.wantR)
+			}
+			if gotCode != tt.wantCode {
+				t.Errorf("GetString() gotCode = %v, want %v", gotCode, tt.wantCode)
+			}
+		})
+	}
+}