package schemas

import (
	"database/sql/driver"
	"encoding/json"
	"time"
)

type CustomTime struct {
	Time *time.Time
}

// MarshalJSON 实现 JSON 序列化
func (ct CustomTime) MarshalJSON() ([]byte, error) {
	if ct.Time == nil {
		return []byte(`""`), nil
	}
	return json.Marshal(ct.Time.Format("2006-01-02 15:04:05"))
}

// UnmarshalJSON 实现 JSON 反序列化
func (ct *CustomTime) UnmarshalJSON(data []byte) error {
	var timeStr string
	if err := json.Unmarshal(data, &timeStr); err != nil {
		return err
	}

	if timeStr == "" {
		ct.Time = nil
		return nil
	}

	parsedTime, err := time.Parse("2006-01-02 15:04:05", timeStr)
	if err != nil {
		return err
	}

	ct.Time = &parsedTime
	return nil
}

// Scan 实现 sql.Scanner 接口，用于数据库扫描
func (ct *CustomTime) Scan(value interface{}) error {
	if value == nil {
		ct.Time = nil
		return nil
	}

	switch v := value.(type) {
	case time.Time:
		ct.Time = &v
	case *time.Time:
		ct.Time = v
	default:
		ct.Time = nil
	}
	return nil
}

// Value 实现 driver.Valuer 接口，用于数据库存储
func (ct CustomTime) Value() (driver.Value, error) {
	if ct.Time == nil {
		return nil, nil
	}
	return *ct.Time, nil
}

// NewCustomTime 创建新的 CustomTime 实例
func NewCustomTime(t *time.Time) CustomTime {
	return CustomTime{Time: t}
}