From 2d5156278efa5f5c499132fc6fc16daec4cc86a2 Mon Sep 17 00:00:00 2001
From: kandrusyak <kandrusyak@inno.tech>
Date: Fri, 30 Jun 2023 17:13:00 +0300
Subject: [PATCH] add config

---
 .gitignore       |  3 ++-
 config/config.go | 31 +++++++++++++++++++++++++++++++
 main.go          |  3 ++-
 3 files changed, 35 insertions(+), 2 deletions(-)
 create mode 100644 config/config.go

diff --git a/.gitignore b/.gitignore
index 723ef36..3ec3f8d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
-.idea
\ No newline at end of file
+.idea
+public
\ No newline at end of file
diff --git a/config/config.go b/config/config.go
new file mode 100644
index 0000000..3e94962
--- /dev/null
+++ b/config/config.go
@@ -0,0 +1,31 @@
+package config
+
+import (
+	"os"
+)
+
+type ServerConfig struct {
+	Port string
+}
+
+type Config struct {
+	Server ServerConfig
+}
+
+func New() *Config {
+	return &Config{
+		Server: ServerConfig{
+			Port: getEnv("PORT", ""),
+		},
+	}
+}
+
+func getEnv(key string, defaultVal string) string {
+	if value, exists := os.LookupEnv(key); exists {
+		return value
+	}
+
+	return defaultVal
+}
+
+var Env = New()
diff --git a/main.go b/main.go
index 1b9bd12..a4e7b07 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"astra-file-storage/config"
 	uuid2 "github.com/google/uuid"
 	"io"
 	"net/http"
@@ -53,5 +54,5 @@ func main() {
 	e.Static("/", "public")
 	e.POST("/upload", upload)
 
-	e.Logger.Fatal(e.Start(":8080"))
+	e.Logger.Fatal(e.Start(":" + config.Env.Server.Port))
 }