mirror of https://github.com/go-gitea/gitea.git
138 lines
3.6 KiB
Go
138 lines
3.6 KiB
Go
![]() |
// Copyright 2020 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
|
||
|
|
||
|
import (
|
||
|
"math"
|
||
|
"strings"
|
||
|
|
||
|
"code.gitea.io/gitea/modules/timeutil"
|
||
|
|
||
|
"github.com/src-d/enry/v2"
|
||
|
)
|
||
|
|
||
|
// LanguageStat describes language statistics of a repository
|
||
|
type LanguageStat struct {
|
||
|
ID int64 `xorm:"pk autoincr"`
|
||
|
RepoID int64 `xorm:"UNIQUE(s) INDEX NOT NULL"`
|
||
|
CommitID string
|
||
|
IsPrimary bool
|
||
|
Language string `xorm:"VARCHAR(30) UNIQUE(s) INDEX NOT NULL"`
|
||
|
Percentage float32 `xorm:"NUMERIC(5,2) NOT NULL DEFAULT 0"`
|
||
|
Color string `xorm:"-"`
|
||
|
CreatedUnix timeutil.TimeStamp `xorm:"INDEX CREATED"`
|
||
|
}
|
||
|
|
||
|
// LanguageStatList defines a list of language statistics
|
||
|
type LanguageStatList []*LanguageStat
|
||
|
|
||
|
func (stats LanguageStatList) loadAttributes() {
|
||
|
for i := range stats {
|
||
|
stats[i].Color = enry.GetColor(stats[i].Language)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (repo *Repository) getLanguageStats(e Engine) (LanguageStatList, error) {
|
||
|
stats := make(LanguageStatList, 0, 6)
|
||
|
if err := e.Where("`repo_id` = ?", repo.ID).Desc("`percentage`").Find(&stats); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
stats.loadAttributes()
|
||
|
return stats, nil
|
||
|
}
|
||
|
|
||
|
// GetLanguageStats returns the language statistics for a repository
|
||
|
func (repo *Repository) GetLanguageStats() (LanguageStatList, error) {
|
||
|
return repo.getLanguageStats(x)
|
||
|
}
|
||
|
|
||
|
// GetTopLanguageStats returns the top language statistics for a repository
|
||
|
func (repo *Repository) GetTopLanguageStats(limit int) (LanguageStatList, error) {
|
||
|
stats, err := repo.getLanguageStats(x)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
topstats := make(LanguageStatList, 0, limit)
|
||
|
var other float32
|
||
|
for i := range stats {
|
||
|
if stats[i].Language == "other" || len(topstats) >= limit {
|
||
|
other += stats[i].Percentage
|
||
|
continue
|
||
|
}
|
||
|
topstats = append(topstats, stats[i])
|
||
|
}
|
||
|
if other > 0 {
|
||
|
topstats = append(topstats, &LanguageStat{
|
||
|
RepoID: repo.ID,
|
||
|
Language: "other",
|
||
|
Color: "#cccccc",
|
||
|
Percentage: float32(math.Round(float64(other)*10) / 10),
|
||
|
})
|
||
|
}
|
||
|
return topstats, nil
|
||
|
}
|
||
|
|
||
|
// UpdateLanguageStats updates the language statistics for repository
|
||
|
func (repo *Repository) UpdateLanguageStats(commitID string, stats map[string]float32) error {
|
||
|
sess := x.NewSession()
|
||
|
if err := sess.Begin(); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer sess.Close()
|
||
|
|
||
|
oldstats, err := repo.getLanguageStats(sess)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
var topLang string
|
||
|
var p float32
|
||
|
for lang, perc := range stats {
|
||
|
if perc > p {
|
||
|
p = perc
|
||
|
topLang = strings.ToLower(lang)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for lang, perc := range stats {
|
||
|
upd := false
|
||
|
llang := strings.ToLower(lang)
|
||
|
for _, s := range oldstats {
|
||
|
// Update already existing language
|
||
|
if strings.ToLower(s.Language) == llang {
|
||
|
s.CommitID = commitID
|
||
|
s.IsPrimary = llang == topLang
|
||
|
s.Percentage = perc
|
||
|
if _, err := sess.ID(s.ID).Cols("`commit_id`", "`percentage`", "`is_primary`").Update(s); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
upd = true
|
||
|
break
|
||
|
}
|
||
|
}
|
||
|
// Insert new language
|
||
|
if !upd {
|
||
|
if _, err := sess.Insert(&LanguageStat{
|
||
|
RepoID: repo.ID,
|
||
|
CommitID: commitID,
|
||
|
IsPrimary: llang == topLang,
|
||
|
Language: lang,
|
||
|
Percentage: perc,
|
||
|
}); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
// Delete old languages
|
||
|
if _, err := sess.Where("`id` IN (SELECT `id` FROM `language_stat` WHERE `repo_id` = ? AND `commit_id` != ?)", repo.ID, commitID).Delete(&LanguageStat{}); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if err = repo.updateIndexerStatus(sess, RepoIndexerTypeStats, commitID); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return sess.Commit()
|
||
|
}
|