Compare commits
	
		
			1 Commits 
		
	
	
		
			2ecac19807
			...
			06b6b6ed9b
		
	
	| Author | SHA1 | Date | 
|---|---|---|
|  | 06b6b6ed9b | 
|  | @ -36,7 +36,7 @@ func init() { | |||
| 	flag.StringVar(&flagAction, "c", "", "cmd to apply") | ||||
| 	flag.StringVar(&flagLogFile, "log", "", "log to file") | ||||
| 	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(&flagVer, "v", false, "show version") | ||||
| 	flag.Usage = func() { | ||||
|  |  | |||
							
								
								
									
										32
									
								
								configui.go
								
								
								
								
							
							
						
						
									
										32
									
								
								configui.go
								
								
								
								
							|  | @ -38,10 +38,9 @@ type Action struct { | |||
| 	Name string `json:"name"` | ||||
| 	Cmd  string `json:"cmd"` | ||||
| 	Dir  string `json:"dir"` | ||||
| 	Args []string `json:"args"` | ||||
| 	Envs []string `json:"envs"` | ||||
| 
 | ||||
| 	run chan struct{} `json:"-"` | ||||
| 	pid int | ||||
| } | ||||
| 
 | ||||
| type Integration struct { | ||||
|  | @ -75,7 +74,8 @@ type ConfigUI struct { | |||
| 	LogLevel klog.Llevel `json:"log_level"` | ||||
| 
 | ||||
| 	// Running commands
 | ||||
| 	Onitachi map[string]*Oni `json:"-"` | ||||
| 	Onitachi []*Oni `json:"-"` | ||||
| 	OniIndex map[string]int | ||||
| 
 | ||||
| 	TmplFS     embed.FS `json:"-"` | ||||
| 	tmpl       *engine.Engine | ||||
|  | @ -98,6 +98,30 @@ func New() *ConfigUI { | |||
| 		"normal": func(name string) string { | ||||
| 			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")) | ||||
| 	return &ConfigUI{ | ||||
| 		fileIndex:  map[string]int{}, | ||||
|  | @ -112,7 +136,7 @@ func New() *ConfigUI { | |||
| 		cmdTimeout: time.Second * 10, | ||||
| 		LogLevel:   klog.Lerror | klog.Linfo, | ||||
| 		log:        klog.Sub("ConfigUI"), | ||||
| 		Onitachi:   make(map[string]*Oni), | ||||
| 		OniIndex:   make(map[string]int), | ||||
| 	} | ||||
| } | ||||
| 
 | ||||
|  |  | |||
|  | @ -15,7 +15,7 @@ var ErrorOniHasNoPID = ksrv.Error{ | |||
| var ErrorOniNotStarted = ksrv.Error{ | ||||
| 	Code: http.StatusConflict, | ||||
| 	ID:   "ErrorOniNotStarted", | ||||
| 	Message: "oni hasn't start", | ||||
| 	Tmpl: "%s hasn't start", | ||||
| } | ||||
| 
 | ||||
| var ErrorOniHasStarted = ksrv.Error{ | ||||
|  | @ -27,7 +27,7 @@ var ErrorOniHasStarted = ksrv.Error{ | |||
| var ErrorOniNotFound = ksrv.Error{ | ||||
| 	Code: http.StatusNotFound, | ||||
| 	ID:   "ErrorOniNotFound", | ||||
| 	Message: "oni not found", | ||||
| 	Tmpl: "%s not found", | ||||
| } | ||||
| 
 | ||||
| var ErrorOniNotValid = ksrv.Error{ | ||||
|  |  | |||
							
								
								
									
										3
									
								
								file.go
								
								
								
								
							
							
						
						
									
										3
									
								
								file.go
								
								
								
								
							|  | @ -58,7 +58,7 @@ func (f *File) Write(data []byte) error { | |||
| 	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 == "" { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | @ -96,6 +96,7 @@ func (f *File) Do(CmdTimeout time.Duration) (string, error) { | |||
| 	} | ||||
| 	go func() { | ||||
| 		f.pid = cmd.Process.Pid | ||||
| 		report <- cmd.Process.Pid | ||||
| 		cmd.Wait() | ||||
| 		done <- b.String() | ||||
| 	}() | ||||
|  |  | |||
							
								
								
									
										57
									
								
								handler.go
								
								
								
								
							
							
						
						
									
										57
									
								
								handler.go
								
								
								
								
							|  | @ -121,7 +121,7 @@ func (cui *ConfigUI) Apply(w http.ResponseWriter, r *http.Request) { | |||
| 		ksrv.Response(w, 404, []byte("file not found")) | ||||
| 		return | ||||
| 	} | ||||
| 	result, err := file.Do(cui.cmdTimeout) | ||||
| 	result, err := file.Do(cui.cmdTimeout, make(chan int)) | ||||
| 	if err != nil { | ||||
| 		panic(err) | ||||
| 	} | ||||
|  | @ -141,23 +141,20 @@ func (cui *ConfigUI) DoAction(w http.ResponseWriter, r *http.Request) { | |||
| 			case cui.Actions[i].run <- struct{}{}: | ||||
| 				defer func() { <-cui.Actions[i].run }() | ||||
| 			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] | ||||
| 			if !ok { | ||||
| 				oni = NewOni(name, cui) | ||||
| 				oni.Cmd = cui.Actions[i].Cmd | ||||
| 				oni.Args = cui.Actions[i].Args | ||||
| 				oni.Envs = cui.Actions[i].Envs | ||||
| 				oni.Dir = cui.Actions[i].Dir | ||||
| 				cui.Onitachi[name] = oni | ||||
| 			} | ||||
| 			err := oni.Start() | ||||
| 			file := &File{Name: name, Cmd: v.Cmd, owner: cui} | ||||
| 			pid := make(chan int) | ||||
| 			go func() { | ||||
| 				cui.Actions[i].pid = <-pid | ||||
| 			}() | ||||
| 			result, err := file.Do(cui.cmdTimeout, pid) | ||||
| 			if err != nil { | ||||
| 				panic(err) | ||||
| 			} | ||||
| 			w.Write([]byte("ok")) | ||||
| 			w.Write([]byte(result)) | ||||
| 
 | ||||
| 			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) { | ||||
| 	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")) | ||||
| } | ||||
|  |  | |||
							
								
								
									
										42
									
								
								muzan.go
								
								
								
								
							
							
						
						
									
										42
									
								
								muzan.go
								
								
								
								
							|  | @ -44,31 +44,28 @@ type Oni struct { | |||
| 
 | ||||
| 	parent  *ConfigUI | ||||
| 	cmd     *exec.Cmd | ||||
| 	start  chan struct{} | ||||
| 	running chan struct{} | ||||
| 	buff    bytes.Buffer | ||||
| 	log     io.Writer | ||||
| 
 | ||||
| 	listeners []io.Writer | ||||
| } | ||||
| 
 | ||||
| func NewOni(name string, cui *ConfigUI) *Oni { | ||||
| func (oni *Oni) Init(cui *ConfigUI) error { | ||||
| 	var err error | ||||
| 	oni := &Oni{ | ||||
| 		Policy:    NO, | ||||
| 		State:     READY, | ||||
| 		parent:    cui, | ||||
| 		start:     make(chan struct{}, 1), | ||||
| 		listeners: make([]io.Writer, 0), | ||||
| 	} | ||||
| 	oni.parent = cui | ||||
| 	oni.running = make(chan struct{}, 1) | ||||
| 	oni.listeners = make([]io.Writer, 0) | ||||
| 	oni.StateChange(READY) | ||||
| 	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.log, err = os.OpenFile(logpath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666) | ||||
| 		if err != nil { | ||||
| 			panic(err) | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	return oni | ||||
| 	return err | ||||
| } | ||||
| 
 | ||||
| func (oni *Oni) Start() error { | ||||
|  | @ -76,7 +73,7 @@ func (oni *Oni) Start() error { | |||
| 		return ErrorOniNotValid | ||||
| 	} | ||||
| 	select { | ||||
| 	case oni.start <- struct{}{}: | ||||
| 	case oni.running <- struct{}{}: | ||||
| 		defer func() { | ||||
| 			err := recover() | ||||
| 			if err != nil { | ||||
|  | @ -105,7 +102,7 @@ func (oni *Oni) Start() error { | |||
| 	} | ||||
| 	go oni.read() | ||||
| 	oni.PID = cmd.Process.Pid | ||||
| 	oni.State = STARTED | ||||
| 	oni.StateChange(STARTED) | ||||
| 	go func() { | ||||
| 		oni.end(cmd.Wait()) | ||||
| 	}() | ||||
|  | @ -122,13 +119,13 @@ func (oni *Oni) Stop() error { | |||
| 
 | ||||
| func (oni *Oni) end(v interface{}) { | ||||
| 	if v == nil { | ||||
| 		oni.State = ENDED | ||||
| 		oni.StateChange(ENDED) | ||||
| 	} else { | ||||
| 		oni.Error = fmt.Sprint(v) | ||||
| 		oni.State = ERROR | ||||
| 		oni.StateChange(ERROR) | ||||
| 	} | ||||
| 	if len(oni.start) == 1 { | ||||
| 		<-oni.start | ||||
| 	if len(oni.running) == 1 { | ||||
| 		<-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
 | ||||
| 	// 	}
 | ||||
| 	// }
 | ||||
| } | ||||
|  |  | |||
|  | @ -0,0 +1,7 @@ | |||
| {{define "action"}} | ||||
| {{template "base/header" .}} | ||||
| {{template "components/error" .}} | ||||
| 
 | ||||
| 
 | ||||
| {{template "base/footer" .}} | ||||
| {{end}} | ||||
|  | @ -14,6 +14,7 @@ | |||
|   </div> | ||||
| 
 | ||||
|   <div id="navbarBasicExample" class="navbar-menu"> | ||||
|     {{/* | ||||
|     <div class="navbar-start"> | ||||
|       {{if .Actions}} | ||||
|       <a class="navbar-item"> | ||||
|  | @ -27,7 +28,6 @@ | |||
|       {{end}} | ||||
|     </div> | ||||
| 
 | ||||
|     {{/* | ||||
|     <div class="navbar-end"> | ||||
|       <div class="navbar-item"> | ||||
|         <div class="buttons"> | ||||
|  | @ -44,6 +44,7 @@ | |||
|   </div> | ||||
| </nav> | ||||
| 
 | ||||
| {{/* not having pages | ||||
| <script> | ||||
| document.addEventListener('DOMContentLoaded', () => { | ||||
| 
 | ||||
|  | @ -70,5 +71,6 @@ document.addEventListener('DOMContentLoaded', () => { | |||
|   } | ||||
| 
 | ||||
| }); | ||||
| */}} | ||||
| </script> | ||||
| {{end}} | ||||
|  | @ -14,6 +14,24 @@ | |||
|       </li> | ||||
|       {{ end }} | ||||
|     </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}} | ||||
|     <p class="menu-label"> | ||||
|       System | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue