ValidateUtil.go

/*
 * Copyright (c) 2020 Devtron Labs
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

package util

import (
	"gopkg.in/go-playground/validator.v9"
	"regexp"
	"strings"
)

func ValidateName(fl validator.FieldLevel) bool {
	hostnameRegexString := `^[a-z]+[a-z0-9\-\?]*[a-z0-9]+$`
	hostnameRegexRFC952 := regexp.MustCompile(hostnameRegexString)
	return hostnameRegexRFC952.MatchString(fl.Field().String())
}

func ValidateCheckoutPath(fl validator.FieldLevel) bool {
	checkoutPath := fl.Field().String()
	if checkoutPath != "" && (!strings.HasPrefix(checkoutPath, "./")) {
		return false
	}
	return true
}

func IntValidator() (*validator.Validate, error) {
	v := validator.New()
	err := v.RegisterValidation("name-component", ValidateName)
	if err != nil {
		return v, err
	}
	err = v.RegisterValidation("checkout-path-component", ValidateCheckoutPath)
	return v, err
}