From 5080c566ac31ec622986c04f1812a1e88c88210e Mon Sep 17 00:00:00 2001
From: Lizzy Hunt <lizzy.hunt@usu.edu>
Date: Fri, 29 Mar 2024 16:35:04 -0600
Subject: guestbook!

---
 args/args.go | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

(limited to 'args/args.go')

diff --git a/args/args.go b/args/args.go
index 3be0abd..40dd1af 100644
--- a/args/args.go
+++ b/args/args.go
@@ -10,11 +10,9 @@ import (
 )
 
 type Arguments struct {
-	DatabasePath    string
-	TemplatePath    string
-	StaticPath      string
-	CloudflareToken string
-	CloudflareZone  string
+	DatabasePath string
+	TemplatePath string
+	StaticPath   string
 
 	Migrate   bool
 	Scheduler bool
@@ -27,6 +25,12 @@ type Arguments struct {
 	Dns          bool
 	DnsRecursion []string
 	DnsPort      int
+
+	CloudflareToken string
+	CloudflareZone  string
+
+	HcaptchaSecret  string
+	HcaptchaSiteKey string
 }
 
 func GetArgs() (*Arguments, error) {
@@ -57,6 +61,9 @@ func GetArgs() (*Arguments, error) {
 	oauthRedirectURI := os.Getenv("OAUTH_REDIRECT_URI")
 	oauthUserInfoURI := os.Getenv("OAUTH_USER_INFO_URI")
 
+	hcaptchaSecret := os.Getenv("HCAPTCHA_SECRET")
+	hcaptchaSiteKey := os.Getenv("HCAPTCHA_SITE_KEY")
+
 	envVars := [][]string{
 		{cloudflareToken, "CLOUDFLARE_TOKEN"},
 		{cloudflareZone, "CLOUDFLARE_ZONE"},
@@ -102,6 +109,9 @@ func GetArgs() (*Arguments, error) {
 
 		OauthConfig:      oauthConfig,
 		OauthUserInfoURI: oauthUserInfoURI,
+
+		HcaptchaSecret:  hcaptchaSecret,
+		HcaptchaSiteKey: hcaptchaSiteKey,
 	}
 
 	return arguments, nil
-- 
cgit v1.2.3-70-g09d2