Evan Chen 2021-11-12 23:09:32 +08:00
parent 56841cc8be
commit 06b6b6ed9b
9 changed files with 140 additions and 51 deletions

View File

@ -36,7 +36,7 @@ func init() {
flag.StringVar(&flagAction, "c", "", "cmd to apply") flag.StringVar(&flagAction, "c", "", "cmd to apply")
flag.StringVar(&flagLogFile, "log", "", "log to file") flag.StringVar(&flagLogFile, "log", "", "log to file")
flag.StringVar(&flagAllow, "allow", "", "IPs to allow, blank to allow all") flag.StringVar(&flagAllow, "allow", "", "IPs to allow, blank to allow all")
flag.StringVar(&flagBind, "bind", "0.0.0.0:8000", "address to bind") flag.StringVar(&flagBind, "bind", "127.0.0.1:8000", "address to bind")
flag.BoolVar(&flagNoReconfig, "static", false, "disable config api") flag.BoolVar(&flagNoReconfig, "static", false, "disable config api")
flag.BoolVar(&flagVer, "v", false, "show version") flag.BoolVar(&flagVer, "v", false, "show version")
flag.Usage = func() { flag.Usage = func() {

View File

@ -35,13 +35,12 @@ var Ext2Mode map[string]string = map[string]string{
} }
type Action struct { type Action struct {
Name string `json:"name"` Name string `json:"name"`
Cmd string `json:"cmd"` Cmd string `json:"cmd"`
Dir string `json:"dir"` Dir string `json:"dir"`
Args []string `json:"args"`
Envs []string `json:"envs"`
run chan struct{} `json:"-"` run chan struct{} `json:"-"`
pid int
} }
type Integration struct { type Integration struct {
@ -75,7 +74,8 @@ type ConfigUI struct {
LogLevel klog.Llevel `json:"log_level"` LogLevel klog.Llevel `json:"log_level"`
// Running commands // Running commands
Onitachi map[string]*Oni `json:"-"` Onitachi []*Oni `json:"-"`
OniIndex map[string]int
TmplFS embed.FS `json:"-"` TmplFS embed.FS `json:"-"`
tmpl *engine.Engine tmpl *engine.Engine
@ -98,6 +98,30 @@ func New() *ConfigUI {
"normal": func(name string) string { "normal": func(name string) string {
return strings.ReplaceAll(name, " ", "-") return strings.ReplaceAll(name, " ", "-")
}, },
"state_class": func(state State) string {
switch state {
case STARTED:
return "is-success"
case ERROR:
return "is-danger"
case ENDED:
return "has-background-grey-light"
default:
return ""
}
},
"state_icon": func(state State) string {
switch state {
case STARTED:
return "play_arrow"
case ERROR:
return "error"
case ENDED:
return "stop"
default:
return "pending"
}
},
}).ParseFS(tmplFS, "templates/*.tmpl", "templates/**/*.tmpl")) }).ParseFS(tmplFS, "templates/*.tmpl", "templates/**/*.tmpl"))
return &ConfigUI{ return &ConfigUI{
fileIndex: map[string]int{}, fileIndex: map[string]int{},
@ -112,7 +136,7 @@ func New() *ConfigUI {
cmdTimeout: time.Second * 10, cmdTimeout: time.Second * 10,
LogLevel: klog.Lerror | klog.Linfo, LogLevel: klog.Lerror | klog.Linfo,
log: klog.Sub("ConfigUI"), log: klog.Sub("ConfigUI"),
Onitachi: make(map[string]*Oni), OniIndex: make(map[string]int),
} }
} }

View File

