bindata/debug.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
}