query-orchestration/internal/client/create.go

package client

import (
	"context"
	"errors"
	"fmt"
	"regexp"
	"strings"

	"queryorchestration/internal/database/repository"
)

type CreateParams struct {
	Name string
	ID   string
}

func (s *Service) Create(ctx context.Context, params CreateParams) (string, error) {
	err := s.normalizeCreate(&params)
	if err != nil {
		return "", err
	}

	err = s.cfg.GetDBQueries().CreateClient(ctx, &repository.CreateClientParams{
		Name:     params.Name,
		Clientid: params.ID,
	})
	if err != nil {
		return "", err
	}

	return params.ID, nil
}

func (s *Service) normalizeCreate(params *CreateParams) error {
	err := normalizeName(&params.Name)
	if err != nil {
		return err
	}

	err = normalizeID(&params.ID)
	if err != nil {
		return err
	}

	return nil
}

// Make sure the char ~ never makes way in as a valid client id character
// Reference bucket key filename
const CLIENT_ID_REGEX = `[a-zA-Z0-9\_#-]+`

func normalizeID(id *string) error {
	if id == nil {
		return nil
	}

	trim := strings.TrimSpace(*id)
	patternStr := fmt.Sprintf("^%s$", CLIENT_ID_REGEX)
	pattern := regexp.MustCompile(patternStr)
	if !pattern.MatchString(trim) {
		return errors.New("invalid external id")
	}

	*id = trim
	return nil
}