From 591ddb99f72b08f19ad56c426b668d893ae8a996 Mon Sep 17 00:00:00 2001
From: Roi Feng <37480123+Rayzggz@users.noreply.github.com>
Date: Tue, 29 Apr 2025 17:38:52 -0400
Subject: [PATCH] feat: health check api

---
 internal/server/torii.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/internal/server/torii.go b/internal/server/torii.go
index ffa1670..830b8fa 100644
--- a/internal/server/torii.go
+++ b/internal/server/torii.go
@@ -18,8 +18,19 @@ func CheckTorii(w http.ResponseWriter, r *http.Request, reqData dataType.UserReq
 	decision.SetCode(action.Continue, []byte("403"))
 	if reqData.Uri == cfg.WebPath+"/captcha" {
 		check.CheckCaptcha(r, reqData, ruleSet, decision)
+	} else if reqData.Uri == cfg.WebPath+"/health_check" {
+		decision.SetResponse(action.Done, []byte("200"), []byte("ok"))
 	}
 	if bytes.Compare(decision.HTTPCode, []byte("200")) == 0 {
+		if bytes.Compare(decision.ResponseData, []byte("ok")) == 0 {
+			w.WriteHeader(http.StatusOK)
+			_, err := w.Write([]byte("ok"))
+			if err != nil {
+				utils.LogError(reqData, "Error writing response: "+err.Error(), "CheckTorii")
+				return
+			}
+			return
+		}
 		if bytes.Compare(decision.ResponseData, []byte("bad")) == 0 {
 			w.WriteHeader(http.StatusOK)
 			_, err := w.Write([]byte("bad"))