Add export utility with support for FHG timesheets (CSV)
This commit is contained in:
268
cmd/gime-export/main.go
Normal file
268
cmd/gime-export/main.go
Normal file
@@ -0,0 +1,268 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"git.bullercodeworks.com/brian/gime"
|
||||
userConfig "github.com/br0xen/user-config"
|
||||
)
|
||||
|
||||
const (
|
||||
AppName = "gime"
|
||||
AppVersion = 1
|
||||
DefDBName = "gime.db"
|
||||
DefArchDBName = "gimearch.db"
|
||||
)
|
||||
|
||||
var validOperations map[string][]string
|
||||
var opFuncs map[string]func([]string) int
|
||||
var timeEntries *gime.TimeEntryCollection
|
||||
var gdb *gime.GimeDB
|
||||
var cfg *userConfig.Config
|
||||
|
||||
var fuzzyFormats []string
|
||||
|
||||
func main() {
|
||||
var ret int
|
||||
initialize()
|
||||
var parms []string
|
||||
|
||||
if len(os.Args) > 1 {
|
||||
parms = os.Args[1:]
|
||||
parms[0] = matchParameter(parms[0])
|
||||
} else {
|
||||
// If no parameters were passed, 'help'
|
||||
parms = append(parms, "help")
|
||||
}
|
||||
|
||||
if fn, ok := opFuncs[parms[0]]; ok {
|
||||
ret = fn(parms[1:])
|
||||
} else {
|
||||
fmt.Println("Unknown command")
|
||||
ret = 1
|
||||
}
|
||||
os.Exit(ret)
|
||||
}
|
||||
|
||||
func loadRecentTimeEntries() {
|
||||
timeEntries = gdb.LoadTimeEntryCollection(gime.TypeRecent)
|
||||
}
|
||||
|
||||
func cmdPrintFHGTimesheet(args []string) int {
|
||||
loadRecentTimeEntries()
|
||||
|
||||
var beg, end time.Time
|
||||
searchTags := []string{}
|
||||
|
||||
for _, opt := range args {
|
||||
var tmpBeg, tmpEnd time.Time
|
||||
|
||||
// Do our best to figure out what timers the user wants to list
|
||||
var err error
|
||||
if strings.Contains(opt, "-") {
|
||||
pts := strings.Split(opt, "-")
|
||||
if len(pts[0]) > 0 {
|
||||
// This should be the starting date
|
||||
tmpBeg, err = parseFuzzyTime(pts[0])
|
||||
if err != nil {
|
||||
// We couldn't parse it as a time,
|
||||
// Probably this is just a tag
|
||||
searchTags = append(searchTags, opt)
|
||||
continue
|
||||
}
|
||||
}
|
||||
if len(pts[1]) > 0 {
|
||||
// This should be the ending date
|
||||
tmpEnd, err = parseFuzzyTime(pts[1])
|
||||
if err != nil {
|
||||
searchTags = append(searchTags, opt)
|
||||
continue
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// Tag filters
|
||||
searchTags = append(searchTags, opt)
|
||||
}
|
||||
if !tmpBeg.IsZero() || !tmpEnd.IsZero() {
|
||||
beg, end = tmpBeg, tmpEnd
|
||||
}
|
||||
}
|
||||
if end.IsZero() {
|
||||
end = time.Now()
|
||||
}
|
||||
|
||||
timeSpanFilter := func(t *gime.TimeEntry) bool {
|
||||
return t.GetStart().After(beg) && t.GetEnd().Before(end)
|
||||
}
|
||||
tagFilter := func(t *gime.TimeEntry) bool {
|
||||
for i := range searchTags {
|
||||
if !t.HasTag(searchTags[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
compoundFilter := func(t *gime.TimeEntry) bool {
|
||||
// Otherwise we want to filter timespan and tags
|
||||
return timeSpanFilter(t) && tagFilter(t)
|
||||
}
|
||||
|
||||
ttl := fmt.Sprintf("%s - %s", beg.Format("2006/01/02"), end.Format("2006/01/02"))
|
||||
if len(searchTags) > 0 {
|
||||
ttl = fmt.Sprint(ttl, " ", searchTags)
|
||||
}
|
||||
fmt.Println(ttl)
|
||||
|
||||
timerCollection := filterTimerCollection(timeEntries, compoundFilter)
|
||||
|
||||
dayTimes := make(map[string]float64)
|
||||
for i := beg; i.Before(end); i = i.Add(time.Hour * 24) {
|
||||
dayTimes[i.Format("2006/01/02")] = 0
|
||||
}
|
||||
|
||||
for i := timerCollection.Length() - 1; i >= 0; i-- {
|
||||
wrkTimer := timerCollection.Get(i)
|
||||
dur := wrkTimer.GetEnd().Sub(wrkTimer.GetStart()).Round(time.Minute * 15)
|
||||
dayTimes[wrkTimer.GetStart().Format("2006/01/02")] += (dur.Minutes() / 60)
|
||||
}
|
||||
|
||||
out := [2]string{"", ""}
|
||||
for i := beg; i.Before(end); i = i.Add(time.Hour * 24) {
|
||||
dayString := i.Format("2006/01/02")
|
||||
out[0] += fmt.Sprint(dayString, ",")
|
||||
out[1] += fmt.Sprint(dayTimes[dayString], ",")
|
||||
}
|
||||
fmt.Println(out[0][:len(out[0])-1])
|
||||
fmt.Println(out[1][:len(out[1])-1])
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
func initialize() {
|
||||
var err error
|
||||
validOperations = make(map[string][]string)
|
||||
opFuncs = make(map[string]func([]string) int)
|
||||
|
||||
opFuncs["fhgts"] = cmdPrintFHGTimesheet
|
||||
validOperations["fhgts"] = []string{
|
||||
"fhgts [duration] [tags] - Output entries in Flint Hills Group timesheet format",
|
||||
" (csv, daily totals, decimal format, 1/4 hour rounding)",
|
||||
}
|
||||
|
||||
// Load the Config
|
||||
cfg, err = userConfig.NewConfig(AppName)
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
fmt.Println("Creating new config")
|
||||
cfg.Save()
|
||||
}
|
||||
// If dbdir isn't set, set it to the config directory
|
||||
if cfg.Get("dbdir") == "" {
|
||||
cfg.Set("dbdir", cfg.GetConfigPath()+"/")
|
||||
}
|
||||
// If dbname isn't set, set it to the default database filename
|
||||
if cfg.Get("dbname") == "" {
|
||||
cfg.Set("dbname", DefDBName)
|
||||
}
|
||||
// If dbarchivename isn't set, set it to the default archive database filename
|
||||
if cfg.Get("dbarchname") == "" {
|
||||
cfg.Set("dbarchname", DefArchDBName)
|
||||
}
|
||||
if gdb, err = gime.LoadDatabase(cfg.Get("dbdir"), cfg.Get("dbname"), cfg.Get("dbarchname")); err != nil {
|
||||
fmt.Println("Error loading the database")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fuzzyFormats = []string{
|
||||
"1504",
|
||||
"15:04", // Kitchen, 24hr
|
||||
time.Kitchen,
|
||||
time.RFC3339,
|
||||
"2006-01-02T15:04:05", // RFC3339 without timezone
|
||||
"2006-01-02T15:04", // RFC3339 without seconds or timezone
|
||||
time.Stamp,
|
||||
"02 Jan 06 15:04:05", // RFC822 with second
|
||||
time.RFC822,
|
||||
"01/02/2006 15:04", // U.S. Format
|
||||
"01/02/2006 15:04:05", // U.S. Format with seconds
|
||||
"01/02/06 15:04", // U.S. Format, short year
|
||||
"01/02/06 15:04:05", // U.S. Format, short year, with seconds
|
||||
"2006-01-02",
|
||||
"2006-01-02 15:04",
|
||||
"2006-01-02 15:04:05",
|
||||
"20060102",
|
||||
"20060102 15:04",
|
||||
"20060102 15:04:05",
|
||||
"20060102 1504",
|
||||
"20060102 150405",
|
||||
"20060102T15:04",
|
||||
"20060102T15:04:05",
|
||||
"20060102T1504",
|
||||
"20060102T150405",
|
||||
}
|
||||
}
|
||||
|
||||
func matchParameter(in string) string {
|
||||
var chkParms []string
|
||||
for k := range validOperations {
|
||||
chkParms = append(chkParms, k)
|
||||
}
|
||||
var nextParms []string
|
||||
for i := range in {
|
||||
for _, p := range chkParms {
|
||||
if p[i] == in[i] {
|
||||
nextParms = append(nextParms, p)
|
||||
}
|
||||
}
|
||||
// If we get here and there is only one parameter left, return it
|
||||
chkParms = nextParms
|
||||
if len(nextParms) == 1 {
|
||||
break
|
||||
}
|
||||
// Otherwise, loop
|
||||
nextParms = []string{}
|
||||
}
|
||||
if len(chkParms) == 0 {
|
||||
return ""
|
||||
}
|
||||
return chkParms[0]
|
||||
}
|
||||
|
||||
func parseFuzzyTime(t string) (time.Time, error) {
|
||||
var ret time.Time
|
||||
var err error
|
||||
for i := range fuzzyFormats {
|
||||
ret, err = time.Parse(fuzzyFormats[i], t)
|
||||
if err == nil {
|
||||
// Make sure it's in the local timezone
|
||||
tz := time.Now().Format("Z07:00")
|
||||
t = ret.Format("2006-01-02T15:04:05") + tz
|
||||
if ret, err = time.Parse(time.RFC3339, t); err != nil {
|
||||
return ret, err
|
||||
}
|
||||
// Check for zero on year/mo/day
|
||||
if ret.Year() == 0 && ret.Month() == time.January && ret.Day() == 1 {
|
||||
ret = ret.AddDate(time.Now().Year(), int(time.Now().Month())-1, time.Now().Day()-1)
|
||||
}
|
||||
return ret, nil
|
||||
}
|
||||
}
|
||||
return time.Time{}, errors.New("Unable to parse time: " + t)
|
||||
}
|
||||
|
||||
// filterTimerCollection takes a collection and a function that it runs every entry through
|
||||
// If the function returns true for the entry, it adds it to a new collection to be returned
|
||||
func filterTimerCollection(c *gime.TimeEntryCollection, fn func(t *gime.TimeEntry) bool) *gime.TimeEntryCollection {
|
||||
ret := new(gime.TimeEntryCollection)
|
||||
for i := 0; i < c.Length(); i++ {
|
||||
if fn(c.Get(i)) {
|
||||
ret.Push(c.Get(i))
|
||||
}
|
||||
}
|
||||
return ret
|
||||
}
|
||||
Reference in New Issue
Block a user