auth/internal/auth.go

package auth

import (
	"context"
	"fmt"
	"net/http"

	"git.slaventius.ru/test3k/auth/internal/config"
	db "git.slaventius.ru/test3k/auth/internal/transport/grpc"
	prometheus "git.slaventius.ru/test3k/auth/internal/transport/prometheus"
	"git.slaventius.ru/test3k/umate/pkg/logger"

	mux "github.com/go-chi/chi/v5"
	"google.golang.org/grpc/status"
)

// ...
type AuthServer struct {
	db         *db.AuthDBClient
	prometheus *prometheus.Prometheus
	Router     *mux.Mux
	config     *config.Config
	logger     *logger.Logger
	ctx        context.Context
}

func NewServer(ctx context.Context, config *config.Config) *AuthServer {
	arouter := mux.NewMux()
	alogger := logger.NewLogger("test3k:authService", config.Sentry.DSN)
	server := &AuthServer{
		db:         db.NewDBClient(ctx, config),
		prometheus: prometheus.NewService(arouter, "/metrics"),
		Router:     arouter,
		config:     config,
		logger:     alogger,
		ctx:        ctx,
	}

	//
	server.Router.Get("/api/v1/healthz", healthz(server))

	//
	server.Router.Post("/api/v1/login", login(server))
	server.Router.Post("/api/v1/registration", registration(server))
	server.Router.Post("/api/v1/confirmation", confirmation(server))

	return server
}

func (s *AuthServer) GracefulStop() error {
	return s.db.Close()
}

func healthz(s *AuthServer) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		// err := s.db.Check()
		// if err != nil {
		// 	status := status.Convert(err)

		// 	//
		// 	s.logger.Error(status.Message())

		// 	//
		// 	w.Write([]byte(status.Message()))

		// 	return
		// }

		w.WriteHeader(http.StatusOK)
		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)

			//
			s.logger.Error(status.Message())

			//
			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)

			//
			s.logger.Error(status.Message())

			//
			w.Write([]byte(status.Message()))

			return
		}

		// Увеличим счетчик заругистрировашихся пользователей
		s.prometheus.IncUsersRegistered()

		//
		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)

			//
			s.logger.Error(status.Message())

			//
			w.Write([]byte(status.Message()))

			return
		}

		// Увеличим счетчик пользователей подтвердивших регистрацию
		s.prometheus.IncUsersConfirmed()

		//
		w.Write([]byte(fmt.Sprintf("Success confirmation for %s", code)))
	}
}