domains.go

/*

Copyright 2020 The Vouch Proxy Authors.
Use of this source code is governed by The MIT License (MIT) that
can be found in the LICENSE file. Software distributed under The
MIT License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
OR CONDITIONS OF ANY KIND, either express or implied.

*/

package domains

import (
	"sort"
	"strings"

	"github.com/vouch/vouch-proxy/pkg/cfg"
	"go.uber.org/zap"
)

var log *zap.SugaredLogger

// Configure see main.go configure()
func Configure() {
	log = cfg.Logging.Logger
	sort.Sort(ByLengthDesc(cfg.Cfg.Domains))
}

// Matches returns one of the domains we're configured for
func Matches(s string) string {
	if strings.Contains(s, ":") {
		// then we have a port and we just want to check the host
		split := strings.Split(s, ":")
		log.Debugf("removing port from %s to test domain %s", s, split[0])
		s = split[0]
	}

	if len(cfg.Cfg.Domains) > 0 {
		for i, v := range cfg.Cfg.Domains {
			if s == v || strings.HasSuffix(s, "."+v) {
				log.Debugf("domain %s matched array value at [%d]=%v", s, i, v)
				return v
			}
		}
		log.Warnf("domain %s not found in any domains %v", s, cfg.Cfg.Domains)
	}
	return ""
}

// IsUnderManagement check if an email is under vouch-managed domain
func IsUnderManagement(email string) bool {
	split := strings.Split(email, "@")
	if len(split) != 2 {
		log.Warnf("not a valid email: %s", email)
		return false
	}

	match := Matches(split[1])
	if match != "" {
		return true
	}
	return false
}

// ByLengthDesc sort from
// https://play.golang.org/p/N6GbEgBffd
type ByLengthDesc []string

func (s ByLengthDesc) Len() int {
	return len(s)
}
func (s ByLengthDesc) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}

// this differs by offing the longest first
func (s ByLengthDesc) Less(i, j int) bool {
	return len(s[j]) < len(s[i])
}