package user

import (
	"budget/internal/storage/dao"
	"budget/internal/storage/dao/postgresql"
	"database/sql"

	"github.com/lib/pq"
)

type Database struct {
	*postgresql.DatabasePostgreSQL
}

func NewDatabase(db *postgresql.DatabasePostgreSQL) *Database {
	return &Database{db}
}

func (db *Database) GetAllUsers() ([]*User, error) {
	q := `
		SELECT u.id, u.created_at, u.updated_at, u.email
		FROM public.user u
	`
	rows, err := db.Session.Query(q)
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	us := make([]*User, 0)
	for rows.Next() {
		u := User{}
		err := rows.Scan(&u.ID, &u.CreatedAt, &u.UpdatedAt, &u.Email)
		if err != nil {
			return nil, err
		}
		us = append(us, &u)
	}
	return us, nil
}

func (db *Database) GetUsersByID(id string) (*User, error) {
	q := `
		SELECT u.id, u.created_at, u.updated_at, u.email
		FROM public.user u
		WHERE u.id = $1
	`
	row := db.Session.QueryRow(q, id)

	u := User{}
	err := row.Scan(&u.ID, &u.CreatedAt, &u.UpdatedAt, &u.Email)
	if errPq, ok := err.(*pq.Error); ok {
		return nil, postgresql.HandlePgError(errPq)
	}
	if err == sql.ErrNoRows {
		return nil, dao.NewDAOError(dao.ErrTypeNotFound, err)
	}
	return &u, err
}

func (db *Database) CreateUser(user *User) error {
	q := `
		INSERT INTO public.user
			(email, google_id)
		VALUES
			($1, $2)
		RETURNING id, created_at
	`

	err := db.Session.
		QueryRow(q, user.Email, user.GoogleID).
		Scan(&user.ID, &user.CreatedAt)
	if errPq, ok := err.(*pq.Error); ok {
		return postgresql.HandlePgError(errPq)
	}
	return err
}

func (db *Database) DeleteUser(id string) error {
	q := `
		DELETE FROM public.user
		WHERE id = $1
	`

	_, err := db.Session.Exec(q, id)
	if errPq, ok := err.(*pq.Error); ok {
		return postgresql.HandlePgError(errPq)
	}
	return err
}

func (db *Database) UpdateUser(user *User) error {
	return nil
}

func (db *Database) GetUsersByGoogleID(id string) (*User, error) {
	q := `
		SELECT u.id, u.created_at, u.updated_at, u.email, u.google_id
		FROM public.user u
		WHERE u.google_id = $1
	`
	row := db.Session.QueryRow(q, id)

	u := User{}
	err := row.Scan(&u.ID, &u.CreatedAt, &u.UpdatedAt, &u.Email, &u.GoogleID)
	if errPq, ok := err.(*pq.Error); ok {
		return nil, postgresql.HandlePgError(errPq)
	}
	if err == sql.ErrNoRows {
		return nil, dao.NewDAOError(dao.ErrTypeNotFound, err)
	}
	return &u, err
}