feat: betterified the contents.Type and context way to set content type.
This commit is contained in:
parent
966c1204d7
commit
8c5726515e
2 changed files with 40 additions and 14 deletions
|
@ -1,25 +1,29 @@
|
||||||
package contents
|
package contents
|
||||||
|
|
||||||
import (
|
type Option interface {
|
||||||
)
|
KeyValue() [2]string
|
||||||
|
}
|
||||||
|
|
||||||
type CharSet string
|
type CharSet string
|
||||||
const (
|
const (
|
||||||
Utf8 = "utf-8"
|
Utf8 = "utf-8"
|
||||||
)
|
)
|
||||||
|
func (cs CharSet) KeyValue() [2]string {
|
||||||
|
return [2]string{
|
||||||
|
"charset",
|
||||||
|
string(cs),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
type Type string
|
type Type string
|
||||||
const (
|
const (
|
||||||
// Using the UTF-8 by default.
|
// Using the UTF-8 by default.
|
||||||
Unknown Type = "application/octet-stream"
|
Unknown Type = "application/octet-stream"
|
||||||
Binary
|
Binary
|
||||||
Plain Type = "text/plain"
|
Plain= "text/plain"
|
||||||
Css Type = "text/css"
|
Css= "text/css"
|
||||||
Html Type = "text/html"
|
Html= "text/html"
|
||||||
Json Type = "application/json"
|
Json= "application/json"
|
||||||
UrlEncoded = "application/x-www-form-urlencoded"
|
UrlEncoded= "application/x-www-form-urlencoded"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (t Type) CharSet(set CharSet) Type {
|
|
||||||
return t + ";" + Type(set)
|
|
||||||
}
|
|
||||||
|
|
30
context.go
30
context.go
|
@ -2,6 +2,7 @@ package bond
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io"
|
"io"
|
||||||
|
"strings"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
@ -31,12 +32,17 @@ func (c *Context) SetStatus(status Status) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set the reply content type.
|
// Set the reply content type.
|
||||||
func (c *Context) SetContentType(typ contents.Type) {
|
func (c *Context) SetContentType(typ contents.Type, opts ...contents.Option) {
|
||||||
c.SetHeader("Content-Type", string(typ))
|
ret := string(typ)
|
||||||
|
for _, opt := range opts {
|
||||||
|
kv := opt.KeyValue()
|
||||||
|
ret += ";"+kv[0]+"="+kv[1]
|
||||||
|
}
|
||||||
|
c.SetHeader("Content-Type", ret)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get the request content type.
|
// Get the request content type.
|
||||||
func (c *Context) ContentType() ContentType {
|
func (c *Context) ContentType() contents.Type {
|
||||||
ret, ok := c.Header("Content-Type")
|
ret, ok := c.Header("Content-Type")
|
||||||
if !ok {
|
if !ok {
|
||||||
return ""
|
return ""
|
||||||
|
@ -44,7 +50,11 @@ func (c *Context) ContentType() ContentType {
|
||||||
if len(ret) < 1 {
|
if len(ret) < 1 {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
return contents.Type(ret[0])
|
splits := strings.SplitN(ret[0], ";", 2)
|
||||||
|
if len(splits) < 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return contents.Type(splits[0])
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Context) SetHeader(k, v string) {
|
func (c *Context) SetHeader(k, v string) {
|
||||||
|
@ -114,6 +124,18 @@ func (c *Context) NotFound() {
|
||||||
http.NotFound(c.W, c.R)
|
http.NotFound(c.W, c.R)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Context) InternalServerError(err error) {
|
||||||
|
c.W.WriteHeader(http.StatusInternalServerError)
|
||||||
|
if err != nil {
|
||||||
|
c.Printf("500 Internal Server Error: %q", err)
|
||||||
|
} else {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Context) Print(v ...any) (int, error) {
|
||||||
|
return fmt.Print(v...)
|
||||||
|
}
|
||||||
|
|
||||||
func (c *Context) Printf(format string, v ...any) (int, error) {
|
func (c *Context) Printf(format string, v ...any) (int, error) {
|
||||||
return fmt.Fprintf(c.W, format, v...)
|
return fmt.Fprintf(c.W, format, v...)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue