Improvements

- Fixed Transport type handling - pass correct goal start date
twitchdev · Oct 27, 2021 · 1602fbf · 1602fbf
1 parent 990c671
commit 1602fbf
Showing 1 changed file with 54 additions and 38 deletions.
diff --git a/internal/events/types/goal/goal_event.go b/internal/events/types/goal/goal_event.go
@@ -33,58 +33,74 @@ func (e Event) GenerateEvent(params events.MockEventParameters) (events.MockEven
 	var event []byte
 	var err error
 	var isAchieved *bool
-	var endDate *string
+	var goalEndDate *string
 	var goalType string
+	var currentAmount int64
+	var targetAmount int64
 
-	createdAt := util.GetTimestamp()
-	switch params.Trigger {
-	case "goal-end":
-		date := createdAt.Add(time.Hour * 24).Format(time.RFC3339)
-		endDate = &date
+	goalStartedAt := util.GetTimestamp()
+	currentAmount = util.RandomInt(10 * 10)
+	targetAmount = util.RandomInt(10 * 100)
+
+	if params.Trigger == "goal-end" {
+		endDate := goalStartedAt.Add(time.Hour * 24).Format(time.RFC3339)
+		goalEndDate = &endDate
 
 		achieved := util.RandomInt(1) == 1
+		if achieved {
+			currentAmount = 100
+			targetAmount = 100
+		}
+
 		isAchieved = &achieved
 	}
 
-	goalType = params.ItemID
+	goalType = params.ItemName
 	if goalType == "" {
 		goalType = "follower"
 	}
 
-	body := *&models.EventsubResponse{
-		Subscription: models.EventsubSubscription{
-			ID:      params.ID,
-			Status:  "enabled",
-			Type:    triggerMapping[params.Transport][params.Trigger],
-			Version: "1",
-			Condition: models.EventsubCondition{
-				BroadcasterUserID: params.ToUserID,
+	switch params.Transport {
+	case models.TransportEventSub:
+
+		body := *&models.EventsubResponse{
+			Subscription: models.EventsubSubscription{
+				ID:      params.ID,
+				Status:  "enabled",
+				Type:    triggerMapping[params.Transport][params.Trigger],
+				Version: "1",
+				Condition: models.EventsubCondition{
+					BroadcasterUserID: params.ToUserID,
+				},
+				Transport: models.EventsubTransport{
+					Method:   "webhook",
+					Callback: "null",
+				},
+				Cost:      0,
+				CreatedAt: util.GetTimestamp().Format(time.RFC3339Nano),
 			},
-			Transport: models.EventsubTransport{
-				Method:   "webhook",
-				Callback: "null",
+			Event: models.GoalEventSubEvent{
+				ID:                   params.ID,
+				BroadcasterUserID:    params.ToUserID,
+				BroadcasterUserLogin: params.ToUserName,
+				BroadcasterUserName:  params.ToUserName,
+				Type:                 goalType,
+				Description:          params.Description,
+				CurrentAmount:        currentAmount,
+				TargetAmount:         targetAmount,
+				StartedAt:            goalStartedAt.Format(time.RFC3339Nano),
+				EndedAt:              goalEndDate,
+				IsAchieved:           isAchieved,
 			},
-			Cost:      0,
-			CreatedAt: util.GetTimestamp().Format(time.RFC3339Nano),
-		},
-		Event: models.GoalEventSubEvent{
-			ID:                   params.ID,
-			BroadcasterUserID:    params.ToUserID,
-			BroadcasterUserLogin: params.ToUserName,
-			BroadcasterUserName:  params.ToUserName,
-			Type:                 goalType,
-			Description:          params.Description,
-			CurrentAmount:        util.RandomInt(10 * 10),
-			TargetAmount:         util.RandomInt(10 * 100),
-			StartedAt:            util.GetTimestamp().Format(time.RFC3339Nano),
-			EndedAt:              endDate,
-			IsAchieved:           isAchieved,
-		},
-	}
+		}
+
+		event, err = json.Marshal(body)
+		if err != nil {
+			return events.MockEventResponse{}, err
+		}
 
-	event, err = json.Marshal(body)
-	if err != nil {
-		return events.MockEventResponse{}, err
+	default:
+		return events.MockEventResponse{}, nil
 	}
 
 	return events.MockEventResponse{