package repo

import (
	"gorm.io/gorm"
	"os/user"
	"work_cation/models"
	"work_cation/pkg/utils"
)

type userRepo struct{}

var User *userRepo

func (*userRepo) GetUserInfo(db *gorm.DB) *models.Users {
	var users models.Users
	ip := utils.IP.Get192Ip()
	db.Where("ip = ?", ip).Find(&users)

	if users.Ip != ip {
		current, err := user.Current()
		if err != nil {
			current = &user.User{Name: ip}
		}
		users = models.Users{
			ID:      utils.Uuid.CreateUUID(),
			Ip:      ip,
			Name:    current.Username,
			Cover:   "",
			Email:   "",
			Phone:   "",
			Address: "友嘉",
		}
		db.Create(&users)
	}
	return &users
}

func (*userRepo) Update(db *gorm.DB, newUser *models.Users) error {
	return db.Model(&models.Users{}).Where("ID = ?", newUser.ID).Updates(newUser).Error
}