From 301113a5606d6d76045f48bf36874e748a9c00b2 Mon Sep 17 00:00:00 2001
From: dalbodeule <11470513+dalbodeule@users.noreply.github.com>
Date: Tue, 14 Oct 2025 05:38:07 +0900
Subject: [PATCH] dotenv configure added.

---
 .gitignore |  4 +++-
 go.mod     |  1 +
 go.sum     |  2 ++
 inc.env    |  1 +
 main.go    | 12 +++++++++++-
 5 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 inc.env

diff --git a/.gitignore b/.gitignore
index 97af23f..6d31474 100644
--- a/.gitignore
+++ b/.gitignore
@@ -240,4 +240,6 @@ $RECYCLE.BIN/
 # Windows shortcuts
 *.lnk
 
-# End of https://www.toptal.com/developers/gitignore/api/go,goland,git,dotenv,database,macos,linux,windows
\ No newline at end of file
+# End of https://www.toptal.com/developers/gitignore/api/go,goland,git,dotenv,database,macos,linux,windows
+
+keys/
\ No newline at end of file
diff --git a/go.mod b/go.mod
index 2ca7531..c129880 100644
--- a/go.mod
+++ b/go.mod
@@ -6,6 +6,7 @@ require github.com/gliderlabs/ssh v0.3.8
 
 require (
 	github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be // indirect
+	github.com/joho/godotenv v1.5.1 // indirect
 	golang.org/x/crypto v0.43.0 // indirect
 	golang.org/x/sys v0.37.0 // indirect
 )
diff --git a/go.sum b/go.sum
index c938159..97ec36b 100644
--- a/go.sum
+++ b/go.sum
@@ -2,6 +2,8 @@ github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be h1:9AeTilPcZAjCFI
 github.com/anmitsu/go-shlex v0.0.0-20200514113438-38f4b401e2be/go.mod h1:ySMOLuWl6zY27l47sB3qLNK6tF2fkHG55UZxx8oIVo4=
 github.com/gliderlabs/ssh v0.3.8 h1:a4YXD1V7xMF9g5nTkdfnja3Sxy1PVDCj1Zg4Wb8vY6c=
 github.com/gliderlabs/ssh v0.3.8/go.mod h1:xYoytBv1sV0aL3CavoDuJIQNURXkkfPA/wxQ1pL1fAU=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 golang.org/x/crypto v0.43.0 h1:dduJYIi3A3KOfdGOHX8AVZ/jGiyPa3IbBozJ5kNuE04=
 golang.org/x/crypto v0.43.0/go.mod h1:BFbav4mRNlXJL4wNeejLpWxB7wMbc79PdRGhWKncxR0=
 golang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ=
diff --git a/inc.env b/inc.env
new file mode 100644
index 0000000..7c98b7c
--- /dev/null
+++ b/inc.env
@@ -0,0 +1 @@
+PORT=2222
\ No newline at end of file
diff --git a/main.go b/main.go
index 52d4b82..d9c923f 100644
--- a/main.go
+++ b/main.go
@@ -3,13 +3,22 @@ package main
 import (
 	"io"
 	"log"
+	"os"
 
 	"sshchat/utils"
 
 	"github.com/gliderlabs/ssh"
+	"github.com/joho/godotenv"
 )
 
 func main() {
+	err := godotenv.Load()
+	if err != nil {
+		log.Fatal("Error loading .env file")
+	}
+
+	port := os.Getenv("PORT")
+
 	keys, err := utils.CheckHostKey()
 	if err != nil {
 		log.Print("Failed to check SSH keys: generate one.\n", err)
@@ -29,12 +38,13 @@ func main() {
 	}
 
 	s := &ssh.Server{
-		Addr:    ":2222",
+		Addr:    ":" + port,
 		Handler: sessionHandler,
 	}
 	for _, key := range keys {
 		s.AddHostKey(key)
 	}
 
+	log.Print("Listening on :" + port)
 	log.Fatal(s.ListenAndServe())
 }