cloudfoundry/cf-k8s-controllers

package manifest

import (
    "math/rand"
    "strings"

    "code.cloudfoundry.org/korifi/api/payloads"
    korifiv1alpha1 "code.cloudfoundry.org/korifi/controllers/api/v1alpha1"
)

type Normalizer struct {
    defaultDomainName string
}

func NewNormalizer(defaultDomainName string) Normalizer {
    return Normalizer{
        defaultDomainName: defaultDomainName,
    }
}

func (n Normalizer) Normalize(appInfo payloads.ManifestApplication, appState AppState) payloads.ManifestApplication {
    fixDeprecatedFields(&appInfo)
    processes := n.normalizeProcesses(appInfo, appState)
    routes := n.normalizeRoutes(appInfo, appState)

    return payloads.ManifestApplication{
        Name:       appInfo.Name,
        Env:        appInfo.Env,
        Buildpacks: appInfo.Buildpacks,
        Processes:  processes,
        Routes:     routes,
        NoRoute:    appInfo.NoRoute,
        Metadata:   appInfo.Metadata,
        Services:   appInfo.Services,
        Docker:     appInfo.Docker,
    }
}

func procValIfSet[T any](appVal, procVal *T) *T {
    if procVal == nil {
        return appVal
    }
    return procVal
}

func fixDeprecatedFields(appInfo *payloads.ManifestApplication) {
    if appInfo.DiskQuota == nil {
        //lint:ignore SA1019 we have to deal with this deprecation
        appInfo.DiskQuota = appInfo.AltDiskQuota
    }

    for i := range appInfo.Processes {
        if appInfo.Processes[i].DiskQuota == nil {
            //lint:ignore SA1019 we have to deal with this deprecation
            appInfo.Processes[i].DiskQuota = appInfo.Processes[i].AltDiskQuota
        }
    }

    //lint:ignore SA1019 we have to deal with this deprecation
    if hasBuildpackSet(appInfo.Buildpack) {
        //lint:ignore SA1019 we have to deal with this deprecation
        appInfo.Buildpacks = append(appInfo.Buildpacks, *appInfo.Buildpack)
    }
}

func hasBuildpackSet(buildpack *string) bool {
    return buildpack != nil && *buildpack != "" && *buildpack != "default" && *buildpack != "null"
}

func (n Normalizer) normalizeProcesses(appInfo payloads.ManifestApplication, appState AppState) []payloads.ManifestApplicationProcess {
    processes := appInfo.Processes

    var webProc *payloads.ManifestApplicationProcess
    for i, p := range processes {
        if p.Type == korifiv1alpha1.ProcessTypeWeb {
            webProc = &processes[i]
            break
        }
    }
    if webProc == nil {
        processes = append(processes, payloads.ManifestApplicationProcess{Type: korifiv1alpha1.ProcessTypeWeb})
        webProc = &processes[len(processes)-1]
    }

    if appInfo.Memory != nil || appInfo.DiskQuota != nil || appInfo.Instances != nil || appInfo.Command != nil ||
        appInfo.HealthCheckHTTPEndpoint != nil || appInfo.HealthCheckType != nil || appInfo.HealthCheckInvocationTimeout != nil || appInfo.Timeout != nil {

        webProc.Memory = procValIfSet(appInfo.Memory, webProc.Memory)
        webProc.DiskQuota = procValIfSet(appInfo.DiskQuota, webProc.DiskQuota)
        webProc.Instances = procValIfSet(appInfo.Instances, webProc.Instances)
        webProc.Command = procValIfSet(appInfo.Command, webProc.Command)
        webProc.HealthCheckHTTPEndpoint = procValIfSet(appInfo.HealthCheckHTTPEndpoint, webProc.HealthCheckHTTPEndpoint)
        webProc.HealthCheckType = procValIfSet(appInfo.HealthCheckType, webProc.HealthCheckType)
        webProc.HealthCheckInvocationTimeout = procValIfSet(appInfo.HealthCheckInvocationTimeout, webProc.HealthCheckInvocationTimeout)
        webProc.Timeout = procValIfSet(appInfo.Timeout, webProc.Timeout)
    }

    return processes
}

func (n Normalizer) normalizeRoutes(appInfo payloads.ManifestApplication, appState AppState) []payloads.ManifestRoute {
    if appInfo.NoRoute {
        return nil
    }

    routes := appInfo.Routes

    needsAutomaticRoutes := len(routes) == 0 && len(appState.Routes) == 0
    if needsAutomaticRoutes {
        if appInfo.DefaultRoute {
            routes = append(routes, n.configureDefaultRoute(appInfo.Name))
        }

        if appInfo.RandomRoute {
            routes = append(routes, n.configureRandomRoute(appInfo.Name))
        }
    }

    return routes
}

func (n Normalizer) configureDefaultRoute(appName string) payloads.ManifestRoute {
    defaultRouteString := appName + "." + n.defaultDomainName
    return payloads.ManifestRoute{
        Route: &defaultRouteString,
    }
}

func (n Normalizer) configureRandomRoute(appName string) payloads.ManifestRoute {
    randomHostname := appName + "-" + generateRandomRoute()
    routeString := randomHostname + "." + n.defaultDomainName
    return payloads.ManifestRoute{
        Route: &routeString,
    }
}

func generateRandomRoute() string {
    suffix := string('a'+rune(rand.Intn(26))) + string('a'+rune(rand.Intn(26)))
    return adjectives[rand.Intn(len(adjectives))] + "-" + nouns[rand.Intn(len(nouns))] + "-" + suffix
}

var adjectives = strings.Split("accountable,active,agile,anxious,appreciative,balanced,bogus,boisterous,bold,boring,brash,brave,bright,busy,chatty,cheerful,chipper,comedic,courteous,daring,delightful,egregious,empathic,excellent,exhausted,fantastic,fearless,fluent,forgiving,friendly,funny,generous,grateful,grouchy,grumpy,happy,hilarious,humble,impressive,insightful,intelligent,industrious,interested,kind,lean,mediating,meditating,nice,noisy,optimistic,palm,patient,persistent,proud,quick,quiet,reflective,relaxed,reliable,resplendent,responsible,responsive,rested,restless,shiny,shy,silly,sleepy,smart,spontaneous,stellar,surprised,sweet,talkative,terrific,thankful,timely,tired,triumphant,turbulent,unexpected,wacky,wise,zany", ",")

var nouns = strings.Split("aardvark,alligator,antelope,armadillo,baboon,badger,bandicoot,bat,bear,bilby,bongo,buffalo,bushbuck,camel,capybara,cassowary,cat,cheetah,chimpanzee,chipmunk,civet,crane,crocodile,dingo,dog,dugong,duiker,echidna,eland,elephant,emu,fossa,fox,gazelle,gecko,gelada,genet,gerenuk,giraffe,gnu,gorilla,grysbok,guanaco,hartebeest,hedgehog,hippopotamus,hyena,hyrax,impala,jackal,jaguar,kangaroo,klipspringer,koala,kob,kookaburra,kudu,lemur,leopard,lion,lizard,llama,lynx,manatee,mandrill,marmot,meerkat,mongoose,mouse,numbat,nyala,okapi,oribi,oryx,ostrich,otter,panda,pangolin,panther,parrot,platypus,porcupine,possum,puku,quokka,quoll,rabbit,ratel,raven,reedbuck,rhinocerous,roan,sable,serval,shark,sitatunga,springhare,squirrel,swan,tiger,topi,toucan,turtle,vicuna,wallaby,warthog,waterbuck,whale,wildebeest,wolf,wolverine,wombat,zebra", ",")