package time

import "time"

// DryccDatetimeFormat is the standard date/time representation used in Drycc.
const DryccDatetimeFormat = "2006-01-02T15:04:05MST"

// PyOpenSSLTimeDateTimeFormat is a different date format to deal with the pyopenssl formatting
// http://www.pyopenssl.org/en/stable/api/crypto.html#OpenSSL.crypto.X509.get_notAfter
const PyOpenSSLTimeDateTimeFormat = "2006-01-02T15:04:05"

// Time represents the standard datetime format used across the Drycc Platform.
type Time struct {
	*time.Time
}

func (t *Time) format() string {
	return t.Format(DryccDatetimeFormat)
}

// MarshalJSON implements the json.Marshaler interface.
// The time is a quoted string in Drycc' datetime format.
func (t *Time) MarshalJSON() ([]byte, error) {
	return []byte(t.Format(`"` + DryccDatetimeFormat + `"`)), nil
}

// UnmarshalText implements the encoding.TextUnmarshaler interface.
// The time is expected to be in Drycc' datetime format.
func (t *Time) UnmarshalText(data []byte) error {
	tt, err := time.Parse(time.RFC3339, string(data))
	if _, ok := err.(*time.ParseError); ok {
		tt, err = time.Parse(DryccDatetimeFormat, string(data))
		if _, ok := err.(*time.ParseError); ok {
			tt, err = time.Parse(PyOpenSSLTimeDateTimeFormat, string(data))
		}
	}
	*t = Time{&tt}
	return err
}

// UnmarshalJSON implements the json.Unmarshaler interface.
// The time is expected to be a quoted string in Drycc' datetime format.
func (t *Time) UnmarshalJSON(data []byte) error {
	// Fractional seconds are handled implicitly by Parse.
	tt, err := time.Parse(`"`+time.RFC3339+`"`, string(data))
	if _, ok := err.(*time.ParseError); ok {
		tt, err = time.Parse(`"`+DryccDatetimeFormat+`"`, string(data))
		if _, ok := err.(*time.ParseError); ok {
			tt, err = time.Parse(`"`+PyOpenSSLTimeDateTimeFormat+`"`, string(data))
		}
	}
	*t = Time{&tt}
	return err
}