package handler

import (
	"baron-sso-backend/internal/domain"
	"baron-sso-backend/internal/repository"
	"baron-sso-backend/internal/service"
	"baron-sso-backend/internal/utils"
	"context"
	"log/slog"
	"strings"
	"time"

	"github.com/gofiber/fiber/v2"
)

type UserHandler struct {
	KratosAdmin    service.KratosAdminService
	OryProvider    *service.OryProvider
	TenantService  service.TenantService
	KetoService    service.KetoService
	KetoOutboxRepo repository.KetoOutboxRepository
	UserRepo       repository.UserRepository
}

func NewUserHandler(kratosAdmin service.KratosAdminService, oryProvider *service.OryProvider, tenantService service.TenantService, ketoService service.KetoService, ketoOutboxRepo repository.KetoOutboxRepository, userRepo repository.UserRepository) *UserHandler {
	return &UserHandler{
		KratosAdmin:    kratosAdmin,
		OryProvider:    oryProvider,
		TenantService:  tenantService,
		KetoService:    ketoService,
		KetoOutboxRepo: ketoOutboxRepo,
		UserRepo:       userRepo,
	}
}

type userSummary struct {
	ID              string         `json:"id"`
	Email           string         `json:"email"`
	Name            string         `json:"name"`
	Phone           string         `json:"phone"`
	Role            string         `json:"role"`
	Status          string         `json:"status"`
	CompanyCode     string         `json:"companyCode"`
	Metadata        domain.JSONMap `json:"metadata,omitempty"`
	Tenant          *domain.Tenant `json:"tenant,omitempty"`
	Department      string         `json:"department"`
	CreatedAt       string         `json:"createdAt"`
	UpdatedAt       string         `json:"updatedAt"`
	InitialPassword string         `json:"initialPassword,omitempty"`
}

type userListResponse struct {
	Items  []userSummary `json:"items"`
	Limit  int           `json:"limit"`
	Offset int           `json:"offset"`
	Total  int64         `json:"total"`
}

func (h *UserHandler) ListUsers(c *fiber.Ctx) error {
	// [New] Get requester profile from middleware
	var requesterRole string
	var requesterCompany string
	if profile, ok := c.Locals("user_profile").(*domain.UserProfileResponse); ok {
		requesterRole = domain.NormalizeRole(profile.Role)
		requesterCompany = profile.CompanyCode
	}

	limit := c.QueryInt("limit", 50)
	offset := c.QueryInt("offset", 0)
	search := strings.TrimSpace(c.Query("search"))
	companyCode := strings.TrimSpace(c.Query("companyCode"))

	if limit <= 0 {
		limit = 50
	}
	if offset < 0 {
		offset = 0
	}

	// 1. Try Kratos First
	identities, err := h.KratosAdmin.ListIdentities(c.Context())
	if err == nil {
		filtered := make([]service.KratosIdentity, 0, len(identities))
		searchLower := strings.ToLower(search)

		for _, identity := range identities {
			email := strings.ToLower(extractTraitString(identity.Traits, "email"))
			name := strings.ToLower(extractTraitString(identity.Traits, "name"))
			compCode := extractTraitString(identity.Traits, "companyCode")

			// Tenant Admin filtering
			if requesterRole == domain.RoleTenantAdmin {
				if requesterCompany == "" || !strings.EqualFold(compCode, requesterCompany) {
					continue
				}
			}

			// Dedicated companyCode filter
			if companyCode != "" && !strings.EqualFold(compCode, companyCode) {
				continue
			}

			// Search filtering (Keyword search in email, name, or companyCode)
			if search != "" {
				if !strings.Contains(email, searchLower) &&
					!strings.Contains(name, searchLower) &&
					!strings.Contains(strings.ToLower(compCode), searchLower) {
					continue
				}
			}
			filtered = append(filtered, identity)
		}

		total := int64(len(filtered))
		if offset > len(filtered) {
			offset = len(filtered)
		}
		end := offset + limit
		if end > len(filtered) {
			end = len(filtered)
		}

		items := make([]userSummary, 0, end-offset)
		for _, identity := range filtered[offset:end] {
			summary := h.mapIdentitySummary(c.Context(), identity)
			items = append(items, summary)
		}

		// [Lazy Sync] Asynchronously update local DB with fresh data from Kratos
		// This ensures that member counts (which use local DB) eventually match reality
		if h.UserRepo != nil {
			go func(ids []service.KratosIdentity) {
				ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
				defer cancel()
				for _, identity := range ids {
					localUser := h.mapToLocalUser(identity)
					_ = h.UserRepo.Update(ctx, localUser)
				}
			}(filtered)
		}

		return c.JSON(userListResponse{Items: items, Limit: limit, Offset: offset, Total: total})
	}

	// 2. Fallback to Local DB if Kratos is down
	slog.Warn("Kratos unavailable, falling back to local DB for user list", "error", err)

	// Fetch from UserRepo
	users, total, err := h.UserRepo.List(c.Context(), offset, limit, search, companyCode)
	if err != nil {
		return errorJSON(c, fiber.StatusInternalServerError, "failed to fetch users from both kratos and local db")
	}

	items := make([]userSummary, 0, len(users))
	for _, u := range users {
		items = append(items, userSummary{
			ID:          u.ID,
			Email:       u.Email,
			Name:        u.Name,
			Phone:       u.Phone,
			Role:        u.Role,
			Status:      u.Status,
			CompanyCode: u.CompanyCode,
			Department:  u.Department,
			CreatedAt:   u.CreatedAt.Format(time.RFC3339),
			UpdatedAt:   u.UpdatedAt.Format(time.RFC3339),
		})
	}

	return c.JSON(userListResponse{
		Items:  items,
		Total:  total,
		Limit:  limit,
		Offset: offset,
	})
}

