package passwordgenerator

import (
	"embed"
	"html/template"
	"net/http"
	"strconv"
	"strings"
)

type Handler struct {
	templates *template.Template
}

type PasswordConfig struct {
	Type           string
	Length         int
	IncludeUpper   bool
	IncludeLower   bool
	NumberCount    int
	SpecialChars   string
	NoConsecutive  bool
	WordCount      int
	UseNumbers     bool
	UseSpecial     bool
	NumberPosition string
	SavePasswords  bool
}

func NewHandler(embeddedFiles embed.FS) *Handler {
	tmpl := template.Must(template.New("").Funcs(template.FuncMap{
		"splitString": func(s, delimiter string) []string {
			return strings.Split(s, delimiter)
		},
		"contains": func(s, substr string) bool {
			return strings.Contains(s, substr)
		},
		"add": func(a, b int) int {
			return a + b
		},
		"ge": func(a, b int) bool {
			return a >= b
		},
		"len": func(v interface{}) int {
			switch s := v.(type) {
			case []string:
				return len(s)
			case map[string]string:
				return len(s)
			case string:
				return len(s)
			default:
				return 0
			}
		},
	}).ParseFS(embeddedFiles, "web/base.html", "web/password.html"))

	return &Handler{
		templates: tmpl,
	}
}

func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	// Parse URL parameters to set default values
	config := PasswordConfig{
		Type:           getStringParam(r, "type", "passphrase"),
		Length:         getIntParam(r, "length", 12),
		IncludeUpper:   getBoolParam(r, "includeUpper", true),
		IncludeLower:   getBoolParam(r, "includeLower", true),
		NumberCount:    getIntParam(r, "numberCount", 1),
		SpecialChars:   getStringParam(r, "specialChars", "!@#$%^&*-_=+"),
		NoConsecutive:  getBoolParam(r, "noConsecutive", false),
		WordCount:      getIntParam(r, "wordCount", 3),
		UseNumbers:     getBoolParam(r, "useNumbers", true),
		UseSpecial:     getBoolParam(r, "useSpecial", false),
		NumberPosition: getStringParam(r, "numberPosition", "end"),
		SavePasswords:  getBoolParam(r, "savePasswords", false),
	}

	data := struct {
		CurrentPage string
		Config      PasswordConfig
	}{
		CurrentPage: "password",
		Config:      config,
	}
	h.templates.ExecuteTemplate(w, "password.html", data)
}

// Helper functions to parse URL parameters
func getStringParam(r *http.Request, key, defaultValue string) string {
	if value := r.URL.Query().Get(key); value != "" {
		return value
	}
	return defaultValue
}

func getIntParam(r *http.Request, key string, defaultValue int) int {
	if value := r.URL.Query().Get(key); value != "" {
		if intValue, err := strconv.Atoi(value); err == nil {
			return intValue
		}
	}
	return defaultValue
}

func getBoolParam(r *http.Request, key string, defaultValue bool) bool {
	if value := r.URL.Query().Get(key); value != "" {
		return value == "true" || value == "1" || value == "on"
	}
	return defaultValue
}