package db

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

	"github.com/sqlc-dev/pqtype"
)

const getUserQ = `-- name: GetUser :one
SELECT
  id,
  COALESCE(email, '') as email,
  password,
  username,
  COALESCE(avatar_picture, '') as avatar_picture,
  banned_at,
  banned_until,
  COALESCE(ban_reason, '') as ban_reason,
  is_permaban,
  is_admin,
  is_critics,
  is_verified,
  social_media
FROM USERS
WHERE username = $1
`

type GetUserRow struct {
	ID            int32                 `json:"id"`
	Email         string                `json:"email"`
	Password      string                `json:"-"`
	Username      string                `json:"username"`
	AvatarPicture string                `json:"avatar_picture"`
	BannedAt      sql.NullTime          `json:"banned_at"`
	BannedUntil   sql.NullTime          `json:"banned_until"`
	BanReason     string                `json:"ban_reason"`
	IsPermaban    bool                  `json:"is_permaban"`
	IsAdmin       bool                  `json:"is_admin"`
	IsCritics     bool                  `json:"is_critics"`
	IsVerified    bool                  `json:"is_verified"`
	SocialMedia   pqtype.NullRawMessage `json:"social_media"`
}

func (q *Queries) GetUser(ctx context.Context, username string) (GetUserRow, error) {
	row := q.db.QueryRowContext(ctx, getUserQ, username)
	var i GetUserRow
	err := row.Scan(
		&i.ID,
		&i.Email,
		&i.Password,
		&i.Username,
		&i.AvatarPicture,
		&i.BannedAt,
		&i.BannedUntil,
		&i.BanReason,
		&i.IsPermaban,
		&i.IsAdmin,
		&i.IsCritics,
		&i.IsVerified,
		&i.SocialMedia,
	)
	return i, err
}

const getUserStatsQ = `
SELECT
json_agg(ur.*) AS reviews,
( SELECT COUNT(id) FROM user_follow u WHERE u.followee_id = $1) as followers,
( SELECT COUNT(id) FROM reviews r WHERE r.submitted_by = $1 ) as score_count,
( SELECT
    json_agg(r1.*) AS scores_distribution
    FROM
        (
        SELECT
            *
            FROM
( SELECT COUNT(id) as "0" FROM reviews r WHERE r.score >= 0 AND r.score <= 9 AND r.submitted_by = $1) as score0,
( SELECT COUNT(id) as "1" FROM reviews r WHERE r.score >= 10 AND r.score <= 19 AND r.submitted_by = $1) as score1,
( SELECT COUNT(id) as "2" FROM reviews r WHERE r.score >= 20 AND r.score <= 29 AND r.submitted_by = $1) as score2,
( SELECT COUNT(id) as "3" FROM reviews r WHERE r.score >= 30 AND r.score <= 39 AND r.submitted_by = $1) as score3,
( SELECT COUNT(id) as "4" FROM reviews r WHERE r.score >= 40 AND r.score <= 49 AND r.submitted_by = $1) as score4,
( SELECT COUNT(id) as "5" FROM reviews r WHERE r.score >= 50 AND r.score <= 59 AND r.submitted_by = $1) as score5,
( SELECT COUNT(id) as "6" FROM reviews r WHERE r.score >= 60 AND r.score <= 69 AND r.submitted_by = $1) as score6,
( SELECT COUNT(id) as "7" FROM reviews r WHERE r.score >= 70 AND r.score <= 79 AND r.submitted_by = $1) as score7,
( SELECT COUNT(id) as "8" FROM reviews r WHERE r.score >= 80 AND r.score <= 89 AND r.submitted_by = $1) as score8,
( SELECT COUNT(id) as "9" FROM reviews r WHERE r.score >= 90 AND r.score <= 99 AND r.submitted_by = $1) as score9,
( SELECT COUNT(id) as "99" FROM reviews r WHERE r.score = 100 AND r.submitted_by = $1) as score10
     ) r1
)
FROM (
    SELECT
    reviews.id,
		reviews.comments,
    score,
    l.name,
    p.province_name,
    COALESCE(l.thumbnail, '') as thumbnail
    FROM
    reviews
    JOIN locations l on reviews.location_id = l.id
    JOIN regencies r on l.regency_id = r.id
    JOIN provinces p on r.province_id = p.id
    WHERE reviews.submitted_by = $1
    ORDER BY score ASC
    LIMIT 10
 ) ur
`

type GetUserStatsRow struct {
	Reviews            []byte `json:"-"`
	Followers          int32  `json:"followers"`
	ScoreCount         int32  `json:"score_count"`
	ScoresDistribution []byte `json:"-"`
}

func (q *Queries) GetUserStats(ctx context.Context, user_id int32) (GetUserStatsRow, error) {
	var i GetUserStatsRow
	row := q.db.QueryRowContext(ctx, getUserStatsQ, user_id)

	err := row.Scan(
		&i.Reviews,
		&i.Followers,
		&i.ScoreCount,
		&i.ScoresDistribution,
	)

	var r []map[string]any

	err = json.Unmarshal(i.ScoresDistribution, &r)
	fmt.Println(r)

	return i, err

}