initial server implementation
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
DBDriver string `mapstructure:"db_driver"`
|
||||
DBConnString string `mapstructure:"db_conn_string"`
|
||||
JWTSecret string `mapstructure:"jwt_secret"`
|
||||
SessionSecret string `mapstructure:"session_secret"`
|
||||
ServerPort int `mapstructure:"server_port"`
|
||||
}
|
||||
|
||||
func Load(path string) (*Config, error) {
|
||||
v := viper.New()
|
||||
v.SetConfigFile(path)
|
||||
|
||||
v.SetDefault("db_driver", "sqlite")
|
||||
v.SetDefault("db_conn_string", "file::memory:?cache=shared")
|
||||
v.SetDefault("server_port", 8080)
|
||||
|
||||
if err := v.ReadInConfig(); err != nil {
|
||||
return nil, fmt.Errorf("read config: %w", err)
|
||||
}
|
||||
|
||||
var cfg Config
|
||||
if err := v.Unmarshal(&cfg); err != nil {
|
||||
return nil, fmt.Errorf("unmarshal config: %w", err)
|
||||
}
|
||||
|
||||
return &cfg, nil
|
||||
}
|
||||
@@ -1,4 +1,4 @@
|
||||
package db
|
||||
package database
|
||||
|
||||
import (
|
||||
"time"
|
||||
@@ -0,0 +1,51 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func Migrate(db *gorm.DB) error {
|
||||
if err := db.AutoMigrate(&User{}, &Role{}); err != nil {
|
||||
return fmt.Errorf("auto-migrate: %w", err)
|
||||
}
|
||||
|
||||
defaultRoles := []Role{
|
||||
{Name: "superadmin"},
|
||||
{Name: "admin"},
|
||||
{Name: "editor"},
|
||||
{Name: "author"},
|
||||
{Name: "viewer"},
|
||||
}
|
||||
|
||||
for _, r := range defaultRoles {
|
||||
if err := db.FirstOrCreate(&r, Role{Name: r.Name}).Error; err != nil {
|
||||
return fmt.Errorf("seed role %q: %w", r.Name, err)
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewDB(driver, dsn string) (*gorm.DB, error) {
|
||||
var dialector gorm.Dialector
|
||||
|
||||
switch driver {
|
||||
case "sqlite":
|
||||
dialector = sqlite.Open(dsn)
|
||||
case "postgres":
|
||||
dialector = postgres.Open(dsn)
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported database driver: %s", driver)
|
||||
}
|
||||
|
||||
db, err := gorm.Open(dialector, &gorm.Config{})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to open database: %w", err)
|
||||
}
|
||||
|
||||
return db, nil
|
||||
}
|
||||
@@ -1,4 +1,4 @@
|
||||
package db
|
||||
package database
|
||||
|
||||
type UserStatus uint8
|
||||
|
||||
Reference in New Issue
Block a user