From 4233090738027d42e1d4e26d7d1bd928e5b6b248 Mon Sep 17 00:00:00 2001 From: "Bo-Yi.Wu" Date: Mon, 10 Oct 2022 17:17:23 +0800 Subject: [PATCH] chore(runner): get token from header Signed-off-by: Bo-Yi.Wu --- routers/api/bots/runner/runner.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/routers/api/bots/runner/runner.go b/routers/api/bots/runner/runner.go index 8b1c4f0ce9..d10a878929 100644 --- a/routers/api/bots/runner/runner.go +++ b/routers/api/bots/runner/runner.go @@ -35,7 +35,9 @@ func (s *Service) Register( ctx context.Context, req *connect.Request[runnerv1.RegisterRequest], ) (*connect.Response[runnerv1.RegisterResponse], error) { - if req.Msg.Token == "" || req.Msg.Name == "" || req.Msg.Url == "" { + token := req.Header().Get("X-Runner-Token") + + if token == "" || req.Msg.Name == "" || req.Msg.Url == "" { return nil, errors.New("missing runner token, name or URL") }