From 4d3a84b6b360752a304f7d88de62f8bd041a4981 Mon Sep 17 00:00:00 2001
From: cdramey <noreply@binarythought.com>
Date: Sun, 18 Nov 2018 01:27:51 +0000
Subject: [PATCH] Added IP and Browser string to submit routines

---
 pages/submit.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/pages/submit.go b/pages/submit.go
index 61359e8..c918b2d 100644
--- a/pages/submit.go
+++ b/pages/submit.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"html/template"
 	"net/http"
+	"net"
 	"qurl/qurl"
 	"qurl/static"
 	"qurl/storage"
@@ -53,6 +54,12 @@ func (ctx *SubmitHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 				URL:     u,
 				Created: time.Now(),
 			}
+			if h, _, err := net.SplitHostPort(r.RemoteAddr); err == nil && h != "" {
+				q.IP = net.ParseIP(h)
+			}
+			if b := r.Header.Get("User-Agent"); b != "" {
+				q.Browser = b
+			}
 
 			err := q.CheckValid()
 			if err != nil {