feat(server): add optional basic auth feature

lukewhrit · Jul 22, 2024 · b66a7be · b66a7be
1 parent 6840957
commit b66a7be
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -31,6 +31,8 @@ type Cfg struct {
 	// Web
 	Headless  bool   `env:"HEADLESS" envDefault:"false" json:"headless"` // Enable website
 	Analytics string `env:"ANALYTICS" envDefault:"" json:"analytics"`    // <script> tag for analytics (leave blank to disable)
+	Username  string `env:"USERNAME" envDefault:"" json:"username"`      // Basic Auth username. Required to enable Basic Auth
+	Password  string `env:"PASSWORD" envDefault:"" json:"password"`      // Basic Auth password. Required to enable Basic Auth
 
 	// Document
 	IDLength      int      `env:"ID_LENGTH" envDefault:"8" json:"id_length"`

diff --git a/internal/server/server.go b/internal/server/server.go
@@ -107,6 +107,13 @@ func (s *Server) MountMiddleware() {
 		AllowCredentials: false,
 		MaxAge:           300, // Maximum value not ignored by any of major browsers
 	}))
+
+	// Basic Auth
+	if s.Config.Username != "" && s.Config.Password != "" {
+		s.Router.Use(middleware.BasicAuth("spacebin", map[string]string{
+			s.Config.Username: s.Config.Password,
+		}))
+	}
 }
 
 func (s *Server) RegisterHeaders() {