You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
96 lines
2.1 KiB
96 lines
2.1 KiB
package auth
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"git.slaventius.ru/test3k/auth/internal/config"
|
|
db "git.slaventius.ru/test3k/auth/internal/transport/grpc"
|
|
|
|
mux "github.com/go-chi/chi/v5"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// ...
|
|
type AuthServer struct {
|
|
db *db.AuthDBClient
|
|
Router *mux.Mux
|
|
config *config.Config
|
|
ctx context.Context
|
|
}
|
|
|
|
func NewServer(ctx context.Context, config *config.Config) *AuthServer {
|
|
s := &AuthServer{
|
|
db: db.NewDBClient(ctx, config),
|
|
Router: mux.NewMux(),
|
|
config: config,
|
|
ctx: ctx,
|
|
}
|
|
|
|
//
|
|
s.Router.Get("/api/v1/ready", ready(s))
|
|
s.Router.Post("/api/v1/login", login(s))
|
|
s.Router.Post("/api/v1/registration", registration(s))
|
|
s.Router.Post("/api/v1/confirmation", confirmation(s))
|
|
|
|
return s
|
|
}
|
|
|
|
func (s *AuthServer) GracefulStop() error {
|
|
return s.db.Close()
|
|
}
|
|
|
|
func ready(s *AuthServer) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("ok"))
|
|
}
|
|
}
|
|
|
|
func login(s *AuthServer) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
uid := r.FormValue("uid")
|
|
password := r.FormValue("password")
|
|
err := s.db.Login(uid, password)
|
|
if err != nil {
|
|
status := status.Convert(err)
|
|
w.Write([]byte(status.Message()))
|
|
|
|
return
|
|
}
|
|
|
|
w.Write([]byte(fmt.Sprintf("Success login for %s", uid)))
|
|
}
|
|
}
|
|
|
|
func registration(s *AuthServer) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
uid := r.FormValue("uid")
|
|
email := r.FormValue("email")
|
|
password := r.FormValue("password")
|
|
_, err := s.db.Registration(uid, email, password)
|
|
if err != nil {
|
|
status := status.Convert(err)
|
|
w.Write([]byte(status.Message()))
|
|
|
|
return
|
|
}
|
|
|
|
w.Write([]byte(fmt.Sprintf("Success registration for %s", uid)))
|
|
}
|
|
}
|
|
|
|
func confirmation(s *AuthServer) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
code := r.FormValue("code")
|
|
err := s.db.Confirmation(code)
|
|
if err != nil {
|
|
status := status.Convert(err)
|
|
w.Write([]byte(status.Message()))
|
|
|
|
return
|
|
}
|
|
|
|
w.Write([]byte(fmt.Sprintf("Success confirmation for %s", code)))
|
|
}
|
|
}
|
|
|