History_Api/pkg/database/seed.go

package database

import (
	"context"
	"database/sql"
	"errors"
	"history-api/internal/gen/sqlc"
	"history-api/pkg/config"
	"history-api/pkg/constants"

	"github.com/jackc/pgx/v5/pgtype"
	"github.com/jackc/pgx/v5/pgxpool"
	"golang.org/x/crypto/bcrypt"
)

func SeedSuperAdmin(pool *pgxpool.Pool) error {
	ctx := context.Background()

	displayName, err := config.GetConfig("ADMIN_DISPLAY_NAME")
	if err != nil {
		return err
	}

	email, err := config.GetConfig("ADMIN_EMAIL")
	if err != nil {
		return err
	}

	password, err := config.GetConfig("ADMIN_PASSWORD")
	if err != nil {
		return err
	}

	q := sqlc.New(pool)

	_, err = q.GetUserByEmail(ctx, email)
	if err == nil {
		return nil
	}

	if !errors.Is(err, sql.ErrNoRows) {
		return err
	}

	hashed, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
	if err != nil {
		return err
	}

	user, err := q.UpsertUser(ctx, sqlc.UpsertUserParams{
		Email: email,
		PasswordHash: pgtype.Text{
			String: string(hashed),
			Valid:  len(hashed) != 0,
		},
		AuthProvider: constants.LocalProvider.String(),
	})
	if err != nil {
		return err
	}

	_, err = q.CreateUserProfile(ctx, sqlc.CreateUserProfileParams{
		UserID: user.ID,
		DisplayName: pgtype.Text{
			String: displayName,
			Valid:  displayName != "",
		},
	})
	if err != nil {
		return err
	}

	role, err := q.GetRoleByName(ctx, constants.ADMIN.String())
	if err != nil {
		return err
	}

	err = q.AddUserRole(
		ctx,
		sqlc.AddUserRoleParams{
			UserID:  user.ID,
			Column2: []pgtype.UUID{role.ID},
		},
	)
	if err != nil {
		return err
	}

	return nil

}