func (h *UserHandler) GetUser(c *fiber.Ctx) error {
	if h.KratosAdmin == nil {
		return errorJSON(c, fiber.StatusServiceUnavailable, "identity provider not available")
	}

	userID := strings.TrimSpace(c.Params("id"))
	if userID == "" {
		return errorJSON(c, fiber.StatusBadRequest, "user id is required")
	}

	identity, err := h.KratosAdmin.GetIdentity(c.Context(), userID)
	if err != nil {
		return errorJSON(c, fiber.StatusInternalServerError, err.Error())
	}
	if identity == nil {
		return errorJSON(c, fiber.StatusNotFound, "user not found")
	}

	// [New] Check access scope
	requester, _ := c.Locals("user_profile").(*domain.UserProfileResponse)
	if requester != nil && domain.NormalizeRole(requester.Role) == domain.RoleTenantAdmin {
		compCode := extractTraitString(identity.Traits, "companyCode")
		if requester.CompanyCode == "" || compCode != requester.CompanyCode {
			return errorJSON(c, fiber.StatusForbidden, "forbidden: access to user in another tenant denied")
		}
	}

	return c.JSON(h.mapIdentitySummary(c.Context(), *identity))
}

func (h *UserHandler) CreateUser(c *fiber.Ctx) error {
	if h.OryProvider == nil || h.KratosAdmin == nil {
		return errorJSON(c, fiber.StatusServiceUnavailable, "identity provider not available")
	}

	var req struct {
		Email       string         `json:"email"`
		Password    string         `json:"password"`
		Name        string         `json:"name"`
		Phone       string         `json:"phone"`
		Role        string         `json:"role"`
		CompanyCode string         `json:"companyCode"`
		Department  string         `json:"department"`
		Metadata    map[string]any `json:"metadata"`
	}
	if err := c.BodyParser(&req); err != nil {
		return errorJSON(c, fiber.StatusBadRequest, "invalid request body")
	}

	email := strings.TrimSpace(req.Email)
	if email == "" {
		return errorJSON(c, fiber.StatusBadRequest, "email is required")
	}
	if !strings.Contains(email, "@") || !strings.Contains(email, ".") {
		return errorJSON(c, fiber.StatusBadRequest, "invalid email format")
	}
	name := strings.TrimSpace(req.Name)
	if name == "" {
		return errorJSON(c, fiber.StatusBadRequest, "name is required")
	}

	password := strings.TrimSpace(req.Password)
	policy, err := h.OryProvider.GetPasswordPolicy()
	if err != nil || policy == nil {
		policy = &domain.PasswordPolicy{
			MinLength:         12,
			Lowercase:         true,
			Uppercase:         false,
			Number:            true,
			NonAlphanumeric:   true,
			MinCharacterTypes: 0,
		}
	}

	generatedPassword := ""
	if password == "" {
		generated, genErr := utils.GeneratePasswordWithPolicy(policy)
		if genErr != nil {
			return errorJSON(c, fiber.StatusInternalServerError, "failed to generate password")
		}
		password = generated
		generatedPassword = generated
	} else {
		if err := utils.ValidatePasswordWithPolicy(policy, password); err != nil {
			return errorJSON(c, fiber.StatusBadRequest, err.Error())
		}
	}

	role := domain.NormalizeRole(req.Role)
	if role == "" {
		role = domain.RoleUser
	}

	attributes := map[string]interface{}{
		"department":      req.Department,
		"affiliationType": "internal",
		"companyCode":     req.CompanyCode,
		"grade":           role,
	}

	// [Resolve TenantID before Kratos creation]
	var tenantID string
	if req.CompanyCode != "" && h.TenantService != nil {
		if tenant, err := h.TenantService.GetTenantBySlug(c.Context(), req.CompanyCode); err == nil && tenant != nil {
			tenantID = tenant.ID
		}
	}
	attributes["role"] = role
	if tenantID != "" {
		attributes["tenant_id"] = tenantID
	}

	// Merge custom metadata into attributes
	for k, v := range req.Metadata {
		// Don't overwrite core fields
		if _, exists := attributes[k]; !exists {
			attributes[k] = v
		}
	}

	brokerUser := &domain.BrokerUser{
		Email:       email,
		Name:        name,
		PhoneNumber: normalizePhoneNumber(req.Phone),
		Attributes:  attributes,
	}

	identityID, err := h.OryProvider.CreateUser(brokerUser, password)
	if err != nil {
		if strings.Contains(err.Error(), "already exists") {
			return errorJSON(c, fiber.StatusConflict, "email already exists")
		}
		return errorJSON(c, fiber.StatusInternalServerError, err.Error())
	}

	// Fetch the newly created identity to ensure we have all traits
	identity, err := h.KratosAdmin.GetIdentity(c.Context(), identityID)
	if err != nil {
		return errorJSON(c, fiber.StatusInternalServerError, err.Error())
	}
	if identity == nil {
		return c.Status(fiber.StatusCreated).JSON(fiber.Map{"id": identityID, "initialPassword": generatedPassword})
	}

	// [New] Local DB Sync - Ensure user exists in read-model
	if h.UserRepo != nil {
		localUser := h.mapToLocalUser(*identity)

		// Sync to local DB
		go func(u *domain.User, role string, tID *string) {
			ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
			defer cancel()

			// Use Update (upsert) instead of Create for robustness
			if err := h.UserRepo.Update(ctx, u); err != nil {
				slog.Error("[UserHandler] Failed to sync new user to local DB", "email", u.Email, "error", err)
				return
			}

			// [Keto] Sync relations via Outbox
			if h.KetoOutboxRepo != nil {
				h.syncKetoRole(ctx, u.ID, role, "", "", tID)
			}
		}(localUser, role, localUser.TenantID)
	}

	response := h.mapIdentitySummary(c.Context(), *identity)
	if generatedPassword != "" {
		response.InitialPassword = generatedPassword
	}
	return c.Status(fiber.StatusCreated).JSON(response)
}

