pixel (#26)

cmd/v3/main.go

@@ -52,6 +52,7 @@ func main() {
 	r.GET("/api/v1/github/profiles/:social_provider_user_id/views/count.json", statsController.GitHubDayWeekMonthTotalCount)
 	r.GET("/api/v1/github/profiles/:social_provider_user_id/views/stats.json", statsController.GitHubStats)
 	r.GET("/api/v1/github/profiles/:social_provider_user_id/views/day-week-month-total-count.svg", statsController.GitHubDayWeekMonthTotalCountBadge)
+	r.GET("/api/v1/github/profiles/:social_provider_user_id/views/pixel.svg", statsController.Pixel)
 	r.GET("/api/v1/github/profiles/:social_provider_user_id/views/total-count.svg", statsController.TotalCountBadge)
 	r.GET("/api/v1/github/profiles/:social_provider_user_id/referrals/stats.json", statsController.ReferralsStats)
 	r.GET("/api/v1/users/stats.json", statsController.UsersCreatedAtStatsByDay)

internal/controllers/stats_controller.go

@@ -88,6 +88,25 @@ func (c *StatsController) GitHubDayWeekMonthTotalCountBadge(ctx *gin.Context) {
 	ctx.Data(http.StatusOK, "image/svg+xml", []byte(tmv2.Badge(statsCount)))
 }
 
+func (c *StatsController) Pixel(ctx *gin.Context) {
+	const (
+		// language=SVG
+		pixel = `<svg xmlns="http://www.w3.org/2000/svg" width="1" height="1">
+  <rect width="1" height="1" fill="white" />
+</svg>`
+	)
+
+	done := c.increment(ctx, dbs.SocialProviderGithub)
+	if done {
+		return
+	}
+
+	ctx.Header("Cache-Control", "no-cache, no-store, must-revalidate")
+	ctx.Header("Pragma", "no-cache")
+	ctx.Header("Expires", "0")
+	ctx.Data(http.StatusOK, "image/svg+xml", []byte(pixel))
+}
+
 func (c *StatsController) GitHubStats(ctx *gin.Context) {
 	socialProviderUserID, done := c.parseSocialProviderUserID(ctx)
 	if done {
@@ -188,6 +207,36 @@ func (c *StatsController) statsCount(ctx *gin.Context, provider dbs.SocialProvid
 	return statsCount, false
 }
 
+func (c *StatsController) increment(ctx *gin.Context, provider dbs.SocialProvider) (done bool) {
+	socialProviderUserID, done := c.parseSocialProviderUserID(ctx)
+	if done {
+		return
+	}
+
+	userID, done := c.toUserID(ctx, provider, socialProviderUserID)
+	if done {
+		return
+	}
+
+	increment := strings.HasPrefix(ctx.GetHeader("User-Agent"), "github-camo")
+	if !increment {
+		return
+	}
+
+	err := c.statsService.Increment(ctx, userID)
+	if err != nil {
+		log.Printf("Database error (stats) %s\n", err)
+
+		ctx.JSON(http.StatusInternalServerError, &ErrorResponse{
+			ErrorMessage: "Database error",
+		})
+
+		return true
+	}
+
+	return false
+}
+
 func (c *StatsController) parseSocialProviderUserID(ctx *gin.Context) (string, bool) {
 	var uri ProfileCountURI
 

internal/services/stats_service.go

@@ -58,6 +58,19 @@ func (s *StatsService) StatsCount(ctx context.Context, userID int64, increment b
 	}, nil
 }
 
+func (s *StatsService) Increment(ctx context.Context, userID int64) (err error) {
+	now := time.Now().UTC().Truncate(time.Hour)
+
+	go func() {
+		err := s.increment(context.Background(), userID, now)
+		if err != nil {
+			log.Printf("Database err %s\n", err)
+		}
+	}()
+
+	return nil
+}
+
 func (s *StatsService) UserDayWeekMonthViewsStatsMap(ctx context.Context, userIDs []int64, now time.Time) (map[int64]models.DayWeekMonthViewsStats, error) {
 	rows, err := s.repository.Queries().ProfileHourlyViewsStats(ctx, dbs.ProfileHourlyViewsStatsParams{
 		Day:     now.AddDate(0, 0, -1),