user.go

package models

import (
	"fmt"
	"time"

	"github.com/aureleoules/epitaf/db"

	"gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"
)

// User structure
type User struct {
	ID bson.ObjectId `json:"id" bson:"_id"`

	Name  string `json:"name" bson:"name"`
	Email string `json:"email" bson:"email"`

	Classroom int `json:"class_room" bson:"class_room"`
	Group  int `json:"group" bson:"group"`
	Promotion int `json:"promotion" bson:"promotion"`

	CreatedAt time.Time `json:"creation_at" bson:"creation_at"`
}

// MicrosoftProfile struct
type MicrosoftProfile struct {
	DisplayName string `json:"displayName"`
	Mail        string `json:"mail"`
}

// SaveSettings saves user's settings
func (u *User) SaveSettings(promo int, class int, group int) error {
	err := db.Epitaf.C(db.UsersCollection).UpdateId(u.ID, bson.M{
		"$set": bson.M{
			"promotion":  promo,
			"class_room": class,
			"group":   group,
		},
	})
	return err
}

// GetUser returns specific user
func GetUser(id bson.ObjectId) (User, error) {
	var user User
	err := db.Epitaf.C(db.UsersCollection).FindId(id).One(&user)
	return user, err
}

// FindUser by microsoft email
func FindUser(profile MicrosoftProfile) (User, error) {
	var user User
	err := db.Epitaf.C(db.UsersCollection).Find(bson.M{
		"email": profile.Mail,
	}).One(&user)
	fmt.Println("user", user)
	fmt.Println("error", err)
	if err == mgo.ErrNotFound {
		user := User{
			ID:        bson.NewObjectId(),
			Email:     profile.Mail,
			Name:      profile.DisplayName,
			CreatedAt: time.Now(),
		}
		err = db.Epitaf.C(db.UsersCollection).Insert(user)
		return user, err
	}
	return user, nil
}