| 380 | } |
| 381 | |
| 382 | func (w *cmdWrapper) wrap(cmd *cobra.Command) { |
| 383 | if fn := cmd.PersistentPreRunE; fn != nil { |
| 384 | cmd.PersistentPreRunE = w.runFunc(fn) |
| 385 | } |
| 386 | if fn := cmd.PreRunE; fn != nil { |
| 387 | cmd.PreRunE = w.runFunc(fn) |
| 388 | } |
| 389 | if fn := cmd.RunE; fn != nil { |
| 390 | cmd.RunE = w.runFunc(fn) |
| 391 | } |
| 392 | if fn := cmd.PostRunE; fn != nil { |
| 393 | cmd.PostRunE = w.runFunc(fn) |
| 394 | } |
| 395 | if fn := cmd.PersistentPostRunE; fn != nil { |
| 396 | cmd.PersistentPostRunE = w.runFunc(fn) |
| 397 | } |
| 398 | } |
| 399 | |
| 400 | func (w *cmdWrapper) runFunc(fn func(cmd *cobra.Command, args []string) error) func(cmd *cobra.Command, args []string) error { |
| 401 | return func(cmd *cobra.Command, args []string) error { |