func (h *UserHandler) UpdateUser(c *fiber.Ctx) error {
	if h.KratosAdmin == nil {
		return errorJSON(c, fiber.StatusServiceUnavailable, "identity provider not available")
	}

	userID := strings.TrimSpace(c.Params("id"))
	if userID == "" {
		return errorJSON(c, fiber.StatusBadRequest, "user id is required")
	}

	identity, err := h.KratosAdmin.GetIdentity(c.Context(), userID)
	if err != nil {
		return errorJSON(c, fiber.StatusInternalServerError, err.Error())
	}
	if identity == nil {
		return errorJSON(c, fiber.StatusNotFound, "user not found")
	}

	// Capture current local state for transition comparison
	var oldRole string
	var oldTenantID string
	if h.UserRepo != nil {
		if local, err := h.UserRepo.FindByID(c.Context(), userID); err == nil && local != nil {
			oldRole = local.Role
			if local.TenantID != nil {
				oldTenantID = *local.TenantID
			}
		}
	}

	// [New] Check access scope
	requester, _ := c.Locals("user_profile").(*domain.UserProfileResponse)
	if requester != nil && domain.NormalizeRole(requester.Role) == domain.RoleTenantAdmin {
		compCode := extractTraitString(identity.Traits, "companyCode")
		if requester.CompanyCode == "" || compCode != requester.CompanyCode {
			return errorJSON(c, fiber.StatusForbidden, "forbidden: cannot update user in another tenant")
		}
	}

	var req struct {
		Password    *string        `json:"password"`
		Name        *string        `json:"name"`
		Phone       *string        `json:"phone"`
		Role        *string        `json:"role"`
		Status      *string        `json:"status"`
		CompanyCode *string        `json:"companyCode"`
		Department  *string        `json:"department"`
		Metadata    map[string]any `json:"metadata"`
	}
	if err := c.BodyParser(&req); err != nil {
		return errorJSON(c, fiber.StatusBadRequest, "invalid request body")
	}

	// [New] Tenant Admin restriction: Cannot change companyCode
	if requester != nil && domain.NormalizeRole(requester.Role) == domain.RoleTenantAdmin {
		if req.CompanyCode != nil && *req.CompanyCode != requester.CompanyCode {
			return errorJSON(c, fiber.StatusForbidden, "forbidden: tenant admins cannot change user's tenant")
		}
	}

	traits := identity.Traits
	if traits == nil {
		traits = map[string]interface{}{}
	}
	if req.Name != nil {
		traits["name"] = strings.TrimSpace(*req.Name)
	}
	if req.Phone != nil {
		phone := normalizePhoneNumber(strings.TrimSpace(*req.Phone))
		if phone == "" {
			delete(traits, "phone_number")
		} else {
			traits["phone_number"] = phone
		}
	}
	if req.CompanyCode != nil {
		code := strings.TrimSpace(*req.CompanyCode)
		traits["companyCode"] = code
		// Resolve TenantID for Kratos Trait
		if h.TenantService != nil && code != "" {
			if tenant, err := h.TenantService.GetTenantBySlug(c.Context(), code); err == nil && tenant != nil {
				traits["tenant_id"] = tenant.ID
			}
		}
	}
	if req.Department != nil {
		traits["department"] = strings.TrimSpace(*req.Department)
	}
	if req.Role != nil {
		role := domain.NormalizeRole(*req.Role)
		if role == "" {
			role = domain.RoleUser
		}
		traits["grade"] = role
		traits["role"] = role
	}

	// [Refined] Metadata synchronization: replace non-core traits with new Metadata
	coreTraits := map[string]bool{
		"email": true, "name": true, "phone_number": true,
		"grade": true, "companyCode": true, "department": true,
		"affiliationType": true, "role": true, "tenant_id": true,
	}

	// 1. Remove existing non-core traits to handle deletions
	for k := range traits {
		if !coreTraits[k] {
			delete(traits, k)
		}
	}

	// 2. Add new metadata fields
	for k, v := range req.Metadata {
		if !coreTraits[k] {
			traits[k] = v
		}
	}

	state := normalizeKratosState(req.Status)
	updated, err := h.KratosAdmin.UpdateIdentity(c.Context(), userID, traits, state)
	if err != nil {
		return errorJSON(c, fiber.StatusInternalServerError, err.Error())
	}

	// [New] Local DB Sync - Sync synchronously to ensure immediate consistency for the caller
	if h.UserRepo != nil {
		updatedLocalUser := h.mapToLocalUser(*updated)

		ctx := context.Background() // Use request context if appropriate, but sync must finish
		if err := h.UserRepo.Update(ctx, updatedLocalUser); err != nil {
			slog.Error("[UserHandler] Failed to sync updated user to local DB", "userID", updatedLocalUser.ID, "error", err)
		}

		// [Keto Sync] asynchronously as it's less critical for immediate UI count
		go h.syncKetoRole(context.Background(), updatedLocalUser.ID,
			extractTraitString(updated.Traits, "grade"), oldRole, oldTenantID, updatedLocalUser.TenantID)
	}

	if req.Password != nil && *req.Password != "" {
		if err := h.KratosAdmin.UpdateIdentityPassword(c.Context(), userID, *req.Password); err != nil {
			return errorJSON(c, fiber.StatusInternalServerError, err.Error())
		}
	}

	return c.JSON(h.mapIdentitySummary(c.Context(), *updated))
}

