package api import ( "fmt" "io" "log" "net/http" "net/url" "text/template" "github.com/ProjectSegfault/segfautils/config" "github.com/ProjectSegfault/segfautils/utils" "github.com/kataras/hcaptcha" "github.com/containrrr/shoutrrr" ) var ( siteKey = config.HCaptchaSiteKey() secretKey = config.HCaptchaSecretKey() webhookURL = config.ShoutrrrURL() client = hcaptcha.New(secretKey) /* See `Client.FailureHandler` too. */ resForm = config.OptForm() ) func FormCheck() { if resForm == "false" { log.Println("[Segfautils] ℹ Contact form is disabled") http.HandleFunc("/form", func(w http.ResponseWriter, r *http.Request) { http.Error(w, "Form is disabled.", http.StatusServiceUnavailable) }) http.HandleFunc("/api/set/form", func(w http.ResponseWriter, r *http.Request) { http.Error(w, "{\"enabled\": \"false\"}", http.StatusOK) }) } else { FormPage() http.HandleFunc("/api/set/form", func(w http.ResponseWriter, r *http.Request) { http.Error(w, "{\"enabled\": \"true\"}", http.StatusOK) }) Form() } } func FormPage() { type StaticThing struct { HCaptchaSiteKey string } tmpl_form := template.Must(template.ParseFiles("static/form.html")) http.HandleFunc("/form/", func(w http.ResponseWriter, r *http.Request) { hcaptcha_site_key := config.HCaptchaSiteKey() data := StaticThing{ HCaptchaSiteKey: hcaptcha_site_key, } tmpl_form.Execute(w, data) }) } func Form() { http.HandleFunc("/api/form", client.HandlerFunc(theActualFormCode)) } func theActualFormCode(w http.ResponseWriter, r *http.Request) { switch r.Method { case http.MethodPost: hcaptchaResp, ok := hcaptcha.Get(r) if !ok { w.WriteHeader(http.StatusUnauthorized) fmt.Fprint(w, "Seems like captcha failed, you didn't complete the captcha or you are a bot. Please try again.\nPlease note that your IP has been logged in our systems for manual review to check if you're an abusive user. If you're seen as abusive, you will be blacklisted.\nYour message has not been sent.") postData := url.Values{ "content": {"IP " + utils.GetUserIP(r) + "failed captcha!\nhttps://abuseipdb.com/check/" + utils.GetUserIP(r)}, } req, err := http.PostForm(webhookURL, postData) if err != nil { log.Fatal("Something went terribly wrong!", err) } fmt.Fprint(io.Discard, req) // I don't want the result of the request in stdout } else { fmt.Fprintf(w, "Thanks for your message, and thanks for doing the captcha!\nPlease ignore how different this page looks to the page you were on earlier. I'll figure it out eventually!\n%#+v", hcaptchaResp) postData := url.Values{ "content": {"IP " + utils.GetUserIP(r) + "\nFrom " + r.FormValue("email") + " with feedback type " + r.FormValue("commentType") + ":\n" + "**" + r.FormValue("message") + "**\n https://abuseipdb.com/check/" + utils.GetUserIP(r)}, } shoutrrr := shoutrrr.Send(config.ShoutrrrURL(), "IP "+utils.GetUserIP(r)+"\nFrom "+r.FormValue("email")+" with feedback type "+r.FormValue("commentType")+":\n"+"**"+r.FormValue("message")+"**\n https://abuseipdb.com/check/"+utils.GetUserIP(r)) if shoutrrr != nil { log.Fatal("Something went terribly wrong!", shoutrrr) } req, err := http.PostForm(webhookURL, postData) if err != nil { log.Fatal("Something went terribly wrong!", err) } fmt.Fprint(io.Discard, req) // Out with your request! I don't want it. } default: http.Error(w, "Method isn't allowed!\nYou may only POST here, not "+r.Method, http.StatusMethodNotAllowed) } log.Println("[HTTP] " + utils.GetUserIP(r) + " accessed /api/form with method " + r.Method) }