124 lines
2.1 KiB
Go
124 lines
2.1 KiB
Go
|
package gotime
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"strings"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
type Timer struct {
|
||
|
Id int
|
||
|
Beg time.Time
|
||
|
End time.Time
|
||
|
Tags []string
|
||
|
}
|
||
|
|
||
|
func (t *Timer) Start() {
|
||
|
if t.Beg.IsZero() {
|
||
|
t.Beg = time.Now()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (t *Timer) Stop() {
|
||
|
if t.End.IsZero() {
|
||
|
t.End = time.Now()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (t *Timer) SetId(i int) {
|
||
|
t.Id = i
|
||
|
}
|
||
|
|
||
|
func (t *Timer) SetStart(tm time.Time) error {
|
||
|
if !t.End.IsZero() && !tm.Before(t.End) {
|
||
|
return errors.New("Beginning time must be before End")
|
||
|
}
|
||
|
t.Beg = tm
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (t *Timer) SetEnd(tm time.Time) error {
|
||
|
if !tm.After(t.Beg) {
|
||
|
return errors.New("End time must be after Beginning")
|
||
|
}
|
||
|
t.End = tm
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (t *Timer) IsRunning() bool {
|
||
|
return !t.Beg.IsZero() && t.End.IsZero()
|
||
|
}
|
||
|
|
||
|
// AddTag adds the tag tg to this timer
|
||
|
func (t *Timer) AddTag(tg string) {
|
||
|
if !t.HasTag(tg) {
|
||
|
t.Tags = append(t.Tags, tg)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// RemoveTag removes the tag tg from this timer
|
||
|
func (t *Timer) RemoveTag(tg string) {
|
||
|
for i := 0; i < len(t.Tags); i++ {
|
||
|
if t.Tags[i] == tg {
|
||
|
t.Tags = append(t.Tags[:i], t.Tags[i+1:]...)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// HasTag returns if this timer has the tag
|
||
|
func (t *Timer) HasTag(tg string) bool {
|
||
|
for i := range t.Tags {
|
||
|
if t.Tags[i] == tg {
|
||
|
return true
|
||
|
}
|
||
|
}
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
func (t *Timer) ToString() string {
|
||
|
if t.Beg.IsZero() {
|
||
|
return ""
|
||
|
}
|
||
|
timeFormat := "20060102T150405Z"
|
||
|
|
||
|
ret := "inc " + t.Beg.Format(timeFormat)
|
||
|
if !t.End.IsZero() {
|
||
|
ret += " - " + t.End.Format(timeFormat)
|
||
|
}
|
||
|
if len(t.Tags) > 0 {
|
||
|
ret += " # "
|
||
|
for i := range t.Tags {
|
||
|
if strings.Contains(t.Tags[i], " ") {
|
||
|
ret += "\"" + t.Tags[i] + "\""
|
||
|
}
|
||
|
ret += " "
|
||
|
}
|
||
|
ret = ret[:len(ret)-1]
|
||
|
}
|
||
|
return ret
|
||
|
}
|
||
|
|
||
|
func (t *Timer) ToJsonString() string {
|
||
|
if t.Beg.IsZero() {
|
||
|
return ""
|
||
|
}
|
||
|
timeFormat := "20060102T150405Z"
|
||
|
|
||
|
ret := "{\"start\":\"" + t.Beg.Format(timeFormat) + "\","
|
||
|
if !t.End.IsZero() {
|
||
|
ret += "\"end\":\"" + t.End.Format(timeFormat) + "\","
|
||
|
}
|
||
|
if len(t.Tags) > 0 {
|
||
|
ret += "\"tags\":["
|
||
|
for i := range t.Tags {
|
||
|
ret += "\"" + t.Tags[i] + "\","
|
||
|
}
|
||
|
// Move trailing ',' to end
|
||
|
ret = ret[:len(ret)-1] + "],"
|
||
|
|
||
|
}
|
||
|
// Trim trailing ,
|
||
|
ret = ret[:len(ret)-1] + "}"
|
||
|
return ret
|
||
|
}
|