update: Container service logic

Author: fressive

server/router/api/v1/container_service.go

@@ -50,6 +50,10 @@ func CreateChallengeContainer(c echo.Context) error {
 		return Failed(&c, "You are not in a team.")
 	}
 
+	if challenge.IsSolvedBy(team, ctx.Store) {
+		return Failed(&c, "You have solved this challenge.")
+	}
+
 	flag := fmt.Sprintf("%s{%s}", challenge.Game.FlagPrefix, util.GenerateFlagContent(challenge.FlagFormat, user.UUID))
 
 	if !ctx.Store.CanCreateContainer(user) {

store/challenge.go

@@ -137,3 +137,9 @@ func (c *Challenge) Expired() bool {
 func (c *Challenge) Ongoing() bool {
 	return c.StartTime == 0 || c.StartTime < time.Now().UnixMilli() && !c.Expired()
 }
+
+func (c *Challenge) IsSolvedBy(team *Team, s *Store) bool {
+	var count int64
+	s.db.Model(Flag{}).Where("team_id = ? AND challenge_id = ? AND state >= 1", team.ID, c.ID).Count(&count)
+	return count > 0
+}