package bd

import (
	"context"
	"fmt"

	"git.ukamnya.ru/stulyaganov/RestApi/internal/bd/model"
)

type Filmsrepo struct {
	db Bd
}

func (f *Filmsrepo) Create(m *model.Films) (*model.Films, error) {
	if err := f.db.db.QueryRow(context.Background(),
		"INSERT INTO films (id, ru_title, orig_title) VALUES($1, $2, $3) RETURNING id",
		m.Id, m.Ru_title, m.Orig_title).Scan(&m.Id); err != nil {
		return nil, err
	}
	return m, nil

}

func FindDyId(id int) (*model.Films, error) {
	return nil, nil
}

func (f *Filmsrepo) FindByAll() (*[]model.Films, error) {
	var buffs []model.Films
	rows, err := f.db.db.Query(context.Background(), "SELECT Id, Ru_title, Orig_title, Imdb_id, Kinopoisk_id, PosterUrl, PosterUrlPreview, Countries, Genres, Year, Description, RatingKinopoisk, RatingImdb, Iframe_src, RatingImdbVoteCount, RatingKinopoiskVoteCount FROM films")
	if err != nil {
		return nil, err
	}

	for rows.Next() {
		var buff model.Films
		err = rows.Scan(
			&buff.Id, &buff.Ru_title, &buff.Orig_title, &buff.Imdb_id, &buff.Kinopoisk_id,
			&buff.PosterUrl, &buff.PosterUrlPreview, &buff.Countries, &buff.Genres, &buff.Year,
			&buff.Description, &buff.RatingKinopoisk, &buff.RatingImdb, &buff.Iframe_src, &buff.RatingImdbVoteCount,
			&buff.RatingKinopoiskVoteCount)
		if err != nil {
			return nil, err
		}
		buffs = append(buffs, buff)
	}
	err = rows.Err()
	if err != nil {
		return nil, err
	}
	return &buffs, nil
}

func (f *Filmsrepo) GetCountFilms() (*int, error) {
	rows, err := f.db.db.Query(context.Background(), "SELECT count(*) FROM films")
	var count int
	if err != nil {
		return nil, err
	}

	for rows.Next() {
		rows.Scan(&count)
	}
	return &count, nil
}

func (f *Filmsrepo) FindById(id string) (*model.Films, error) {
	query := fmt.Sprintf("SELECT * FROM films WHERE Id = %s;", id)
	var buff model.Films
	rows, err := f.db.db.Query(context.Background(), query)
	if err != nil {

		return nil, err
	}
	for rows.Next() {
		rows.Scan(&buff.Id, &buff.Ru_title, &buff.Orig_title, &buff.Imdb_id, &buff.Kinopoisk_id,
			&buff.PosterUrl, &buff.PosterUrlPreview, &buff.Countries, &buff.Genres, &buff.Year,
			&buff.Description, &buff.RatingKinopoisk, &buff.RatingImdb, &buff.Iframe_src, &buff.RatingImdbVoteCount,
			&buff.RatingKinopoiskVoteCount, &buff.Media)

	}

	if rows.Err() != nil {
		return nil, err
	}

	return &buff, nil
}