func (h *UserHandler) DeleteUser(c *fiber.Ctx) error {
	if h.KratosAdmin == nil {
		return errorJSON(c, fiber.StatusServiceUnavailable, "identity provider not available")
	}

	userID := strings.TrimSpace(c.Params("id"))
	if userID == "" {
		return errorJSON(c, fiber.StatusBadRequest, "user id is required")
	}

	// [New] Check access scope before deletion
	requester, _ := c.Locals("user_profile").(*domain.UserProfileResponse)
	if requester != nil && domain.NormalizeRole(requester.Role) == domain.RoleTenantAdmin {
		identity, err := h.KratosAdmin.GetIdentity(c.Context(), userID)
		if err == nil && identity != nil {
			compCode := extractTraitString(identity.Traits, "companyCode")
			if requester.CompanyCode == "" || compCode != requester.CompanyCode {
				return errorJSON(c, fiber.StatusForbidden, "forbidden: cannot delete user in another tenant")
			}
		}
	}

	if err := h.KratosAdmin.DeleteIdentity(c.Context(), userID); err != nil {
		return errorJSON(c, fiber.StatusInternalServerError, err.Error())
	}

	// [Keto] Cleanup relations via Outbox
	if h.KetoOutboxRepo != nil {
		ctx := context.Background()
		_ = h.KetoOutboxRepo.Create(ctx, &domain.KetoOutbox{
			Namespace: "System",
			Object:    "global",
			Relation:  "super_admins",
			Subject:   "User:" + userID,
			Action:    domain.KetoOutboxActionDelete,
		})
		// Additional cleanup for tenants could be added here if we keep track of user's current tenants
	}

	return c.SendStatus(fiber.StatusNoContent)
}

