Add team and org specific estimation scale update methods

StevenWeathers · Sep 17, 2024 · 8984eb4 · 8984eb4
1 parent 373774c
commit 8984eb4
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 2 deletions.
diff --git a/internal/db/poker/estimationscale.go b/internal/db/poker/estimationscale.go
@@ -184,6 +184,60 @@ func (d *Service) UpdateEstimationScale(ctx context.Context, scale *thunderdome.
 	return scale, nil
 }
 
+// UpdateTeamEstimationScale updates an existing team estimation scale
+func (d *Service) UpdateTeamEstimationScale(ctx context.Context, scale *thunderdome.EstimationScale) (*thunderdome.EstimationScale, error) {
+	query := `
+		UPDATE thunderdome.estimation_scale
+		SET name = $2, description = $3, scale_type = $4, values = $5, is_public = $6, 
+			default_scale = $8, updated_at = CURRENT_TIMESTAMP
+		WHERE id = $1 AND team_id = $7
+		RETURNING updated_at;
+	`
+	err := d.DB.QueryRowContext(ctx, query,
+		scale.ID,
+		scale.Name,
+		scale.Description,
+		scale.ScaleType,
+		scale.Values,
+		scale.IsPublic,
+		scale.TeamID,
+		scale.DefaultScale,
+	).Scan(&scale.UpdatedAt)
+
+	if err != nil {
+		return nil, fmt.Errorf("error updating estimation scale: %v", err)
+	}
+
+	return scale, nil
+}
+
+// UpdateOrganizationEstimationScale updates an existing organization estimation scale
+func (d *Service) UpdateOrganizationEstimationScale(ctx context.Context, scale *thunderdome.EstimationScale) (*thunderdome.EstimationScale, error) {
+	query := `
+		UPDATE thunderdome.estimation_scale
+		SET name = $2, description = $3, scale_type = $4, values = $5, is_public = $6, 
+			default_scale = $8, updated_at = CURRENT_TIMESTAMP
+		WHERE id = $1 AND organization_id = $7
+		RETURNING updated_at;
+	`
+	err := d.DB.QueryRowContext(ctx, query,
+		scale.ID,
+		scale.Name,
+		scale.Description,
+		scale.ScaleType,
+		scale.Values,
+		scale.IsPublic,
+		scale.OrganizationID,
+		scale.DefaultScale,
+	).Scan(&scale.UpdatedAt)
+
+	if err != nil {
+		return nil, fmt.Errorf("error updating estimation scale: %v", err)
+	}
+
+	return scale, nil
+}
+
 // DeleteEstimationScale deletes an estimation scale
 func (d *Service) DeleteEstimationScale(ctx context.Context, scaleID string) error {
 	query := `DELETE FROM thunderdome.estimation_scale WHERE id = $1;`

diff --git a/internal/http/estimationscale.go b/internal/http/estimationscale.go
@@ -602,7 +602,7 @@ func (s *Service) handleTeamEstimationScaleUpdate() http.HandlerFunc {
 			IsPublic:     false,
 		}
 
-		updatedScale, err := s.PokerDataSvc.UpdateEstimationScale(ctx, &es)
+		updatedScale, err := s.PokerDataSvc.UpdateTeamEstimationScale(ctx, &es)
 		if err != nil {
 			s.Logger.Ctx(ctx).Error("handleEstimationScaleUpdate error", zap.Error(err),
 				zap.String("scale_id", ID),
@@ -669,7 +669,7 @@ func (s *Service) handleOrganizationEstimationScaleUpdate() http.HandlerFunc {
 			IsPublic:       false,
 		}
 
-		updatedScale, err := s.PokerDataSvc.UpdateEstimationScale(ctx, &es)
+		updatedScale, err := s.PokerDataSvc.UpdateOrganizationEstimationScale(ctx, &es)
 		if err != nil {
 			s.Logger.Ctx(ctx).Error("handleEstimationScaleUpdate error", zap.Error(err),
 				zap.String("scale_id", ID),

diff --git a/thunderdome/poker.go b/thunderdome/poker.go
@@ -172,4 +172,8 @@ type PokerDataSvc interface {
 	DeleteOrganizationEstimationScale(ctx context.Context, orgID string, scaleID string) error
 	// DeleteTeamEstimationScale deletes a team's estimation scale by its ID
 	DeleteTeamEstimationScale(ctx context.Context, teamID string, scaleID string) error
+	// UpdateOrganizationEstimationScale updates an existing organization estimation scale
+	UpdateOrganizationEstimationScale(ctx context.Context, scale *EstimationScale) (*EstimationScale, error)
+	// UpdateTeamEstimationScale updates an existing team estimation scale
+	UpdateTeamEstimationScale(ctx context.Context, scale *EstimationScale) (*EstimationScale, error)
 }