160 lines
3.8 KiB
Go
160 lines
3.8 KiB
Go
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) ConfigGetGlobal(k string) (ret string, err error) {
|
|
err = c.HttpCli().JsonGet("/api/v1/config/global/"+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
|
|
}
|