package repository

import (
	"users_management/m/model/entity"

	"github.com/google/uuid"
	"gorm.io/gorm"
)

type FishboneRepo interface {
	Post(fishbone entity.Fishbone) error
	GetAll() ([]entity.Fishbone, error)
	Update(id uuid.UUID,updates map[string]interface{}) error

    GetByID(id uuid.UUID) (entity.Fishbone, error)
}

type fishboneRepo struct {
    db *gorm.DB
}

func NewFishboneRepo(db *gorm.DB) FishboneRepo {
    return &fishboneRepo{
        db: db,
    }
}

func (r *fishboneRepo) Post(fishbone entity.Fishbone) error {
    err := r.db.Create(&fishbone).Error
    if err != nil {
        return err
    }
    return nil
}

func (r *fishboneRepo) GetAll() ([]entity.Fishbone, error) {
    var fishbones []entity.Fishbone
    err := r.db.Preload("DeviceStart").Preload("DeviceEnd").Preload("Backbone").Preload("Backbone.DeviceStart").Preload("Backbone.DeviceEnd").Find(&fishbones).Error
    if err != nil {
        return fishbones, err   
    }
    return fishbones, nil
}

func (r *fishboneRepo) Update(id uuid.UUID,updates map[string]interface{}) error {
    err := r.db.Model(&entity.Fishbone{}).Where("id = ?", id).Updates(updates).Error
    if err != nil {
        return err
    }
    return nil
}

func (r *fishboneRepo) GetByID(id uuid.UUID) (entity.Fishbone, error) {    
    var fishbone entity.Fishbone
    err := r.db.Where("id = ?", id).Preload("DeviceStart").Preload("DeviceEnd").Preload("Backbone.DeviceStart").Preload("Backbone.DeviceEnd").First(&fishbone).Error
    if err != nil {
        return fishbone, err
    }
    return fishbone, nil
}