jpictl/pkg/zones/machine_file.go

package zones

import (
	"fmt"
	"io"
	"os"
	"path/filepath"

	fs "github.com/hack-pad/hackpadfs"
)

// OpenFile opens a file on the machine's config directory with the specified flags
func (m *Machine) OpenFile(name string, flags int, args ...any) (fs.File, error) {
	base := m.zone.zones.dir
	fullName := m.getFilename(name, args...)

	return fs.OpenFile(base, fullName, flags, 0644)
}

// CreateTruncFile creates or truncates a file on the machine's config directory
func (m *Machine) CreateTruncFile(name string, args ...any) (io.WriteCloser, error) {
	return m.openWriter(name, os.O_CREATE|os.O_TRUNC, args...)
}

// CreateFile creates a file on the machine's config directory
func (m *Machine) CreateFile(name string, args ...any) (io.WriteCloser, error) {
	return m.openWriter(name, os.O_CREATE, args...)
}

func (m *Machine) openWriter(name string, flags int, args ...any) (io.WriteCloser, error) {
	f, err := m.OpenFile(name, os.O_WRONLY|flags, args...)
	if err != nil {
		return nil, err
	}

	return f.(io.WriteCloser), nil
}

// RemoveFile deletes a file from the machine's config directory
func (m *Machine) RemoveFile(name string, args ...any) error {
	base := m.zone.zones.dir
	fullName := m.getFilename(name, args...)
	err := fs.Remove(base, fullName)

	switch {
	case os.IsNotExist(err):
		return nil
	default:
		return err
	}
}

// ReadFile reads a file from the machine's config directory
func (m *Machine) ReadFile(name string, args ...any) ([]byte, error) {
	base := m.zone.zones.dir
	fullName := m.getFilename(name, args...)

	return fs.ReadFile(base, fullName)
}

func (m *Machine) getFilename(name string, args ...any) string {
	if len(args) > 0 {
		name = fmt.Sprintf(name, args...)
	}

	s := []string{
		m.zone.Name,
		m.Name,
		name,
	}

	return filepath.Join(s...)
}