package models import ( "errors" "time" "github.com/arnaucube/gogame/database" "gopkg.in/mgo.v2/bson" ) type Resources struct { Metal int64 Crystal int64 Deuterium int64 Energy int64 } // UserDb is the data in DB type UserDb struct { Id bson.ObjectId `json:"id" bson:"_id,omitempty"` Name string Password string Email string LastUpdated time.Time Resources Resources Planets []bson.ObjectId } // User is the data in memory, after getting it from DB type User struct { Id bson.ObjectId `json:"id" bson:"_id,omitempty"` Name string LastUpdated time.Time db *database.Db Resources Resources } func NewUser(db *database.Db, name, password, email string) (*User, error) { newUser := UserDb{ Id: bson.NewObjectId(), Name: name, Password: password, Email: email, LastUpdated: time.Now(), Resources: Resources{ Metal: 500, Crystal: 500, Deuterium: 500, Energy: 500, }, } err := db.Users.Insert(newUser) if err != nil { return nil, err } user := UserDbToUser(db, newUser) return user, nil } func UserDbToUser(db *database.Db, u UserDb) *User { return &User{ Id: u.Id, Name: u.Name, LastUpdated: u.LastUpdated, db: db, Resources: u.Resources, } } func (u *User) StoreInDb() error { userDb := UserDb{ Id: u.Id, Name: u.Name, LastUpdated: time.Now(), Resources: u.Resources, } err := u.db.Users.Update(bson.M{"_id": u.Id}, userDb) return err } func (u *User) GetFromDb() error { var userDb UserDb err := u.db.Users.Find(bson.M{"_id": u.Id}).One(&userDb) if err != nil { return err } u = UserDbToUser(u.db, userDb) return nil } func (u *User) GetPlanets() ([]Planet, error) { var planets []Planet err := u.db.Planets.Find(bson.M{"ownerid": u.Id}).All(&planets) if err != nil { return planets, err } return planets, nil } // GetResources updates the values of resources and returns the value, also updates the user.Resources // Resource types: metal, crystal, deuterium, energy func (u *User) GetResources() (*Resources, error) { // get current values err := u.GetFromDb() if err != nil { return nil, err } // calculate Delta time = currentTime - u.LastUpdated delta := time.Since(u.LastUpdated) // get planets planets, err := u.GetPlanets() if err != nil { return nil, err } // get Resource-Plant level in each planet // and calculate growth = ResourcePlant.Level for each planet var metalGrowth, crystalGrowth, deuteriumGrowth, energyGrowth int64 for _, planet := range planets { // TODO find correct formulas metalGrowth = metalGrowth + ((1 + planet.Buildings["metalmine"]) * int64(delta)) crystalGrowth = crystalGrowth + ((1 + planet.Buildings["crystalmine"]) * int64(delta)) deuteriumGrowth = deuteriumGrowth + ((1 + planet.Buildings["deuteriummine"]) * int64(delta)) energyGrowth = energyGrowth + ((1 + planet.Buildings["energymine"]) * int64(delta)) } // calculate newAmount = oldAmount + growth u.Resources.Metal = u.Resources.Metal + metalGrowth u.Resources.Crystal = u.Resources.Crystal + crystalGrowth u.Resources.Deuterium = u.Resources.Deuterium + deuteriumGrowth u.Resources.Energy = u.Resources.Energy + energyGrowth // store new amount to user db err = u.StoreInDb() if err != nil { return nil, err } // return user return &u.Resources, nil } // SpendResources checks if user has enough resources, then substracts the resources, and updates the amounts in the database func (u *User) SpendResources(r Resources) error { err := u.GetFromDb() if err != nil { return err } if u.Resources.Metal < r.Metal { return errors.New("not enough metal resources") } if u.Resources.Crystal < r.Crystal { return errors.New("not enough crystal resources") } if u.Resources.Deuterium < r.Deuterium { return errors.New("not enough deuterium resources") } if u.Resources.Energy < r.Energy { return errors.New("not enough energy resources") } u.Resources.Metal = u.Resources.Metal - r.Metal u.Resources.Crystal = u.Resources.Crystal - r.Crystal u.Resources.Deuterium = u.Resources.Deuterium - r.Deuterium u.Resources.Energy = u.Resources.Energy - r.Energy err = u.StoreInDb() if err != nil { return err } return nil }