From 878ce0b3a6b78bd590288b9ce3b1f418659d97a5 Mon Sep 17 00:00:00 2001
From: shijie <810833920@qq.com>
Date: Mon, 8 Dec 2025 17:13:59 +0800
Subject: [PATCH] [feat][meta] Configured with a token

---
 internal/model/query/query.go          | 3 ++-
 internal/service/repository_service.go | 8 ++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/internal/model/query/query.go b/internal/model/query/query.go
index d4c06b0..625f279 100644
--- a/internal/model/query/query.go
+++ b/internal/model/query/query.go
@@ -114,7 +114,8 @@ type ModelQuery struct {
 }
 
 type RepositoryReq struct {
-	Id int64 `json:"id"`
+	Id    int64  `json:"id"`
+	Token string `json:"token"`
 }
 
 type TagQuery struct {
diff --git a/internal/service/repository_service.go b/internal/service/repository_service.go
index d8e96b2..103c7c2 100644
--- a/internal/service/repository_service.go
+++ b/internal/service/repository_service.go
@@ -257,6 +257,14 @@ func (s *RepositoryService) MountRepository(repoReq *query.RepositoryReq) error
 	if err != nil {
 		return err
 	}
+
+	authHeaders := make(map[string]string)
+	if repoReq.Token != "" {
+		authHeaders["Authorization"] = fmt.Sprintf("Bearer %s", repoReq.Token)
+	} else {
+		authHeaders = s.hfTokenDao.GetHeaders()
+	}
+
 	_, err = util.PostForDomain(speedDomain, "/api/cacheJob/create", "application/json", b, s.hfTokenDao.GetHeaders())
 	if err != nil {
 		return err