Subversion Repositories openrisc

// Copyright 2009 The Go Authors. All rights reserved.

// Use of this source code is governed by a BSD-style

// license that can be found in the LICENSE file.

// Template library: default formatters

package template

import (

        "bytes"

        "fmt"

        "io"

)

// StringFormatter formats into the default string representation.

// It is stored under the name "str" and is the default formatter.

// You can override the default formatter by storing your default

// under the name "" in your custom formatter map.

func StringFormatter(w io.Writer, format string, value ...interface{}) {

        if len(value) == 1 {

                if b, ok := value[0].([]byte); ok {

                        w.Write(b)

                        return

                }

        }

        fmt.Fprint(w, value...)

}

var (

        esc_quot = []byte(""") // shorter than """

        esc_apos = []byte("'") // shorter than "'"

        esc_amp  = []byte("&")

        esc_lt   = []byte("<")

        esc_gt   = []byte(">")

)

// HTMLEscape writes to w the properly escaped HTML equivalent

// of the plain text data s.

func HTMLEscape(w io.Writer, s []byte) {

        var esc []byte

        last := 0

        for i, c := range s {

                switch c {

                case '"':

                        esc = esc_quot

                case '\'':

                        esc = esc_apos

                case '&':

                        esc = esc_amp

                case '<':

                        esc = esc_lt

                case '>':

                        esc = esc_gt

                default:

                        continue

                }

                w.Write(s[last:i])

                w.Write(esc)

                last = i + 1

        }

        w.Write(s[last:])

}

// HTMLFormatter formats arbitrary values for HTML

func HTMLFormatter(w io.Writer, format string, value ...interface{}) {

        ok := false

        var b []byte

        if len(value) == 1 {

                b, ok = value[0].([]byte)

        }

        if !ok {

                var buf bytes.Buffer

                fmt.Fprint(&buf, value...)

                b = buf.Bytes()

        }

        HTMLEscape(w, b)

}