func (h *UserHandler) mapIdentitySummary(ctx context.Context, identity service.KratosIdentity) userSummary {
	traits := identity.Traits
	role := extractTraitString(traits, "grade")
	if role == "" {
		role = extractTraitString(traits, "role")
	}
	role = domain.NormalizeRole(role)
	if role == "" {
		role = domain.RoleUser
	}

	compCode := extractTraitString(traits, "companyCode")
	slog.Debug("Mapping identity", "email", extractTraitString(traits, "email"), "compCode", compCode)
	summary := userSummary{
		ID:          identity.ID,
		Email:       extractTraitString(traits, "email"),
		Name:        extractTraitString(traits, "name"),
		Phone:       extractTraitString(traits, "phone_number"),
		Role:        role,
		Status:      normalizeStatus(identity.State),
		CompanyCode: compCode,
		Department:  extractTraitString(traits, "department"),
		Metadata:    make(domain.JSONMap),
		CreatedAt:   formatTime(identity.CreatedAt),
		UpdatedAt:   formatTime(identity.UpdatedAt),
	}

	// Filter out core traits and put everything else in Metadata
	coreTraits := map[string]bool{
		"email": true, "name": true, "phone_number": true,
		"grade": true, "companyCode": true, "department": true,
		"affiliationType": true,
	}
	for k, v := range traits {
		if !coreTraits[k] {
			summary.Metadata[k] = v
		}
	}

	if compCode != "" && h.TenantService != nil {
		if tenant, err := h.TenantService.GetTenantBySlug(ctx, compCode); err == nil && tenant != nil {
			summary.Tenant = tenant
		}
	}

	return summary
}

func (h *UserHandler) normalizePhoneNumber(phone string) string {
	return normalizePhoneNumber(phone)
}

