forked from Nixius/authelia
76 lines
2.1 KiB
Go
76 lines
2.1 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
)
|
|
|
|
func (a *App) handleStackManage(w http.ResponseWriter, r *http.Request) {
|
|
remoteUser := r.Header.Get("Remote-User")
|
|
remoteGroups := r.Header.Get("Remote-Groups")
|
|
if remoteUser == "" {
|
|
http.Redirect(w, r, a.cfg.AutheliaURL, http.StatusSeeOther)
|
|
return
|
|
}
|
|
if !contains(remoteGroups, "customers") {
|
|
http.Error(w, "forbidden", http.StatusForbidden)
|
|
return
|
|
}
|
|
|
|
action := r.FormValue("action")
|
|
stackName := fmt.Sprintf("customer-%s", remoteUser)
|
|
|
|
switch action {
|
|
case "stop":
|
|
if err := a.swarm.ScaleStack(stackName, 0); err != nil {
|
|
log.Printf("stack-manage stop %s: %v", remoteUser, err)
|
|
http.Error(w, "failed to stop stack", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
case "start":
|
|
exists, _ := a.swarm.StackExists(stackName)
|
|
if !exists {
|
|
if err := a.swarm.DeployStack(stackName, remoteUser, a.cfg.TraefikDomain); err != nil {
|
|
log.Printf("stack-manage start (deploy) %s: %v", remoteUser, err)
|
|
http.Error(w, "failed to start stack", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
} else {
|
|
if err := a.swarm.ScaleStack(stackName, 1); err != nil {
|
|
log.Printf("stack-manage start (scale) %s: %v", remoteUser, err)
|
|
http.Error(w, "failed to start stack", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
case "restart":
|
|
if err := a.swarm.RestartStack(stackName); err != nil {
|
|
log.Printf("stack-manage restart %s: %v", remoteUser, err)
|
|
http.Error(w, "failed to restart stack", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
case "rebuild":
|
|
if err := a.swarm.DeployStack(stackName, remoteUser, a.cfg.TraefikDomain); err != nil {
|
|
log.Printf("stack-manage rebuild %s: %v", remoteUser, err)
|
|
http.Error(w, "failed to rebuild stack", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
case "destroy":
|
|
if err := a.swarm.RemoveStack(stackName); err != nil {
|
|
log.Printf("stack-manage destroy %s: %v", remoteUser, err)
|
|
http.Error(w, "failed to destroy stack", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
default:
|
|
http.Error(w, "unknown action", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, a.cfg.AppURL+"/dashboard", http.StatusSeeOther)
|
|
}
|