Merge lp:~kyrofa/crackly/desktop_file_support into lp:crackly

[Kyle Fazzari]

Add install/remove wrappers to deal with .desktop files.

Add install/remove wrappers to deal with .desktop files.

=== added file 'main.go'
--- main.go	1970-01-01 00:00:00 +0000
+++ main.go	2015-10-30 14:11:44 +0000
@@ -0,0 +1,43 @@
+package main
+
+import (
+	"fmt"
+	"launchpad.net/crackly/operations"
+	"log"
+	"os"
+)
+
+func main() {
+	arguments := os.Args[1:]
+
+	if len(arguments) != 2 {
+		printUsageAndExit("Invalid number of arguments!")
+	}
+
+	switch arguments[0] {
+	case "install":
+		err := operations.Install(arguments[1])
+		if err != nil {
+			log.Fatalf(`Unable to install "%s": %s`, arguments[1], err)
+		}
+	case "remove":
+		err := operations.Remove(arguments[1])
+		if err != nil {
+			log.Fatalf(`Unable to remove "%s": %s`, arguments[1], err)
+		}
+	default:
+		printUsageAndExit("Unsupported action: %s", arguments[0])
+	}
+}
+
+func printUsageAndExit(format string, a ...interface{}) {
+	fmt.Printf(format, a...)
+	fmt.Println("\n")
+	printUsage()
+	os.Exit(1)
+}
+
+func printUsage() {
+	fmt.Println("Usage:")
+	fmt.Printf("%s install <snap>\n", os.Args[0])
+}
 
=== added directory 'operations'
=== added file 'operations/generate_desktop_files.go'
--- operations/generate_desktop_files.go	1970-01-01 00:00:00 +0000
+++ operations/generate_desktop_files.go	2015-10-30 14:11:44 +0000
@@ -0,0 +1,104 @@
+package operations
+
+import (
+	"fmt"
+	"path"
+	"os"
+	"io/ioutil"
+	"path/filepath"
+
+	"launchpad.net/crackly/snap"
+	"launchpad.net/snappy/snappy"
+)
+
+func GenerateDesktopFiles(part snappy.Part, metadata snap.Metadata) error {
+	rootDirectory := os.Getenv("SNAPPY_GLOBAL_ROOT")
+	if rootDirectory == "" {
+		rootDirectory = "/"
+	}
+
+	desktopFileRoot, err := getDesktopFileRoot()
+	if err != nil {
+		return fmt.Errorf("Unable to determine .desktop file destination: %s",
+			err)
+	}
+
+	qualifiedPartName := snappy.QualifiedName(part)
+	for _, binary := range metadata.Binaries {
+		qualifiedName := qualifiedPartName+"."+binary.Name
+		desktopFileName := fmt.Sprintf("%s.desktop", qualifiedName)
+
+		desktopFilePath := path.Join(desktopFileRoot, desktopFileName)
+
+		// .desktop file contents
+		desktopFile := "[Desktop Entry]\n"
+		desktopFile += "Type=Application\n"
+		desktopFile += "Terminal=false\n"
+		desktopFile += "X-Ubuntu-Touch=true\n"
+		desktopFile += fmt.Sprintf("Name=%s\n", binary.Name)
+		desktopFile += fmt.Sprintf("Icon=%s\n", part.Icon())
+//		desktopFile += fmt.Sprintf("Exec=%s/%s -- --desktop_file_hint=%s",
+//			rootDirectory+"apps/bin", metadata.Name+"."+binary.Name,
+//			desktopFilePath)
+
+		desktopFile += fmt.Sprintf("Exec=%s/%s",
+			rootDirectory+"apps/bin", metadata.Name+"."+binary.Name)
+
+		err = ioutil.WriteFile(desktopFilePath, []byte(desktopFile), 0644)
+		if err != nil {
+			return fmt.Errorf(`Unable to write .desktop file "%s": %s`, desktopFilePath, err)
+		}
+	}
+
+	return nil
+}
+
+func RemoveDesktopFiles(part snappy.Part) error {
+	desktopFileRoot, err := getDesktopFileRoot()
+	if err != nil {
+		return fmt.Errorf("Unable to determine .desktop file destination: %s",
+			err)
+	}
+
+	desktopFiles, err := filepath.Glob(
+		filepath.Join(desktopFileRoot, snappy.QualifiedName(part) + ".*"))
+
+	for _, desktopFilePath := range desktopFiles {
+		err = os.Remove(desktopFilePath)
+		if err != nil {
+			return fmt.Errorf(`Unable to remove "%s": %s`, desktopFilePath, err)
+		}
+	}
+
+	return nil
+}
+
+func getDesktopFileRoot() (string, error) {
+//	currentUser, err := user.Current()
+//	if err != nil {
+//		return "", fmt.Errorf("Unable to find current user: %s", err)
+//	}
+
+//	// Don't use root, use the person who called `sudo`
+//	if currentUser.Uid == "0" {
+//		userName := os.Getenv("SUDO_USER")
+//		if userName == "" {
+//			return "", fmt.Errorf("Unable to determine appropriate user")
+//		}
+
+//		currentUser, err = user.Lookup(userName)
+//		if err != nil {
+//			return "", fmt.Errorf(`Unable to find user with username "%s": %s`,
+//				userName, err)
+//		}
+//	}
+
+//	desktopFileRoot := path.Join(currentUser.HomeDir,
+//		".local/share/applications")
+
+//	fmt.Println("Current user:", currentUser)
+
+//	return desktopFileRoot, nil
+
+	return "/usr/share/applications", nil
+}
 
