107 lines
2.0 KiB
Go
107 lines
2.0 KiB
Go
// This work is subject to the CC0 1.0 Universal (CC0 1.0) Public Domain Dedication
|
|
// license. Its contents can be found at:
|
|
// http://creativecommons.org/publicdomain/zero/1.0/
|
|
|
|
package bindata
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// writeDebug writes the debug code file.
|
|
func writeDebug(c *Config, toc []Asset) error {
|
|
fd, err := os.Create(filepath.Join(c.Output, "bindata_debug.go"))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer fd.Close()
|
|
|
|
err = writeDebugHeader(fd, c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for i := range toc {
|
|
err = writeDebugAsset(fd, c, &toc[i])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// writeDebugHeader writes output file headers.
|
|
// This targets debug builds.
|
|
func writeDebugHeader(w io.Writer, c *Config) error {
|
|
var err error
|
|
|
|
// Write build tags, if applicable.
|
|
if len(c.Tags) > 0 {
|
|
_, err = fmt.Fprintf(w, "// +build !release %s\n\n", c.Tags)
|
|
} else {
|
|
_, err = fmt.Fprintf(w, "// +build !release\n\n")
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Write package declaration
|
|
_, err = fmt.Fprintf(w, "package %s\n\n", c.Package)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Define packages we need to import.
|
|
// And add the asset_read function. This is called
|
|
// from asset-specific functions.
|
|
_, err = fmt.Fprintf(w, `import (
|
|
"bytes"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
// bindata_read reads the given file from disk.
|
|
// It panics if anything went wrong.
|
|
func bindata_read(path, name string) []byte {
|
|
fd, err := os.Open(path)
|
|
if err != nil {
|
|
log.Fatalf("Read %%s: %%v", name, err)
|
|
}
|
|
|
|
defer fd.Close()
|
|
|
|
var buf bytes.Buffer
|
|
_, err = io.Copy(&buf, fd)
|
|
if err != nil {
|
|
log.Fatalf("Read %%s: %%v", name, err)
|
|
}
|
|
|
|
return buf.Bytes()
|
|
}
|
|
|
|
`)
|
|
return err
|
|
}
|
|
|
|
// writeDebugAsset write a debug entry for the given asset.
|
|
// A debug entry is simply a function which reads the asset from
|
|
// the original file (e.g.: from disk).
|
|
func writeDebugAsset(w io.Writer, c *Config, asset *Asset) error {
|
|
_, err := fmt.Fprintf(w, `func %s() []byte {
|
|
return bindata_read(
|
|
%q,
|
|
%q,
|
|
)
|
|
}
|
|
|
|
`, asset.Func, asset.Path, asset.Name)
|
|
return err
|
|
}
|