implemented protocol for web

ritsec · Dec 13, 2023 · a505114 · a505114
1 parent 1b04ac8
commit a505114
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/commands/slash/vote.go b/commands/slash/vote.go
@@ -264,7 +264,7 @@ func Vote() (*discordgo.ApplicationCommand, func(s *discordgo.Session, i *discor
 				return
 			}
 
-			out := "```\n" + entVoteResults.Plain + "\n```" + "\nResults Graph:\nhttp://" + config.Web.Hostname + ":" + config.Web.Port + "/vote/" + voteSlug + "\n"
+			out := "```\n" + entVoteResults.Plain + "\n```" + "\nResults Graph:\n" + config.Web.Protocol + "://" + config.Web.Hostname + ":" + config.Web.Port + "/vote/" + voteSlug + "\n"
 
 			_, err = s.InteractionResponseEdit(i.Interaction, &discordgo.WebhookEdit{
 				Content:    &out,

diff --git a/config/main.go b/config/main.go
@@ -74,6 +74,7 @@ func web() structs.WebConfig {
 	return structs.WebConfig{
 		Port:     viper.GetString("web.port"),
 		Hostname: viper.GetString("web.hostname"),
+		Protocol: viper.GetString("web.protocol"),
 	}
 }
 

diff --git a/structs/webConfig.go b/structs/webConfig.go
@@ -7,4 +7,7 @@ type WebConfig struct {
 
 	// Hostname is the hostname to listen on
 	Hostname string
+
+	// Protocol is the protocol to use
+	Protocol string
 }