package main

import (
	"context"
	"time"

	"codeberg.org/pronounscc/pronouns.cc/backend/db"
	"emperror.dev/errors"
	"github.com/georgysavva/scany/v2/pgxscan"
)

func getUsers() (u []db.User, err error) {
	q := sq.Select("*", "(SELECT instance FROM fediverse_apps WHERE id = users.fediverse_app_id) AS fediverse_instance").
		From("users").OrderBy("snowflake_id ASC")

	if *exportLimit != 0 {
		q = q.Limit(uint64(*exportLimit))
	}

	sql, args, err := q.ToSql()
	if err != nil {
		return u, errors.Wrap(err, "building sql")
	}

	err = pgxscan.Select(context.Background(), oldDB, &u, sql, args...)
	if err != nil {
		return u, errors.Wrap(err, "getting users from db")
	}

	return u, nil
}

func userToNewUser(u db.User) (NewUser, error) {
	fields, err := oldDB.UserFields(context.Background(), u.ID)
	if err != nil {
		return NewUser{}, errors.Wrap(err, "getting fields")
	}

	new := NewUser{
		ID:          u.SnowflakeID.String(),
		SID:         u.SID,
		Username:    u.Username,
		DisplayName: u.DisplayName,
		Bio:         u.Bio,
		MemberTitle: u.MemberTitle,
		LastActive:  u.LastActive,
		Avatar:      u.Avatar,
		Links:       u.Links,
		Names:       u.Names,

		Discord:           u.Discord,
		DiscordUsername:   u.DiscordUsername,
		Fediverse:         u.Fediverse,
		FediverseUsername: u.FediverseUsername,
		FediverseAppID:    u.FediverseAppID,
		Tumblr:            u.Tumblr,
		TumblrUsername:    u.TumblrUsername,
		Google:            u.Google,
		GoogleUsername:    u.GoogleUsername,

		MemberListHidden: u.ListPrivate,
		Timezone:         u.Timezone,
		Role:             "USER",

		Deleted:      u.DeletedAt != nil,
		DeletedAt:    u.DeletedAt,
		DeleteReason: u.DeleteReason,

		CustomPreferences: u.CustomPreferences,
	}

	if u.IsAdmin {
		new.Role = "ADMIN"
	}

	for _, p := range u.Pronouns {
		new.Pronouns = append(new.Pronouns, NewPronoun{Value: p.Pronouns, Status: string(p.Status), DisplayText: p.DisplayText})
	}

	for _, f := range fields {
		new.Fields = append(new.Fields, NewField{Name: f.Name, Entries: f.Entries})
	}

	return new, nil
}

type NewUser struct {
	ID          string          `json:"id"`
	SID         string          `json:"sid"`
	Username    string          `json:"username"`
	DisplayName *string         `json:"display_name"`
	Bio         *string         `json:"bio"`
	MemberTitle *string         `json:"member_title"`
	LastActive  time.Time       `json:"last_active"`
	Avatar      *string         `json:"avatar_hash"`
	Links       []string        `json:"links"`
	Names       []db.FieldEntry `json:"names"`
	Pronouns    []NewPronoun    `json:"pronouns"`
	Fields      []NewField      `json:"fields"`

	Discord         *string `json:"discord_id"`
	DiscordUsername *string `json:"discord_username"`

	Fediverse         *string `json:"fediverse_id"`
	FediverseUsername *string `json:"fediverse_username"`
	FediverseAppID    *int64  `json:"fediverse_app_id"`

	Tumblr         *string `json:"tumblr_id"`
	TumblrUsername *string `json:"tumblr_username"`

	Google         *string `json:"google_id"`
	GoogleUsername *string `json:"google_username"`

	MemberListHidden bool    `json:"member_list_hidden"`
	Timezone         *string `json:"timezone"`
	Role             string  `json:"role"` // one of USER or ADMIN

	Deleted      bool       `json:"deleted"`
	DeletedAt    *time.Time `json:"deleted_at"`
	DeleteReason *string    `json:"delete_reason"` // TODO: this should be imported as a warning

	CustomPreferences db.CustomPreferences `json:"custom_preferences"`
}

type NewPronoun struct {
	Value       string  `json:"value"`
	Status      string  `json:"status"`
	DisplayText *string `json:"display_text"`
}

type NewField struct {
	Name    string          `json:"name"`
	Entries []db.FieldEntry `json:"entries"`
}