From ecb0b510c4f790407ff94ca822aabe14d08f98db Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=BD=99=E6=B3=93=E9=93=AE?= <yuhongzheng@bytedance.com>
Date: Wed, 14 Aug 2024 16:28:30 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=B8=8B=E8=BD=BDeapi=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=EF=BC=8C=E5=B9=B6=E6=B7=BB=E5=8A=A0token?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 hrp/internal/builtin/utils.go      | 15 ++++++++++++++-
 hrp/internal/builtin/utils_test.go | 12 ++++++++++++
 2 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 hrp/internal/builtin/utils_test.go

diff --git a/hrp/internal/builtin/utils.go b/hrp/internal/builtin/utils.go
index dcfb8bf1..08f264e4 100644
--- a/hrp/internal/builtin/utils.go
+++ b/hrp/internal/builtin/utils.go
@@ -520,7 +520,19 @@ func DownloadFile(filePath string, url string) error {
 	}
 	defer out.Close()
 
-	resp, err := http.Get(url)
+	// 创建一个新的 HTTP 请求
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return err
+	}
+
+	// 添加自定义头部
+	req.Header.Add("accessKey", "ies.vedem.video")
+	req.Header.Add("token", "***REMOVED***")
+
+	// 创建一个 HTTP 客户端并发送请求
+	client := &http.Client{}
+	resp, err := client.Do(req)
 	if err != nil {
 		return err
 	}
@@ -530,6 +542,7 @@ func DownloadFile(filePath string, url string) error {
 		return fmt.Errorf("bad status: %s, download failed", resp.Status)
 	}
 
+	// 将响应主体写入文件
 	_, err = io.Copy(out, resp.Body)
 	if err != nil {
 		return err
diff --git a/hrp/internal/builtin/utils_test.go b/hrp/internal/builtin/utils_test.go
new file mode 100644
index 00000000..e8f70220
--- /dev/null
+++ b/hrp/internal/builtin/utils_test.go
@@ -0,0 +1,12 @@
+package builtin
+
+import (
+	"testing"
+)
+
+func TestDownload(t *testing.T) {
+	err := DownloadFile("/tmp/bytedance.ds.zip", "https://gtf-eapi-cn.bytedance.com/cn/mostRecent/bytedance.ds.zip")
+	if err != nil {
+		t.Fatal(err)
+	}
+}