=== added file 'operations/install.go'
--- operations/install.go	1970-01-01 00:00:00 +0000
+++ operations/install.go	2015-10-30 14:11:44 +0000
@@ -0,0 +1,51 @@
+package operations
+
+import (
+	"fmt"
+	"launchpad.net/crackly/snap"
+	"launchpad.net/snappy/clickdeb"
+	"launchpad.net/snappy/progress"
+	"launchpad.net/snappy/snappy"
+)
+
+func Install(snapFile string) error {
+	deb, err := clickdeb.Open(snapFile)
+	if err != nil {
+		return fmt.Errorf("Unable to open snap: %s", err)
+	}
+
+	defer deb.Close()
+
+	yamlData, err := deb.MetaMember("package.yaml")
+	if err != nil {
+		return fmt.Errorf("Unable to obtain package.yaml: %s", err)
+	}
+
+	metadata, err := snap.ParseMetadata(yamlData)
+	if err != nil {
+		return fmt.Errorf("Unable to parse package.yaml: %s", err)
+	}
+
+	if len(metadata.Services) > 0 {
+		return fmt.Errorf("Cannot install a snap that contains services")
+	}
+
+	packageName, err := snappy.Install(snapFile, snappy.AllowUnauthenticated|snappy.DoInstallGC, progress.MakeProgressBar())
+	if err != nil {
+		return fmt.Errorf(`Error installing snap "%s": %s`, snapFile, err)
+	}
+
+	fmt.Println("Package name:", packageName)
+
+	part := snappy.ActiveSnapByName(metadata.Name)
+	if part == nil {
+		return fmt.Errorf("Unable to find snap after install")
+	}
+
+	err = GenerateDesktopFiles(part, metadata)
+	if err != nil {
+		return fmt.Errorf("Error generating .desktop files: %s", err)
+	}
+
+	return nil
+}
 
=== added file 'operations/remove.go'
--- operations/remove.go	1970-01-01 00:00:00 +0000
+++ operations/remove.go	2015-10-30 14:11:44 +0000
@@ -0,0 +1,26 @@
+package operations
+
+import (
+	"fmt"
+	"launchpad.net/snappy/progress"
+	"launchpad.net/snappy/snappy"
+)
+
+func Remove(snapName string) error {
+	part := snappy.ActiveSnapByName(snapName)
+	if part == nil {
+		return fmt.Errorf(`No snap named "%s"`, snapName)
+	}
+
+	err := snappy.Remove(snapName, snappy.DoRemoveGC, progress.MakeProgressBar())
+	if err != nil {
+		return fmt.Errorf(`Error removing snap "%s": %s`, snapName, err)
+	}
+
+	err = RemoveDesktopFiles(part)
+	if err != nil {
+		return fmt.Errorf("Error removing .desktop files: %s", err)
+	}
+
+	return nil
+}
 
=== added directory 'snap'
=== added file 'snap/metadata.go'
--- snap/metadata.go	1970-01-01 00:00:00 +0000
+++ snap/metadata.go	2015-10-30 14:11:44 +0000
@@ -0,0 +1,21 @@
+package snap
+
+import (
+	"gopkg.in/yaml.v2"
+	"launchpad.net/snappy/snappy"
+)
+
+type Metadata struct {
+	Name    string
+	Version string
+	Vendor  string
+	Icon    string
+
+	Services []snappy.ServiceYaml `yaml:"services,omitempty"`
+	Binaries []snappy.Binary  `yaml:"binaries,omitempty"`
+}
+
+func ParseMetadata(data []byte) (metadata Metadata, err error) {
+	err = yaml.Unmarshal(data, &metadata)
+	return
+}