wakapi/models/heartbeat.go

package models

import (
	"regexp"
	"time"
)

type CustomTime time.Time

type Heartbeat struct {
	ID              uint       `gorm:"primary_key"`
	User            *User      `json:"-" gorm:"not null"`
	UserID          string     `json:"-" gorm:"not null; index:idx_time_user"`
	Entity          string     `json:"entity" gorm:"not null; index:idx_entity"`
	Type            string     `json:"type"`
	Category        string     `json:"category"`
	Project         string     `json:"project"`
	Branch          string     `json:"branch"`
	Language        string     `json:"language" gorm:"index:idx_language"`
	IsWrite         bool       `json:"is_write"`
	Editor          string     `json:"editor"`
	OperatingSystem string     `json:"operating_system"`
	Time            CustomTime `json:"time" gorm:"type:timestamp; default:CURRENT_TIMESTAMP; index:idx_time,idx_time_user"`
	languageRegex   *regexp.Regexp
}

func (h *Heartbeat) Valid() bool {
	return h.User != nil && h.UserID != "" && h.Time != CustomTime(time.Time{})
}

func (h *Heartbeat) Augment(customLangs map[string]string) {
	if h.Language == "" {
		if h.languageRegex == nil {
			h.languageRegex = regexp.MustCompile(`^.+\.(.+)$`)
		}
		groups := h.languageRegex.FindAllStringSubmatch(h.Entity, -1)
		if len(groups) == 0 || len(groups[0]) != 2 {
			return
		}
		ending := groups[0][1]
		if _, ok := customLangs[ending]; !ok {
			return
		}
		h.Language, _ = customLangs[ending]
	}
}