2018-10-23 04:57:42 +02:00
// Copyright 2018 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.package models
package models
import (
"code.gitea.io/gitea/modules/setting"
2019-08-15 22:46:21 +08:00
"code.gitea.io/gitea/modules/timeutil"
2018-10-23 04:57:42 +02:00
)
// UserHeatmapData represents the data needed to create a heatmap
type UserHeatmapData struct {
2019-08-15 22:46:21 +08:00
Timestamp timeutil . TimeStamp ` json:"timestamp" `
Contributions int64 ` json:"contributions" `
2018-10-23 04:57:42 +02:00
}
// GetUserHeatmapDataByUser returns an array of UserHeatmapData
2021-03-15 02:52:12 +08:00
func GetUserHeatmapDataByUser ( user , doer * User ) ( [ ] * UserHeatmapData , error ) {
2020-12-27 20:58:03 +01:00
return getUserHeatmapData ( user , nil , doer )
}
// GetUserHeatmapDataByUserTeam returns an array of UserHeatmapData
func GetUserHeatmapDataByUserTeam ( user * User , team * Team , doer * User ) ( [ ] * UserHeatmapData , error ) {
return getUserHeatmapData ( user , team , doer )
}
func getUserHeatmapData ( user * User , team * Team , doer * User ) ( [ ] * UserHeatmapData , error ) {
2018-10-24 15:17:21 +02:00
hdata := make ( [ ] * UserHeatmapData , 0 )
2020-06-05 22:01:53 +02:00
2020-12-22 02:53:37 +00:00
if ! activityReadable ( user , doer ) {
2020-06-05 22:01:53 +02:00
return hdata , nil
}
2018-10-23 04:57:42 +02:00
var groupBy string
2021-03-15 02:52:12 +08:00
groupByName := "timestamp" // We need this extra case because mssql doesn't allow grouping by alias
2018-10-23 04:57:42 +02:00
switch {
2019-08-24 17:24:45 +08:00
case setting . Database . UseSQLite3 :
2018-10-23 04:57:42 +02:00
groupBy = "strftime('%s', strftime('%Y-%m-%d', created_unix, 'unixepoch'))"
2019-08-24 17:24:45 +08:00
case setting . Database . UseMySQL :
2018-10-31 23:56:32 +03:00
groupBy = "UNIX_TIMESTAMP(DATE(FROM_UNIXTIME(created_unix)))"
2019-08-24 17:24:45 +08:00
case setting . Database . UsePostgreSQL :
2018-10-23 04:57:42 +02:00
groupBy = "extract(epoch from date_trunc('day', to_timestamp(created_unix)))"
2019-08-24 17:24:45 +08:00
case setting . Database . UseMSSQL :
2018-12-11 03:05:24 +01:00
groupBy = "datediff(SECOND, '19700101', dateadd(DAY, 0, datediff(day, 0, dateadd(s, created_unix, '19700101'))))"
2018-11-01 19:12:17 +01:00
groupByName = groupBy
2018-10-23 04:57:42 +02:00
}
2020-12-22 02:53:37 +00:00
cond , err := activityQueryCondition ( GetFeedsOptions {
RequestedUser : user ,
2020-12-27 20:58:03 +01:00
RequestedTeam : team ,
2020-12-22 02:53:37 +00: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-06 20:29:05 +01:00
}
2020-12-22 02:53:37 +00:00
return hdata , x .
Select ( groupBy + " AS timestamp, count(user_id) as contributions" ) .
Table ( "action" ) .
Where ( cond ) .
And ( "created_unix > ?" , ( timeutil . TimeStampNow ( ) - 31536000 ) ) .
GroupBy ( groupByName ) .
2018-10-23 04:57:42 +02:00
OrderBy ( "timestamp" ) .
Find ( & hdata )
}