885 lines
26 KiB
Go
885 lines
26 KiB
Go
package tmpl
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"log"
|
|
"strconv"
|
|
"strings"
|
|
//"regexp"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"reflect"
|
|
"text/template/parse"
|
|
)
|
|
|
|
// TODO: Turn this file into a library
|
|
var textOverlapList = make(map[string]int)
|
|
|
|
// nolint
|
|
type VarItem struct {
|
|
Name string
|
|
Destination string
|
|
Type string
|
|
}
|
|
type VarItemReflect struct {
|
|
Name string
|
|
Destination string
|
|
Value reflect.Value
|
|
}
|
|
type CTemplateSet struct {
|
|
tlist map[string]*parse.Tree
|
|
dir string
|
|
funcMap map[string]interface{}
|
|
importMap map[string]string
|
|
Fragments map[string]int
|
|
FragmentCursor map[string]int
|
|
FragOut string
|
|
varList map[string]VarItem
|
|
localVars map[string]map[string]VarItemReflect
|
|
stats map[string]int
|
|
pVarList string
|
|
pVarPosition int
|
|
previousNode parse.NodeType
|
|
currentNode parse.NodeType
|
|
nextNode parse.NodeType
|
|
//tempVars map[string]string
|
|
doImports bool
|
|
minify bool
|
|
debug bool
|
|
superDebug bool
|
|
skipHandles bool
|
|
expectsInt interface{}
|
|
}
|
|
|
|
func (c *CTemplateSet) Minify(on bool) {
|
|
c.minify = on
|
|
}
|
|
|
|
func (c *CTemplateSet) Debug(on bool) {
|
|
c.debug = on
|
|
}
|
|
|
|
func (c *CTemplateSet) SuperDebug(on bool) {
|
|
c.superDebug = on
|
|
}
|
|
|
|
func (c *CTemplateSet) SkipHandles(on bool) {
|
|
c.skipHandles = on
|
|
}
|
|
|
|
func (c *CTemplateSet) Compile(name string, dir string, expects string, expectsInt interface{}, varList map[string]VarItem, imports ...string) (out string, err error) {
|
|
if c.debug {
|
|
fmt.Println("Compiling template '" + name + "'")
|
|
}
|
|
|
|
c.dir = dir
|
|
c.doImports = true
|
|
c.funcMap = map[string]interface{}{
|
|
"and": "&&",
|
|
"not": "!",
|
|
"or": "||",
|
|
"eq": true,
|
|
"ge": true,
|
|
"gt": true,
|
|
"le": true,
|
|
"lt": true,
|
|
"ne": true,
|
|
"add": true,
|
|
"subtract": true,
|
|
"multiply": true,
|
|
"divide": true,
|
|
}
|
|
|
|
c.importMap = map[string]string{
|
|
"net/http": "net/http",
|
|
"./common": "./common",
|
|
}
|
|
if len(imports) > 0 {
|
|
for _, importItem := range imports {
|
|
c.importMap[importItem] = importItem
|
|
}
|
|
}
|
|
|
|
c.varList = varList
|
|
//c.pVarList = ""
|
|
//c.pVarPosition = 0
|
|
c.stats = make(map[string]int)
|
|
c.expectsInt = expectsInt
|
|
holdreflect := reflect.ValueOf(expectsInt)
|
|
|
|
res, err := ioutil.ReadFile(dir + name)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
content := string(res)
|
|
if c.minify {
|
|
content = minify(content)
|
|
}
|
|
|
|
tree := parse.New(name, c.funcMap)
|
|
var treeSet = make(map[string]*parse.Tree)
|
|
tree, err = tree.Parse(content, "{{", "}}", treeSet, c.funcMap)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
c.log(name)
|
|
|
|
out = ""
|
|
fname := strings.TrimSuffix(name, filepath.Ext(name))
|
|
c.tlist = make(map[string]*parse.Tree)
|
|
c.tlist[fname] = tree
|
|
varholder := "tmpl_" + fname + "_vars"
|
|
|
|
c.log(c.tlist)
|
|
c.localVars = make(map[string]map[string]VarItemReflect)
|
|
c.localVars[fname] = make(map[string]VarItemReflect)
|
|
c.localVars[fname]["."] = VarItemReflect{".", varholder, holdreflect}
|
|
if c.Fragments == nil {
|
|
c.Fragments = make(map[string]int)
|
|
}
|
|
c.FragmentCursor = make(map[string]int)
|
|
c.FragmentCursor[fname] = 0
|
|
|
|
out += c.rootIterate(c.tlist[fname], varholder, holdreflect, fname)
|
|
|
|
var importList string
|
|
if c.doImports {
|
|
for _, item := range c.importMap {
|
|
importList += "import \"" + item + "\"\n"
|
|
}
|
|
}
|
|
|
|
var varString string
|
|
for _, varItem := range c.varList {
|
|
varString += "var " + varItem.Name + " " + varItem.Type + " = " + varItem.Destination + "\n"
|
|
}
|
|
|
|
fout := "// +build !no_templategen\n\n// Code generated by Gosora. More below:\n/* This file was automatically generated by the software. Please don't edit it as your changes may be overwritten at any moment. */\n"
|
|
|
|
fout += "package main\n" + importList + c.pVarList + "\n"
|
|
fout += "// nolint\nfunc init() {\n"
|
|
|
|
if !c.skipHandles {
|
|
fout += "\tcommon.Template_" + fname + "_handle = Template_" + fname + "\n"
|
|
|
|
fout += "\tcommon.Ctemplates = append(common.Ctemplates,\"" + fname + "\")\n\tcommon.TmplPtrMap[\"" + fname + "\"] = &common.Template_" + fname + "_handle\n"
|
|
}
|
|
|
|
fout += "\tcommon.TmplPtrMap[\"o_" + fname + "\"] = Template_" + fname + "\n}\n\n"
|
|
|
|
fout += "// nolint\nfunc Template_" + fname + "(tmpl_" + fname + "_vars " + expects + ", w http.ResponseWriter) error {\n" + varString + out + "\treturn nil\n}\n"
|
|
|
|
fout = strings.Replace(fout, `))
|
|
w.Write([]byte(`, " + ", -1)
|
|
fout = strings.Replace(fout, "` + `", "", -1)
|
|
//spstr := "`([:space:]*)`"
|
|
//whitespaceWrites := regexp.MustCompile(`(?s)w.Write\(\[\]byte\(`+spstr+`\)\)`)
|
|
//fout = whitespaceWrites.ReplaceAllString(fout,"")
|
|
|
|
if c.debug {
|
|
for index, count := range c.stats {
|
|
fmt.Println(index+": ", strconv.Itoa(count))
|
|
}
|
|
fmt.Println(" ")
|
|
}
|
|
|
|
c.log("Output!")
|
|
c.log(fout)
|
|
//log.Fatal("remove the log.Fatal line")
|
|
return fout, nil
|
|
}
|
|
|
|
func (c *CTemplateSet) rootIterate(tree *parse.Tree, varholder string, holdreflect reflect.Value, fname string) (out string) {
|
|
c.log(tree.Root)
|
|
treeLength := len(tree.Root.Nodes)
|
|
for index, node := range tree.Root.Nodes {
|
|
c.log("Node:", node.String())
|
|
c.previousNode = c.currentNode
|
|
c.currentNode = node.Type()
|
|
if treeLength != (index + 1) {
|
|
c.nextNode = tree.Root.Nodes[index+1].Type()
|
|
}
|
|
out += c.compileSwitch(varholder, holdreflect, fname, node)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) compileSwitch(varholder string, holdreflect reflect.Value, templateName string, node parse.Node) (out string) {
|
|
c.log("in compileSwitch")
|
|
switch node := node.(type) {
|
|
case *parse.ActionNode:
|
|
c.log("Action Node")
|
|
if node.Pipe == nil {
|
|
break
|
|
}
|
|
for _, cmd := range node.Pipe.Cmds {
|
|
out += c.compileSubswitch(varholder, holdreflect, templateName, cmd)
|
|
}
|
|
case *parse.IfNode:
|
|
c.log("If Node:")
|
|
c.log("node.Pipe", node.Pipe)
|
|
|
|
var expr string
|
|
for _, cmd := range node.Pipe.Cmds {
|
|
c.log("If Node Bit:", cmd)
|
|
c.log("If Node Bit Type:", reflect.ValueOf(cmd).Type().Name())
|
|
expr += c.compileVarswitch(varholder, holdreflect, templateName, cmd)
|
|
c.log("If Node Expression Step:", c.compileVarswitch(varholder, holdreflect, templateName, cmd))
|
|
}
|
|
|
|
c.log("If Node Expression:", expr)
|
|
c.previousNode = c.currentNode
|
|
c.currentNode = parse.NodeList
|
|
c.nextNode = -1
|
|
if node.ElseList == nil {
|
|
c.log("Selected Branch 1")
|
|
return "if " + expr + " {\n" + c.compileSwitch(varholder, holdreflect, templateName, node.List) + "}\n"
|
|
}
|
|
|
|
c.log("Selected Branch 2")
|
|
return "if " + expr + " {\n" + c.compileSwitch(varholder, holdreflect, templateName, node.List) + "} else {\n" + c.compileSwitch(varholder, holdreflect, templateName, node.ElseList) + "}\n"
|
|
case *parse.ListNode:
|
|
c.log("List Node")
|
|
for _, subnode := range node.Nodes {
|
|
out += c.compileSwitch(varholder, holdreflect, templateName, subnode)
|
|
}
|
|
case *parse.RangeNode:
|
|
return c.compileRangeNode(varholder, holdreflect, templateName, node)
|
|
case *parse.TemplateNode:
|
|
return c.compileSubtemplate(varholder, holdreflect, node)
|
|
case *parse.TextNode:
|
|
c.previousNode = c.currentNode
|
|
c.currentNode = node.Type()
|
|
c.nextNode = 0
|
|
tmpText := bytes.TrimSpace(node.Text)
|
|
if len(tmpText) == 0 {
|
|
return ""
|
|
}
|
|
|
|
fragmentName := templateName + "_" + strconv.Itoa(c.FragmentCursor[templateName])
|
|
_, ok := c.Fragments[fragmentName]
|
|
if !ok {
|
|
c.Fragments[fragmentName] = len(node.Text)
|
|
c.FragOut += "var " + fragmentName + " = []byte(`" + string(node.Text) + "`)\n"
|
|
}
|
|
c.FragmentCursor[templateName] = c.FragmentCursor[templateName] + 1
|
|
return "w.Write(" + fragmentName + ")\n"
|
|
default:
|
|
return c.unknownNode(node)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) compileRangeNode(varholder string, holdreflect reflect.Value, templateName string, node *parse.RangeNode) (out string) {
|
|
c.log("Range Node!")
|
|
c.log(node.Pipe)
|
|
|
|
var outVal reflect.Value
|
|
for _, cmd := range node.Pipe.Cmds {
|
|
c.log("Range Bit:", cmd)
|
|
out, outVal = c.compileReflectSwitch(varholder, holdreflect, templateName, cmd)
|
|
}
|
|
c.log("Returned:", out)
|
|
c.log("Range Kind Switch!")
|
|
|
|
switch outVal.Kind() {
|
|
case reflect.Map:
|
|
var item reflect.Value
|
|
for _, key := range outVal.MapKeys() {
|
|
item = outVal.MapIndex(key)
|
|
}
|
|
if c.debug {
|
|
fmt.Println("Range item:", item)
|
|
}
|
|
if !item.IsValid() {
|
|
panic("item" + "^\n" + "Invalid map. Maybe, it doesn't have any entries for the template engine to analyse?")
|
|
}
|
|
|
|
if node.ElseList != nil {
|
|
out = "if len(" + out + ") != 0 {\nfor _, item := range " + out + " {\n" + c.compileSwitch("item", item, templateName, node.List) + "}\n} else {\n" + c.compileSwitch("item", item, templateName, node.ElseList) + "}\n"
|
|
} else {
|
|
out = "if len(" + out + ") != 0 {\nfor _, item := range " + out + " {\n" + c.compileSwitch("item", item, templateName, node.List) + "}\n}"
|
|
}
|
|
case reflect.Slice:
|
|
if outVal.Len() == 0 {
|
|
panic("The sample data needs at-least one or more elements for the slices. We're looking into removing this requirement at some point!")
|
|
}
|
|
item := outVal.Index(0)
|
|
out = "if len(" + out + ") != 0 {\nfor _, item := range " + out + " {\n" + c.compileSwitch("item", item, templateName, node.List) + "}\n}"
|
|
case reflect.Invalid:
|
|
return ""
|
|
}
|
|
|
|
if node.ElseList != nil {
|
|
out += " else {\n" + c.compileSwitch(varholder, holdreflect, templateName, node.ElseList) + "}"
|
|
}
|
|
return out + "\n"
|
|
}
|
|
|
|
func (c *CTemplateSet) compileSubswitch(varholder string, holdreflect reflect.Value, templateName string, node *parse.CommandNode) (out string) {
|
|
c.log("in compileSubswitch")
|
|
firstWord := node.Args[0]
|
|
switch n := firstWord.(type) {
|
|
case *parse.FieldNode:
|
|
c.log("Field Node:", n.Ident)
|
|
|
|
/* Use reflect to determine if the field is for a method, otherwise assume it's a variable. Variable declarations are coming soon! */
|
|
cur := holdreflect
|
|
|
|
var varbit string
|
|
if cur.Kind() == reflect.Interface {
|
|
cur = cur.Elem()
|
|
varbit += ".(" + cur.Type().Name() + ")"
|
|
}
|
|
|
|
for _, id := range n.Ident {
|
|
c.log("Data Kind:", cur.Kind().String())
|
|
c.log("Field Bit:", id)
|
|
|
|
if cur.Kind() == reflect.Ptr {
|
|
c.log("Looping over pointer")
|
|
for cur.Kind() == reflect.Ptr {
|
|
cur = cur.Elem()
|
|
}
|
|
c.log("Data Kind:", cur.Kind().String())
|
|
c.log("Field Bit:", id)
|
|
}
|
|
|
|
if !cur.IsValid() {
|
|
if c.debug {
|
|
fmt.Println("Debug Data:")
|
|
fmt.Println("Holdreflect:", holdreflect)
|
|
fmt.Println("Holdreflect.Kind():", holdreflect.Kind())
|
|
if !c.superDebug {
|
|
fmt.Println("cur.Kind():", cur.Kind().String())
|
|
}
|
|
fmt.Println("")
|
|
}
|
|
|
|
panic(varholder + varbit + "^\n" + "Invalid value. Maybe, it doesn't exist?")
|
|
}
|
|
|
|
cur = cur.FieldByName(id)
|
|
if cur.Kind() == reflect.Interface {
|
|
cur = cur.Elem()
|
|
// TODO: Surely, there's a better way of detecting this?
|
|
/*if cur.Kind() == reflect.String && cur.Type().Name() != "string" {
|
|
varbit = "string(" + varbit + "." + id + ")"*/
|
|
//if cur.Kind() == reflect.String && cur.Type().Name() != "string" {
|
|
if cur.Type().PkgPath() != "main" && cur.Type().PkgPath() != "" {
|
|
c.importMap["html/template"] = "html/template"
|
|
varbit += "." + id + ".(" + strings.TrimPrefix(cur.Type().PkgPath(), "html/") + "." + cur.Type().Name() + ")"
|
|
} else {
|
|
varbit += "." + id + ".(" + cur.Type().Name() + ")"
|
|
}
|
|
} else {
|
|
varbit += "." + id
|
|
}
|
|
c.log("End Cycle")
|
|
}
|
|
out = c.compileVarsub(varholder+varbit, cur)
|
|
|
|
for _, varItem := range c.varList {
|
|
if strings.HasPrefix(out, varItem.Destination) {
|
|
out = strings.Replace(out, varItem.Destination, varItem.Name, 1)
|
|
}
|
|
}
|
|
return out
|
|
case *parse.DotNode:
|
|
c.log("Dot Node:", node.String())
|
|
return c.compileVarsub(varholder, holdreflect)
|
|
case *parse.NilNode:
|
|
panic("Nil is not a command x.x")
|
|
case *parse.VariableNode:
|
|
c.log("Variable Node:", n.String())
|
|
c.log(n.Ident)
|
|
varname, reflectVal := c.compileIfVarsub(n.String(), varholder, templateName, holdreflect)
|
|
return c.compileVarsub(varname, reflectVal)
|
|
case *parse.StringNode:
|
|
return n.Quoted
|
|
case *parse.IdentifierNode:
|
|
c.log("Identifier Node:", node)
|
|
c.log("Identifier Node Args:", node.Args)
|
|
return c.compileVarsub(c.compileIdentSwitch(varholder, holdreflect, templateName, node))
|
|
default:
|
|
return c.unknownNode(node)
|
|
}
|
|
}
|
|
|
|
func (c *CTemplateSet) compileVarswitch(varholder string, holdreflect reflect.Value, templateName string, node *parse.CommandNode) (out string) {
|
|
c.log("in compileVarswitch")
|
|
firstWord := node.Args[0]
|
|
switch n := firstWord.(type) {
|
|
case *parse.FieldNode:
|
|
if c.superDebug {
|
|
fmt.Println("Field Node:", n.Ident)
|
|
for _, id := range n.Ident {
|
|
fmt.Println("Field Bit:", id)
|
|
}
|
|
}
|
|
|
|
/* Use reflect to determine if the field is for a method, otherwise assume it's a variable. Coming Soon. */
|
|
return c.compileBoolsub(n.String(), varholder, templateName, holdreflect)
|
|
case *parse.ChainNode:
|
|
c.log("Chain Node:", n.Node)
|
|
c.log("Chain Node Args:", node.Args)
|
|
case *parse.IdentifierNode:
|
|
c.log("Identifier Node:", node)
|
|
c.log("Identifier Node Args:", node.Args)
|
|
return c.compileIdentSwitchN(varholder, holdreflect, templateName, node)
|
|
case *parse.DotNode:
|
|
return varholder
|
|
case *parse.VariableNode:
|
|
c.log("Variable Node:", n.String())
|
|
c.log("Variable Node Identifier:", n.Ident)
|
|
out, _ = c.compileIfVarsub(n.String(), varholder, templateName, holdreflect)
|
|
case *parse.NilNode:
|
|
panic("Nil is not a command x.x")
|
|
case *parse.PipeNode:
|
|
c.log("Pipe Node!")
|
|
c.log(n)
|
|
c.log("Args:", node.Args)
|
|
out += c.compileIdentSwitchN(varholder, holdreflect, templateName, node)
|
|
|
|
c.log("Out:", out)
|
|
default:
|
|
return c.unknownNode(firstWord)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) unknownNode(node parse.Node) (out string) {
|
|
fmt.Println("Unknown Kind:", reflect.ValueOf(node).Elem().Kind())
|
|
fmt.Println("Unknown Type:", reflect.ValueOf(node).Elem().Type().Name())
|
|
panic("I don't know what node this is! Grr...")
|
|
}
|
|
|
|
func (c *CTemplateSet) compileIdentSwitchN(varholder string, holdreflect reflect.Value, templateName string, node *parse.CommandNode) (out string) {
|
|
c.log("in compileIdentSwitchN")
|
|
out, _ = c.compileIdentSwitch(varholder, holdreflect, templateName, node)
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) dumpSymbol(pos int, node *parse.CommandNode, symbol string) {
|
|
c.log("symbol: ", symbol)
|
|
c.log("node.Args[pos + 1]", node.Args[pos+1])
|
|
c.log("node.Args[pos + 2]", node.Args[pos+2])
|
|
}
|
|
|
|
func (c *CTemplateSet) compareFunc(varholder string, holdreflect reflect.Value, templateName string, pos int, node *parse.CommandNode, compare string) (out string) {
|
|
c.dumpSymbol(pos, node, compare)
|
|
return c.compileIfVarsubN(node.Args[pos+1].String(), varholder, templateName, holdreflect) + " " + compare + " " + c.compileIfVarsubN(node.Args[pos+2].String(), varholder, templateName, holdreflect)
|
|
}
|
|
|
|
func (c *CTemplateSet) simpleMath(varholder string, holdreflect reflect.Value, templateName string, pos int, node *parse.CommandNode, symbol string) (out string, val reflect.Value) {
|
|
leftParam, val2 := c.compileIfVarsub(node.Args[pos+1].String(), varholder, templateName, holdreflect)
|
|
rightParam, val3 := c.compileIfVarsub(node.Args[pos+2].String(), varholder, templateName, holdreflect)
|
|
|
|
if val2.IsValid() {
|
|
val = val2
|
|
} else if val3.IsValid() {
|
|
val = val3
|
|
} else {
|
|
// TODO: What does this do?
|
|
numSample := 1
|
|
val = reflect.ValueOf(numSample)
|
|
}
|
|
|
|
c.dumpSymbol(pos, node, symbol)
|
|
return leftParam + " " + symbol + " " + rightParam, val
|
|
}
|
|
|
|
func (c *CTemplateSet) compareJoin(varholder string, holdreflect reflect.Value, templateName string, pos int, node *parse.CommandNode, symbol string) (pos2 int, out string) {
|
|
c.log("Building " + symbol + " function")
|
|
if pos == 0 {
|
|
fmt.Println("pos:", pos)
|
|
panic(symbol + " is missing a left operand")
|
|
}
|
|
if len(node.Args) <= pos {
|
|
fmt.Println("post pos:", pos)
|
|
fmt.Println("len(node.Args):", len(node.Args))
|
|
panic(symbol + " is missing a right operand")
|
|
}
|
|
|
|
left := c.compileBoolsub(node.Args[pos-1].String(), varholder, templateName, holdreflect)
|
|
_, funcExists := c.funcMap[node.Args[pos+1].String()]
|
|
|
|
var right string
|
|
if !funcExists {
|
|
right = c.compileBoolsub(node.Args[pos+1].String(), varholder, templateName, holdreflect)
|
|
}
|
|
out = left + " " + symbol + " " + right
|
|
|
|
c.log("Left operand:", node.Args[pos-1])
|
|
c.log("Right operand:", node.Args[pos+1])
|
|
if !funcExists {
|
|
pos++
|
|
}
|
|
c.log("pos:", pos)
|
|
c.log("len(node.Args):", len(node.Args))
|
|
|
|
return pos, out
|
|
}
|
|
|
|
func (c *CTemplateSet) compileIdentSwitch(varholder string, holdreflect reflect.Value, templateName string, node *parse.CommandNode) (out string, val reflect.Value) {
|
|
c.log("in compileIdentSwitch")
|
|
//var outbuf map[int]string
|
|
ArgLoop:
|
|
for pos := 0; pos < len(node.Args); pos++ {
|
|
id := node.Args[pos]
|
|
c.log("pos:", pos)
|
|
c.log("ID:", id)
|
|
switch id.String() {
|
|
case "not":
|
|
out += "!"
|
|
case "or", "and":
|
|
var rout string
|
|
pos, rout = c.compareJoin(varholder, holdreflect, templateName, pos, node, c.funcMap[id.String()].(string)) // TODO: Test this
|
|
out += rout
|
|
case "le": // TODO: Can we condense these comparison cases down into one?
|
|
out += c.compareFunc(varholder, holdreflect, templateName, pos, node, "<=")
|
|
break ArgLoop
|
|
case "lt":
|
|
out += c.compareFunc(varholder, holdreflect, templateName, pos, node, "<")
|
|
break ArgLoop
|
|
case "gt":
|
|
out += c.compareFunc(varholder, holdreflect, templateName, pos, node, ">")
|
|
break ArgLoop
|
|
case "ge":
|
|
out += c.compareFunc(varholder, holdreflect, templateName, pos, node, ">=")
|
|
break ArgLoop
|
|
case "eq":
|
|
out += c.compareFunc(varholder, holdreflect, templateName, pos, node, "==")
|
|
break ArgLoop
|
|
case "ne":
|
|
out += c.compareFunc(varholder, holdreflect, templateName, pos, node, "!=")
|
|
break ArgLoop
|
|
case "add":
|
|
rout, rval := c.simpleMath(varholder, holdreflect, templateName, pos, node, "+")
|
|
out += rout
|
|
val = rval
|
|
break ArgLoop
|
|
case "subtract":
|
|
rout, rval := c.simpleMath(varholder, holdreflect, templateName, pos, node, "-")
|
|
out += rout
|
|
val = rval
|
|
break ArgLoop
|
|
case "divide":
|
|
rout, rval := c.simpleMath(varholder, holdreflect, templateName, pos, node, "/")
|
|
out += rout
|
|
val = rval
|
|
break ArgLoop
|
|
case "multiply":
|
|
rout, rval := c.simpleMath(varholder, holdreflect, templateName, pos, node, "*")
|
|
out += rout
|
|
val = rval
|
|
break ArgLoop
|
|
default:
|
|
c.log("Variable!")
|
|
if len(node.Args) > (pos + 1) {
|
|
nextNode := node.Args[pos+1].String()
|
|
if nextNode == "or" || nextNode == "and" {
|
|
continue
|
|
}
|
|
}
|
|
out += c.compileIfVarsubN(id.String(), varholder, templateName, holdreflect)
|
|
}
|
|
}
|
|
return out, val
|
|
}
|
|
|
|
func (c *CTemplateSet) compileReflectSwitch(varholder string, holdreflect reflect.Value, templateName string, node *parse.CommandNode) (out string, outVal reflect.Value) {
|
|
c.log("in compileReflectSwitch")
|
|
firstWord := node.Args[0]
|
|
switch n := firstWord.(type) {
|
|
case *parse.FieldNode:
|
|
if c.superDebug {
|
|
fmt.Println("Field Node:", n.Ident)
|
|
for _, id := range n.Ident {
|
|
fmt.Println("Field Bit:", id)
|
|
}
|
|
}
|
|
/* Use reflect to determine if the field is for a method, otherwise assume it's a variable. Coming Soon. */
|
|
return c.compileIfVarsub(n.String(), varholder, templateName, holdreflect)
|
|
case *parse.ChainNode:
|
|
c.log("Chain Node:", n.Node)
|
|
c.log("node.Args:", node.Args)
|
|
case *parse.DotNode:
|
|
return varholder, holdreflect
|
|
case *parse.NilNode:
|
|
panic("Nil is not a command x.x")
|
|
default:
|
|
//panic("I don't know what node this is")
|
|
}
|
|
return "", outVal
|
|
}
|
|
|
|
func (c *CTemplateSet) compileIfVarsubN(varname string, varholder string, templateName string, cur reflect.Value) (out string) {
|
|
c.log("in compileIfVarsubN")
|
|
out, _ = c.compileIfVarsub(varname, varholder, templateName, cur)
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) compileIfVarsub(varname string, varholder string, templateName string, cur reflect.Value) (out string, val reflect.Value) {
|
|
c.log("in compileIfVarsub")
|
|
if varname[0] != '.' && varname[0] != '$' {
|
|
return varname, cur
|
|
}
|
|
|
|
bits := strings.Split(varname, ".")
|
|
if varname[0] == '$' {
|
|
var res VarItemReflect
|
|
if varname[1] == '.' {
|
|
res = c.localVars[templateName]["."]
|
|
} else {
|
|
res = c.localVars[templateName][strings.TrimPrefix(bits[0], "$")]
|
|
}
|
|
out += res.Destination
|
|
cur = res.Value
|
|
|
|
if cur.Kind() == reflect.Interface {
|
|
cur = cur.Elem()
|
|
}
|
|
} else {
|
|
out += varholder
|
|
if cur.Kind() == reflect.Interface {
|
|
cur = cur.Elem()
|
|
out += ".(" + cur.Type().Name() + ")"
|
|
}
|
|
}
|
|
bits[0] = strings.TrimPrefix(bits[0], "$")
|
|
|
|
c.log("Cur Kind:", cur.Kind())
|
|
c.log("Cur Type:", cur.Type().Name())
|
|
for _, bit := range bits {
|
|
c.log("Variable Field:", bit)
|
|
if bit == "" {
|
|
continue
|
|
}
|
|
|
|
// TODO: Fix this up so that it works for regular pointers and not just struct pointers. Ditto for the other cur.Kind() == reflect.Ptr we have in this file
|
|
if cur.Kind() == reflect.Ptr {
|
|
c.log("Looping over pointer")
|
|
for cur.Kind() == reflect.Ptr {
|
|
cur = cur.Elem()
|
|
}
|
|
c.log("Data Kind:", cur.Kind().String())
|
|
c.log("Field Bit:", bit)
|
|
}
|
|
|
|
cur = cur.FieldByName(bit)
|
|
out += "." + bit
|
|
if cur.Kind() == reflect.Interface {
|
|
cur = cur.Elem()
|
|
out += ".(" + cur.Type().Name() + ")"
|
|
}
|
|
if !cur.IsValid() {
|
|
panic(out + "^\n" + "Invalid value. Maybe, it doesn't exist?")
|
|
}
|
|
|
|
c.log("Data Kind:", cur.Kind())
|
|
c.log("Data Type:", cur.Type().Name())
|
|
}
|
|
|
|
c.log("Out Value:", out)
|
|
c.log("Out Kind:", cur.Kind())
|
|
c.log("Out Type:", cur.Type().Name())
|
|
|
|
for _, varItem := range c.varList {
|
|
if strings.HasPrefix(out, varItem.Destination) {
|
|
out = strings.Replace(out, varItem.Destination, varItem.Name, 1)
|
|
}
|
|
}
|
|
|
|
c.log("Out Value:", out)
|
|
c.log("Out Kind:", cur.Kind())
|
|
c.log("Out Type:", cur.Type().Name())
|
|
|
|
_, ok := c.stats[out]
|
|
if ok {
|
|
c.stats[out]++
|
|
} else {
|
|
c.stats[out] = 1
|
|
}
|
|
|
|
return out, cur
|
|
}
|
|
|
|
func (c *CTemplateSet) compileBoolsub(varname string, varholder string, templateName string, val reflect.Value) string {
|
|
c.log("in compileBoolsub")
|
|
out, val := c.compileIfVarsub(varname, varholder, templateName, val)
|
|
// TODO: What if it's a pointer or an interface? I *think* we've got pointers handled somewhere, but not interfaces which we don't know the types of at compile time
|
|
switch val.Kind() {
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Float32, reflect.Float64:
|
|
out += " > 0"
|
|
case reflect.Bool: // Do nothing
|
|
case reflect.String:
|
|
out += " != \"\""
|
|
case reflect.Slice, reflect.Map:
|
|
out = "len(" + out + ") != 0"
|
|
default:
|
|
fmt.Println("Variable Name:", varname)
|
|
fmt.Println("Variable Holder:", varholder)
|
|
fmt.Println("Variable Kind:", val.Kind())
|
|
panic("I don't know what this variable's type is o.o\n")
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) compileVarsub(varname string, val reflect.Value) string {
|
|
c.log("in compileVarsub")
|
|
for _, varItem := range c.varList {
|
|
if strings.HasPrefix(varname, varItem.Destination) {
|
|
varname = strings.Replace(varname, varItem.Destination, varItem.Name, 1)
|
|
}
|
|
}
|
|
|
|
_, ok := c.stats[varname]
|
|
if ok {
|
|
c.stats[varname]++
|
|
} else {
|
|
c.stats[varname] = 1
|
|
}
|
|
|
|
if val.Kind() == reflect.Interface {
|
|
val = val.Elem()
|
|
}
|
|
|
|
switch val.Kind() {
|
|
case reflect.Int:
|
|
c.importMap["strconv"] = "strconv"
|
|
return "w.Write([]byte(strconv.Itoa(" + varname + ")))\n"
|
|
case reflect.Bool:
|
|
return "if " + varname + " {\nw.Write([]byte(\"true\"))} else {\nw.Write([]byte(\"false\"))\n}\n"
|
|
case reflect.String:
|
|
if val.Type().Name() != "string" && !strings.HasPrefix(varname, "string(") {
|
|
return "w.Write([]byte(string(" + varname + ")))\n"
|
|
}
|
|
return "w.Write([]byte(" + varname + "))\n"
|
|
case reflect.Int64:
|
|
c.importMap["strconv"] = "strconv"
|
|
return "w.Write([]byte(strconv.FormatInt(" + varname + ", 10)))"
|
|
default:
|
|
if !val.IsValid() {
|
|
panic(varname + "^\n" + "Invalid value. Maybe, it doesn't exist?")
|
|
}
|
|
fmt.Println("Unknown Variable Name:", varname)
|
|
fmt.Println("Unknown Kind:", val.Kind())
|
|
fmt.Println("Unknown Type:", val.Type().Name())
|
|
panic("// I don't know what this variable's type is o.o\n")
|
|
}
|
|
}
|
|
|
|
func (c *CTemplateSet) compileSubtemplate(pvarholder string, pholdreflect reflect.Value, node *parse.TemplateNode) (out string) {
|
|
c.log("in compileSubtemplate")
|
|
c.log("Template Node:", node.Name)
|
|
|
|
fname := strings.TrimSuffix(node.Name, filepath.Ext(node.Name))
|
|
varholder := "tmpl_" + fname + "_vars"
|
|
var holdreflect reflect.Value
|
|
if node.Pipe != nil {
|
|
for _, cmd := range node.Pipe.Cmds {
|
|
firstWord := cmd.Args[0]
|
|
switch firstWord.(type) {
|
|
case *parse.DotNode:
|
|
varholder = pvarholder
|
|
holdreflect = pholdreflect
|
|
case *parse.NilNode:
|
|
panic("Nil is not a command x.x")
|
|
default:
|
|
out = "var " + varholder + " := false\n"
|
|
out += c.compileCommand(cmd)
|
|
}
|
|
}
|
|
}
|
|
|
|
// TODO: Cascade errors back up the tree to the caller?
|
|
res, err := ioutil.ReadFile(c.dir + node.Name)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
content := string(res)
|
|
if c.minify {
|
|
content = minify(content)
|
|
}
|
|
|
|
tree := parse.New(node.Name, c.funcMap)
|
|
var treeSet = make(map[string]*parse.Tree)
|
|
tree, err = tree.Parse(content, "{{", "}}", treeSet, c.funcMap)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
c.tlist[fname] = tree
|
|
subtree := c.tlist[fname]
|
|
c.log("subtree.Root", subtree.Root)
|
|
|
|
c.localVars[fname] = make(map[string]VarItemReflect)
|
|
c.localVars[fname]["."] = VarItemReflect{".", varholder, holdreflect}
|
|
c.FragmentCursor[fname] = 0
|
|
|
|
out += c.rootIterate(subtree, varholder, holdreflect, fname)
|
|
return out
|
|
}
|
|
|
|
func (c *CTemplateSet) log(args ...interface{}) {
|
|
if c.superDebug {
|
|
fmt.Println(args...)
|
|
}
|
|
}
|
|
|
|
func (c *CTemplateSet) compileCommand(*parse.CommandNode) (out string) {
|
|
panic("Uh oh! Something went wrong!")
|
|
}
|
|
|
|
// TODO: Write unit tests for this
|
|
func minify(data string) string {
|
|
data = strings.Replace(data, "\t", "", -1)
|
|
data = strings.Replace(data, "\v", "", -1)
|
|
data = strings.Replace(data, "\n", "", -1)
|
|
data = strings.Replace(data, "\r", "", -1)
|
|
data = strings.Replace(data, " ", " ", -1)
|
|
return data
|
|
}
|
|
|
|
// TODO: Strip comments
|
|
// TODO: Handle CSS nested in <style> tags?
|
|
// TODO: Write unit tests for this
|
|
func minifyHTML(data string) string {
|
|
return minify(data)
|
|
}
|
|
|
|
// TODO: Have static files use this
|
|
// TODO: Strip comments
|
|
// TODO: Convert the rgb()s to hex codes?
|
|
// TODO: Write unit tests for this
|
|
func minifyCSS(data string) string {
|
|
return minify(data)
|
|
}
|
|
|
|
// TODO: Convert this to three character hex strings whenever possible?
|
|
// TODO: Write unit tests for this
|
|
// nolint
|
|
func rgbToHexstr(red int, green int, blue int) string {
|
|
return strconv.FormatInt(int64(red), 16) + strconv.FormatInt(int64(green), 16) + strconv.FormatInt(int64(blue), 16)
|
|
}
|
|
|
|
/*
|
|
// TODO: Write unit tests for this
|
|
func hexstrToRgb(hexstr string) (red int, blue int, green int, err error) {
|
|
// Strip the # at the start
|
|
if hexstr[0] == '#' {
|
|
hexstr = strings.TrimPrefix(hexstr,"#")
|
|
}
|
|
if len(hexstr) != 3 && len(hexstr) != 6 {
|
|
return 0, 0, 0, errors.New("Hex colour codes may only be three or six characters long")
|
|
}
|
|
|
|
if len(hexstr) == 3 {
|
|
hexstr = hexstr[0] + hexstr[0] + hexstr[1] + hexstr[1] + hexstr[2] + hexstr[2]
|
|
}
|
|
}*/
|