2018-10-23 11:57:42 +09:00
|
|
|
// Copyright 2018 The Gitea Authors. All rights reserved.
|
2022-11-28 03:20:29 +09:00
|
|
|
// SPDX-License-Identifier: MIT
|
2018-10-23 11:57:42 +09:00
|
|
|
|
2022-08-25 11:31:57 +09:00
|
|
|
package activities
|
2018-10-23 11:57:42 +09:00
|
|
|
|
|
|
|
import (
|
2023-09-25 22:17:37 +09:00
|
|
|
"context"
|
|
|
|
|
2021-09-19 20:49:59 +09:00
|
|
|
"code.gitea.io/gitea/models/db"
|
2022-03-29 15:29:02 +09:00
|
|
|
"code.gitea.io/gitea/models/organization"
|
2021-11-24 18:49:20 +09:00
|
|
|
user_model "code.gitea.io/gitea/models/user"
|
2018-10-23 11:57:42 +09:00
|
|
|
"code.gitea.io/gitea/modules/setting"
|
2019-08-15 23:46:21 +09:00
|
|
|
"code.gitea.io/gitea/modules/timeutil"
|
2018-10-23 11:57:42 +09:00
|
|
|
)
|
|
|
|
|
|
|
|
// UserHeatmapData represents the data needed to create a heatmap
|
|
|
|
type UserHeatmapData struct {
|
2019-08-15 23:46:21 +09:00
|
|
|
Timestamp timeutil.TimeStamp `json:"timestamp"`
|
|
|
|
Contributions int64 `json:"contributions"`
|
2018-10-23 11:57:42 +09:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetUserHeatmapDataByUser returns an array of UserHeatmapData
|
2023-09-25 22:17:37 +09:00
|
|
|
func GetUserHeatmapDataByUser(ctx context.Context, user, doer *user_model.User) ([]*UserHeatmapData, error) {
|
|
|
|
return getUserHeatmapData(ctx, user, nil, doer)
|
2020-12-28 04:58:03 +09:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetUserHeatmapDataByUserTeam returns an array of UserHeatmapData
|
2023-09-25 22:17:37 +09:00
|
|
|
func GetUserHeatmapDataByUserTeam(ctx context.Context, user *user_model.User, team *organization.Team, doer *user_model.User) ([]*UserHeatmapData, error) {
|
|
|
|
return getUserHeatmapData(ctx, user, team, doer)
|
2020-12-28 04:58:03 +09:00
|
|
|
}
|
|
|
|
|
2023-09-25 22:17:37 +09:00
|
|
|
func getUserHeatmapData(ctx context.Context, user *user_model.User, team *organization.Team, doer *user_model.User) ([]*UserHeatmapData, error) {
|
2018-10-24 22:17:21 +09:00
|
|
|
hdata := make([]*UserHeatmapData, 0)
|
2020-06-06 05:01:53 +09:00
|
|
|
|
2022-08-25 11:31:57 +09:00
|
|
|
if !ActivityReadable(user, doer) {
|
2020-06-06 05:01:53 +09:00
|
|
|
return hdata, nil
|
|
|
|
}
|
|
|
|
|
2021-06-26 01:59:25 +09:00
|
|
|
// Group by 15 minute intervals which will allow the client to accurately shift the timestamp to their timezone.
|
|
|
|
// The interval is based on the fact that there are timezones such as UTC +5:30 and UTC +12:45.
|
|
|
|
groupBy := "created_unix / 900 * 900"
|
2024-04-05 01:02:24 +09:00
|
|
|
if setting.Database.Type.IsMySQL() {
|
2021-06-26 01:59:25 +09:00
|
|
|
groupBy = "created_unix DIV 900 * 900"
|
2018-10-23 11:57:42 +09:00
|
|
|
}
|
|
|
|
|
2023-09-29 21:12:54 +09:00
|
|
|
cond, err := activityQueryCondition(ctx, GetFeedsOptions{
|
2020-12-22 11:53:37 +09:00
|
|
|
RequestedUser: user,
|
2020-12-28 04:58:03 +09:00
|
|
|
RequestedTeam: team,
|
2020-12-22 11:53:37 +09:00
|
|
|
Actor: doer,
|
|
|
|
IncludePrivate: true, // don't filter by private, as we already filter by repo access
|
|
|
|
IncludeDeleted: true,
|
|
|
|
// * Heatmaps for individual users only include actions that the user themself did.
|
|
|
|
// * For organizations actions by all users that were made in owned
|
|
|
|
// repositories are counted.
|
|
|
|
OnlyPerformedBy: !user.IsOrganization(),
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
2019-01-07 04:29:05 +09:00
|
|
|
}
|
|
|
|
|
2023-09-25 22:17:37 +09:00
|
|
|
return hdata, db.GetEngine(ctx).
|
2020-12-22 11:53:37 +09:00
|
|
|
Select(groupBy+" AS timestamp, count(user_id) as contributions").
|
|
|
|
Table("action").
|
|
|
|
Where(cond).
|
2021-04-09 16:40:34 +09:00
|
|
|
And("created_unix > ?", timeutil.TimeStampNow()-31536000).
|
2024-04-05 01:02:24 +09:00
|
|
|
GroupBy("timestamp").
|
2018-10-23 11:57:42 +09:00
|
|
|
OrderBy("timestamp").
|
|
|
|
Find(&hdata)
|
|
|
|
}
|
2023-04-18 03:26:01 +09:00
|
|
|
|
|
|
|
// GetTotalContributionsInHeatmap returns the total number of contributions in a heatmap
|
|
|
|
func GetTotalContributionsInHeatmap(hdata []*UserHeatmapData) int64 {
|
|
|
|
var total int64
|
|
|
|
for _, v := range hdata {
|
|
|
|
total += v.Contributions
|
|
|
|
}
|
|
|
|
return total
|
|
|
|
}
|