From ad898b2d334c682477a59b8b6b2093b2356f3dd2 Mon Sep 17 00:00:00 2001
From: morganamilo <morganamilo@gmail.com>
Date: Thu, 15 Feb 2018 05:36:42 +0000
Subject: [PATCH] Regenerate .SRCINFO if editing the PKGBUILD

---
 install.go | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/install.go b/install.go
index b2c27cdd..9b33a069 100644
--- a/install.go
+++ b/install.go
@@ -101,8 +101,15 @@ func install(parser *arguments) error {
 			return err
 		}
 
-		askEditPkgBuilds(dc.AurMake)
-		askEditPkgBuilds(dc.Aur)
+		err = askEditPkgBuilds(dc.AurMake)
+		if err != nil {
+			return err
+		}
+		err = askEditPkgBuilds(dc.Aur)
+		if err != nil {
+			return err
+		}
+		
 		if _, ok := arguments.options["gendb"]; ok {
 			fmt.Println("GenDB finished. No packages were installed")
 			return nil
@@ -240,7 +247,7 @@ func checkForConflicts(aur []*rpc.Pkg, aurMake []*rpc.Pkg, repo []*alpm.Package,
 	return nil
 }
 
-func askEditPkgBuilds(pkgs []*rpc.Pkg) {
+func askEditPkgBuilds(pkgs []*rpc.Pkg) (error)  {
 	for _, pkg := range pkgs {
 		dir := config.BuildDir + pkg.PackageBase + "/"
 
@@ -248,8 +255,24 @@ func askEditPkgBuilds(pkgs []*rpc.Pkg) {
 			editcmd := exec.Command(editor(), dir+"PKGBUILD")
 			editcmd.Stdin, editcmd.Stdout, editcmd.Stderr = os.Stdin, os.Stdout, os.Stderr
 			editcmd.Run()
+
+			file, err := os.OpenFile(dir + ".SRCINFO", os.O_TRUNC|os.O_WRONLY|os.O_CREATE, 0666)
+			if err != nil {
+				return err
+			}
+			defer file.Close()
+
+			cmd := exec.Command(config.MakepkgBin, "--printsrcinfo")
+			cmd.Stdout, cmd.Stderr = file, os.Stderr
+			cmd.Dir = dir
+			err = cmd.Run()
+
+			if err != nil {
+				return err
+			}
 		}
 
+
 		pkgbuild, err := gopkg.ParseSRCINFO(dir + ".SRCINFO")
 		if err == nil {
 			for _, pkgsource := range pkgbuild.Source {
@@ -264,6 +287,8 @@ func askEditPkgBuilds(pkgs []*rpc.Pkg) {
 		}
 
 	}
+
+	return nil
 }
 
 func dowloadPkgBuilds(pkgs []*rpc.Pkg) (err error) {