mirror of
https://github.com/kaaninchen/Laterna.git
synced 2026-09-17 19:12:48 +00:00
QoL: Revamp project structure
This commit is contained in:
@@ -0,0 +1,37 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
type HTTPConfig struct {
|
||||
AdminToken string `json:"adminToken"`
|
||||
Port int `json:"port"`
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
HTTP HTTPConfig `json:"http"`
|
||||
FileLogging bool `json:"fileLogging"`
|
||||
VerboseLogging bool `json:"verboseLogging"`
|
||||
}
|
||||
|
||||
func LoadConfig(path string) *Config {
|
||||
file, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
fmt.Println("An error occured while reading the config file:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
var cfg Config
|
||||
err = json.Unmarshal(file, &cfg)
|
||||
if err != nil {
|
||||
fmt.Println("An error occured while parsing the config file:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
return &cfg
|
||||
}
|
||||
|
||||
var AppConfig *Config
|
||||
@@ -0,0 +1,17 @@
|
||||
package db
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/mizuw/laterna/server/cmd/internal/logger"
|
||||
)
|
||||
|
||||
func ConnectDB() {
|
||||
db, err := sql.Open("sqlite3", "./db.sql")
|
||||
if err != nil {
|
||||
logger.DBLogger.Printf("An Error occured while connecting to the database: %v", err)
|
||||
}
|
||||
logger.DBLogger.Printf("Connected")
|
||||
defer db.Close()
|
||||
}
|
||||
@@ -0,0 +1 @@
|
||||
package db
|
||||
@@ -0,0 +1,35 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
|
||||
"github.com/mizuw/laterna/server/cmd/internal/config"
|
||||
"github.com/mizuw/laterna/server/cmd/internal/logger"
|
||||
)
|
||||
|
||||
func StartHTTPServer() {
|
||||
router := http.NewServeMux()
|
||||
|
||||
router.HandleFunc("GET /api/v1/admin/token/new", createToken)
|
||||
|
||||
port := config.AppConfig.HTTP.Port
|
||||
logger.HTTPLogger.Printf("HTTP Server running on :%v", port)
|
||||
http.ListenAndServe(fmt.Sprintf(":%d", port), router)
|
||||
}
|
||||
|
||||
func createToken(w http.ResponseWriter, r *http.Request) {
|
||||
authHeader := r.Header.Get("Authorization")
|
||||
adminToken := config.AppConfig.HTTP.AdminToken
|
||||
|
||||
if authHeader == "" {
|
||||
http.Error(w, "Authorization header missing", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
if authHeader != adminToken {
|
||||
http.Error(w, "nuh uh", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
print(authHeader)
|
||||
fmt.Fprintf(w, "Success!!! Token: %v, authHeader: %v", adminToken, authHeader)
|
||||
}
|
||||
@@ -0,0 +1,44 @@
|
||||
package logger
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/mizuw/laterna/server/cmd/internal/config"
|
||||
)
|
||||
|
||||
var (
|
||||
HTTPLogger *log.Logger
|
||||
DBLogger *log.Logger
|
||||
flags int
|
||||
)
|
||||
|
||||
func InitLoggers() {
|
||||
var logOutput io.Writer = os.Stdout
|
||||
|
||||
if config.AppConfig.FileLogging {
|
||||
|
||||
os.Mkdir("logs", os.ModePerm)
|
||||
|
||||
timestamp := time.Now().Format("2006-01-02_15-04-05")
|
||||
filename := fmt.Sprintf("logs/log_%s.txt", timestamp)
|
||||
|
||||
file, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0666)
|
||||
if err != nil {
|
||||
log.Fatal("An error occured while loading logs.txt")
|
||||
}
|
||||
|
||||
logOutput = io.MultiWriter(os.Stdout, file)
|
||||
log.SetOutput(logOutput)
|
||||
}
|
||||
|
||||
if config.AppConfig.VerboseLogging {
|
||||
flags = log.Ldate | log.Ltime | log.Lshortfile
|
||||
}
|
||||
|
||||
HTTPLogger = log.New(logOutput, "[HTTP] ", flags)
|
||||
DBLogger = log.New(logOutput, "[DATABASE] ", flags)
|
||||
}
|
||||
Reference in New Issue
Block a user