wings/server/state.go

package server

import (
	"encoding/json"
	"github.com/pkg/errors"
	"os"
	"sync"
)

var (
	statesLock sync.Mutex
	statesFile = "data/states.json"
)

// DoesStatesFileExist .
func DoesStatesFileExist() (bool, error) {
	statesLock.Lock()
	defer statesLock.Unlock()

	if _, err := os.Stat(statesFile); err != nil {
		if !os.IsNotExist(err) {
			return false, errors.WithStack(err)
		}

		return false, nil
	}

	return true, nil
}

// FetchServerStates .
func FetchServerStates() (map[string]string, error) {
	// Check if the states file exists.
	exists, err := DoesStatesFileExist()
	if err != nil {
		return nil, errors.WithStack(err)
	}

	// Request a lock after we check if the file exists.
	statesLock.Lock()
	defer statesLock.Unlock()

	// Return an empty map if the file does not exist.
	if !exists {
		return map[string]string{}, nil
	}

	// Open the states file.
	f, err := os.Open(statesFile)
	if err != nil {
		return nil, errors.WithStack(err)
	}
	defer f.Close()

	// Convert the json object to a map.
	states := map[string]string{}
	if err := json.NewDecoder(f).Decode(states); err != nil {
		return nil, errors.WithStack(err)
	}

	return states, nil
}

// SaveServerStates .
func SaveServerStates() error {
	// Get the states of all servers on the daemon.
	states := map[string]string{}
	for _, s := range GetServers().All() {
		states[s.Uuid] = s.State
	}

	// Convert the map to a json object.
	data, err := json.Marshal(states)
	if err != nil {
		return errors.WithStack(err)
	}

	// Check if the states file exists.
	exists, err := DoesStatesFileExist()
	if err != nil {
		return errors.WithStack(err)
	}

	// Request a lock after we check if the file exists.
	statesLock.Lock()
	defer statesLock.Unlock()

	// Create the file if it doesn't exist or open it if it already does.
	var f *os.File
	if !exists {
		f, err = os.Create(statesFile)
		if err != nil {
			return errors.WithStack(err)
		}
	} else {
		f, err = os.Open(statesFile)
		if err != nil {
			return errors.WithStack(err)
		}
	}
	defer f.Close()

	// Write the data to the file
	if _, err := f.Write(data); err != nil {
		return errors.WithStack(err)
	}

	// Save the file basically.
	return f.Sync()
}