From 992894c87bfdb565d90ff09b774afb07553c5896 Mon Sep 17 00:00:00 2001
From: arnaucube <arnaucube@gmail.com>
Date: Mon, 17 Jun 2019 19:50:49 +0200
Subject: [PATCH] add countdown indicator on building

---
 models/planet.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/models/planet.go b/models/planet.go
index 5439664..de51caf 100644
--- a/models/planet.go
+++ b/models/planet.go
@@ -11,9 +11,10 @@ import (
 
 type Process struct {
 	// if Title == "", is not active, and can build other buildings/research
-	Title    string // building name / research name + level
-	Building string
-	Ends     time.Time
+	Title     string // building name / research name + level
+	Building  string
+	Ends      time.Time
+	CountDown int64
 }
 
 type Resources struct {
@@ -221,6 +222,8 @@ func (p *Planet) CheckCurrentBuild() (bool, error) {
 			}
 			return false, nil
 		}
+		p.CurrentBuild.CountDown = p.CurrentBuild.Ends.Unix() - time.Now().Unix()
+
 		return true, nil
 	}
 	return false, nil
@@ -253,6 +256,7 @@ func (p *Planet) UpgradeBuilding(building string) error {
 	p.CurrentBuild.Building = building
 	p.CurrentBuild.Title = building + " - Level " + strconv.Itoa(int(p.Buildings[building]))
 	p.CurrentBuild.Ends = endsTime
+	p.CurrentBuild.CountDown = p.CurrentBuild.Ends.Unix() - time.Now().Unix()
 
 	// store planet in db
 	err = p.Db.Planets.Update(bson.M{"_id": p.Id}, p)