plugin.go

package plugin

import (
	"github.com/cloudnative-pg/cnpg-i-machinery/pkg/pluginhelper/http"
	"github.com/cloudnative-pg/cnpg-i/pkg/lifecycle"
	"github.com/cloudnative-pg/cnpg-i/pkg/operator"
	"github.com/spf13/cobra"
	"google.golang.org/grpc"

	"github.com/cloudnative-pg/cnpg-i-hello-world/internal/identity"
	lifecycleImpl "github.com/cloudnative-pg/cnpg-i-hello-world/internal/lifecycle"
	operatorImpl "github.com/cloudnative-pg/cnpg-i-hello-world/internal/operator"
)

// NewCmd creates the `plugin` command
func NewCmd() *cobra.Command {
	cmd := http.CreateMainCmd(identity.Implementation{}, func(server *grpc.Server) error {
		// Register the declared implementations
		operator.RegisterOperatorServer(server, operatorImpl.Implementation{})
		lifecycle.RegisterOperatorLifecycleServer(server, lifecycleImpl.Implementation{})
		return nil
	})

	// If you want to provide your own logr.Logger here, inject it into a context.Context
	// with logr.NewContext(ctx, logger) and pass it to cmd.SetContext(ctx)

	// Additional custom behaviour can be added by wrapping cmd.PersistentPreRun or cmd.Run

	cmd.Use = "plugin"

	return cmd
}