main.go

package main

import (
	"errors"
	"log"
	"net/http"

	"github.com/tentsk8s/k8s-claimer/handlers"
	"github.com/tentsk8s/k8s-claimer/htp"

	// "github.com/tentsk8s/k8s-claimer/providers/azure"
	"github.com/tentsk8s/k8s-claimer/providers/gke"
)

const (
	appName      = "k8s-claimer"
	authTokenKey = "Authorization"
)

var (
	errNilConfig = errors.New("nil config")
)

// func kubeNamespacesFromConfig() func(*k8s.KubeConfig) (k8s.NamespaceListerDeleter, error) {
// 	return func(conf *k8s.KubeConfig) (k8s.NamespaceListerDeleter, error) {
// 		if conf == nil {
// 			return nil, errNilConfig
// 		}
// 		cl, err := k8s.CreateKubeClientFromConfig(conf)
// 		if err != nil {
// 			return nil, err
// 		}
// 		return cl.Namespaces(), nil
// 	}
// }

func configureRoutesWithAuth(
	serveMux *http.ServeMux,
	createLeaseHandler http.Handler,
	deleteLeaseHandler http.Handler,
	authToken string,
) {
	createLeaseHandler = htp.MethodMux(
		map[htp.Method]http.Handler{htp.Post: createLeaseHandler},
	)
	deleteLeaseHandler = htp.MethodMux(
		map[htp.Method]http.Handler{htp.Delete: deleteLeaseHandler},
	)

	serveMux.Handle("/lease", handlers.WithAuth(authToken, authTokenKey, createLeaseHandler))
	serveMux.Handle("/lease/", handlers.WithAuth(authToken, authTokenKey, deleteLeaseHandler))
}

//CreateHealthzHandler returns an http.Handler
func CreateHealthzHandler() http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
	})
}

func main() {
	leaseStorage := leases.NewStorage()
	serverConf, err := parseServerConfig(appName)
	if err != nil {
		log.Fatalf("Error getting server config (%s)", err)
	}
	serverConf.Print()
	googleConfig, err := parseGoogleConfig(appName)
	if err != nil {
		log.Fatalf("Error getting google cloud config (%s) -- %+v", err, googleConfig)
	}

	azureConfig, err := parseAzureConfig(appName)
	if err != nil {
		log.Fatalf("Error getting azure config (%s) -- %+v", err, azureConfig)
	}

	containerService, err := gke.GetContainerService(
		googleConfig.AccountFile.ClientEmail,
		gke.PrivateKey(googleConfig.AccountFile.PrivateKey),
	)
	if err != nil {
		log.Fatalf("Error creating GKE client (%s)", err)
	}
	gkeClusterLister := gke.NewGKEClusterLister(containerService)
	azureClusterLister := azure.NewAzureClusterLister(azureConfig)

	mux := http.NewServeMux()
	createLeaseHandler := handlers.CreateLease(
		// services,
		serverConf.ServiceName,
		gkeClusterLister,
		azureClusterLister,
		azureConfig,
		googleConfig,
	)
	deleteLeaseHandler := handlers.DeleteLease(
		// services,
		serverConf.ServiceName,
		gkeClusterLister,
		azureClusterLister,
		azureConfig,
		googleConfig,
		serverConf.ClearNamespaces,
		// kubeNamespacesFromConfig(),
	)

	mux.Handle("/healthz", CreateHealthzHandler())

	configureRoutesWithAuth(mux, createLeaseHandler, deleteLeaseHandler, serverConf.AuthToken)

	log.Println("k8s claimer started!")
	http.ListenAndServe(serverConf.HostStr(), mux)
}