From 18dbcd4818d8a9d53aaac3cdd6869ddedb5d5a58 Mon Sep 17 00:00:00 2001
From: samwaf <samwafgo@gmail.com>
Date: Thu, 24 Oct 2024 08:44:56 +0800
Subject: [PATCH] feat:add robots.txt

---
 wafmangeweb/static/static.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/wafmangeweb/static/static.go b/wafmangeweb/static/static.go
index 7d77a0f..a5f3c2f 100644
--- a/wafmangeweb/static/static.go
+++ b/wafmangeweb/static/static.go
@@ -59,6 +59,19 @@ func Static(r *gin.Engine, noRoute func(handlers ...gin.HandlerFunc)) {
 		c.Status(http.StatusOK)
 		_, _ = io.Copy(c.Writer, faviconFile)
 	})
+	r.GET("/robots.txt", func(c *gin.Context) {
+		robotFile, err := static.Open("robots.txt")
+		if err != nil {
+			zlog.Error("can't find robots.txt")
+			c.Status(http.StatusNotFound)
+			return
+		}
+		defer robotFile.Close()
+
+		c.Header("Content-Type", "text/plain")
+		c.Status(http.StatusOK)
+		_, _ = io.Copy(c.Writer, robotFile)
+	})
 
 	noRoute(func(c *gin.Context) {
 		c.Header("Content-Type", "text/html")