astfile.go

package template

import (
	"fmt"
	"io"
	"sort"
	"strings"
)

type astFile struct {
	header, macroses iAstNode
}

func (n *astFile) GetImports() []string {
	res := []string{}
	if n.macroses != nil {
		res = append(res, n.macroses.GetImports()...)
	}

	return res
}

func (n *astFile) GetStrings() []string {
	res := []string{}
	if n.macroses != nil {
		res = append(res, n.macroses.GetStrings()...)
	}

	return res
}

func (n *astFile) WriteGo(w io.Writer, opts *GenGoOpts) {
	io.WriteString(w, "package "+opts.PackageName+"\n")

	if n.header == nil {
		n.header = &astHeader{}
	}

	opts.Imports = append(opts.Imports, n.GetImports()...)

	n.header.WriteGo(w, opts)

	stringsSet := make(map[string]struct{})
	for _, s := range n.GetStrings() {
		stringsSet[s] = struct{}{}
	}

	if len(stringsSet) > 0 {
		stringsSlice := make([]string, 0, len(stringsSet))
		for s := range stringsSet {
			stringsSlice = append(stringsSlice, s)
		}
		sort.Strings(stringsSlice)

		io.WriteString(w, "var (\n")
		for _, s := range stringsSlice {
			io.WriteString(w, strVarName(s, opts.FileName)+" = []byte{")
			s := strings.Replace(s, `\"`, `"`, -1)
			s = strings.Replace(s, `\\`, `\`, -1)

			for i, b := range []byte(s) {
				if i > 0 {
					if i%50 == 0 {
						io.WriteString(w, ",\n")
					} else {
						io.WriteString(w, ", ")
					}
				}
				fmt.Fprintf(w, "0x%X", b)
			}
			io.WriteString(w, "}\n")
		}
		io.WriteString(w, ")\n\n")
	}

	if n.macroses != nil {
		n.macroses.WriteGo(w, opts)
	}
}