initial commit

models/account.go

@@ -0,0 +1,86 @@
+package models
+
+import (
+	"net/url"
+	"strings"
+	"time"
+
+	validation "github.com/go-ozzo/ozzo-validation"
+	"github.com/go-ozzo/ozzo-validation/is"
+	"github.com/go-pg/pg/orm"
+)
+
+type Account struct {
+	ID        int       `json:"id"`
+	CreatedAt time.Time `json:"created_at,omitempty"`
+	UpdatedAt time.Time `json:"updated_at,omitempty"`
+	LastLogin time.Time `json:"last_login,omitempty"`
+
+	Email  string   `json:"email"`
+	Name   string   `json:"name"`
+	Active bool     `sql:",notnull" json:"active"`
+	Roles  []string `pg:",array" json:"roles,omitempty"`
+
+	Profile *Profile `json:"profile,omitempty"`
+	Token   []*Token `json:"token,omitempty"`
+}
+
+func (a *Account) BeforeInsert(db orm.DB) error {
+	now := time.Now()
+	if a.CreatedAt.IsZero() {
+		a.CreatedAt = now
+		a.UpdatedAt = now
+	}
+	if err := a.Validate(); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (a *Account) BeforeUpdate(db orm.DB) error {
+	if err := a.Validate(); err != nil {
+		return err
+	}
+	a.UpdatedAt = time.Now()
+	return nil
+}
+
+func (a *Account) BeforeDelete(db orm.DB) error {
+	return nil
+}
+
+func (a *Account) Validate() error {
+	a.Email = strings.TrimSpace(a.Email)
+	a.Email = strings.ToLower(a.Email)
+	a.Name = strings.TrimSpace(a.Name)
+
+	return validation.ValidateStruct(a,
+		validation.Field(&a.Email, validation.Required, is.Email, is.LowerCase),
+		validation.Field(&a.Name, validation.Required, is.ASCII),
+	)
+}
+
+func (a *Account) CanLogin() bool {
+	return a.Active
+}
+
+type AccountFilter struct {
+	orm.Pager
+	Filters url.Values
+	Order   []string
+}
+
+func (f *AccountFilter) Filter(q *orm.Query) (*orm.Query, error) {
+	q = q.Apply(f.Pager.Paginate)
+	q = q.Apply(orm.URLFilters(f.Filters))
+	q = q.Order(f.Order...)
+	return q, nil
+}
+
+func NewAccountFilter(v url.Values) AccountFilter {
+	var f AccountFilter
+	f.SetURLValues(v)
+	f.Filters = v
+	f.Order = v["order"]
+	return f
+}

models/profile.go

@@ -0,0 +1,42 @@
+package models
+
+import (
+	"time"
+
+	"github.com/go-ozzo/ozzo-validation"
+
+	"github.com/go-pg/pg/orm"
+)
+
+type Profile struct {
+	ID        int       `json:"id,omitempty"`
+	AccountID int       `json:"-"`
+	CreatedAt time.Time `json:"created_at,omitempty"`
+	UpdatedAt time.Time `json:"updated_at,omitempty"`
+
+	Theme string `json:"theme,omitempty"`
+}
+
+func (p *Profile) BeforeInsert(db orm.DB) error {
+	now := time.Now()
+	if p.CreatedAt.IsZero() {
+		p.CreatedAt = now
+		p.UpdatedAt = now
+	}
+	return nil
+}
+
+func (p *Profile) BeforeUpdate(db orm.DB) error {
+	if err := p.Validate(); err != nil {
+		return err
+	}
+	p.UpdatedAt = time.Now()
+	return nil
+}
+
+func (p *Profile) Validate() error {
+
+	return validation.ValidateStruct(p,
+		validation.Field(&p.Theme, validation.Required, validation.In("default", "dark")),
+	)
+}

models/token.go

@@ -0,0 +1,33 @@
+package models
+
+import (
+	"time"
+
+	"github.com/go-pg/pg/orm"
+)
+
+type Token struct {
+	ID        int       `json:"id,omitempty"`
+	CreatedAt time.Time `json:"created_at,omitempty"`
+	UpdatedAt time.Time `json:"updated_at,omitempty"`
+	AccountID int       `json:"-"`
+
+	Token      string    `json:"-"`
+	Expiry     time.Time `json:"-"`
+	Mobile     bool      `sql:",notnull" json:"mobile"`
+	Identifier string    `json:"identifier,omitempty"`
+}
+
+func (t *Token) BeforeInsert(db orm.DB) error {
+	now := time.Now()
+	if t.CreatedAt.IsZero() {
+		t.CreatedAt = now
+		t.UpdatedAt = now
+	}
+	return nil
+}
+
+func (t *Token) BeforeUpdate(db orm.DB) error {
+	t.UpdatedAt = time.Now()
+	return nil
+}