First commit

2020-02-04 18:46:10 +01:00
commit 190e1eb01d
91 changed files with 13411 additions and 0 deletions
@@ -0,0 +1,137 @@
+package api
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/gisle/stufflog/database"
+	"github.com/gisle/stufflog/models"
+	"github.com/gisle/stufflog/services"
+	"github.com/gisle/stufflog/slerrors"
+	"sort"
+)
+
+func Activity(g *gin.RouterGroup, db database.Database, auth *services.AuthService) {
+	type resObj struct {
+		Activity   *models.Activity   `json:"activity,omitempty"`
+		Activities []*models.Activity `json:"activities,omitempty"`
+	}
+
+	g.Use(auth.GinSessionMiddleware(true))
+
+	// GET / – List own activities.
+	g.GET("/", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+
+		activities, err := db.Activities().ListUser(c.Request.Context(), *user)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		sort.Sort(models.ActivitiesByName(activities))
+
+		c.JSON(200, resObj{Activities: activities})
+	})
+
+	// GET / — Find an activity
+	g.GET("/:id", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+
+		activity, err := db.Activities().FindID(c.Request.Context(), c.Param("id"))
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+		if activity.UserID != user.ID {
+			slerrors.GinRespond(c, slerrors.NotFound("Activity"))
+			return
+		}
+
+		c.JSON(200, resObj{Activity: activity})
+	})
+
+	// POST / – Create an activity
+	g.POST("/", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+		activity := models.Activity{}
+
+		err := c.BindJSON(&activity)
+		if err != nil {
+			slerrors.GinRespond(c, &slerrors.SLError{Code: 400, Text: err.Error()})
+			return
+		}
+
+		activity.UserID = user.ID
+		activity.GenerateIDs()
+
+		err = db.Activities().Insert(c.Request.Context(), activity)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		c.JSON(200, resObj{Activity: &activity})
+	})
+
+	// PATCH /:id – Update one activity
+	g.PATCH("/:id", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+		updates := make([]*models.ActivityUpdate, 0, 8)
+
+		err := c.BindJSON(&updates)
+		if err != nil {
+			slerrors.GinRespond(c, &slerrors.SLError{Code: 400, Text: err.Error()})
+			return
+		}
+
+		activity, err := db.Activities().FindID(c.Request.Context(), c.Param("id"))
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+		if activity.UserID != user.ID {
+			slerrors.GinRespond(c, slerrors.NotFound("Activity"))
+			return
+		}
+
+		activity, err = db.Activities().Update(c.Request.Context(), *activity, updates)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		c.JSON(200, resObj{Activity: activity})
+	})
+
+	// DELETE /:id – Delete an activity
+	g.DELETE("/:id", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+
+		activity, err := db.Activities().FindID(c.Request.Context(), c.Param("id"))
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+		if activity.UserID != user.ID {
+			slerrors.GinRespond(c, slerrors.NotFound("Activity"))
+			return
+		}
+
+		periods, err := db.Periods().ListActivity(c.Request.Context(), *activity)
+		if err != nil {
+			slerrors.GinRespond(c, slerrors.Wrap(err))
+			return
+		}
+		if len(periods) > 0 {
+			slerrors.GinRespond(c, slerrors.PreconditionFailed("You cannot delete an activity that's in use."))
+			return
+		}
+
+		err = db.Activities().Remove(c.Request.Context(), *activity)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		c.JSON(200, resObj{Activity: activity})
+	})
+}
@@ -0,0 +1,178 @@
+package api
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/gisle/stufflog/database"
+	"github.com/gisle/stufflog/models"
+	"github.com/gisle/stufflog/services"
+	"github.com/gisle/stufflog/slerrors"
+	"sort"
+)
+
+func Period(g *gin.RouterGroup, db database.Database, scoring *services.ScoringService, auth *services.AuthService) {
+	type resObj struct {
+		Period     *models.Period     `json:"period,omitempty"`
+		Periods    []*models.Period   `json:"periods,omitempty"`
+		Activities []*models.Activity `json:"activities,omitempty"`
+	}
+
+	g.Use(auth.GinSessionMiddleware(true))
+
+	// GET / – List own periods.
+	g.GET("/", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+
+		periods, err := db.Periods().ListUser(c.Request.Context(), *user)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		sort.Sort(models.PeriodsByFrom(periods))
+
+		c.JSON(200, resObj{Periods: periods})
+	})
+
+	// GET / — Find a period
+	g.GET("/:id", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+
+		period, err := db.Periods().FindID(c.Request.Context(), c.Param("id"))
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+		if period.UserID != user.ID {
+			slerrors.GinRespond(c, slerrors.NotFound("Period"))
+			return
+		}
+
+		activities := make([]*models.Activity, 0, 8)
+		activityAdded := make(map[string]bool)
+		for _, goal := range period.Goals {
+			if !activityAdded[goal.ActivityID] {
+				activityAdded[goal.ActivityID] = true
+
+				activity, err := db.Activities().FindID(c.Request.Context(), goal.ActivityID)
+				if err != nil {
+					slerrors.GinRespond(c, err)
+					return
+				}
+
+				activities = append(activities, activity)
+			}
+		}
+
+		c.JSON(200, resObj{Period: period, Activities: activities})
+	})
+
+	// POST / – Create a period
+	g.POST("/", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+		period := models.Period{}
+
+		err := c.BindJSON(&period)
+		if err != nil {
+			slerrors.GinRespond(c, &slerrors.SLError{Code: 400, Text: err.Error()})
+			return
+		}
+
+		period.GenerateIDs()
+		period.RemoveDuplicateTags()
+		period.UserID = user.ID
+		period.Logs = make([]models.PeriodLog, 0)
+		period.ShouldReScore = false
+
+		if period.Tags == nil {
+			period.Tags = make([]string, 0)
+		}
+		if period.Goals == nil {
+			period.Goals = make([]models.PeriodGoal, 0)
+		}
+		for i := range period.Goals {
+			if period.Goals[i].SubGoals == nil {
+				period.Goals[i].SubGoals = make([]models.PeriodSubGoal, 0)
+			}
+		}
+
+		err = db.Periods().Insert(c.Request.Context(), period)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		c.JSON(200, resObj{Period: &period})
+	})
+
+	// PATCH /:id – Update one Period
+	g.PATCH("/:id", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+		updates := make([]*models.PeriodUpdate, 0, 8)
+
+		err := c.BindJSON(&updates)
+		if err != nil {
+			slerrors.GinRespond(c, &slerrors.SLError{Code: 400, Text: err.Error()})
+			return
+		}
+
+		period, err := db.Periods().FindID(c.Request.Context(), c.Param("id"))
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+		if period.UserID != user.ID {
+			slerrors.GinRespond(c, slerrors.NotFound("Period"))
+			return
+		}
+
+		for _, update := range updates {
+			if update.AddLog != nil {
+				score, err := scoring.ScoreOne(c.Request.Context(), *period, *update.AddLog)
+				if err != nil {
+					slerrors.GinRespond(c, err)
+					return
+				}
+				update.AddLog.Score = score
+			}
+		}
+
+		period, err = db.Periods().Update(c.Request.Context(), *period, updates)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		if period.ShouldReScore {
+			period, err = scoring.ScoreAll(c.Request.Context(), *period)
+			if err != nil {
+				slerrors.GinRespond(c, err)
+				return
+			}
+		}
+
+		c.JSON(200, resObj{Period: period})
+	})
+
+	// DELETE /:id – Delete a period
+	g.DELETE("/:id", func(c *gin.Context) {
+		user := auth.UserFromContext(c)
+
+		period, err := db.Periods().FindID(c.Request.Context(), c.Param("id"))
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+		if period.UserID != user.ID {
+			slerrors.GinRespond(c, slerrors.NotFound("Period"))
+			return
+		}
+
+		err = db.Periods().Remove(c.Request.Context(), *period)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		c.JSON(200, resObj{Period: period})
+	})
+}
@@ -0,0 +1,75 @@
+package api
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/gisle/stufflog/models"
+	"github.com/gisle/stufflog/services"
+	"github.com/gisle/stufflog/slerrors"
+)
+
+func User(g *gin.RouterGroup, auth *services.AuthService) {
+	type loginRegisterRequest struct {
+		Username string `json:"username"`
+		Password string `json:"password"`
+	}
+	type resObj struct {
+		User *models.User `json:"user,omitempty"`
+	}
+
+	g.Use(auth.GinSessionMiddleware(false))
+
+	// GET / – Check session
+	g.GET("/", func(c *gin.Context) {
+		c.JSON(200, resObj{
+			User: auth.UserFromContext(c.Request.Context()),
+		})
+	})
+
+	g.POST("/register", func(c *gin.Context) {
+		data := loginRegisterRequest{}
+		err := c.BindJSON(&data)
+		if err != nil {
+			slerrors.GinRespond(c, &slerrors.SLError{Code: 400, Text: err.Error()})
+			return
+		}
+
+		user, err := auth.Register(c.Request.Context(), data.Username, data.Password)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		c.JSON(200, resObj{User: user})
+	})
+
+	g.POST("/login", func(c *gin.Context) {
+		data := loginRegisterRequest{}
+		err := c.BindJSON(&data)
+		if err != nil {
+			slerrors.GinRespond(c, &slerrors.SLError{Code: 400, Text: err.Error()})
+			return
+		}
+
+		user, session, err := auth.Login(c.Request.Context(), data.Username, data.Password)
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		auth.GinSaveSession(c, *session)
+
+		c.JSON(200, resObj{User: user})
+	})
+
+	g.POST("/logout", func(c *gin.Context) {
+		err := auth.Logout(c.Request.Context())
+		if err != nil {
+			slerrors.GinRespond(c, err)
+			return
+		}
+
+		auth.GinClearSession(c)
+
+		c.JSON(200, resObj{})
+	})
+}