Hein
174 lines
3.3 KiB
Go
174 lines
3.3 KiB
Go
// Copyright (C) 2016 Kohei YOSHIDA. All rights reserved.
|
|
//
|
|
// This program is free software; you can redistribute it and/or
|
|
// modify it under the terms of The BSD 3-Clause License
|
|
// that can be found in the LICENSE file.
|
|
|
|
package uritemplate
|
|
|
|
import (
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type template interface {
|
|
expand(*strings.Builder, Values) error
|
|
regexp(*strings.Builder)
|
|
}
|
|
|
|
type literals string
|
|
|
|
func (l literals) expand(b *strings.Builder, _ Values) error {
|
|
b.WriteString(string(l))
|
|
return nil
|
|
}
|
|
|
|
func (l literals) regexp(b *strings.Builder) {
|
|
b.WriteString("(?:")
|
|
b.WriteString(regexp.QuoteMeta(string(l)))
|
|
b.WriteByte(')')
|
|
}
|
|
|
|
type varspec struct {
|
|
name string
|
|
maxlen int
|
|
explode bool
|
|
}
|
|
|
|
type expression struct {
|
|
vars []varspec
|
|
op parseOp
|
|
first string
|
|
sep string
|
|
named bool
|
|
ifemp string
|
|
escape escapeFunc
|
|
allow runeClass
|
|
}
|
|
|
|
func (e *expression) init() {
|
|
switch e.op {
|
|
case parseOpSimple:
|
|
e.sep = ","
|
|
e.escape = escapeExceptU
|
|
e.allow = runeClassU
|
|
case parseOpPlus:
|
|
e.sep = ","
|
|
e.escape = escapeExceptUR
|
|
e.allow = runeClassUR
|
|
case parseOpCrosshatch:
|
|
e.first = "#"
|
|
e.sep = ","
|
|
e.escape = escapeExceptUR
|
|
e.allow = runeClassUR
|
|
case parseOpDot:
|
|
e.first = "."
|
|
e.sep = "."
|
|
e.escape = escapeExceptU
|
|
e.allow = runeClassU
|
|
case parseOpSlash:
|
|
e.first = "/"
|
|
e.sep = "/"
|
|
e.escape = escapeExceptU
|
|
e.allow = runeClassU
|
|
case parseOpSemicolon:
|
|
e.first = ";"
|
|
e.sep = ";"
|
|
e.named = true
|
|
e.escape = escapeExceptU
|
|
e.allow = runeClassU
|
|
case parseOpQuestion:
|
|
e.first = "?"
|
|
e.sep = "&"
|
|
e.named = true
|
|
e.ifemp = "="
|
|
e.escape = escapeExceptU
|
|
e.allow = runeClassU
|
|
case parseOpAmpersand:
|
|
e.first = "&"
|
|
e.sep = "&"
|
|
e.named = true
|
|
e.ifemp = "="
|
|
e.escape = escapeExceptU
|
|
e.allow = runeClassU
|
|
}
|
|
}
|
|
|
|
func (e *expression) expand(w *strings.Builder, values Values) error {
|
|
first := true
|
|
for _, varspec := range e.vars {
|
|
value := values.Get(varspec.name)
|
|
if !value.Valid() {
|
|
continue
|
|
}
|
|
|
|
if first {
|
|
w.WriteString(e.first)
|
|
first = false
|
|
} else {
|
|
w.WriteString(e.sep)
|
|
}
|
|
|
|
if err := value.expand(w, varspec, e); err != nil {
|
|
return err
|
|
}
|
|
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (e *expression) regexp(b *strings.Builder) {
|
|
if e.first != "" {
|
|
b.WriteString("(?:") // $1
|
|
b.WriteString(regexp.QuoteMeta(e.first))
|
|
}
|
|
b.WriteByte('(') // $2
|
|
runeClassToRegexp(b, e.allow, e.named || e.vars[0].explode)
|
|
if len(e.vars) > 1 || e.vars[0].explode {
|
|
max := len(e.vars) - 1
|
|
for i := 0; i < len(e.vars); i++ {
|
|
if e.vars[i].explode {
|
|
max = -1
|
|
break
|
|
}
|
|
}
|
|
|
|
b.WriteString("(?:") // $3
|
|
b.WriteString(regexp.QuoteMeta(e.sep))
|
|
runeClassToRegexp(b, e.allow, e.named || max < 0)
|
|
b.WriteByte(')') // $3
|
|
if max > 0 {
|
|
b.WriteString("{0,")
|
|
b.WriteString(strconv.Itoa(max))
|
|
b.WriteByte('}')
|
|
} else {
|
|
b.WriteByte('*')
|
|
}
|
|
}
|
|
b.WriteByte(')') // $2
|
|
if e.first != "" {
|
|
b.WriteByte(')') // $1
|
|
}
|
|
b.WriteByte('?')
|
|
}
|
|
|
|
func runeClassToRegexp(b *strings.Builder, class runeClass, named bool) {
|
|
b.WriteString("(?:(?:[")
|
|
if class&runeClassR == 0 {
|
|
b.WriteString(`\x2c`)
|
|
if named {
|
|
b.WriteString(`\x3d`)
|
|
}
|
|
}
|
|
if class&runeClassU == runeClassU {
|
|
b.WriteString(reUnreserved)
|
|
}
|
|
if class&runeClassR == runeClassR {
|
|
b.WriteString(reReserved)
|
|
}
|
|
b.WriteString("]")
|
|
b.WriteString("|%[[:xdigit:]][[:xdigit:]]")
|
|
b.WriteString(")*)")
|
|
}
|