162 lines
3.6 KiB
Go
162 lines
3.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
var funcs = map[string]interface{}{
|
|
"formatBytes": formatBytes,
|
|
}
|
|
|
|
var (
|
|
tmpl = template.Must(template.New("").Funcs(funcs).Parse(`{{- if .Tag -}} // +build {{ .Tag }} {{- end }}
|
|
|
|
// Code generated by "embed"; DO NOT EDIT.
|
|
package {{ .Package }}
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
var assets = map[string][]byte{}
|
|
|
|
var FS = &fs{}
|
|
|
|
type fs struct {}
|
|
|
|
func (fs *fs) Open(name string) (http.File, error) {
|
|
if name == "/" {
|
|
return fs, nil;
|
|
}
|
|
b, ok := assets[name]
|
|
if !ok {
|
|
return nil, os.ErrNotExist
|
|
}
|
|
return &file{name: name, size: len(b), Reader: bytes.NewReader(b)}, nil
|
|
}
|
|
|
|
func (fs *fs) Close() error { return nil }
|
|
func (fs *fs) Read(p []byte) (int, error) { return 0, nil }
|
|
func (fs *fs) Seek(offset int64, whence int) (int64, error) { return 0, nil }
|
|
func (fs *fs) Stat() (os.FileInfo, error) { return fs, nil }
|
|
func (fs *fs) Name() string { return "/" }
|
|
func (fs *fs) Size() int64 { return 0 }
|
|
func (fs *fs) Mode() os.FileMode { return 0755}
|
|
func (fs *fs) ModTime() time.Time{ return time.Time{} }
|
|
func (fs *fs) IsDir() bool { return true }
|
|
func (fs *fs) Sys() interface{} { return nil }
|
|
func (fs *fs) Readdir(count int) ([]os.FileInfo, error) {
|
|
files := []os.FileInfo{}
|
|
for name, data := range assets {
|
|
files = append(files, &file{name: name, size: len(data), Reader: bytes.NewReader(data)})
|
|
}
|
|
return files, nil
|
|
}
|
|
|
|
type file struct {
|
|
name string
|
|
size int
|
|
*bytes.Reader
|
|
}
|
|
|
|
func (f *file) Close() error { return nil }
|
|
func (f *file) Readdir(count int) ([]os.FileInfo, error) { return nil, errors.New("readdir is not supported") }
|
|
func (f *file) Stat() (os.FileInfo, error) { return f, nil }
|
|
func (f *file) Name() string { return f.name }
|
|
func (f *file) Size() int64 { return int64(f.size) }
|
|
func (f *file) Mode() os.FileMode { return 0644 }
|
|
func (f *file) ModTime() time.Time{ return time.Time{} }
|
|
func (f *file) IsDir() bool { return false }
|
|
func (f *file) Sys() interface{} { return nil }
|
|
|
|
`))
|
|
)
|
|
|
|
// Embed is a helper function that embeds assets from the given directories
|
|
// into a Go source file. It is designed to be called from some generator
|
|
// script, see example project to find out how it can be used.
|
|
func Embed(file string, dirs ...string) error {
|
|
var buf bytes.Buffer
|
|
|
|
// execute template
|
|
if err := tmpl.Execute(&buf, struct {
|
|
Package string
|
|
Tag string
|
|
}{
|
|
Package: packageName,
|
|
Tag: tag,
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
w, err := os.Create(file)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer w.Close()
|
|
|
|
fmt.Fprintln(w, buf.String())
|
|
fmt.Fprintln(w, `func init() {`)
|
|
|
|
for _, dir := range dirs {
|
|
filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
log.Printf("packing %s...", path)
|
|
|
|
b, err := ioutil.ReadFile(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
path = filepath.ToSlash(path)
|
|
fmt.Fprintf(w, ` assets[%q] = []byte{`, strings.TrimPrefix(path, dir))
|
|
fmt.Fprintf(w, formatBytes(b))
|
|
fmt.Fprintln(w, `}`)
|
|
return nil
|
|
})
|
|
}
|
|
|
|
fmt.Fprintln(w, `}`)
|
|
|
|
return nil
|
|
}
|
|
|
|
func formatBytes(s []byte) string {
|
|
var builder strings.Builder
|
|
for _, v := range s {
|
|
builder.WriteString(fmt.Sprintf("0x%02x, ", int(v)))
|
|
}
|
|
return builder.String()
|
|
}
|
|
|
|
var packageName string
|
|
var outputFile string
|
|
var tag string
|
|
|
|
func main() {
|
|
flag.StringVar(&packageName, "package", "", "package name")
|
|
flag.StringVar(&tag, "tag", "", "build tag in the generated file")
|
|
flag.StringVar(&outputFile, "output", "assets.go", "output filename")
|
|
flag.Parse()
|
|
args := flag.Args()
|
|
if err := Embed(outputFile, args...); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|