func (h *UserHandler) mapToLocalUser(identity service.KratosIdentity) *domain.User {
	traits := identity.Traits
	role := extractTraitString(traits, "grade")
	if role == "" {
		role = extractTraitString(traits, "role")
	}
	role = domain.NormalizeRole(role)
	if role == "" {
		role = domain.RoleUser
	}
	compCode := extractTraitString(traits, "companyCode")

	user := &domain.User{
		ID:              identity.ID,
		Email:           extractTraitString(traits, "email"),
		Name:            extractTraitString(traits, "name"),
		Phone:           extractTraitString(traits, "phone_number"),
		Role:            role,
		Status:          normalizeStatus(identity.State),
		CompanyCode:     compCode,
		Department:      extractTraitString(traits, "department"),
		AffiliationType: extractTraitString(traits, "affiliationType"),
		CreatedAt:       identity.CreatedAt,
		UpdatedAt:       identity.UpdatedAt,
	}

	if compCode != "" && h.TenantService != nil {
		// Use a background context or a timeout-limited context for tenant lookup
		ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
		defer cancel()
		if tenant, err := h.TenantService.GetTenantBySlug(ctx, compCode); err == nil && tenant != nil {
			user.TenantID = &tenant.ID
		}
	}

	// Metadata
	user.Metadata = make(domain.JSONMap)
	coreTraits := map[string]bool{
		"email": true, "name": true, "phone_number": true,
		"grade": true, "companyCode": true, "department": true,
		"affiliationType": true, "role": true, "tenant_id": true,
	}
	for k, v := range traits {
		if !coreTraits[k] {
			user.Metadata[k] = v
		}
	}

	return user
}

func (h *UserHandler) syncKetoRole(ctx context.Context, userID, newRole, oldRole, oldTenantID string, newTenantID *string) {
	newRole = domain.NormalizeRole(newRole)
	oldRole = domain.NormalizeRole(oldRole)

	// Remove old roles
	if oldRole == domain.RoleSuperAdmin {
		_ = h.KetoOutboxRepo.Create(ctx, &domain.KetoOutbox{
			Namespace: "System",
			Object:    "global",
			Relation:  "super_admins",
			Subject:   "User:" + userID,
			Action:    domain.KetoOutboxActionDelete,
		})
	} else if oldRole == domain.RoleTenantAdmin && oldTenantID != "" {
		_ = h.KetoOutboxRepo.Create(ctx, &domain.KetoOutbox{
			Namespace: "Tenant",
			Object:    oldTenantID,
			Relation:  "admins",
			Subject:   "User:" + userID,
			Action:    domain.KetoOutboxActionDelete,
		})
	}

	// Add new roles
	if newRole == domain.RoleSuperAdmin {
		_ = h.KetoOutboxRepo.Create(ctx, &domain.KetoOutbox{
			Namespace: "System",
			Object:    "global",
			Relation:  "super_admins",
			Subject:   "User:" + userID,
			Action:    domain.KetoOutboxActionCreate,
		})
	} else if newRole == domain.RoleTenantAdmin && newTenantID != nil {
		_ = h.KetoOutboxRepo.Create(ctx, &domain.KetoOutbox{
			Namespace: "Tenant",
			Object:    *newTenantID,
			Relation:  "admins",
			Subject:   "User:" + userID,
			Action:    domain.KetoOutboxActionCreate,
		})
	}
}

func extractTraitString(traits map[string]interface{}, key string) string {
	if traits == nil {
		return ""
	}
	if raw, ok := traits[key]; ok {
		if value, ok := raw.(string); ok {
			return value
		}
	}
	return ""
}

func formatTime(value time.Time) string {
	if value.IsZero() {
		return ""
	}
	return value.Format(time.RFC3339)
}

func normalizeStatus(state string) string {
	state = strings.ToLower(strings.TrimSpace(state))
	if state == "inactive" || state == "blocked" || state == "active" {
		return state
	}
	if state == "" {
		return "active"
	}
	return state
}

func normalizeKratosState(status *string) string {
	if status == nil {
		return ""
	}
	value := strings.ToLower(strings.TrimSpace(*status))
	if value == "blocked" {
		return "inactive"
	}
	if value == "active" || value == "inactive" {
		return value
	}
	return ""
}

func normalizePhoneNumber(phone string) string {
	normalized := strings.ReplaceAll(phone, "-", "")
	normalized = strings.ReplaceAll(normalized, " ", "")
	if normalized == "" {
		return ""
	}
	if strings.HasPrefix(normalized, "010") {
		return "+82" + normalized[1:]
	}
	if strings.HasPrefix(normalized, "82") {
		return "+" + normalized
	}
	return normalized
}