flags.go

package main

import (
	"flag"
	"os"

	"github.com/AOSC-Dev/ciel/systemd-api/nspawn"
)

func flagCielDir() *string {
	basePath := getEnv("CIEL_DIR", ".")
	flag.StringVar(&basePath, "C", basePath, "Ciel work `directory`; CIEL_DIR")
	return &basePath
}
func saveCielDir(basePath string) {
	saveEnv("CIEL_DIR", basePath)
}

func flagInstance() *string {
	instName := getEnv("CIEL_INST", "")
	flag.StringVar(&instName, "i", instName, "instance `name`; CIEL_INST")
	return &instName
}
func saveInstance(instName string) {
	saveEnv("CIEL_INST", instName)
}

func flagNetwork() *bool {
	network := getEnv("CIEL_NET", "false") == "true"
	flag.BoolVar(&network, "net", network, "create a network zone; CIEL_NET")
	return &network
}
func saveNetwork(network bool) {
	saveEnv("CIEL_NET", network)
}

func flagNoBooting() *bool {
	noBooting := getEnv("CIEL_BOOT", "false") == "true"
	flag.BoolVar(&noBooting, "n", noBooting, "do not boot the container; CIEL_BOOT")
	return &noBooting
}
func saveNoBooting(noBooting bool) {
	saveEnv("CIEL_BOOT", noBooting)
}

func flagBatch() *bool {
	batch := getEnv("CIEL_BATCH_MODE", "false") == "true"
	flag.BoolVar(&batch, "batch", batch, "do not ask; CIEL_BATCH_MODE")
	return &batch
}
func saveBatch(batch bool) {
	saveEnv("CIEL_BATCH_MODE", batch)
}

func flagLocalRepo() *bool {
	localRepo := getEnv("CIEL_LOCAL_REPO", "false") == "true"
	return &localRepo
}
func saveLocalRepo(localRepo bool) {
	saveEnv("CIEL_LOCAL_REPO", localRepo)
}

func getEnv(key, def string) string {
	v, ok := os.LookupEnv(key)
	if !ok {
		return def
	}
	return v
}
func saveEnv(key string, value interface{}) {
	switch v := value.(type) {
	case bool:
		if v {
			os.Setenv(key, "true")
		} else {
			os.Setenv(key, "false")
		}
	case string:
		os.Setenv(key, v)
	}
}

func buildContainerInfo(boot bool, network bool) *nspawn.ContainerInfo {
	ci := &nspawn.ContainerInfo{
		Init: boot,
	}
	if network {
		ci.Network = &nspawn.NetworkInfo{
			Zone: "ciel",
		}
	}
	return ci
}

func buildRunInfo(args []string) *nspawn.RunInfo {
	ri := &nspawn.RunInfo{
		App: args[0],
	}
	if len(args) > 1 {
		ri.Args = args[1:]
	}
	return ri
}