mirror of
https://github.com/cwinfo/matterbridge.git
synced 2024-12-11 12:25:49 +00:00
173 lines
4.5 KiB
Go
173 lines
4.5 KiB
Go
package main
|
|
|
|
import (
|
|
"archive/zip"
|
|
"fmt"
|
|
"go/build"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/daaku/go.zipexe"
|
|
)
|
|
|
|
func operationAppend(pkgs []*build.Package) {
|
|
if runtime.GOOS == "windows" {
|
|
_, err := exec.LookPath("zip")
|
|
if err != nil {
|
|
fmt.Println("#### WARNING ! ####")
|
|
fmt.Println("`rice append` is known not to work under windows because the `zip` command is not available. Please let me know if you got this to work (and how).")
|
|
}
|
|
}
|
|
|
|
// MARKED FOR DELETION
|
|
// This is actually not required, the append command now has the option --exec required.
|
|
// // check if package is a command
|
|
// if !pkg.IsCommand() {
|
|
// fmt.Println("Error: can not append to non-main package. Please follow instructions at github.com/GeertJohan/go.rice")
|
|
// os.Exit(1)
|
|
// }
|
|
|
|
// create tmp zipfile
|
|
tmpZipfileName := filepath.Join(os.TempDir(), fmt.Sprintf("ricebox-%d-%s.zip", time.Now().Unix(), randomString(10)))
|
|
verbosef("Will create tmp zipfile: %s\n", tmpZipfileName)
|
|
tmpZipfile, err := os.Create(tmpZipfileName)
|
|
if err != nil {
|
|
fmt.Printf("Error creating tmp zipfile: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
defer func() {
|
|
tmpZipfile.Close()
|
|
os.Remove(tmpZipfileName)
|
|
}()
|
|
|
|
// find abs path for binary file
|
|
binfileName, err := filepath.Abs(flags.Append.Executable)
|
|
if err != nil {
|
|
fmt.Printf("Error finding absolute path for executable to append: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
verbosef("Will append to file: %s\n", binfileName)
|
|
|
|
// check that command doesn't already have zip appended
|
|
if rd, _ := zipexe.Open(binfileName); rd != nil {
|
|
fmt.Printf("Cannot append to already appended executable. Please remove %s and build a fresh one.\n", binfileName)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// open binfile
|
|
binfile, err := os.OpenFile(binfileName, os.O_WRONLY, os.ModeAppend)
|
|
if err != nil {
|
|
fmt.Printf("Error: unable to open executable file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// create zip.Writer
|
|
zipWriter := zip.NewWriter(tmpZipfile)
|
|
|
|
for _, pkg := range pkgs {
|
|
// find boxes for this command
|
|
boxMap := findBoxes(pkg)
|
|
|
|
// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ?
|
|
if len(boxMap) == 0 {
|
|
fmt.Printf("no calls to rice.FindBox() or rice.MustFindBox() found in import path `%s`\n", pkg.ImportPath)
|
|
continue
|
|
}
|
|
|
|
verbosef("\n")
|
|
|
|
for boxname := range boxMap {
|
|
appendedBoxName := strings.Replace(boxname, `/`, `-`, -1)
|
|
|
|
// walk box path's and insert files
|
|
boxPath := filepath.Clean(filepath.Join(pkg.Dir, boxname))
|
|
filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error {
|
|
if info == nil {
|
|
fmt.Printf("Error: box \"%s\" not found on disk\n", path)
|
|
os.Exit(1)
|
|
}
|
|
// create zipFilename
|
|
zipFileName := filepath.Join(appendedBoxName, strings.TrimPrefix(path, boxPath))
|
|
// write directories as empty file with comment "dir"
|
|
if info.IsDir() {
|
|
_, err := zipWriter.CreateHeader(&zip.FileHeader{
|
|
Name: zipFileName,
|
|
Comment: "dir",
|
|
})
|
|
if err != nil {
|
|
fmt.Printf("Error creating dir in tmp zip: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// create zipFileWriter
|
|
zipFileHeader, err := zip.FileInfoHeader(info)
|
|
if err != nil {
|
|
fmt.Printf("Error creating zip FileHeader: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
zipFileHeader.Name = zipFileName
|
|
zipFileWriter, err := zipWriter.CreateHeader(zipFileHeader)
|
|
if err != nil {
|
|
fmt.Printf("Error creating file in tmp zip: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
srcFile, err := os.Open(path)
|
|
if err != nil {
|
|
fmt.Printf("Error opening file to append: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
_, err = io.Copy(zipFileWriter, srcFile)
|
|
if err != nil {
|
|
fmt.Printf("Error copying file contents to zip: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
srcFile.Close()
|
|
|
|
return nil
|
|
})
|
|
}
|
|
}
|
|
|
|
err = zipWriter.Close()
|
|
if err != nil {
|
|
fmt.Printf("Error closing tmp zipfile: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
err = tmpZipfile.Sync()
|
|
if err != nil {
|
|
fmt.Printf("Error syncing tmp zipfile: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
_, err = tmpZipfile.Seek(0, 0)
|
|
if err != nil {
|
|
fmt.Printf("Error seeking tmp zipfile: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
_, err = binfile.Seek(0, 2)
|
|
if err != nil {
|
|
fmt.Printf("Error seeking bin file: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
_, err = io.Copy(binfile, tmpZipfile)
|
|
if err != nil {
|
|
fmt.Printf("Error appending zipfile to executable: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
zipA := exec.Command("zip", "-A", binfileName)
|
|
err = zipA.Run()
|
|
if err != nil {
|
|
fmt.Printf("Error setting zip offset: %s\n", err)
|
|
os.Exit(1)
|
|
}
|
|
}
|