init from gitlab

internal/server/api/setup/setup.go

@@ -0,0 +1,65 @@
+package setup
+
+import (
+	"fmt"
+	"github.com/labstack/echo/v4"
+	"github.com/rs/zerolog/log"
+	"gitlab.com/texm/shokku/internal/env"
+	"gitlab.com/texm/shokku/internal/models"
+	"gitlab.com/texm/shokku/internal/server/auth"
+	"gitlab.com/texm/shokku/internal/server/dto"
+	"net/http"
+)
+
+func GetStatus(e *env.Env, c echo.Context) error {
+	method := string(e.Auth.GetMethod())
+	log.Debug().
+		Bool("is_setup", e.SetupCompleted).
+		Str("method", method).
+		Msg("get setup status")
+	return c.JSON(http.StatusOK, dto.GetSetupStatusResponse{
+		IsSetup: e.SetupCompleted,
+		Method:  method,
+	})
+}
+
+func GetSetupKeyValid(e *env.Env, c echo.Context) error {
+	return c.NoContent(http.StatusOK)
+}
+
+func setupServerWithAuthMethod(e *env.Env, method auth.Method) error {
+	var state models.Server
+	e.DB.FirstOrCreate(&state)
+
+	state.IsSetup = true
+	state.AuthMethod = method
+	if err := e.DB.Save(&state).Error; err != nil {
+		log.Error().Err(err).Msg("failed to save setup state")
+		return echo.ErrInternalServerError
+	}
+
+	newAuth, authErr := createAuthenticator(e, method)
+	if authErr != nil {
+		log.Error().Err(authErr).Msg("failed to init new authenticator")
+		return echo.ErrInternalServerError
+	}
+	e.Auth = newAuth
+	e.SetupCompleted = true
+
+	return nil
+}
+
+func createAuthenticator(e *env.Env, method auth.Method) (auth.Authenticator, error) {
+	config := auth.Config{
+		SigningKey:    e.Auth.GetSigningKey(),
+		CookieDomain:  e.Auth.GetCookieDomain(),
+		TokenLifetime: e.Auth.GetTokenLifetime(),
+	}
+	switch method {
+	case auth.MethodGithub:
+		return auth.NewGithubAuthenticator(config)
+	case auth.MethodPassword:
+		return auth.NewPasswordAuthenticator(config, auth.DefaultBCryptCost)
+	}
+	return nil, fmt.Errorf("unknown method %s", method)
+}