added support to list like headers ie: x-forwarded-for

CONTRIBUTING.md

@@ -0,0 +1,12 @@
+# How to contribute
+
+Thank you for your interest in improving droneip.
+
+Here is how to contribute back some code or documentation:
+
+- Fork repo
+- Create a feature branch off of the `dev` branch
+- Make some useful change
+- Submit a pull request against the `dev` branch.
+- Be kind
+

handler.go

@@ -30,7 +30,7 @@ func (h *DroneHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 	inspectHeader := config.Get("INSPECT_HEADER")
 	if inspectHeader != "" {
-		remoteIP = r.Header.Get(inspectHeader)
+		remoteIP = getRemoteIP(r.Header.Get(inspectHeader))
 	}
 	cacheKey := fmt.Sprintf("droneip-%s", remoteIP)
 
@@ -79,3 +79,13 @@ func (h *DroneHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	io.Copy(w, res.Body)
 
 }
+
+func getRemoteIP(ips string) string {
+	if !strings.Contains(ips, ",") {
+		return ips
+	}
+
+	ipList := strings.Split(ips, ",")
+
+	return strings.Trim(ipList[0], " ")
+}

logger/README.md

@@ -1,9 +1,8 @@
 [![Go Report Card](https://goreportcard.com/badge/github.com/diegohce/logger)](https://goreportcard.com/report/github.com/diegohce/logger)
 [![GoDoc](https://godoc.org/github.com/diegohce/logger?status.svg)](https://godoc.org/github.com/diegohce/logger)
-[![GPLv3 license](https://img.shields.io/badge/License-GPLv3-blue.svg)](https://github.com/diegohce/logger/blob/master/LICENSE)
+[![GPLv3 license](https://img.shields.io/badge/License-apache_2.0-blue.svg)](https://github.com/diegohce/logger/blob/master/LICENSE)
 [![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://github.com/diegohce/logger/graphs/commit-activity)
-[![HitCount](http://hits.dwyl.io/diegohce/logger.svg)](http://hits.dwyl.io/diegohce/logger)
-[![Sourcegraph](https://sourcegraph.com/github.com/diegohce/logger/-/badge.svg)](https://sourcegraph.com/github.com/diegohce/logger?badge)
+[![Sourcegraph](https://sourcegraph.com/github.com/diegohce/droneip/logger/-/badge.svg)](https://sourcegraph.com/github.com/diegohce/droneip/logger?badge)
 
 # logger
 

service_test.go

@@ -93,3 +93,29 @@ func TestAdmin(t *testing.T) {
 	}
 
 }
+
+func TestGetRemoteIP(t *testing.T) {
+
+	cases := []struct {
+		ips  string
+		want string
+	}{
+		{
+			ips:  "10.10.0.1, 10.10.0.2, 10.10.0.3, 10.10.0.4, 10.10.0.5, 10.10.0.6",
+			want: "10.10.0.1",
+		},
+		{
+			ips:  "10.10.0.1",
+			want: "10.10.0.1",
+		},
+	}
+
+	for _, c := range cases {
+		got := getRemoteIP(c.ips)
+		if got != c.want {
+			t.Errorf("got %s want %s", got, c.want)
+		}
+
+	}
+
+}