From 1fcb5ee549fa7ba4b7bfa7e62c15dbb8a01c38b5 Mon Sep 17 00:00:00 2001
From: Mitchell Riedstra <mitch@riedstra.dev>
Date: Sat, 3 Apr 2021 13:09:45 -0400
Subject: Some changes to make supporting a demo environment easier. Docker
 compose demo.

---
 cmd/web/util.go | 34 ++++++++++++++++++++++++++++++++--
 1 file changed, 32 insertions(+), 2 deletions(-)

(limited to 'cmd/web/util.go')

diff --git a/cmd/web/util.go b/cmd/web/util.go
index 2c32922..454912f 100644
--- a/cmd/web/util.go
+++ b/cmd/web/util.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"io"
 	"net"
 	"net/http"
@@ -22,9 +23,38 @@ func UnauthorizedIfNotLocal(h http.Handler) http.Handler {
 	})
 }
 
+var isLocalCIDR = "127.0.0.1/8"
+
 func isLocal(addr string) bool {
-	_, localNet, _ := net.ParseCIDR("127.0.0.1/8")
-	return localNet.Contains(net.ParseIP(strings.Split(addr, ":")[0]))
+	if strings.Contains(isLocalCIDR, ",") {
+		for _, cidr := range strings.Split(isLocalCIDR, ",") {
+			_, localNet, err := net.ParseCIDR(cidr)
+			if err != nil {
+				panic(err)
+			}
+			if localNet.Contains(net.ParseIP(strings.Split(addr, ":")[0])) {
+				return true
+			}
+		}
+
+		return false
+	} else {
+		_, localNet, err := net.ParseCIDR(isLocalCIDR)
+		if err != nil {
+			panic(err)
+		}
+		return localNet.Contains(net.ParseIP(strings.Split(addr, ":")[0]))
+	}
+}
+
+var shareLink = ""
+
+func getShareLink() string {
+	if shareLink != "" {
+		return shareLink
+	}
+
+	return fmt.Sprintf("http://%s:%s/", GetHostIP(), getPort())
 }
 
 // GetHostIP attempts to guess the IP address of the current machine and
-- 
cgit v1.2.3