forked from hyperledger/fabric
/
package.go
108 lines (87 loc) · 2.41 KB
/
package.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
Copyright IBM Corp All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package nwo
import (
"archive/tar"
"compress/gzip"
"encoding/json"
"io"
"io/ioutil"
"os"
. "github.com/onsi/gomega"
)
// PackageChaincodeBinary is a helper function to package
// an already built chaincode and write it to the location
// specified by Chaincode.PackageFile.
func PackageChaincodeBinary(c Chaincode) {
file, err := os.Create(c.PackageFile)
Expect(err).NotTo(HaveOccurred())
defer file.Close()
writeTarGz(c, file)
}
func writeTarGz(c Chaincode, w io.Writer) {
gw := gzip.NewWriter(w)
tw := tar.NewWriter(gw)
defer closeAll(tw, gw)
writeMetadataJSON(tw, c.Path, c.Lang, c.Label)
writeCodeTarGz(tw, c.CodeFiles)
}
// packageMetadata holds the path, type, and label for a chaincode package
type packageMetadata struct {
Path string `json:"path"`
Type string `json:"type"`
Label string `json:"label"`
}
func writeMetadataJSON(tw *tar.Writer, path, ccType, label string) {
metadata, err := json.Marshal(&packageMetadata{
Path: path,
Type: ccType,
Label: label,
})
Expect(err).NotTo(HaveOccurred())
// write it to the package as metadata.json
err = tw.WriteHeader(&tar.Header{
Name: "metadata.json",
Size: int64(len(metadata)),
Mode: 0100644,
})
_, err = tw.Write(metadata)
Expect(err).NotTo(HaveOccurred())
}
func writeCodeTarGz(tw *tar.Writer, codeFiles map[string]string) {
// create temp file to hold code.tar.gz
tempfile, err := ioutil.TempFile("", "code.tar.gz")
Expect(err).NotTo(HaveOccurred())
defer os.Remove(tempfile.Name())
gzipWriter := gzip.NewWriter(tempfile)
tarWriter := tar.NewWriter(gzipWriter)
for source, target := range codeFiles {
file, err := os.Open(source)
Expect(err).NotTo(HaveOccurred())
writeFileToTar(tarWriter, file, target)
file.Close()
}
// close down the inner tar
closeAll(tarWriter, gzipWriter)
writeFileToTar(tw, tempfile, "code.tar.gz")
}
func writeFileToTar(tw *tar.Writer, file *os.File, name string) {
_, err := file.Seek(0, 0)
Expect(err).NotTo(HaveOccurred())
fi, err := file.Stat()
Expect(err).NotTo(HaveOccurred())
header, err := tar.FileInfoHeader(fi, "")
Expect(err).NotTo(HaveOccurred())
header.Name = name
err = tw.WriteHeader(header)
Expect(err).NotTo(HaveOccurred())
_, err = io.Copy(tw, file)
Expect(err).NotTo(HaveOccurred())
}
func closeAll(closers ...io.Closer) {
for _, c := range closers {
Expect(c.Close()).To(Succeed())
}
}