package replay import ( "encoding/json" "fmt" "os" "time" "go.digitalcircle.com.br/open/replaycli-go/api" "go.digitalcircle.com.br/open/replaycli-go/types" "go.digitalcircle.com.br/open/replaycli-go/util" ) type Cli struct { *api.ApiCli } func (c *Cli) Log(l *types.Log) error { l.Alias = c.ReplayEnvAlias() l.Dtlogend = time.Now() return c.HttpCli().JsonPost("/api/v1/log/add", l, nil) } func (c *Cli) NewLog() *types.Log { ret := &types.Log{} ret.Alias = c.ReplayEnvAlias() ret.Dtlog = time.Now() return ret } func (c *Cli) ReplayEnvQueueId() string { return os.Getenv("REPLAY_QUEUEID") } func (c *Cli) ReplayEnvAlias() string { return os.Getenv("REPLAY_ALIAS") } func (c *Cli) ReplayEnvFeatureDir() string { return os.Getenv("REPLAY_FEATUREDIR") } func (c *Cli) ReplayEnvRoot() string { return os.Getenv("REPLAY_ROOT") } func (c *Cli) ReplayEnvAddr() string { return os.Getenv("REPLAY_ADDR") } func (c *Cli) ReplayEnvHomeDir() string { return os.Getenv("REPLAY_HOMEDIR") } func (c *Cli) ReplayEnvRepo() string { return os.Getenv("REPLAY_REPO") } func (c *Cli) ReplayEnvVer() string { return os.Getenv("REPLAY_VER") } func (c *Cli) ReplayEnvRepoDir() string { return os.Getenv("REPLAY_REPODIR") } func (c *Cli) ReplayEnvDataDir() string { return os.Getenv("REPLAY_DATADIR") } func (c *Cli) ReplayEnvInstanceAlias() string { return os.Getenv("REPLAY_INSTANCE_ALIAS") } func (c *Cli) ReplayEnvApiKey() string { return os.Getenv("REPLAY_APIKEY") } func (c *Cli) OpenApp(id uint) error { err := c.HttpCli().JsonGet(fmt.Sprintf("/api/v1/app/run/%d", id), nil) return err } func (c *Cli) ConfigGet(k string) (ret string, err error) { err = c.HttpCli().JsonGet("/api/v1/config/byrepo/"+util.Repo()+"/"+k, &ret) return } func (c *Cli) ConfigGetAll() (ret []types.Config, err error) { ret = make([]types.Config, 0) err = c.HttpCli().JsonGet("/api/v1/config/byrepo/"+util.Repo(), &ret) return } func (c *Cli) ConfigGetMap() (ret map[string]types.Config, err error) { cfgs, err := c.ConfigGetAll() if err != nil { return } ret = make(map[string]types.Config) for _, v := range cfgs { ret[v.K] = v } return } func (c *Cli) CronReload() error { return c.HttpCli().JsonGet("/ipc/cron/reload", nil) } func (c *Cli) Exit() error { err := c.HttpCli().JsonGet("/api/v1/exit", nil) return err } func (c *Cli) MenuGetAllEnabled() (ret []types.Menu, err error) { ret = make([]types.Menu, 0) err = c.HttpCli().JsonGet("/api/v1/menu/enabled", nil) return ret, err } func (c *Cli) QueueAdd(job string, bs []byte) error { _, err := c.HttpCli().RawPost("/api/v1/queue/add/"+job, bs) return err } func (c *Cli) QueueGetData(id string) (ret []byte, err error) { ret, err = c.HttpCli().RawGet("/api/v1/queue/getrequest/" + id) return } func (c *Cli) QueueGetMyData() ([]byte, error) { return c.QueueGetData(os.Getenv("REPLAY_QUEUEID")) } func (c *Cli) QueueGetMyJson(i interface{}) error { bs, err := c.QueueGetData(os.Getenv("REPLAY_QUEUEID")) if err != nil { return err } err = json.Unmarshal(bs, i) return err } func (c *Cli) QueueAbort() error { return c.HttpCli().JsonGet("/api/v1/queue/abort", nil) } func (c *Cli) QueueEnqueue(id uint) error { return c.HttpCli().JsonGet(fmt.Sprintf("/api/v1/robots/op/enqueue/%d", id), nil) } func (c *Cli) ServiceStopAll() error { return c.HttpCli().JsonGet("/api/v1/service/op/stopall", nil) } type SQLReturn struct { Data []map[string]interface{} `json:"data"` Err string `json:"err"` } func (c *Cli) SQL(s string) (*SQLReturn, error) { in := &struct { Sql string `json:"sql"` }{s} out := &SQLReturn{} err := c.HttpCli().JsonPost("/api/v1/sql", in, &out) return out, err } func NewCli() *Cli { ret := &Cli{ApiCli: api.NewApiCli()} return ret }