ingresshostgen.go

package ingresshostgen

import (
	"fmt"
	"strings"

	"github.com/rancher/rancher/pkg/settings"
	v1beta12 "github.com/rancher/types/apis/extensions/v1beta1"
	"github.com/rancher/types/config"
	"k8s.io/api/extensions/v1beta1"
)

type IngressHostGen struct {
	ingress v1beta12.IngressInterface
}

func Register(userOnlyContext *config.UserOnlyContext) {
	c := &IngressHostGen{
		ingress: userOnlyContext.Extensions.Ingresses(""),
	}
	userOnlyContext.Extensions.Ingresses("").AddHandler("ingress-host-gen", c.sync)
}

func isGeneratedDomain(obj *v1beta1.Ingress, host, domain string) bool {
	parts := strings.Split(host, ".")
	return strings.HasSuffix(host, "."+domain) && len(parts) == 8 && parts[1] == obj.Namespace
}

func (i *IngressHostGen) sync(key string, obj *v1beta1.Ingress) error {
	if obj == nil {
		return nil
	}

	ipDomain := settings.IngressIPDomain.Get()
	if ipDomain == "" {
		return nil
	}

	var xipHost string
	for _, status := range obj.Status.LoadBalancer.Ingress {
		if status.IP != "" {
			xipHost = fmt.Sprintf("%s.%s.%s.%s", obj.Name, obj.Namespace, status.IP, ipDomain)
			break
		}
	}

	if xipHost == "" {
		return nil
	}

	changed := false
	for _, rule := range obj.Spec.Rules {
		if (isGeneratedDomain(obj, rule.Host, ipDomain) || rule.Host == ipDomain) && rule.Host != xipHost {
			changed = true
			break
		}
	}

	if !changed {
		return nil
	}

	obj = obj.DeepCopy()
	for i, rule := range obj.Spec.Rules {
		if strings.HasSuffix(rule.Host, ipDomain) {
			obj.Spec.Rules[i].Host = xipHost
		}
	}

	_, err := i.ingress.Update(obj)
	return err
}