59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package context
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
type key string
|
|
|
|
const (
|
|
keyCliCtx = key("cmd_cli")
|
|
)
|
|
|
|
// Context fx context
|
|
type Context struct {
|
|
context.Context
|
|
}
|
|
|
|
// NewContext new a context
|
|
func NewContext() *Context {
|
|
ctx := context.Background()
|
|
return &Context{ctx}
|
|
}
|
|
|
|
// FromCliContext create context from cli.Context
|
|
func FromCliContext(c *cli.Context) *Context {
|
|
ctx := NewContext()
|
|
ctx.WithCliContext(c)
|
|
return ctx
|
|
}
|
|
|
|
// WithCliContext set cli.Context
|
|
func (ctx *Context) WithCliContext(c *cli.Context) {
|
|
newCtx := context.WithValue(ctx.Context, keyCliCtx, c)
|
|
ctx.Context = newCtx
|
|
}
|
|
|
|
// GetCliContext get cli.Context
|
|
func (ctx *Context) GetCliContext() *cli.Context {
|
|
return ctx.Value(keyCliCtx).(*cli.Context)
|
|
}
|
|
|
|
// Set a value with name
|
|
func (ctx *Context) Set(name string, value interface{}) {
|
|
newCtx := context.WithValue(ctx.Context, name, value)
|
|
ctx.Context = newCtx
|
|
}
|
|
|
|
// Get a value
|
|
func (ctx *Context) Get(name string) interface{} {
|
|
return ctx.Context.Value(name)
|
|
}
|
|
|
|
// Use invole a middle
|
|
func (ctx *Context) Use(fn func(ctx *Context) error) error {
|
|
return fn(ctx)
|
|
}
|