From 84e7a0a230faefd7fb2f3ee607b56a67c5877a1d Mon Sep 17 00:00:00 2001
From: Ulna <30731799+ulnasensei@users.noreply.github.com>
Date: Sat, 2 Sep 2023 14:48:11 +0300
Subject: [PATCH 1/2] changed hardcoded url to use hostname

---
 docs/index.html | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/docs/index.html b/docs/index.html
index 147438f..b29fdec 100644
--- a/docs/index.html
+++ b/docs/index.html
@@ -7,22 +7,26 @@
 	<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/water.css@2/out/water.css">
 	<title>PrivTracker - Private BitTorrent tracker for everyone</title>
 	<script>
+		document.addEventListener("DOMContentLoaded", () => {
+			document.querySelectorAll(".url").forEach(url => {
+				url.innerText = url.innerText.replace('{{hostname}}', window.location.hostname);
+			})
+		})
 		function makeRoom(length) {
-			var roomID = '';
-			var characters = 'abcdefghijklmnopqrstuvwxyz0123456789';
-			var charactersLength = characters.length;
-			for (var i = 0; i < length; i++) {
+			let roomID = '';
+			const characters = 'abcdefghijklmnopqrstuvwxyz0123456789';
+			const charactersLength = characters.length;
+			for (let i = 0; i < length; i++) {
 				roomID += characters.charAt(Math.floor(Math.random() * charactersLength));
 			}
-			var urls = document.getElementsByClassName('url');
-			for (var i = 0; i < urls.length; i++) {
-				urls[i].innerText = urls[i].innerText.replace('{{room}}', roomID);
-			}
+			document.querySelectorAll(".url").forEach(url => {
+				url.innerText = url.innerText.replace('{{room}}', roomID);
+			})
 
-			document.getElementById('hidden').style.display = 'block';
+			document.querySelector('#hidden').style.display = 'block';
 		}
 		function copyToClipboard() {
-			navigator.clipboard.writeText(document.getElementsByClassName('url')[0].innerText);
+			navigator.clipboard.writeText(document.querySelector('.url').innerText);
 		}
 	</script>
 </head>
@@ -44,7 +48,7 @@
 
 		<div id="hidden" style="display:none">
 			<h3>Your announce URL:</h3>
-			<code class="url">https://privtracker.com/{{room}}/announce</code>
+			<code class="url">https://{{hostname}}/{{room}}/announce</code>
 			<button onclick="copyToClipboard()">Copy to clipboard</button>
 			<p>(You can also use any random string as the Room ID. We don't store it anywhere.)</p>
 		</div>
@@ -56,7 +60,7 @@
 		<ul>
 			<li>File <b>&rarr;</b> New&hellip;</li>
 			<li>Select file to share</li>
-			<li>In <b>Trackers</b> field enter <code class="url">https://privtracker.com/{{room}}/announce</code></li>
+			<li>In <b>Trackers</b> field enter <code class="url">https://{{hostname}}/{{room}}/announce</code></li>
 			<li>Select <b>Private torrent</b></li>
 			<li>Click <b>New</b></li>
 			<li>Click <b>Add</b> in next window</li>
@@ -68,7 +72,8 @@
 		<ul>
 			<li>File <b>&rarr;</b> Create Torrent File&hellip;</li>
 			<li>Select file to share</li>
-			<li>Click <b>&plus;</b> under <b>Trackers</b> field and enter <code class="url">https://privtracker.com/{{room}}/announce</code></li>
+			<li>Click <b>&plus;</b> under <b>Trackers</b> field and enter <code
+					class="url">https://{{hostname}}/{{room}}/announce</code></li>
 			<li>Select <b>Private</b>
 			<li>Select <b>Open when created</b></li>
 			<li>Click <b>Create</b></li>
@@ -80,4 +85,4 @@
 	</div>
 </body>
 
-</html>
+</html>
\ No newline at end of file

From 02f320379e8c4ec6be7736903acbf47365498ac4 Mon Sep 17 00:00:00 2001
From: Ulna <30731799+ulnasensei@users.noreply.github.com>
Date: Sat, 2 Sep 2023 15:26:47 +0300
Subject: [PATCH 2/2] changed hardcoded url to use environment variables

---
 README.md | 22 ++++++++++++++++++++++
 main.go   | 40 ++++++++++++++++++++++++++++++++--------
 2 files changed, 54 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index a3557d8..442f044 100644
--- a/README.md
+++ b/README.md
@@ -3,3 +3,25 @@
 PrivTracker allows to share torrent files just with your friends, nobody else.
 Unlike public trackers, it shares peers only within a group which is using the same Announce URL.
 It really works like a private tracker, but can be generated with one click of a button. 
+
+---
+### Build
+```bash
+# Clone this repository.
+$ git clone https://github.com/meehow/privtracker.git
+
+# cd into the directory
+$ cd privtracker
+
+# Run go build
+$ go build
+```
+### Usage
+```bash
+# Runs on port 1337 and redirects to privtracker.com by default.
+$ ./privtracker
+```
+```bash
+# Export PORT and DOMAIN variables to use custom values.
+$ export PORT=12345 DOMAIN=customprivtracker.com; ./privtracker
+```
\ No newline at end of file
diff --git a/main.go b/main.go
index a3ac20e..11876c3 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"crypto/tls"
+	"fmt"
 	"log"
 	"net"
 	"os"
@@ -16,6 +17,8 @@ import (
 	"golang.org/x/crypto/acme/autocert"
 )
 
+var envConfig = getEnvConfig()
+
 func main() {
 	go Cleanup()
 	config := fiber.Config{
@@ -30,6 +33,7 @@ func main() {
 		config.TrustedProxies = []string{"127.0.0.1"}
 		config.ProxyHeader = fiber.HeaderXForwardedFor
 	}
+
 	app := fiber.New(config)
 	app.Use(recover.New())
 	// app.Use(pprof.New())
@@ -50,14 +54,34 @@ func main() {
 		split := strings.Split(domains, ",")
 		log.Fatal(app.Listener(myListener(split...)))
 	} else {
-		port := os.Getenv("PORT")
-		if port == "" {
-			port = "1337"
-		}
-		log.Fatal(app.Listen(":" + port))
+		log.Fatal(app.Listen(":" + envConfig.port))
 	}
 }
 
+type EnvConfig struct {
+	domain string
+	port   string
+}
+
+func getEnvConfig() EnvConfig {
+
+	config := EnvConfig{
+		domain: "privtracker.com",
+		port:   "1337",
+	}
+
+	port := os.Getenv("PORT")
+	domain := os.Getenv("DOMAIN")
+	if domain != "" {
+		config.domain = domain
+	}
+	if port != "" {
+		config.port = port
+	}
+
+	return config
+}
+
 func myListener(domains ...string) net.Listener {
 	homeDir, err := os.UserHomeDir()
 	if err != nil {
@@ -98,7 +122,7 @@ func redirect80(config fiber.Config) {
 	config.DisableStartupMessage = true
 	app := fiber.New(config)
 	app.Use(func(c *fiber.Ctx) error {
-		return c.Redirect("https://privtracker.com/", fiber.StatusMovedPermanently)
+		return c.Redirect(fmt.Sprintf("https://%s/", envConfig.domain), fiber.StatusMovedPermanently)
 	})
 	log.Print(app.Listen(":80"))
 }
@@ -115,8 +139,8 @@ func hsts(c *fiber.Ctx) error {
 }
 
 func docs(c *fiber.Ctx) error {
-	if c.Hostname() != "privtracker.com" {
-		return c.Redirect("https://privtracker.com/", fiber.StatusMovedPermanently)
+	if c.Hostname() != envConfig.domain {
+		return c.Redirect(fmt.Sprintf("https://%s/", envConfig.domain), fiber.StatusMovedPermanently)
 	}
 	return c.Next()
 }