Golang db.DB類代碼示例

func active(d db.DB) ([]*db.Entry, error) {
	if itr, err := d.Query(db.Query{Active: true}); err != nil {
		return nil, err
	} else {
		return db.IteratorEntries(itr)
	}
}

func cmdStart(d db.DB, resume bool, categoryS string) {
	entries, err := active(d)
	if err != nil {
		fatal(err)
	}
	now := time.Now()
	path, err := d.CategoryPath(ParseCategory(categoryS), true)
	if err != nil {
		fatal(err)
	}
	entry := &db.Entry{CategoryID: path.CategoryID(), Start: now}
	if resume {
		last, err := Last(d)
		if err != nil {
			fatal(err)
		}
		if !last.End.IsZero() {
			entry.Start = last.End
		}
		if entry.CategoryID == "" {
			entry.CategoryID = last.CategoryID
		}
	}
	if err := d.SaveEntry(entry); err != nil {
		fatal(err)
	}
	FprintEntry(os.Stdout, entry, path, PrintHideDuration|PrintHideEnd)
	if err := endAt(d, entries, now); err != nil {
		fatal(err)
	}
}

// Last returns the last entry or an error.
func Last(d db.DB) (*db.Entry, error) {
	itr, err := d.Query(db.Query{})
	if err != nil {
		return nil, err
	}
	defer itr.Close()
	if entry, err := itr.Next(); err == io.EOF {
		return nil, errors.New("db is empty")
	} else {
		return entry, err
	}
}

// ById returns the entry with the given id, or an error.
func ById(d db.DB, id string) (*db.Entry, error) {
	itr, err := d.Query(db.Query{IDs: []string{id}})
	if err != nil {
		return nil, err
	}
	defer itr.Close()
	if entry, err := itr.Next(); err == io.EOF {
		return nil, fmt.Errorf("entry does not exist: %s", id)
	} else {
		return entry, err
	}
}

// NewSummaryIterator returns a new summary iterator producing summaries for
// the given period and firstDay of the week. If the period is datetime.Day,
// it is is ignored. Callers are required to call Close once they are done
// with the iterator.
func NewSummaryIterator(d db.DB, period datetime.Period, firstDay time.Weekday, now time.Time) (*SummaryIterator, error) {
	entries, err := d.Query(db.Query{})
	if err != nil {
		return nil, err
	}
	return &SummaryIterator{
		now:      now,
		entries:  entries,
		period:   period,
		firstDay: firstDay,
	}, nil
}

func cmdRm(d db.DB, id string) {
	entry, err := ById(d, id)
	if err != nil {
		fatal(err)
	} else if err := d.Remove(id); err != nil {
		fatal(err)
	}
	categories, err := d.Categories()
	if err != nil {
		fatal(err)
	}
	FprintEntry(os.Stdout, entry, categories.Path(entry.CategoryID), PrintDefault)
}

func endAt(d db.DB, entries []*db.Entry, t time.Time) error {
	categories, err := d.Categories()
	if err != nil {
		return err
	}
	for _, entry := range entries {
		entry.End = t
		if err := d.SaveEntry(entry); err != nil {
			return err
		}
		FprintEntry(os.Stdout, entry, categories.Path(entry.CategoryID), PrintDefault)
	}
	return nil
}

func cmdSummary(d db.DB, periodS, firstDayS string) {
	period, err := datetime.ParsePeriod(periodS)
	if err != nil {
		fatal(err)
	}
	firstDay, err := datetime.ParseWeekday(firstDayS)
	if err != nil {
		fatal(err)
	}
	categories, err := d.Categories()
	if err != nil {
		fatal(err)
	}
	itr, err := NewSummaryIterator(d, period, firstDay, time.Now())
	if err != nil {
		fatal(err)
	}
	defer itr.Close()
	for {
		if summary, err := itr.Next(); err == io.EOF {
			break
		} else if err != nil {
			fatal(err)
		} else {
			fmt.Printf("%s\n\n", PeriodHeadline(summary.From, summary.To, period))
			names := make(map[string]string)
			order := make([]string, 0, len(summary.Categories))
			for id, _ := range summary.Categories {
				names[id] = FormatCategory(categories.Path(id))
				order = append(order, id)
			}
			slice.Sort(order, func(i, j int) bool {
				return summary.Categories[order[i]] > summary.Categories[order[j]]
			})
			t := table.New().Padding(" ")
			for _, id := range order {
				d := FormatDuration(summary.Categories[id])
				t.Add(table.String(names[id]), table.String(d).Align(table.Right))
			}
			fmt.Printf("%s\n", Indent(t.String(), "  "))
		}
	}
}

