upgrade: add ifdefines

This makes it possible to correctly link against the system sqlite. Also, don't
use defers with log.Fatal() since they won't be run on exit.

Signed-off-by: Christian Brauner <[email protected]>

Author: Christian Brauner

tool/upgrade.go

@@ -37,18 +37,21 @@ func main() {
 	if err != nil {
 		log.Fatal(err)
 	}
-	defer resp.Body.Close()
 
 	b, err := ioutil.ReadAll(resp.Body)
 	if err != nil {
+		resp.Body.Close()
 		log.Fatal(err)
 	}
 
 	fmt.Printf("extracting %v\n", path.Base(url))
 	r, err := zip.NewReader(bytes.NewReader(b), resp.ContentLength)
 	if err != nil {
+		resp.Body.Close()
 		log.Fatal(err)
 	}
+	resp.Body.Close()
+
 	for _, zf := range r.File {
 		var f *os.File
 		switch path.Base(zf.Name) {
@@ -68,11 +71,27 @@ func main() {
 		if err != nil {
 			log.Fatal(err)
 		}
+
+		_, err = io.WriteString(f, "#ifndef USE_LIBSQLITE3\n")
+		if err != nil {
+			zr.Close()
+			f.Close()
+			log.Fatal(err)
+		}
 		_, err = io.Copy(f, zr)
-		f.Close()
 		if err != nil {
+			zr.Close()
+			f.Close()
 			log.Fatal(err)
 		}
+		_, err = io.WriteString(f, "#else // USE_LIBSQLITE3\n // If users really want to link against the system sqlite3 we\n// need to make this file a noop.\n #endif")
+		if err != nil {
+			zr.Close()
+			f.Close()
+			log.Fatal(err)
+		}
+		zr.Close()
+		f.Close()
 		fmt.Printf("extracted %v\n", filepath.Base(f.Name()))
 	}
 }