Function wrap

internal/command/inject.go:40–56

wrap wraps a function with middleware using a new context. It returns a cli.ActionFunc with the cli.Context added to the Context.

(fn func(context.Context) error, middleware ...func(context.Context) (context.Context, error))

// wrap wraps a function with middleware using a new context. It returns a cli.ActionFunc with
// the cli.Context added to the Context.
func wrap(fn func(context.Context) error, middleware ...func(context.Context) (context.Context, error)) cli.ActionFunc {
	return func(clictx *cli.Context) error {
		ctx := context.Background()

		// apply middleware to the new context
		for _, fn := range middleware {
			var err error
			if ctx, err = fn(ctx); err != nil {
				return err
			}
		}

		ctx = withCLIContext(ctx, clictx)

		return fn(ctx)
	}
}