mirror of
https://github.com/astaxie/beego.git
synced 2024-11-25 20:51:29 +00:00
add BeeTemplateExt
This commit is contained in:
astaxie
parent
512d115639
commit
281747ab43
325
template.go
325
template.go
@ -1,156 +1,169 @@
|
|||||||
package beego
|
package beego
|
||||||
|
|
||||||
//@todo add template funcs
|
//@todo add template funcs
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/russross/blackfriday"
|
"github.com/russross/blackfriday"
|
||||||
"html/template"
|
"html/template"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
beegoTplFuncMap template.FuncMap
|
beegoTplFuncMap template.FuncMap
|
||||||
BeeTemplates map[string]*template.Template
|
BeeTemplates map[string]*template.Template
|
||||||
BeeTemplateExt string
|
BeeTemplateExt []string
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
BeeTemplates = make(map[string]*template.Template)
|
BeeTemplates = make(map[string]*template.Template)
|
||||||
beegoTplFuncMap = make(template.FuncMap)
|
beegoTplFuncMap = make(template.FuncMap)
|
||||||
BeeTemplateExt = "tpl"
|
BeeTemplateExt = make([]string, 0)
|
||||||
beegoTplFuncMap["markdown"] = MarkDown
|
BeeTemplateExt = append(BeeTemplateExt, "tpl", "html")
|
||||||
beegoTplFuncMap["dateformat"] = DateFormat
|
beegoTplFuncMap["markdown"] = MarkDown
|
||||||
beegoTplFuncMap["date"] = Date
|
beegoTplFuncMap["dateformat"] = DateFormat
|
||||||
beegoTplFuncMap["compare"] = Compare
|
beegoTplFuncMap["date"] = Date
|
||||||
}
|
beegoTplFuncMap["compare"] = Compare
|
||||||
|
}
|
||||||
// MarkDown parses a string in MarkDown format and returns HTML. Used by the template parser as "markdown"
|
|
||||||
func MarkDown(raw string) (output template.HTML) {
|
// MarkDown parses a string in MarkDown format and returns HTML. Used by the template parser as "markdown"
|
||||||
input := []byte(raw)
|
func MarkDown(raw string) (output template.HTML) {
|
||||||
bOutput := blackfriday.MarkdownBasic(input)
|
input := []byte(raw)
|
||||||
output = template.HTML(string(bOutput))
|
bOutput := blackfriday.MarkdownBasic(input)
|
||||||
return
|
output = template.HTML(string(bOutput))
|
||||||
}
|
return
|
||||||
|
}
|
||||||
// DateFormat takes a time and a layout string and returns a string with the formatted date. Used by the template parser as "dateformat"
|
|
||||||
func DateFormat(t time.Time, layout string) (datestring string) {
|
// DateFormat takes a time and a layout string and returns a string with the formatted date. Used by the template parser as "dateformat"
|
||||||
datestring = t.Format(layout)
|
func DateFormat(t time.Time, layout string) (datestring string) {
|
||||||
return
|
datestring = t.Format(layout)
|
||||||
}
|
return
|
||||||
|
}
|
||||||
// Date takes a PHP like date func to Go's time fomate
|
|
||||||
func Date(t time.Time, format string) (datestring string) {
|
// Date takes a PHP like date func to Go's time fomate
|
||||||
patterns := []string{
|
func Date(t time.Time, format string) (datestring string) {
|
||||||
// year
|
patterns := []string{
|
||||||
"Y", "2006", // A full numeric representation of a year, 4 digits Examples: 1999 or 2003
|
// year
|
||||||
"y", "06", //A two digit representation of a year Examples: 99 or 03
|
"Y", "2006", // A full numeric representation of a year, 4 digits Examples: 1999 or 2003
|
||||||
|
"y", "06", //A two digit representation of a year Examples: 99 or 03
|
||||||
// month
|
|
||||||
"m", "01", // Numeric representation of a month, with leading zeros 01 through 12
|
// month
|
||||||
"n", "1", // Numeric representation of a month, without leading zeros 1 through 12
|
"m", "01", // Numeric representation of a month, with leading zeros 01 through 12
|
||||||
"M", "Jan", // A short textual representation of a month, three letters Jan through Dec
|
"n", "1", // Numeric representation of a month, without leading zeros 1 through 12
|
||||||
"F", "January", // A full textual representation of a month, such as January or March January through December
|
"M", "Jan", // A short textual representation of a month, three letters Jan through Dec
|
||||||
|
"F", "January", // A full textual representation of a month, such as January or March January through December
|
||||||
// day
|
|
||||||
"d", "02", // Day of the month, 2 digits with leading zeros 01 to 31
|
// day
|
||||||
"j", "2", // Day of the month without leading zeros 1 to 31
|
"d", "02", // Day of the month, 2 digits with leading zeros 01 to 31
|
||||||
|
"j", "2", // Day of the month without leading zeros 1 to 31
|
||||||
// week
|
|
||||||
"D", "Mon", // A textual representation of a day, three letters Mon through Sun
|
// week
|
||||||
"l", "Monday", // A full textual representation of the day of the week Sunday through Saturday
|
"D", "Mon", // A textual representation of a day, three letters Mon through Sun
|
||||||
|
"l", "Monday", // A full textual representation of the day of the week Sunday through Saturday
|
||||||
// time
|
|
||||||
"g", "3", // 12-hour format of an hour without leading zeros 1 through 12
|
// time
|
||||||
"G", "15", // 24-hour format of an hour without leading zeros 0 through 23
|
"g", "3", // 12-hour format of an hour without leading zeros 1 through 12
|
||||||
"h", "03", // 12-hour format of an hour with leading zeros 01 through 12
|
"G", "15", // 24-hour format of an hour without leading zeros 0 through 23
|
||||||
"H", "15", // 24-hour format of an hour with leading zeros 00 through 23
|
"h", "03", // 12-hour format of an hour with leading zeros 01 through 12
|
||||||
|
"H", "15", // 24-hour format of an hour with leading zeros 00 through 23
|
||||||
"a", "pm", // Lowercase Ante meridiem and Post meridiem am or pm
|
|
||||||
"A", "PM", // Uppercase Ante meridiem and Post meridiem AM or PM
|
"a", "pm", // Lowercase Ante meridiem and Post meridiem am or pm
|
||||||
|
"A", "PM", // Uppercase Ante meridiem and Post meridiem AM or PM
|
||||||
"i", "04", // Minutes with leading zeros 00 to 59
|
|
||||||
"s", "05", // Seconds, with leading zeros 00 through 59
|
"i", "04", // Minutes with leading zeros 00 to 59
|
||||||
}
|
"s", "05", // Seconds, with leading zeros 00 through 59
|
||||||
replacer := strings.NewReplacer(patterns...)
|
}
|
||||||
format = replacer.Replace(format)
|
replacer := strings.NewReplacer(patterns...)
|
||||||
datestring = t.Format(format)
|
format = replacer.Replace(format)
|
||||||
return
|
datestring = t.Format(format)
|
||||||
}
|
return
|
||||||
|
}
|
||||||
// Compare is a quick and dirty comparison function. It will convert whatever you give it to strings and see if the two values are equal.
|
|
||||||
// Whitespace is trimmed. Used by the template parser as "eq"
|
// Compare is a quick and dirty comparison function. It will convert whatever you give it to strings and see if the two values are equal.
|
||||||
func Compare(a, b interface{}) (equal bool) {
|
// Whitespace is trimmed. Used by the template parser as "eq"
|
||||||
equal = false
|
func Compare(a, b interface{}) (equal bool) {
|
||||||
if strings.TrimSpace(fmt.Sprintf("%v", a)) == strings.TrimSpace(fmt.Sprintf("%v", b)) {
|
equal = false
|
||||||
equal = true
|
if strings.TrimSpace(fmt.Sprintf("%v", a)) == strings.TrimSpace(fmt.Sprintf("%v", b)) {
|
||||||
}
|
equal = true
|
||||||
return
|
}
|
||||||
}
|
return
|
||||||
|
}
|
||||||
// AddFuncMap let user to register a func in the template
|
|
||||||
func AddFuncMap(key string, funname interface{}) error {
|
// AddFuncMap let user to register a func in the template
|
||||||
if _, ok := beegoTplFuncMap[key]; ok {
|
func AddFuncMap(key string, funname interface{}) error {
|
||||||
return errors.New("funcmap already has the key")
|
if _, ok := beegoTplFuncMap[key]; ok {
|
||||||
}
|
return errors.New("funcmap already has the key")
|
||||||
beegoTplFuncMap[key] = funname
|
}
|
||||||
return nil
|
beegoTplFuncMap[key] = funname
|
||||||
}
|
return nil
|
||||||
|
}
|
||||||
type templatefile struct {
|
|
||||||
root string
|
type templatefile struct {
|
||||||
files map[string][]string
|
root string
|
||||||
}
|
files map[string][]string
|
||||||
|
}
|
||||||
func (self *templatefile) visit(paths string, f os.FileInfo, err error) error {
|
|
||||||
if f == nil {
|
func (self *templatefile) visit(paths string, f os.FileInfo, err error) error {
|
||||||
return err
|
if f == nil {
|
||||||
}
|
return err
|
||||||
if f.IsDir() {
|
}
|
||||||
return nil
|
if f.IsDir() {
|
||||||
} else if (f.Mode() & os.ModeSymlink) > 0 {
|
return nil
|
||||||
return nil
|
} else if (f.Mode() & os.ModeSymlink) > 0 {
|
||||||
} else {
|
return nil
|
||||||
if strings.HasSuffix(paths, BeeTemplateExt) {
|
} else {
|
||||||
a := []byte(paths)
|
hasExt := false
|
||||||
a = a[len([]byte(self.root)):]
|
for _, v := range BeeTemplateExt {
|
||||||
subdir := path.Dir(strings.TrimLeft(string(a), "/"))
|
if strings.HasSuffix(paths, v) {
|
||||||
if _, ok := self.files[subdir]; ok {
|
hasExt = true
|
||||||
self.files[subdir] = append(self.files[subdir], paths)
|
break
|
||||||
} else {
|
}
|
||||||
m := make([]string, 1)
|
}
|
||||||
m[0] = paths
|
if hasExt {
|
||||||
self.files[subdir] = m
|
a := []byte(paths)
|
||||||
}
|
a = a[len([]byte(self.root)):]
|
||||||
|
subdir := path.Dir(strings.TrimLeft(string(a), "/"))
|
||||||
}
|
if _, ok := self.files[subdir]; ok {
|
||||||
}
|
self.files[subdir] = append(self.files[subdir], paths)
|
||||||
return nil
|
} else {
|
||||||
}
|
m := make([]string, 1)
|
||||||
|
m[0] = paths
|
||||||
func SetGlobalTemplateExt(ext string) {
|
self.files[subdir] = m
|
||||||
BeeTemplateExt = ext
|
}
|
||||||
}
|
|
||||||
|
}
|
||||||
func BuildTemplate(dir string) error {
|
}
|
||||||
self := templatefile{
|
return nil
|
||||||
root: dir,
|
}
|
||||||
files: make(map[string][]string),
|
|
||||||
}
|
func AddTemplateExt(ext string) {
|
||||||
err := filepath.Walk(dir, func(path string, f os.FileInfo, err error) error {
|
for _, v := range BeeTemplateExt {
|
||||||
return self.visit(path, f, err)
|
if v == ext {
|
||||||
})
|
return
|
||||||
if err != nil {
|
}
|
||||||
fmt.Printf("filepath.Walk() returned %v\n", err)
|
}
|
||||||
return err
|
BeeTemplateExt = append(BeeTemplateExt, ext)
|
||||||
}
|
}
|
||||||
for k, v := range self.files {
|
|
||||||
BeeTemplates[k] = template.Must(template.New("beegoTemplate" + k).Funcs(beegoTplFuncMap).ParseFiles(v...))
|
func BuildTemplate(dir string) error {
|
||||||
}
|
self := templatefile{
|
||||||
return nil
|
root: dir,
|
||||||
}
|
files: make(map[string][]string),
|
||||||
|
}
|
||||||
|
err := filepath.Walk(dir, func(path string, f os.FileInfo, err error) error {
|
||||||
|
return self.visit(path, f, err)
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("filepath.Walk() returned %v\n", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
for k, v := range self.files {
|
||||||
|
BeeTemplates[k] = template.Must(template.New("beegoTemplate" + k).Funcs(beegoTplFuncMap).ParseFiles(v...))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user