@ -13,9 +13,9 @@ var ErrorOniHasNoPID = ksrv.Error{
} }
var ErrorOniNotStarted = ksrv.Error{ var ErrorOniNotStarted = ksrv.Error{
Code: http.StatusConflict, Code: http.StatusConflict,
ID: "ErrorOniNotStarted", ID: "ErrorOniNotStarted",
Message: "oni hasn't start", Tmpl: "%s hasn't start",
} }
var ErrorOniHasStarted = ksrv.Error{ var ErrorOniHasStarted = ksrv.Error{
@ -25,9 +25,9 @@ var ErrorOniHasStarted = ksrv.Error{
} }
var ErrorOniNotFound = ksrv.Error{ var ErrorOniNotFound = ksrv.Error{
Code: http.StatusNotFound, Code: http.StatusNotFound,
ID: "ErrorOniNotFound", ID: "ErrorOniNotFound",
Message: "oni not found", Tmpl: "%s not found",
} }
var ErrorOniNotValid = ksrv.Error{ var ErrorOniNotValid = ksrv.Error{

View File

@ -58,7 +58,7 @@ func (f *File) Write(data []byte) error {
return os.WriteFile(f.Path, data, info.Mode()) return os.WriteFile(f.Path, data, info.Mode())
} }
func (f *File) Do(CmdTimeout time.Duration) (string, error) { func (f *File) Do(CmdTimeout time.Duration, report chan int) (string, error) {
if f.Cmd == "" { if f.Cmd == "" {
return "", nil return "", nil
} }
@ -96,6 +96,7 @@ func (f *File) Do(CmdTimeout time.Duration) (string, error) {
} }
go func() { go func() {
f.pid = cmd.Process.Pid f.pid = cmd.Process.Pid
report <- cmd.Process.Pid
cmd.Wait() cmd.Wait()
done <- b.String() done <- b.String()
}() }()

View File

@ -121,7 +121,7 @@ func (cui *ConfigUI) Apply(w http.ResponseWriter, r *http.Request) {
ksrv.Response(w, 404, []byte("file not found")) ksrv.Response(w, 404, []byte("file not found"))
return return
} }
result, err := file.Do(cui.cmdTimeout) result, err := file.Do(cui.cmdTimeout, make(chan int))
if err != nil { if err != nil {
panic(err) panic(err)
} }
@ -141,23 +141,20 @@ func (cui *ConfigUI) DoAction(w http.ResponseWriter, r *http.Request) {
case cui.Actions[i].run <- struct{}{}: case cui.Actions[i].run <- struct{}{}:
defer func() { <-cui.Actions[i].run }() defer func() { <-cui.Actions[i].run }()
default: default:
panic(fmt.Errorf("another task of %s is running with", name)) panic(fmt.Errorf("another task of %s is running with pid: %d", name, v.pid))
} }
oni, ok := cui.Onitachi[name] file := &File{Name: name, Cmd: v.Cmd, owner: cui}
if !ok { pid := make(chan int)
oni = NewOni(name, cui) go func() {
oni.Cmd = cui.Actions[i].Cmd cui.Actions[i].pid = <-pid
oni.Args = cui.Actions[i].Args }()
oni.Envs = cui.Actions[i].Envs result, err := file.Do(cui.cmdTimeout, pid)
oni.Dir = cui.Actions[i].Dir
cui.Onitachi[name] = oni
}
err := oni.Start()
if err != nil { if err != nil {
panic(err) panic(err)
} }
w.Write([]byte("ok")) w.Write([]byte(result))
return return
} }
} }
@ -266,3 +263,37 @@ func (cui *ConfigUI) GetConfig(w http.ResponseWriter, r *http.Request) {
func (cui *ConfigUI) GetRunning(w http.ResponseWriter, r *http.Request) { func (cui *ConfigUI) GetRunning(w http.ResponseWriter, r *http.Request) {
ksrv.JSON(w, cui.Onitachi) ksrv.JSON(w, cui.Onitachi)
} }
func (cui *ConfigUI) OniStart(w http.ResponseWriter, r *http.Request) {
name := r.URL.Query().Get("name")
i, ok := cui.OniIndex[name]
if !ok {
panic(ErrorOniNotFound.New(name))
}
oni := cui.Onitachi[i]
switch oni.State {
case "":
oni.Init(cui)
case STARTED:
panic(ErrorOniHasStarted)
}
oni.Start()
w.Write([]byte("ok"))
}
func (cui *ConfigUI) OniStop(w http.ResponseWriter, r *http.Request) {
name := r.URL.Query().Get("name")
i, ok := cui.OniIndex[name]
if !ok {
panic(ErrorOniNotFound.New(name))
}
oni := cui.Onitachi[i]
switch oni.State {
case STARTED:
oni.Stop()
default:
panic(ErrorOniNotStarted.New(name))
}
oni.Start()
w.Write([]byte("ok"))
}

View File

@ -42,33 +42,30 @@ type Oni struct {
Error string `json:"error"` Error string `json:"error"`
ManStop bool `json:"manual_stop"` ManStop bool `json:"manual_stop"`
parent *ConfigUI parent *ConfigUI
cmd *exec.Cmd cmd *exec.Cmd
start chan struct{} running chan struct{}
buff bytes.Buffer buff bytes.Buffer
log io.Writer log io.Writer
listeners []io.Writer listeners []io.Writer
} }
func NewOni(name string, cui *ConfigUI) *Oni { func (oni *Oni) Init(cui *ConfigUI) error {
var err error var err error
oni := &Oni{ oni.parent = cui
Policy: NO, oni.running = make(chan struct{}, 1)
State: READY, oni.listeners = make([]io.Writer, 0)
parent: cui, oni.StateChange(READY)
start: make(chan struct{}, 1),
listeners: make([]io.Writer, 0),
}
if cui.LogPath != "" { if cui.LogPath != "" {
logpath := filepath.Join(filepath.Dir(cui.LogPath), name+".log") logpath := filepath.Join(filepath.Dir(cui.LogPath), oni.Name+".log")
oni.LogFile = logpath oni.LogFile = logpath
oni.log, err = os.OpenFile(logpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666) oni.log, err = os.OpenFile(logpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil { if err != nil {
panic(err) return err
} }
} }
return oni return err
} }
func (oni *Oni) Start() error { func (oni *Oni) Start() error {
@ -76,7 +73,7 @@ func (oni *Oni) Start() error {
return ErrorOniNotValid return ErrorOniNotValid
} }
select { select {
case oni.start <- struct{}{}: case oni.running <- struct{}{}:
defer func() { defer func() {
err := recover() err := recover()
if err != nil { if err != nil {
@ -105,7 +102,7 @@ func (oni *Oni) Start() error {
} }
go oni.read() go oni.read()
oni.PID = cmd.Process.Pid oni.PID = cmd.Process.Pid
oni.State = STARTED oni.StateChange(STARTED)
go func() { go func() {
oni.end(cmd.Wait()) oni.end(cmd.Wait())
}() }()
@ -122,13 +119,13 @@ func (oni *Oni) Stop() error {
func (oni *Oni) end(v interface{}) { func (oni *Oni) end(v interface{}) {
if v == nil { if v == nil {
oni.State = ENDED oni.StateChange(ENDED)
} else { } else {
oni.Error = fmt.Sprint(v) oni.Error = fmt.Sprint(v)
oni.State = ERROR oni.StateChange(ERROR)
} }
if len(oni.start) == 1 { if len(oni.running) == 1 {
<-oni.start <-oni.running
} }
} }
@ -144,3 +141,12 @@ func (oni *Oni) read() {
} }
}() }()
} }
func (oni *Oni) StateChange(state State) {
oni.State = state
// for i := range oni.parent.Actions {
// if oni.parent.Actions[i].Name == oni.Name {
// oni.parent.Actions[i].State = state
// }
// }
}

7
templates/action.tmpl Normal file
View File

@ -0,0 +1,7 @@
{{define "action"}}
{{template "base/header" .}}
{{template "components/error" .}}
{{template "base/footer" .}}
{{end}}

View File

@ -14,6 +14,7 @@
</div> </div>
<div id="navbarBasicExample" class="navbar-menu"> <div id="navbarBasicExample" class="navbar-menu">
{{/*
<div class="navbar-start"> <div class="navbar-start">
{{if .Actions}} {{if .Actions}}
<a class="navbar-item"> <a class="navbar-item">
@ -27,7 +28,6 @@
{{end}} {{end}}
</div> </div>
{{/*
<div class="navbar-end"> <div class="navbar-end">
<div class="navbar-item"> <div class="navbar-item">
<div class="buttons"> <div class="buttons">
@ -44,6 +44,7 @@
</div> </div>
</nav> </nav>
{{/* not having pages
<script> <script>
document.addEventListener('DOMContentLoaded', () => { document.addEventListener('DOMContentLoaded', () => {
@ -70,5 +71,6 @@ document.addEventListener('DOMContentLoaded', () => {
} }
}); });
*/}}
</script> </script>
{{end}} {{end}}

View File

@ -14,6 +14,24 @@
</li> </li>
{{ end }} {{ end }}
</ul> </ul>
<!-- {{if .Actions}}
<p class="menu-label">
Actions
</p>
<ul class="menu-list">
{{- range .Actions -}}
<li>
<a class="button has-tooltip-arrow {{.State|state_class}}" id="actbtn-{{.Name|normal}}"
data-tooltip="{{.Cmd}}" onclick="toolDoAction('{{.Name}}')">
<span>{{- .Name -}}</span>
<span class="icon is-small"><span class="material-icons">
{{.State|state_icon}}
</span></span>
</a>
</li>
{{- end -}}
</ul>
{{end}} -->
{{if not .HideConfig}} {{if not .HideConfig}}
<p class="menu-label"> <p class="menu-label">
System System