package gorm

import (
	"context"
	"fmt"
	"time"

	"gitea.neatflow.kr/biosvos/seven-skies/internal/pkg/domain"
	"gitea.neatflow.kr/biosvos/seven-skies/internal/pkg/repository"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

var _ repository.Repository = (*Repository)(nil)

type Repository struct {
	db *gorm.DB
}

func NewRepository(dsn string) (*Repository, error) {
	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{}) //nolint:exhaustruct
	if err != nil {
		return nil, fmt.Errorf("failed to open database: %w", err)
	}

	err = db.AutoMigrate(&Weather{}) //nolint:exhaustruct
	if err != nil {
		return nil, fmt.Errorf("failed to migrate database: %w", err)
	}

	return &Repository{
		db: db,
	}, nil
}

func (r *Repository) CreateWeather(ctx context.Context, weather *domain.Weather) error {
	modelWeather := newModelWeather(weather)

	err := gorm.G[Weather](r.db).Create(ctx, modelWeather)
	if err != nil {
		return fmt.Errorf("failed to create weather: %w", err)
	}

	return nil
}

func (r *Repository) ListWeathers(ctx context.Context, from, to time.Time) ([]*domain.Weather, error) {
	weathers, err := gorm.G[*Weather](r.db).Where("target_date BETWEEN ? AND ?", from, to).Find(ctx)
	if err != nil {
		return nil, fmt.Errorf("failed to list weathers: %w", err)
	}

	var ret []*domain.Weather
	for _, weather := range weathers {
		ret = append(ret, weather.toDomain())
	}

	return ret, nil
}