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)
}