Added configuration
This commit is contained in:
+3
-1
@@ -5,6 +5,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/ESilva15/PTInvCertSWDB/config"
|
||||||
sw "github.com/ESilva15/PTInvCertSWDB/sw"
|
sw "github.com/ESilva15/PTInvCertSWDB/sw"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
@@ -48,6 +49,7 @@ func count(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func RunServer() error {
|
func RunServer() error {
|
||||||
|
cfg := config.GetInstance()
|
||||||
gin.SetMode(ginMode)
|
gin.SetMode(ginMode)
|
||||||
router := gin.Default()
|
router := gin.Default()
|
||||||
|
|
||||||
@@ -64,7 +66,7 @@ func RunServer() error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
err = router.Run(":8086")
|
err = router.Run(":" + cfg.Port)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
"github.com/ESilva15/PTInvCertSWDB/config"
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/golang-jwt/jwt/v4"
|
"github.com/golang-jwt/jwt/v4"
|
||||||
)
|
)
|
||||||
@@ -12,11 +13,11 @@ type TokenData struct {
|
|||||||
Token string `json:"JWTToken"`
|
Token string `json:"JWTToken"`
|
||||||
}
|
}
|
||||||
|
|
||||||
const key = "stronk"
|
|
||||||
|
|
||||||
func validateToken(tStr string) (bool, error) {
|
func validateToken(tStr string) (bool, error) {
|
||||||
|
cfg := config.GetInstance()
|
||||||
|
|
||||||
token, err := jwt.Parse(tStr, func(token *jwt.Token) (any, error) {
|
token, err := jwt.Parse(tStr, func(token *jwt.Token) (any, error) {
|
||||||
return []byte(key), nil
|
return []byte(cfg.AuthKey), nil
|
||||||
}, jwt.WithValidMethods([]string{jwt.SigningMethodHS256.Alg()}))
|
}, jwt.WithValidMethods([]string{jwt.SigningMethodHS256.Alg()}))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return false, err
|
return false, err
|
||||||
|
|||||||
@@ -0,0 +1,5 @@
|
|||||||
|
port: 8086
|
||||||
|
authKey: "stronk"
|
||||||
|
dataBackend: "YAML"
|
||||||
|
yamlBackend:
|
||||||
|
path: "./swdb.yaml"
|
||||||
@@ -0,0 +1,56 @@
|
|||||||
|
// Package config defines the configuration of our app
|
||||||
|
package config
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"gopkg.in/yaml.v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
type YamlBackend struct {
|
||||||
|
Path string `yaml:"path"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Configuration defines the base configuration of the expenses app
|
||||||
|
type Configuration struct {
|
||||||
|
Port string `yaml:"port"`
|
||||||
|
AuthKey string `yaml:"authKey"`
|
||||||
|
DataBackend string `yaml:"dataBackend"`
|
||||||
|
YamlBackend YamlBackend `yaml:"yamlBackend"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
instance *Configuration
|
||||||
|
once sync.Once
|
||||||
|
confPath string
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetInstance returns the instance of the configuration of the app.
|
||||||
|
func GetInstance() *Configuration {
|
||||||
|
once.Do(func() {
|
||||||
|
instance = &Configuration{}
|
||||||
|
instance.loadConfiguration()
|
||||||
|
})
|
||||||
|
|
||||||
|
return instance
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetConfig sets the path for the configuration file.
|
||||||
|
func SetConfig(path string) {
|
||||||
|
confPath = path
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Configuration) loadConfiguration() {
|
||||||
|
file, err := os.ReadFile(confPath)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Unable to open configuration file [%s]: %s", confPath,
|
||||||
|
err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
err = yaml.Unmarshal(file, &instance)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Error parsing JSON: %s", err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user