package networkmap_pgsql

import (
	"context"
	"database/sql"
	"encoding/json"
	"time"

	"github.com/jackc/pgx/v5"
	"github.com/netbirdio/netbird/shared/management/networkmap/nmdata"
)

const (
	GetAccountSettingsQuery = `
	select settings_peer_login_expiration_enabled as peer_login_expiration_enabled,
	settings_peer_login_expiration as peer_login_expiration,
	settings_peer_inactivity_expiration_enabled as peer_inactivity_expiration_enabled,
	settings_peer_inactivity_expiration as peer_inactivity_expiration,
	settings_dns_domain as dns_domain,
	settings_ipv6_enabled_groups as ipv6_enabled_groups,
	settings_routing_peer_dns_resolution_enabled as routing_peer_dns_resolution_enabled,
	settings_lazy_connection_enabled as lazy_connection_enabled,
	settings_auto_update_version as auto_update_version,
	settings_auto_update_always as auto_update_always,
	settings_metrics_push_enabled as metrics_push_enabled
	from accounts
	where id=$1
	`
)

func (pg *PgStore) GetAccountSettings(ctx context.Context, accountId string) (nmdata.AccountSettingsInfo, error) {
	c, err := pg.Pool.Acquire(ctx)
	if err != nil {
		return nmdata.AccountSettingsInfo{}, err
	}
	return GetAccountSettingsViaPgxConnection(ctx, c.Conn(), accountId)

}

func GetAccountSettingsViaPgxConnection(ctx context.Context, con *pgx.Conn, accountId string) (nmdata.AccountSettingsInfo, error) {
	rows, err := con.Query(ctx, GetAccountSettingsQuery, accountId)
	if err != nil {
		return nmdata.AccountSettingsInfo{}, err
	}

	settings, err := pgx.CollectOneRow(rows, pgx.RowToStructByName[account])
	if err != nil {
		return nmdata.AccountSettingsInfo{}, err
	}

	settingsInfo := nmdata.AccountSettingsInfo{
		PeerLoginExpirationEnabled:      settings.PeerLoginExpirationEnabled.Bool,
		PeerLoginExpiration:             time.Duration(settings.PeerLoginExpiration.Int64),
		PeerInactivityExpirationEnabled: settings.PeerInactivityExpirationEnabled.Bool,
		PeerInactivityExpiration:        time.Duration(settings.PeerInactivityExpiration.Int64),
		DNSDomain:                       settings.DNSDomain.String,
		RoutingPeerDNSResolutionEnabled: settings.RoutingPeerDNSResolutionEnabled.Bool,
		LazyConnectionEnabled:           settings.LazyConnectionEnabled.Bool,
		AutoUpdateVersion:               settings.AutoUpdateVersion.String,
		AutoUpdateAlways:                settings.AutoUpdateAlways.Bool,
		MetricsPushEnabled:              settings.MetricsPushEnabled.Bool,
	}
	if settings.IPv6EnabledGroups != nil {
		if err := json.Unmarshal(settings.IPv6EnabledGroups, &settingsInfo.IPv6EnabledGroups); err != nil {
			return nmdata.AccountSettingsInfo{}, err
		}
	}

	return settingsInfo, nil
}

type account struct {
	PeerLoginExpirationEnabled      sql.NullBool
	PeerLoginExpiration             sql.NullInt64
	PeerInactivityExpirationEnabled sql.NullBool
	PeerInactivityExpiration        sql.NullInt64
	DNSDomain                       sql.NullString
	IPv6EnabledGroups               json.RawMessage
	RoutingPeerDNSResolutionEnabled sql.NullBool
	LazyConnectionEnabled           sql.NullBool
	AutoUpdateVersion               sql.NullString
	AutoUpdateAlways                sql.NullBool
	MetricsPushEnabled              sql.NullBool
}