diff --git a/z/entry.go b/z/entry.go index c033976..2d70a2a 100644 --- a/z/entry.go +++ b/z/entry.go @@ -2,6 +2,7 @@ package z import ( "errors" + "strings" "time" "fmt" "github.com/gookit/color" @@ -49,6 +50,17 @@ func NewEntry( return newEntry, nil } +func (entry *Entry) SetIDFromDatabaseKey(key string) (error) { + splitKey := strings.Split(key, ":") + + if len(splitKey) < 3 || len(splitKey) > 3 { + return errors.New("not a valid database key") + } + + entry.ID = splitKey[2] + return nil +} + func (entry *Entry) SetBeginFromString(begin string) (time.Time, error) { var beginTime time.Time var err error