From 5fc793ac211e6324db1ec1fbd0e043fd244fb05d Mon Sep 17 00:00:00 2001
From: VnPower <vnpower@disroot.org>
Date: Thu, 18 May 2023 20:18:44 +0700
Subject: [PATCH] Feature: Added config file

---
 .gitignore         |  1 +
 config.example.yml |  6 ++++++
 configs/config.go  | 29 +++++++++++++++++++++++++++++
 main.go            |  4 ++++
 4 files changed, 40 insertions(+)
 create mode 100644 config.example.yml
 create mode 100644 configs/config.go

diff --git a/.gitignore b/.gitignore
index 2dcffb4..928f824 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 .vercel
 *.map
+config.yml
diff --git a/config.example.yml b/config.example.yml
new file mode 100644
index 0000000..e2fb9aa
--- /dev/null
+++ b/config.example.yml
@@ -0,0 +1,6 @@
+# This is required for the API. See https://github.com/Nandaka/PixivUtil2/wiki#pixiv-login-using-cookie
+# Please keep this carefully, there is a risk of account theft if somebody got your cookie.
+# It is better to use a decoy account, instead of using your main account's cookie.
+PHPSESSID:
+
+userAgent: Mozilla/5.0
diff --git a/configs/config.go b/configs/config.go
new file mode 100644
index 0000000..f436739
--- /dev/null
+++ b/configs/config.go
@@ -0,0 +1,29 @@
+package configs
+
+import (
+	"embed"
+	"gopkg.in/yaml.v3"
+	"os"
+)
+
+var Config embed.FS
+var Configs Conf
+
+type Conf struct {
+	PHPSESSID string `yaml:"PHPSESSID"`
+	userAgent int    `yaml:"userAgent"`
+}
+
+func (conf *Conf) ReadConfig() {
+	f, err := os.Open("config.yml")
+	if err != nil {
+		panic(err)
+	}
+	defer f.Close()
+
+	decoder := yaml.NewDecoder(f)
+	err = decoder.Decode(conf)
+	if err != nil {
+		panic(err)
+	}
+}
diff --git a/main.go b/main.go
index 9aca425..b45b196 100644
--- a/main.go
+++ b/main.go
@@ -2,6 +2,7 @@ package main
 
 import (
 	"html/template"
+	"pixivfe/configs"
 	"pixivfe/views"
 
 	"github.com/gin-gonic/gin"
@@ -31,6 +32,9 @@ func setupRouter() *gin.Engine {
 }
 
 func main() {
+	configs.Configs.ReadConfig()
+	println(configs.Configs.PHPSESSID)
+
 	r := setupRouter()
 
 	r.Run(":8080")