From 25a2f45e590410d80ded6409b50a4e82f8bd1e6a Mon Sep 17 00:00:00 2001
From: Mikhail Klementev <blame@dumpstack.io>
Date: Thu, 2 Jan 2020 17:09:58 +0000
Subject: [PATCH] Fix generating for packages inside modules

---
 generate.go | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/generate.go b/generate.go
index 0f3ed6b..1bd0275 100644
--- a/generate.go
+++ b/generate.go
@@ -131,7 +131,9 @@ func generate(l *libvirt.Libvirt, pkg, bin, vmname string) (err error) {
 
 		if !found {
 			for _, f := range files {
-				if f.Name() == pkg {
+				parts := strings.Split(pkg, ".")
+				log.Println("pkg", parts[len(parts)-1])
+				if f.Name() == parts[len(parts)-1] {
 					log.Println("Use", f.Name())
 					bin = f.Name()
 					found = true
@@ -161,7 +163,16 @@ func generate(l *libvirt.Libvirt, pkg, bin, vmname string) (err error) {
 		bin = files[0].Name()
 	}
 
-	realName := strings.Split(name, ".")[1]
+	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 != "" {