mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
package models
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"gorm.io/gorm"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
UserKey = "user"
|
|
ImprintKey = "imprint"
|
|
AuthCookieKey = "wakapi_auth"
|
|
)
|
|
|
|
type MigrationFunc func(db *gorm.DB) error
|
|
|
|
type KeyStringValue struct {
|
|
Key string `gorm:"primary_key"`
|
|
Value string `gorm:"type:text"`
|
|
}
|
|
|
|
type Interval struct {
|
|
Start time.Time
|
|
End time.Time
|
|
}
|
|
|
|
type KeyedInterval struct {
|
|
Interval
|
|
Key *IntervalKey
|
|
}
|
|
|
|
// CustomTime is a wrapper type around time.Time, mainly used for the purpose of transparently unmarshalling Python timestamps in the format <sec>.<nsec> (e.g. 1619335137.3324468)
|
|
type CustomTime time.Time
|
|
|
|
func (j *CustomTime) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(j.T())
|
|
}
|
|
|
|
func (j *CustomTime) UnmarshalJSON(b []byte) error {
|
|
s := strings.Trim(string(b), "\"")
|
|
ts, err := strconv.ParseFloat(s, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
t := time.Unix(0, int64(ts*1e9)) // ms to ns
|
|
*j = CustomTime(t)
|
|
return nil
|
|
}
|
|
|
|
func (j *CustomTime) Scan(value interface{}) error {
|
|
var (
|
|
t time.Time
|
|
err error
|
|
)
|
|
|
|
switch value.(type) {
|
|
case string:
|
|
// with sqlite, some queries (like GetLastByUser()) return dates as strings,
|
|
// however, most of the time they are returned as time.Time
|
|
t, err = time.Parse("2006-01-02 15:04:05-07:00", value.(string))
|
|
if err != nil {
|
|
return errors.New(fmt.Sprintf("unsupported date time format: %s", value))
|
|
}
|
|
case time.Time:
|
|
t = value.(time.Time)
|
|
break
|
|
default:
|
|
return errors.New(fmt.Sprintf("unsupported type: %T", value))
|
|
}
|
|
|
|
t = time.Unix(0, (t.UnixNano()/int64(time.Millisecond))*int64(time.Millisecond)) // round to millisecond precision
|
|
*j = CustomTime(t)
|
|
|
|
return nil
|
|
}
|
|
|
|
func (j CustomTime) Value() (driver.Value, error) {
|
|
t := time.Unix(0, j.T().UnixNano()/int64(time.Millisecond)*int64(time.Millisecond)) // round to millisecond precision
|
|
return t, nil
|
|
}
|
|
|
|
func (j *CustomTime) Hash() (uint64, error) {
|
|
return uint64((j.T().UnixNano() / 1000) / 1000), nil
|
|
}
|
|
|
|
func (j CustomTime) String() string {
|
|
return j.T().String()
|
|
}
|
|
|
|
func (j CustomTime) T() time.Time {
|
|
return time.Time(j)
|
|
}
|
|
|
|
func (j CustomTime) Valid() bool {
|
|
return j.T().Unix() >= 0
|
|
}
|