113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"regexp"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
log "github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Addr string
|
||
|
|
|
||
|
|
var (
|
||
|
|
reBox = regexp.MustCompile(`^[a-zA-Z0-9\-_\.]+$`)
|
||
|
|
rePath = regexp.MustCompile(`^(/){0,1}[a-zA-Z0-9\-_\.]+(/[a-zA-Z0-9\-_\.]+)+$`)
|
||
|
|
)
|
||
|
|
|
||
|
|
func (a Addr) Box() string {
|
||
|
|
s := strings.Split(string(a), `:`)
|
||
|
|
box := s[0]
|
||
|
|
if reBox.MatchString(box) {
|
||
|
|
return box
|
||
|
|
} else {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) Path() string {
|
||
|
|
s := strings.Split(string(a), `:`)
|
||
|
|
path := s[1]
|
||
|
|
if rePath.MatchString(path) {
|
||
|
|
return path
|
||
|
|
} else {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) Append(path string) Addr {
|
||
|
|
newPath := a.Path() + path
|
||
|
|
if rePath.MatchString(newPath) {
|
||
|
|
return Addr(a.Box() + ":" + newPath)
|
||
|
|
} else {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) BoxExec(cmd string) (string, error) {
|
||
|
|
log.WithFields(log.Fields{"addr": a}).Debugf("starting")
|
||
|
|
defer log.WithFields(log.Fields{"addr": a}).Debugf("done")
|
||
|
|
|
||
|
|
if b, ok := cfg.box[a.Box()]; !ok {
|
||
|
|
err := errors.New("box doesn't exist")
|
||
|
|
log.WithFields(log.Fields{"addr": a, "box": a.Box()}).Errorf("")
|
||
|
|
return "", err
|
||
|
|
} else {
|
||
|
|
return b.Exec(cmd)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) Exec() (string, error) {
|
||
|
|
log.WithFields(log.Fields{"addr": a}).Debugf("starting")
|
||
|
|
defer log.WithFields(log.Fields{"addr": a}).Debugf("done")
|
||
|
|
|
||
|
|
return a.BoxExec(a.Path())
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) ValidSnapshots() ([]*ZfsSnapshot, error) {
|
||
|
|
log.WithFields(log.Fields{"addr": a}).Debugf("starting")
|
||
|
|
defer log.WithFields(log.Fields{"addr": a}).Debugf("done")
|
||
|
|
|
||
|
|
if b, ok := cfg.box[a.Box()]; !ok {
|
||
|
|
err := errors.New("box doesn't exist")
|
||
|
|
log.WithFields(log.Fields{"addr": a, "box": a.Box()}).Errorf("")
|
||
|
|
return nil, err
|
||
|
|
} else {
|
||
|
|
if fs, ok := b.zfs.filesystems[a.Path()]; ok {
|
||
|
|
return fs.ValidSnapshots(), nil
|
||
|
|
} else {
|
||
|
|
err := errors.New("path doesn't exist")
|
||
|
|
log.WithFields(log.Fields{"addr": a}).Errorf("")
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) Mkdir() error {
|
||
|
|
log.WithFields(log.Fields{"addr": a}).Debugf("starting")
|
||
|
|
defer log.WithFields(log.Fields{"addr": a}).Debugf("done")
|
||
|
|
|
||
|
|
if b, ok := cfg.box[a.Box()]; !ok {
|
||
|
|
err := errors.New("box doesn't exist")
|
||
|
|
log.WithFields(log.Fields{"addr": a, "box": a.Box()}).Errorf("")
|
||
|
|
return err
|
||
|
|
} else {
|
||
|
|
return b.zfs.Mkdir(a.Path())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) String() string {
|
||
|
|
return string(a)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (a Addr) Online() bool {
|
||
|
|
log.WithFields(log.Fields{"addr": a}).Debugf("starting")
|
||
|
|
defer log.WithFields(log.Fields{"addr": a}).Debugf("done")
|
||
|
|
|
||
|
|
if b, ok := cfg.box[a.Box()]; !ok {
|
||
|
|
return false
|
||
|
|
} else {
|
||
|
|
return b.online
|
||
|
|
}
|
||
|
|
}
|