bindata/debug.go

107 lines
2.0 KiB
Go
Raw Normal View History

// 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
}