package repository

import (
	"context"

	"evobgp/internal/store"
)

func (p *Postgres) fillModuleDohFields(ctx context.Context, m *store.Module) error {
	if m == nil {
		return nil
	}
	return p.batchFillModuleDohFields(ctx, map[string]*store.Module{m.ID: m})
}

func (p *Postgres) batchFillModuleDohFields(ctx context.Context, modules map[string]*store.Module) error {
	if len(modules) == 0 {
		return nil
	}
	ids := make([]string, 0, len(modules))
	for id := range modules {
		ids = append(ids, id)
	}
	rows, err := p.pool.Query(ctx, `
		SELECT module_id::text, doh_profile_id::text
		FROM module_doh_profile
		WHERE module_id = ANY($1::uuid[])
		ORDER BY module_id, sort_order, doh_profile_id`, ids)
	if err != nil {
		return err
	}
	defer rows.Close()
	byModule := make(map[string][]string, len(modules))
	for rows.Next() {
		var moduleID, profileID string
		if err := rows.Scan(&moduleID, &profileID); err != nil {
			return err
		}
		byModule[moduleID] = append(byModule[moduleID], profileID)
	}
	if err := rows.Err(); err != nil {
		return err
	}
	for id, m := range modules {
		m.DohProfileIDs = store.NormalizeDohProfileIDList(byModule[id])
		m.SyncLegacyDohProfileID()
	}
	return nil
}

func (p *Postgres) setModuleDohProfiles(ctx context.Context, moduleID string, ids []string) error {
	ids = store.NormalizeDohProfileIDList(ids)
	if _, err := p.pool.Exec(ctx, `DELETE FROM module_doh_profile WHERE module_id = $1`, moduleID); err != nil {
		return err
	}
	for i, id := range ids {
		if _, err := p.pool.Exec(ctx, `
			INSERT INTO module_doh_profile (module_id, doh_profile_id, sort_order)
			VALUES ($1, $2, $3)`, moduleID, id, i); err != nil {
			return err
		}
	}
	var dohArg any
	if len(ids) > 0 {
		dohArg = ids[0]
	}
	_, err := p.pool.Exec(ctx, `UPDATE module SET doh_profile_id = $2, updated_at = now() WHERE id = $1`, moduleID, dohArg)
	return err
}

func (p *Postgres) moduleDohProfileInUse(ctx context.Context, dohProfileID string) (bool, error) {
	var n int
	if err := p.pool.QueryRow(ctx, `
		SELECT COUNT(*) FROM module_doh_profile mdp
		JOIN module m ON m.id = mdp.module_id
		WHERE mdp.doh_profile_id = $1::uuid AND m.deleted_at IS NULL`, dohProfileID).Scan(&n); err != nil {
		return false, err
	}
	return n > 0, nil
}