// unitNameMangle tries to turn a string that might not be a unit name into a
// sensible unit name.
func unitNameMangle(baseName string) string {
name := path.Base(baseName)
if !unit.RecognizedUnitType(name) {
return unit.DefaultUnitType(name)
}
return name
}
func lsUnitsDir(dir string) ([]string, error) {
filterFunc := func(name string) bool {
if !unit.RecognizedUnitType(name) {
log.Warningf("Found unrecognized file in %s, ignoring", path.Join(dir, name))
return true
}
return false
}
return pkg.ListDirectory(dir, filterFunc)
}
// Units enumerates all files recognized as valid systemd units in
// this manager's units directory.
func (m *SystemdUnitManager) Units() (units []string, err error) {
fis, err := ioutil.ReadDir(m.UnitsDir)
if err != nil {
return
}
for _, fi := range fis {
name := fi.Name()
if !unit.RecognizedUnitType(name) {
log.Warningf("Found unrecognized file in %s, ignoring", path.Join(m.UnitsDir, name))
continue
}
units = append(units, name)
}
return
}
func maybeAppendDefaultUnitType(arg string) string {
if !unit.RecognizedUnitType(arg) {
arg = unit.DefaultUnitType(arg)
}
return arg
}
