compute-blade-agent/cmd/bladectl/main.go

package main

import (
	"context"
	"log"
	"strings"

	bladeapiv1alpha1 "github.com/compute-blade-community/compute-blade-agent/api/bladeapi/v1alpha1"
	"github.com/spf13/viper"
)

type grpcClientContextKey int

const (
	defaultGrpcClientContextKey  grpcClientContextKey = 0
	defaultGrpcClientsContextKey grpcClientContextKey = 1
)

var (
	Version string
	Commit  string
	Date    string
)

func clientIntoContext(ctx context.Context, client bladeapiv1alpha1.BladeAgentServiceClient) context.Context {
	return context.WithValue(ctx, defaultGrpcClientContextKey, client)
}

func clientsIntoContext(ctx context.Context, clients []bladeapiv1alpha1.BladeAgentServiceClient) context.Context {
	return context.WithValue(ctx, defaultGrpcClientsContextKey, clients)
}

func clientFromContext(ctx context.Context) bladeapiv1alpha1.BladeAgentServiceClient {
	client, ok := ctx.Value(defaultGrpcClientContextKey).(bladeapiv1alpha1.BladeAgentServiceClient)
	if !ok {
		panic("grpc client not found in context")
	}
	return client
}

func clientsFromContext(ctx context.Context) []bladeapiv1alpha1.BladeAgentServiceClient {
	clients, ok := ctx.Value(defaultGrpcClientsContextKey).([]bladeapiv1alpha1.BladeAgentServiceClient)
	if !ok {
		panic("grpc client not found in context")
	}
	return clients
}

func main() {
	// Setup configuration
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
	viper.AutomaticEnv()
	viper.SetConfigName("config")
	viper.SetConfigType("yaml")
	viper.AddConfigPath("$HOME/.config/bladectl")

	if err := rootCmd.Execute(); err != nil {
		log.Fatal(err)
	}
}