|
|
|
@ -1,9 +1,12 @@
|
|
|
|
|
package cluster |
|
|
|
|
|
|
|
|
|
import ( |
|
|
|
|
"bufio" |
|
|
|
|
"bytes" |
|
|
|
|
"fmt" |
|
|
|
|
"io" |
|
|
|
|
"os" |
|
|
|
|
"strings" |
|
|
|
|
|
|
|
|
|
fs "github.com/hack-pad/hackpadfs" |
|
|
|
|
) |
|
|
|
@ -40,6 +43,21 @@ func (m *Cluster) openWriter(name string, flags int, args ...any) (io.WriteClose
|
|
|
|
|
panic("unreachable") |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// RemoveFile deletes a file from the cluster's config directory
|
|
|
|
|
func (m *Cluster) RemoveFile(name string, args ...any) error { |
|
|
|
|
if len(args) > 0 { |
|
|
|
|
name = fmt.Sprintf(name, args...) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
err := fs.Remove(m.dir, name) |
|
|
|
|
switch { |
|
|
|
|
case os.IsNotExist(err): |
|
|
|
|
return nil |
|
|
|
|
default: |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// ReadFile reads a file from the cluster's config directory
|
|
|
|
|
func (m *Cluster) ReadFile(name string, args ...any) ([]byte, error) { |
|
|
|
|
if len(args) > 0 { |
|
|
|
@ -48,3 +66,41 @@ func (m *Cluster) ReadFile(name string, args ...any) ([]byte, error) {
|
|
|
|
|
|
|
|
|
|
return fs.ReadFile(m.dir, name) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// ReadLines reads a file from the cluster's config directory,
|
|
|
|
|
// split by lines, trimmed, and accepting `#` to comment lines out.
|
|
|
|
|
func (m *Cluster) ReadLines(name string, args ...any) ([]string, error) { |
|
|
|
|
var out []string |
|
|
|
|
|
|
|
|
|
data, err := m.ReadFile(name, args...) |
|
|
|
|
if err != nil { |
|
|
|
|
return nil, err |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
sc := bufio.NewScanner(bytes.NewReader(data)) |
|
|
|
|
for sc.Scan() { |
|
|
|
|
s := strings.TrimSpace(sc.Text()) |
|
|
|
|
switch { |
|
|
|
|
case s == "", strings.HasPrefix(s, "#"): |
|
|
|
|
// ignore
|
|
|
|
|
default: |
|
|
|
|
// accepted
|
|
|
|
|
out = append(out, s) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
return out, nil |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// WriteStringFile writes the given content to a file on the machine's config directory
|
|
|
|
|
func (m *Cluster) WriteStringFile(value string, name string, args ...any) error { |
|
|
|
|
f, err := m.CreateTruncFile(name, args...) |
|
|
|
|
if err != nil { |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|
defer f.Close() |
|
|
|
|
|
|
|
|
|
buf := bytes.NewBufferString(value) |
|
|
|
|
_, err = buf.WriteTo(f) |
|
|
|
|
return err |
|
|
|
|
} |
|
|
|
|