Add version to headers

1 files changed, 4 insertions, 3 deletions

diff --git a/cmd/web/server.go b/cmd/web/server.go
index e76b6b2..d13cd58 100644
--- a/cmd/web/server.go
+++ b/cmd/web/server.go
@@ -25,12 +25,12 @@ import (
         "golang.org/x/time/rate"
 )
 
-func Register(root *cobra.Command, embeddedTemplates fs.FS) {
+func Register(root *cobra.Command, embeddedTemplates fs.FS, version string) {
         webCmd := &cobra.Command{
                 Use:   "web [options]",
                 Short: "Run web server",
                 Run: func(c *cobra.Command, args []string) {
-                        webMain(app.NewConfigFromCmd(c), embeddedTemplates)
+                        webMain(app.NewConfigFromCmd(c), embeddedTemplates, version)
                 },
         }
 
@@ -53,7 +53,7 @@ func Register(root *cobra.Command, embeddedTemplates fs.FS) {
         root.AddCommand(webCmd)
 }
 
-func webMain(cfg app.Config, embeddedTemplates fs.FS) {
+func webMain(cfg app.Config, embeddedTemplates fs.FS, version string) {
         ctx := context.Background()
 
         s, err := glecho.NewDefaultEchoWithConfig(glecho.EchoConfig{
@@ -86,6 +86,7 @@ func webMain(cfg app.Config, embeddedTemplates fs.FS) {
         if err = s.Init(); err != nil {
                 log.Fatalf("Error initializing echo: %w", err)
         }
+        s.Use(middleware.AddServerHeader(version))
 
         mongo, err := mongodb.Connect(ctx, cfg.MongoDbUri, cfg.MongodbVaultPath)
         if err != nil {