func cmdLs(d db.DB, categoryS string, asc bool) {
	if categories, err := d.Categories(); err != nil {
		fatal(err)
	} else if path, err := d.CategoryPath(ParseCategory(categoryS), false); err != nil {
		fatal(err)
	} else if itr, err := d.Query(db.Query{Asc: asc, CategoryID: path.CategoryID()}); err != nil {
		fatal(err)
	} else {
		FprintIterator(os.Stdout, itr, categories, PrintDefault)
	}
}

func cmdEdit(d db.DB, id string) {
	var (
		entry *db.Entry
		err   error
	)
	if id != "" {
		entry, err = ById(d, id)
	} else {
		entry, err = Last(d)
	}
	if err != nil {
		fatal(err)
	}
	categories, err := d.Categories()
	if err != nil {
		fatal(err)
	}
	e := term.NewEditor()
	FprintEntry(e, entry, categories.Path(entry.CategoryID), PrintSeparator|PrintHideDuration)
	if err := e.Run(); err != nil {
		fatal(err)
	} else if doc, err := ParseEntryDocument(e); err != nil {
		fatal(err)
	} else if doc == nil {
		return
	} else {
		entry := &db.Entry{
			ID:    doc.ID,
			Start: doc.Start,
			End:   doc.End,
			Note:  doc.Note,
		}
		if path, err := d.CategoryPath(doc.Category, false); err != nil {
			fatal(err)
		} else {
			entry.CategoryID = path.CategoryID()
		}
		if err := d.SaveEntry(entry); err != nil {
			fatal(err)
		} else {
			FprintIterator(os.Stdout, db.EntryIterator([]*db.Entry{entry}), categories, PrintDefault)
		}
	}
}

func cmdReport(d db.DB, categoryS, periodS, firstDayS string) {
	period, err := datetime.ParsePeriod(periodS)
	if err != nil {
		fatal(err)
	} else if period == datetime.Day {
		fatal(errors.New("bad period: day"))
	}
	firstDay, err := datetime.ParseWeekday(firstDayS)
	if err != nil {
		fatal(err)
	}
	path, err := d.CategoryPath(ParseCategory(categoryS), false)
	if err != nil {
		fatal(err)
	}
	entryItr, err := d.Query(db.Query{CategoryID: path.CategoryID()})
	if err != nil {
		fatal(err)
	}
	defer entryItr.Close()
	entry, err := entryItr.Next()
	if err == io.EOF {
		return
	} else if err != nil {
		fatal(err)
	}
	// @TODO move logic into separate function
	now := time.Now()
	reportItr := datetime.NewIterator(entry.Start, period, false, firstDay)
	report := &Report{Duration: period}
	report.From, report.To = reportItr.Next()
	dayItr := datetime.NewIterator(report.To, datetime.Day, false, firstDay)
	day := &ReportDay{}
	day.From, day.To = dayItr.Next()
	report.Days = append([]*ReportDay{day}, report.Days...)
	noteAssigned := false

outer:
	for {
		var overlap time.Duration
		for {
			overlap = entry.PartialDuration(now, day.From, day.To)
			if overlap > 0 {
				day.Tracked += overlap
				if !noteAssigned {
					note := strings.Trim(entry.Note, "\n")
					if note != "" {
						day.Notes = append([]string{note}, day.Notes...)
					}
					noteAssigned = true
				}
			}
			if !entry.Start.Before(day.From) {
				entry, err = entryItr.Next()
				if err == io.EOF {
					fmt.Fprint(os.Stdout, FormatReport(report))
					break outer
				} else if err != nil {
					fatal(err)
				}
				noteAssigned = false
				continue
			}
			day = &ReportDay{}
			day.From, day.To = dayItr.Next()
			if day.To.Before(report.From) {
				fmt.Fprint(os.Stdout, FormatReport(report))
				report = &Report{Duration: period}
				report.From, report.To = reportItr.Next()
			}
			report.Days = append([]*ReportDay{day}, report.Days...)
		}
	}
}