Use only nixos.org channels if guess

This reverts commit 02dda8bcf9.

appvm.go

@@ -13,6 +13,7 @@ import (
 	"io"
 	"io/ioutil"
 	"log"
+	"math/rand"
 	"net"
 	"os"
 	"os/exec"
@@ -165,23 +166,21 @@ func isRunning(l *libvirt.Libvirt, name string) bool {
 	return err == nil
 }
 
-func generateAppVM(l *libvirt.Libvirt, appvmPath, name string, verbose,
-	online bool) (err error) {
+func generateAppVM(l *libvirt.Libvirt,
+	nixName, vmName, appvmPath, sharedDir string,
+	verbose, online bool) (err error) {
 
 	err = os.Chdir(appvmPath)
 	if err != nil {
 		return
 	}
 
-	realpath, reginfo, qcow2, err := generateVM(name, verbose)
+	realpath, reginfo, qcow2, err := generateVM(nixName, verbose)
 	if err != nil {
 		return
 	}
 
-	sharedDir := fmt.Sprintf(os.Getenv("HOME") + "/appvm/" + name)
-	os.MkdirAll(sharedDir, 0700)
-
-	xml := generateXML(name, online, realpath, reginfo, qcow2, sharedDir)
+	xml := generateXML(vmName, online, realpath, reginfo, qcow2, sharedDir)
 	_, err = l.DomainCreateXML(xml, libvirt.DomainStartValidate)
 	return
 }
@@ -210,14 +209,31 @@ func isAppvmConfigurationExists(appvmPath, name string) bool {
 	return fileExists(appvmPath + "/nix/" + name + ".nix")
 }
 
-func start(l *libvirt.Libvirt, name string, verbose, online bool,
+func start(l *libvirt.Libvirt, name string, verbose, online, stateless bool,
 	args, open string) {
 
 	appvmPath := configDir
-	vmHomePath := os.Getenv("HOME") + "/appvm/" + name + "/"
+
+	statelessName := fmt.Sprintf("tmp_%d_%s", rand.Int(), name)
+
+	sharedDir := os.Getenv("HOME") + "/appvm/"
+	if stateless {
+		sharedDir += statelessName
+	} else {
+		sharedDir += name
+	}
+
+	os.MkdirAll(sharedDir, 0700)
+
+	vmName := "appvm_"
+	if stateless {
+		vmName += statelessName
+	} else {
+		vmName += name
+	}
 
 	if open != "" {
-		filename := vmHomePath + filepath.Base(open)
+		filename := sharedDir + "/" + filepath.Base(open)
 		err := copyFile(open, filename)
 		if err != nil {
 			log.Println("Can't copy file")
@@ -228,7 +244,7 @@ func start(l *libvirt.Libvirt, name string, verbose, online bool,
 	}
 
 	if args != "" {
-		err := ioutil.WriteFile(vmHomePath+".args", []byte(args), 0700)
+		err := ioutil.WriteFile(sharedDir+"/"+".args", []byte(args), 0700)
 		if err != nil {
 			log.Println("Can't write args")
 			return
@@ -251,17 +267,19 @@ func start(l *libvirt.Libvirt, name string, verbose, online bool,
 		log.Fatal(err)
 	}
 
-	if !isRunning(l, name) {
+	if !isRunning(l, vmName) {
 		if !verbose {
 			go stupidProgressBar()
 		}
-		err = generateAppVM(l, appvmPath, name, verbose, online)
+
+		err = generateAppVM(l, name, vmName, appvmPath, sharedDir,
+			verbose, online)
 		if err != nil {
 			log.Fatal(err)
 		}
 	}
 
-	cmd := exec.Command("virt-viewer", "-c", "qemu:///system", "appvm_"+name)
+	cmd := exec.Command("virt-viewer", "-c", "qemu:///system", vmName)
 	cmd.Start()
 }
 
@@ -365,9 +383,40 @@ func sync() {
 	log.Println("Done")
 }
 
+func cleanupStatelessVMs(l *libvirt.Libvirt) {
+	domains, err := l.Domains()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	dirs, err := ioutil.ReadDir(appvmHomesDir)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for _, f := range dirs {
+		if !strings.HasPrefix(f.Name(), "tmp_") {
+			continue
+		}
+
+		alive := false
+		for _, d := range domains {
+			if d.Name == "appvm_"+f.Name() {
+				alive = true
+			}
+		}
+		if !alive {
+			os.RemoveAll(appvmHomesDir + f.Name())
+		}
+	}
+}
+
 var configDir = os.Getenv("HOME") + "/.config/appvm/"
+var appvmHomesDir = os.Getenv("HOME") + "/appvm/"
 
 func main() {
+	rand.Seed(time.Now().UnixNano())
+
 	os.Mkdir(os.Getenv("HOME")+"/appvm", 0700)
 
 	os.MkdirAll(configDir+"/nix", 0700)
@@ -393,6 +442,8 @@ func main() {
 	}
 	defer l.Disconnect()
 
+	cleanupStatelessVMs(l)
+
 	kingpin.Command("list", "List applications")
 	autoballonCommand := kingpin.Command("autoballoon", "Automatically adjust/reduce app vm memory")
 	minMemory := autoballonCommand.Flag("min-memory", "Set minimal memory (megabytes)").Default("1024").Uint64()
@@ -404,6 +455,7 @@ func main() {
 	startArgs := startCommand.Flag("args", "Command line arguments").String()
 	startOpen := startCommand.Flag("open", "Pass file to application").String()
 	startOffline := startCommand.Flag("offline", "Disconnect").Bool()
+	startStateless := startCommand.Flag("stateless", "Do not use default state directory").Bool()
 
 	stopName := kingpin.Command("stop", "Stop application").Arg("name", "Application name").Required().String()
 	dropName := kingpin.Command("drop", "Remove application data").Arg("name", "Application name").Required().String()
@@ -426,7 +478,8 @@ func main() {
 	case "generate":
 		generate(l, *generateName, *generateBin, *generateVMName)
 	case "start":
-		start(l, *startName, !*startQuiet, !*startOffline,
+		start(l, *startName,
+			!*startQuiet, !*startOffline, *startStateless,
 			*startArgs, *startOpen)
 	case "stop":
 		stop(l, *stopName)

base.nix.go

@@ -37,7 +37,7 @@ main = xmonad defaultConfig
 
 startup :: X ()
 startup = do
-  spawn "${pkgs.spice-vdagent}/bin/spice-vdagent"
+  spawn "while [ 1 ]; do ${pkgs.spice-vdagent}/bin/spice-vdagent -x; done &"
   '';
 
   systemd.services.home-user-build-xmonad = {

generate.go

@@ -34,12 +34,8 @@ in {
 }
 `
 
-func isPackageExists(name string) bool {
-	cmd := exec.Command("nix-env", "-iA", name)
-	cmd.Stdout = os.Stdout
-	cmd.Stderr = os.Stderr
-	err := cmd.Run()
-	return err == nil
+func isPackageExists(channel, name string) bool {
+	return nil == exec.Command("nix-build", "<"+channel+">", "-A", name).Run()
 }
 
 func nixPath(name string) (path string, err error) {
@@ -62,10 +58,12 @@ func guessChannel() (channel string, err error) {
 	for _, line := range channels {
 		fields := strings.Fields(line)
 		if len(fields) == 2 {
+			if strings.Contains(fields[1], "nixos.org/channels") {
 				channel = fields[0]
 				return
 			}
 		}
+	}
 
 	err = errors.New("No channel found")
 	return
@@ -81,33 +79,37 @@ func filterDotfiles(files []os.FileInfo) (notHiddenFiles []os.FileInfo) {
 }
 
 func generate(l *libvirt.Libvirt, pkg, bin, vmname string) (err error) {
-	var name string
+	// TODO refactor
+	var name, channel string
 
 	if strings.Contains(pkg, ".") {
-		name = pkg
+		channel = strings.Split(pkg, ".")[0]
+		name = strings.Join(strings.Split(pkg, ".")[1:], ".")
 	} else {
 		log.Println("Package name does not contains channel")
 		log.Println("Trying to guess")
-		var channel string
+
 		channel, err = guessChannel()
 		if err != nil {
 			log.Println("Cannot guess channel")
 			log.Println("Check nix-channel --list")
-			return
+			log.Println("Will try <nixpkgs>")
+			channel = "nixpkgs"
+			err = nil
 		}
 
-		name = channel + "." + pkg
-		log.Println("Use", name)
+		name = pkg
+		log.Println("Use", channel+"."+pkg)
 	}
 
-	if !isPackageExists(name) {
+	if !isPackageExists(channel, name) {
 		s := "Package " + name + " does not exists"
 		err = errors.New(s)
 		log.Println(s)
 		return
 	}
 
-	path, err := nixPath(name)
+	path, err := nixPath(channel + "." + name)
 	if err != nil {
 		log.Println("Cannot find nix path")
 		return
@@ -171,25 +173,14 @@ func generate(l *libvirt.Libvirt, pkg, bin, vmname string) (err error) {
 		bin = files[0].Name()
 	}
 
-	var realName string
-	for i, s := range strings.Split(name, ".") {
-		if i == 0 {
-			continue
-		}
-		if i != 1 {
-			realName += "."
-		}
-		realName += s
-	}
-
 	var appFilename string
 	if vmname != "" {
 		appFilename = configDir + "/nix/" + vmname + ".nix"
 	} else {
-		appFilename = configDir + "/nix/" + realName + ".nix"
+		appFilename = configDir + "/nix/" + name + ".nix"
 	}
 
-	appNixConfig := fmt.Sprintf(template, realName, bin)
+	appNixConfig := fmt.Sprintf(template, name, bin)
 
 	err = ioutil.WriteFile(appFilename, []byte(appNixConfig), 0600)
 	if err != nil {

xml.go

@@ -5,7 +5,9 @@ import "fmt"
 // You may think that you want to rewrite to proper golang structures.
 // Believe me, you shouldn't.
 
-func generateXML(name string, online bool, vmNixPath, reginfo, img, sharedDir string) string {
+func generateXML(vmName string, online bool,
+	vmNixPath, reginfo, img, sharedDir string) string {
+
 	qemuParams := `
           <qemu:commandline>
             <qemu:arg value='-device'/>
@@ -24,7 +26,7 @@ func generateXML(name string, online bool, vmNixPath, reginfo, img, sharedDir st
                 `
 	}
 
-	return fmt.Sprintf(xmlTmpl, "appvm_"+name, vmNixPath, vmNixPath, vmNixPath,
+	return fmt.Sprintf(xmlTmpl, vmName, vmNixPath, vmNixPath, vmNixPath,
 		reginfo, img, sharedDir, sharedDir, sharedDir, qemuParams)
 }