Add users
This commit is contained in:
86
main.go
86
main.go
@@ -2,7 +2,6 @@ package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"github.com/dgraph-io/ristretto"
|
||||
"github.com/docker/docker/api/types"
|
||||
"github.com/docker/docker/client"
|
||||
@@ -10,12 +9,14 @@ import (
|
||||
"github.com/eko/gocache/store"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/go-co-op/gocron"
|
||||
"github.com/jackc/pgx/v4/pgxpool"
|
||||
"go.uber.org/zap"
|
||||
"go.uber.org/zap/zapcore"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"os"
|
||||
"serverctl/pkg/db"
|
||||
"serverctl/pkg/db/postgres"
|
||||
"serverctl/pkg/users"
|
||||
"time"
|
||||
)
|
||||
|
||||
@@ -46,12 +47,61 @@ func setupLogger() *zap.Logger {
|
||||
defer logger.Sync()
|
||||
return logger
|
||||
}
|
||||
func setupApi(l *zap.Logger, cc *cache.Cache) {
|
||||
func BasicAuthMiddleware(l *zap.Logger, us *users.Service) gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
username, password, hasAuth := c.Request.BasicAuth()
|
||||
if !hasAuth {
|
||||
l.Info("user could not be authenticated",
|
||||
zap.String("username", username))
|
||||
c.Header("WWW-Authenticate", "Basic realm=serverctl")
|
||||
c.Abort()
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"message": "credentials were invalid (authorization header missing)"})
|
||||
return
|
||||
}
|
||||
|
||||
user, err := us.Authenticate(c.Request.Context(), username, password)
|
||||
if err != nil {
|
||||
l.Info("user could not be authenticated",
|
||||
zap.String("username", username))
|
||||
c.Abort()
|
||||
c.Header("WWW-Authenticate", "Basic realm=serverctl")
|
||||
c.JSON(http.StatusUnauthorized, gin.H{"message": "credentials were invalid (credentials didn't match)"})
|
||||
return
|
||||
}
|
||||
|
||||
l.Debug("user has been authenticated",
|
||||
zap.Int("userId", user.Id),
|
||||
zap.String("email", user.Email))
|
||||
c.Set("userId", user.Id)
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
func setupApi(l *zap.Logger, cc *cache.Cache, us *users.Service) {
|
||||
l.Info("Setting up serverctl setupApi (using gin)")
|
||||
|
||||
r := gin.Default()
|
||||
|
||||
r.GET("/containers", func(c *gin.Context) {
|
||||
r.POST("/auth/register", func(c *gin.Context) {
|
||||
type RegisterUser struct {
|
||||
Email string `json:"email" binding:"required"`
|
||||
Password string `json:"password" binding:"required"`
|
||||
}
|
||||
var registerUser RegisterUser
|
||||
if err := c.BindJSON(®isterUser); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
createUser, err := us.Create(registerUser.Email, registerUser.Password)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"message": "you have provided invalid input"})
|
||||
return
|
||||
}
|
||||
|
||||
c.JSON(http.StatusCreated, gin.H{"message": "user has been registered", "userId": createUser})
|
||||
})
|
||||
|
||||
containers := r.Group("/containers", BasicAuthMiddleware(l, us))
|
||||
containers.GET("/", func(c *gin.Context) {
|
||||
type container struct {
|
||||
Name string `json:"name"`
|
||||
}
|
||||
@@ -126,27 +176,6 @@ func setupCron(l *zap.Logger, cm *cache.Cache, cc *client.Client) {
|
||||
|
||||
s.StartAsync()
|
||||
}
|
||||
func setupDatabase(l *zap.Logger) *pgxpool.Pool {
|
||||
l.Info("Setting up database connection")
|
||||
dbUrl := os.Getenv("DATABASE_URL")
|
||||
if dbUrl == "" {
|
||||
panic(errors.New("DATABASE_URL is not set"))
|
||||
}
|
||||
dbpool, err := pgxpool.Connect(context.Background(), dbUrl)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
var greeting string
|
||||
err = dbpool.QueryRow(context.Background(), "select 'Hello, world!'").Scan(&greeting)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
l.Info("Database successfully connected")
|
||||
|
||||
return dbpool
|
||||
}
|
||||
|
||||
func main() {
|
||||
logger := setupLogger()
|
||||
@@ -155,7 +184,10 @@ func main() {
|
||||
cacheM := setupCache(logger)
|
||||
containerClient := setupDocker(logger)
|
||||
setupCron(logger, cacheM, containerClient)
|
||||
dbpool := setupDatabase(logger)
|
||||
|
||||
setupApi(logger, cacheM)
|
||||
database := db.NewClient(logger)
|
||||
usersRepository := postgres.NewUsersRepository(database)
|
||||
usersService := users.NewService(logger, usersRepository, cacheM)
|
||||
|
||||
setupApi(logger, cacheM, usersService)
|
||||
}
|
||||
|
Reference in New Issue
Block a user