feat(helpers): unsafe StringToBytes and BytesToString

Author: oddyamill

internal/client/client.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"encoding/json"
 	"errors"
+	"github.com/kiasuo/bot/internal/helpers"
 	"github.com/kiasuo/bot/internal/users"
 	"net/http"
 )
@@ -39,7 +40,7 @@ func httpRequest[T any](client Client, request *http.Request) (*http.Response, *
 }
 
 func RefreshToken(client *Client) error {
-	body := []byte(`{"refresh-token":"` + client.User.RefreshToken + `"}`)
+	body := helpers.StringToBytes(`{"refresh-token":"` + client.User.RefreshToken + `"}`)
 
 	request, err := http.NewRequest("POST", BaseUrl+"/refresh", bytes.NewBuffer(body))
 	request.Header.Set("Content-Type", "application/json")

internal/helpers/helpers.go

@@ -3,6 +3,7 @@ package helpers
 import (
 	"os"
 	"strings"
+	"unsafe"
 )
 
 func If[T any](condition bool, truth, falsity T) T {
@@ -13,6 +14,14 @@ func If[T any](condition bool, truth, falsity T) T {
 	return falsity
 }
 
+func StringToBytes(str string) []byte {
+	return unsafe.Slice(unsafe.StringData(str), len(str))
+}
+
+func BytesToString(bytes []byte) string {
+	return unsafe.String(unsafe.SliceData(bytes), len(bytes))
+}
+
 func GetEnv(key string) string {
 	value, ok := os.LookupEnv(key)
 
@@ -29,7 +38,7 @@ func GetEnv(key string) string {
 			panic(err)
 		}
 
-		return strings.TrimSpace(string(buffer))
+		return strings.TrimSpace(BytesToString(buffer))
 	}
 
 	panic("Environment variable " + key + " not set")

internal/helpers/helpers_test.go

@@ -21,6 +21,22 @@ func TestIfFalse(t *testing.T) {
 	}
 }
 
+func TestStringToBytes(t *testing.T) {
+	result := StringToBytes("test")
+
+	if string(result) != "test" {
+		t.Errorf("StringToBytes() = %s; want test", result)
+	}
+}
+
+func TestBytesToString(t *testing.T) {
+	result := BytesToString([]byte("test"))
+
+	if result != "test" {
+		t.Errorf("BytesToString() = %s; want test", result)
+	}
+}
+
 func TestGetEnvPath(t *testing.T) {
 	// I think PATH variable is set on every system
 	GetEnv("PATH")