package libcontainer

import (
	"encoding/json"
	"os"
	"path/filepath"

	"github.com/docker/libcontainer/network"
)

// State represents a running container's state
type State struct {
	// InitPid is the init process id in the parent namespace
	InitPid int `json:"init_pid,omitempty"`

	// InitStartTime is the init process start time
	InitStartTime string `json:"init_start_time,omitempty"`

	// Network runtime state.
	NetworkState network.NetworkState `json:"network_state,omitempty"`

	// Path to all the cgroups setup for a container. Key is cgroup subsystem name.
	CgroupPaths map[string]string `json:"cgroup_paths,omitempty"`
}

// The running state of the container.
type RunState int

const (
	// The name of the runtime state file
	stateFile = "state.json"

	// The container exists and is running.
	Running RunState = iota

	// The container exists, it is in the process of being paused.
	Pausing

	// The container exists, but all its processes are paused.
	Paused

	// The container does not exist.
	Destroyed
)

// SaveState writes the container's runtime state to a state.json file
// in the specified path
func SaveState(basePath string, state *State) error {
	f, err := os.Create(filepath.Join(basePath, stateFile))
	if err != nil {
		return err
	}
	defer f.Close()

	return json.NewEncoder(f).Encode(state)
}

// GetState reads the state.json file for a running container
func GetState(basePath string) (*State, error) {
	f, err := os.Open(filepath.Join(basePath, stateFile))
	if err != nil {
		return nil, err
	}
	defer f.Close()

	var state *State
	if err := json.NewDecoder(f).Decode(&state); err != nil {
		return nil, err
	}

	return state, nil
}

// DeleteState deletes the state.json file
func DeleteState(basePath string) error {
	return os.Remove(filepath.Join(basePath, stateFile))
}