mirror of
https://github.com/cwinfo/matterbridge.git
synced 2024-11-25 21:21:35 +00:00
Update vendor
This commit is contained in:
parent
58483ea70c
commit
930b639cc9
22
vendor/github.com/GeertJohan/go.rice/LICENSE
generated
vendored
Normal file
22
vendor/github.com/GeertJohan/go.rice/LICENSE
generated
vendored
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
Copyright (c) 2013, Geert-Johan Riemer
|
||||||
|
All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without
|
||||||
|
modification, are permitted provided that the following conditions are met:
|
||||||
|
|
||||||
|
1. Redistributions of source code must retain the above copyright notice, this
|
||||||
|
list of conditions and the following disclaimer.
|
||||||
|
2. Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
this list of conditions and the following disclaimer in the documentation
|
||||||
|
and/or other materials provided with the distribution.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
||||||
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||||
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||||
|
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
138
vendor/github.com/GeertJohan/go.rice/appended.go
generated
vendored
Normal file
138
vendor/github.com/GeertJohan/go.rice/appended.go
generated
vendored
Normal file
@ -0,0 +1,138 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"archive/zip"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/daaku/go.zipexe"
|
||||||
|
"github.com/kardianos/osext"
|
||||||
|
)
|
||||||
|
|
||||||
|
// appendedBox defines an appended box
|
||||||
|
type appendedBox struct {
|
||||||
|
Name string // box name
|
||||||
|
Files map[string]*appendedFile // appended files (*zip.File) by full path
|
||||||
|
}
|
||||||
|
|
||||||
|
type appendedFile struct {
|
||||||
|
zipFile *zip.File
|
||||||
|
dir bool
|
||||||
|
dirInfo *appendedDirInfo
|
||||||
|
children []*appendedFile
|
||||||
|
content []byte
|
||||||
|
}
|
||||||
|
|
||||||
|
// appendedBoxes is a public register of appendes boxes
|
||||||
|
var appendedBoxes = make(map[string]*appendedBox)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// find if exec is appended
|
||||||
|
thisFile, err := osext.Executable()
|
||||||
|
if err != nil {
|
||||||
|
return // not appended or cant find self executable
|
||||||
|
}
|
||||||
|
closer, rd, err := zipexe.OpenCloser(thisFile)
|
||||||
|
if err != nil {
|
||||||
|
return // not appended
|
||||||
|
}
|
||||||
|
defer closer.Close()
|
||||||
|
|
||||||
|
for _, f := range rd.File {
|
||||||
|
// get box and file name from f.Name
|
||||||
|
fileParts := strings.SplitN(strings.TrimLeft(filepath.ToSlash(f.Name), "/"), "/", 2)
|
||||||
|
boxName := fileParts[0]
|
||||||
|
var fileName string
|
||||||
|
if len(fileParts) > 1 {
|
||||||
|
fileName = fileParts[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
// find box or create new one if doesn't exist
|
||||||
|
box := appendedBoxes[boxName]
|
||||||
|
if box == nil {
|
||||||
|
box = &appendedBox{
|
||||||
|
Name: boxName,
|
||||||
|
Files: make(map[string]*appendedFile),
|
||||||
|
}
|
||||||
|
appendedBoxes[boxName] = box
|
||||||
|
}
|
||||||
|
|
||||||
|
// create and add file to box
|
||||||
|
af := &appendedFile{
|
||||||
|
zipFile: f,
|
||||||
|
}
|
||||||
|
if f.Comment == "dir" {
|
||||||
|
af.dir = true
|
||||||
|
af.dirInfo = &appendedDirInfo{
|
||||||
|
name: filepath.Base(af.zipFile.Name),
|
||||||
|
//++ TODO: use zip modtime when that is set correctly: af.zipFile.ModTime()
|
||||||
|
time: time.Now(),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// this is a file, we need it's contents so we can create a bytes.Reader when the file is opened
|
||||||
|
// make a new byteslice
|
||||||
|
af.content = make([]byte, af.zipFile.FileInfo().Size())
|
||||||
|
// ignore reading empty files from zip (empty file still is a valid file to be read though!)
|
||||||
|
if len(af.content) > 0 {
|
||||||
|
// open io.ReadCloser
|
||||||
|
rc, err := af.zipFile.Open()
|
||||||
|
if err != nil {
|
||||||
|
af.content = nil // this will cause an error when the file is being opened or seeked (which is good)
|
||||||
|
// TODO: it's quite blunt to just log this stuff. but this is in init, so rice.Debug can't be changed yet..
|
||||||
|
log.Printf("error opening appended file %s: %v", af.zipFile.Name, err)
|
||||||
|
} else {
|
||||||
|
_, err = rc.Read(af.content)
|
||||||
|
rc.Close()
|
||||||
|
if err != nil {
|
||||||
|
af.content = nil // this will cause an error when the file is being opened or seeked (which is good)
|
||||||
|
// TODO: it's quite blunt to just log this stuff. but this is in init, so rice.Debug can't be changed yet..
|
||||||
|
log.Printf("error reading data for appended file %s: %v", af.zipFile.Name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// add appendedFile to box file list
|
||||||
|
box.Files[fileName] = af
|
||||||
|
|
||||||
|
// add to parent dir (if any)
|
||||||
|
dirName := filepath.Dir(fileName)
|
||||||
|
if dirName == "." {
|
||||||
|
dirName = ""
|
||||||
|
}
|
||||||
|
if fileName != "" { // don't make box root dir a child of itself
|
||||||
|
if dir := box.Files[dirName]; dir != nil {
|
||||||
|
dir.children = append(dir.children, af)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// implements os.FileInfo.
|
||||||
|
// used for Readdir()
|
||||||
|
type appendedDirInfo struct {
|
||||||
|
name string
|
||||||
|
time time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
func (adi *appendedDirInfo) Name() string {
|
||||||
|
return adi.name
|
||||||
|
}
|
||||||
|
func (adi *appendedDirInfo) Size() int64 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
func (adi *appendedDirInfo) Mode() os.FileMode {
|
||||||
|
return os.ModeDir
|
||||||
|
}
|
||||||
|
func (adi *appendedDirInfo) ModTime() time.Time {
|
||||||
|
return adi.time
|
||||||
|
}
|
||||||
|
func (adi *appendedDirInfo) IsDir() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
func (adi *appendedDirInfo) Sys() interface{} {
|
||||||
|
return nil
|
||||||
|
}
|
337
vendor/github.com/GeertJohan/go.rice/box.go
generated
vendored
Normal file
337
vendor/github.com/GeertJohan/go.rice/box.go
generated
vendored
Normal file
@ -0,0 +1,337 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/GeertJohan/go.rice/embedded"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Box abstracts a directory for resources/files.
|
||||||
|
// It can either load files from disk, or from embedded code (when `rice --embed` was ran).
|
||||||
|
type Box struct {
|
||||||
|
name string
|
||||||
|
absolutePath string
|
||||||
|
embed *embedded.EmbeddedBox
|
||||||
|
appendd *appendedBox
|
||||||
|
}
|
||||||
|
|
||||||
|
var defaultLocateOrder = []LocateMethod{LocateEmbedded, LocateAppended, LocateFS}
|
||||||
|
|
||||||
|
func findBox(name string, order []LocateMethod) (*Box, error) {
|
||||||
|
b := &Box{name: name}
|
||||||
|
|
||||||
|
// no support for absolute paths since gopath can be different on different machines.
|
||||||
|
// therefore, required box must be located relative to package requiring it.
|
||||||
|
if filepath.IsAbs(name) {
|
||||||
|
return nil, errors.New("given name/path is absolute")
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
for _, method := range order {
|
||||||
|
switch method {
|
||||||
|
case LocateEmbedded:
|
||||||
|
if embed := embedded.EmbeddedBoxes[name]; embed != nil {
|
||||||
|
b.embed = embed
|
||||||
|
return b, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
case LocateAppended:
|
||||||
|
appendedBoxName := strings.Replace(name, `/`, `-`, -1)
|
||||||
|
if appendd := appendedBoxes[appendedBoxName]; appendd != nil {
|
||||||
|
b.appendd = appendd
|
||||||
|
return b, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
case LocateFS:
|
||||||
|
// resolve absolute directory path
|
||||||
|
err := b.resolveAbsolutePathFromCaller()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// check if absolutePath exists on filesystem
|
||||||
|
info, err := os.Stat(b.absolutePath)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// check if absolutePath is actually a directory
|
||||||
|
if !info.IsDir() {
|
||||||
|
err = errors.New("given name/path is not a directory")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return b, nil
|
||||||
|
case LocateWorkingDirectory:
|
||||||
|
// resolve absolute directory path
|
||||||
|
err := b.resolveAbsolutePathFromWorkingDirectory()
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// check if absolutePath exists on filesystem
|
||||||
|
info, err := os.Stat(b.absolutePath)
|
||||||
|
if err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// check if absolutePath is actually a directory
|
||||||
|
if !info.IsDir() {
|
||||||
|
err = errors.New("given name/path is not a directory")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
return b, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
err = fmt.Errorf("could not locate box %q", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindBox returns a Box instance for given name.
|
||||||
|
// When the given name is a relative path, it's base path will be the calling pkg/cmd's source root.
|
||||||
|
// When the given name is absolute, it's absolute. derp.
|
||||||
|
// Make sure the path doesn't contain any sensitive information as it might be placed into generated go source (embedded).
|
||||||
|
func FindBox(name string) (*Box, error) {
|
||||||
|
return findBox(name, defaultLocateOrder)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustFindBox returns a Box instance for given name, like FindBox does.
|
||||||
|
// It does not return an error, instead it panics when an error occurs.
|
||||||
|
func MustFindBox(name string) *Box {
|
||||||
|
box, err := findBox(name, defaultLocateOrder)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return box
|
||||||
|
}
|
||||||
|
|
||||||
|
// This is injected as a mutable function literal so that we can mock it out in
|
||||||
|
// tests and return a fixed test file.
|
||||||
|
var resolveAbsolutePathFromCaller = func(name string, nStackFrames int) (string, error) {
|
||||||
|
_, callingGoFile, _, ok := runtime.Caller(nStackFrames)
|
||||||
|
if !ok {
|
||||||
|
return "", errors.New("couldn't find caller on stack")
|
||||||
|
}
|
||||||
|
|
||||||
|
// resolve to proper path
|
||||||
|
pkgDir := filepath.Dir(callingGoFile)
|
||||||
|
// fix for go cover
|
||||||
|
const coverPath = "_test/_obj_test"
|
||||||
|
if !filepath.IsAbs(pkgDir) {
|
||||||
|
if i := strings.Index(pkgDir, coverPath); i >= 0 {
|
||||||
|
pkgDir = pkgDir[:i] + pkgDir[i+len(coverPath):] // remove coverPath
|
||||||
|
pkgDir = filepath.Join(os.Getenv("GOPATH"), "src", pkgDir) // make absolute
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return filepath.Join(pkgDir, name), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Box) resolveAbsolutePathFromCaller() error {
|
||||||
|
path, err := resolveAbsolutePathFromCaller(b.name, 4)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b.absolutePath = path
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Box) resolveAbsolutePathFromWorkingDirectory() error {
|
||||||
|
path, err := os.Getwd()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b.absolutePath = filepath.Join(path, b.name)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsEmbedded indicates wether this box was embedded into the application
|
||||||
|
func (b *Box) IsEmbedded() bool {
|
||||||
|
return b.embed != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsAppended indicates wether this box was appended to the application
|
||||||
|
func (b *Box) IsAppended() bool {
|
||||||
|
return b.appendd != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Time returns how actual the box is.
|
||||||
|
// When the box is embedded, it's value is saved in the embedding code.
|
||||||
|
// When the box is live, this methods returns time.Now()
|
||||||
|
func (b *Box) Time() time.Time {
|
||||||
|
if b.IsEmbedded() {
|
||||||
|
return b.embed.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
//++ TODO: return time for appended box
|
||||||
|
|
||||||
|
return time.Now()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open opens a File from the box
|
||||||
|
// If there is an error, it will be of type *os.PathError.
|
||||||
|
func (b *Box) Open(name string) (*File, error) {
|
||||||
|
if Debug {
|
||||||
|
fmt.Printf("Open(%s)\n", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
if b.IsEmbedded() {
|
||||||
|
if Debug {
|
||||||
|
fmt.Println("Box is embedded")
|
||||||
|
}
|
||||||
|
|
||||||
|
// trim prefix (paths are relative to box)
|
||||||
|
name = strings.TrimLeft(name, "/")
|
||||||
|
if Debug {
|
||||||
|
fmt.Printf("Trying %s\n", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
// search for file
|
||||||
|
ef := b.embed.Files[name]
|
||||||
|
if ef == nil {
|
||||||
|
if Debug {
|
||||||
|
fmt.Println("Didn't find file in embed")
|
||||||
|
}
|
||||||
|
// file not found, try dir
|
||||||
|
ed := b.embed.Dirs[name]
|
||||||
|
if ed == nil {
|
||||||
|
if Debug {
|
||||||
|
fmt.Println("Didn't find dir in embed")
|
||||||
|
}
|
||||||
|
// dir not found, error out
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "open",
|
||||||
|
Path: name,
|
||||||
|
Err: os.ErrNotExist,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if Debug {
|
||||||
|
fmt.Println("Found dir. Returning virtual dir")
|
||||||
|
}
|
||||||
|
vd := newVirtualDir(ed)
|
||||||
|
return &File{virtualD: vd}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// box is embedded
|
||||||
|
if Debug {
|
||||||
|
fmt.Println("Found file. Returning virtual file")
|
||||||
|
}
|
||||||
|
vf := newVirtualFile(ef)
|
||||||
|
return &File{virtualF: vf}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if b.IsAppended() {
|
||||||
|
// trim prefix (paths are relative to box)
|
||||||
|
name = strings.TrimLeft(name, "/")
|
||||||
|
|
||||||
|
// search for file
|
||||||
|
appendedFile := b.appendd.Files[name]
|
||||||
|
if appendedFile == nil {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "open",
|
||||||
|
Path: name,
|
||||||
|
Err: os.ErrNotExist,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// create new file
|
||||||
|
f := &File{
|
||||||
|
appendedF: appendedFile,
|
||||||
|
}
|
||||||
|
|
||||||
|
// if this file is a directory, we want to be able to read and seek
|
||||||
|
if !appendedFile.dir {
|
||||||
|
// looks like malformed data in zip, error now
|
||||||
|
if appendedFile.content == nil {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "open",
|
||||||
|
Path: "name",
|
||||||
|
Err: errors.New("error reading data from zip file"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// create new bytes.Reader
|
||||||
|
f.appendedFileReader = bytes.NewReader(appendedFile.content)
|
||||||
|
}
|
||||||
|
|
||||||
|
// all done
|
||||||
|
return f, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// perform os open
|
||||||
|
if Debug {
|
||||||
|
fmt.Printf("Using os.Open(%s)", filepath.Join(b.absolutePath, name))
|
||||||
|
}
|
||||||
|
file, err := os.Open(filepath.Join(b.absolutePath, name))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &File{realF: file}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bytes returns the content of the file with given name as []byte.
|
||||||
|
func (b *Box) Bytes(name string) ([]byte, error) {
|
||||||
|
file, err := b.Open(name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
content, err := ioutil.ReadAll(file)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return content, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustBytes returns the content of the file with given name as []byte.
|
||||||
|
// panic's on error.
|
||||||
|
func (b *Box) MustBytes(name string) []byte {
|
||||||
|
bts, err := b.Bytes(name)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return bts
|
||||||
|
}
|
||||||
|
|
||||||
|
// String returns the content of the file with given name as string.
|
||||||
|
func (b *Box) String(name string) (string, error) {
|
||||||
|
// check if box is embedded, optimized fast path
|
||||||
|
if b.IsEmbedded() {
|
||||||
|
// find file in embed
|
||||||
|
ef := b.embed.Files[name]
|
||||||
|
if ef == nil {
|
||||||
|
return "", os.ErrNotExist
|
||||||
|
}
|
||||||
|
// return as string
|
||||||
|
return ef.Content, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
bts, err := b.Bytes(name)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return string(bts), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustString returns the content of the file with given name as string.
|
||||||
|
// panic's on error.
|
||||||
|
func (b *Box) MustString(name string) string {
|
||||||
|
str, err := b.String(name)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return str
|
||||||
|
}
|
||||||
|
|
||||||
|
// Name returns the name of the box
|
||||||
|
func (b *Box) Name() string {
|
||||||
|
return b.name
|
||||||
|
}
|
39
vendor/github.com/GeertJohan/go.rice/config.go
generated
vendored
Normal file
39
vendor/github.com/GeertJohan/go.rice/config.go
generated
vendored
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
// LocateMethod defines how a box is located.
|
||||||
|
type LocateMethod int
|
||||||
|
|
||||||
|
const (
|
||||||
|
LocateFS = LocateMethod(iota) // Locate on the filesystem according to package path.
|
||||||
|
LocateAppended // Locate boxes appended to the executable.
|
||||||
|
LocateEmbedded // Locate embedded boxes.
|
||||||
|
LocateWorkingDirectory // Locate on the binary working directory
|
||||||
|
)
|
||||||
|
|
||||||
|
// Config allows customizing the box lookup behavior.
|
||||||
|
type Config struct {
|
||||||
|
// LocateOrder defines the priority order that boxes are searched for. By
|
||||||
|
// default, the package global FindBox searches for embedded boxes first,
|
||||||
|
// then appended boxes, and then finally boxes on the filesystem. That
|
||||||
|
// search order may be customized by provided the ordered list here. Leaving
|
||||||
|
// out a particular method will omit that from the search space. For
|
||||||
|
// example, []LocateMethod{LocateEmbedded, LocateAppended} will never search
|
||||||
|
// the filesystem for boxes.
|
||||||
|
LocateOrder []LocateMethod
|
||||||
|
}
|
||||||
|
|
||||||
|
// FindBox searches for boxes using the LocateOrder of the config.
|
||||||
|
func (c *Config) FindBox(boxName string) (*Box, error) {
|
||||||
|
return findBox(boxName, c.LocateOrder)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MustFindBox searches for boxes using the LocateOrder of the config, like
|
||||||
|
// FindBox does. It does not return an error, instead it panics when an error
|
||||||
|
// occurs.
|
||||||
|
func (c *Config) MustFindBox(boxName string) *Box {
|
||||||
|
box, err := findBox(boxName, c.LocateOrder)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return box
|
||||||
|
}
|
4
vendor/github.com/GeertJohan/go.rice/debug.go
generated
vendored
Normal file
4
vendor/github.com/GeertJohan/go.rice/debug.go
generated
vendored
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
// Debug can be set to true to enable debugging.
|
||||||
|
var Debug = false
|
90
vendor/github.com/GeertJohan/go.rice/embedded.go
generated
vendored
Normal file
90
vendor/github.com/GeertJohan/go.rice/embedded.go
generated
vendored
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/GeertJohan/go.rice/embedded"
|
||||||
|
)
|
||||||
|
|
||||||
|
// re-type to make exported methods invisible to user (godoc)
|
||||||
|
// they're not required for the user
|
||||||
|
// embeddedDirInfo implements os.FileInfo
|
||||||
|
type embeddedDirInfo embedded.EmbeddedDir
|
||||||
|
|
||||||
|
// Name returns the base name of the directory
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ed *embeddedDirInfo) Name() string {
|
||||||
|
return ed.Filename
|
||||||
|
}
|
||||||
|
|
||||||
|
// Size always returns 0
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ed *embeddedDirInfo) Size() int64 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mode returns the file mode bits
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ed *embeddedDirInfo) Mode() os.FileMode {
|
||||||
|
return os.FileMode(0555 | os.ModeDir) // dr-xr-xr-x
|
||||||
|
}
|
||||||
|
|
||||||
|
// ModTime returns the modification time
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ed *embeddedDirInfo) ModTime() time.Time {
|
||||||
|
return ed.DirModTime
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsDir returns the abbreviation for Mode().IsDir() (always true)
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ed *embeddedDirInfo) IsDir() bool {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sys returns the underlying data source (always nil)
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ed *embeddedDirInfo) Sys() interface{} {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// re-type to make exported methods invisible to user (godoc)
|
||||||
|
// they're not required for the user
|
||||||
|
// embeddedFileInfo implements os.FileInfo
|
||||||
|
type embeddedFileInfo embedded.EmbeddedFile
|
||||||
|
|
||||||
|
// Name returns the base name of the file
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ef *embeddedFileInfo) Name() string {
|
||||||
|
return ef.Filename
|
||||||
|
}
|
||||||
|
|
||||||
|
// Size returns the length in bytes for regular files; system-dependent for others
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ef *embeddedFileInfo) Size() int64 {
|
||||||
|
return int64(len(ef.Content))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mode returns the file mode bits
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ef *embeddedFileInfo) Mode() os.FileMode {
|
||||||
|
return os.FileMode(0555) // r-xr-xr-x
|
||||||
|
}
|
||||||
|
|
||||||
|
// ModTime returns the modification time
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ef *embeddedFileInfo) ModTime() time.Time {
|
||||||
|
return ef.FileModTime
|
||||||
|
}
|
||||||
|
|
||||||
|
// IsDir returns the abbreviation for Mode().IsDir() (always false)
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ef *embeddedFileInfo) IsDir() bool {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sys returns the underlying data source (always nil)
|
||||||
|
// (implementing os.FileInfo)
|
||||||
|
func (ef *embeddedFileInfo) Sys() interface{} {
|
||||||
|
return nil
|
||||||
|
}
|
80
vendor/github.com/GeertJohan/go.rice/embedded/embedded.go
generated
vendored
Normal file
80
vendor/github.com/GeertJohan/go.rice/embedded/embedded.go
generated
vendored
Normal file
@ -0,0 +1,80 @@
|
|||||||
|
// Package embedded defines embedded data types that are shared between the go.rice package and generated code.
|
||||||
|
package embedded
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
EmbedTypeGo = 0
|
||||||
|
EmbedTypeSyso = 1
|
||||||
|
)
|
||||||
|
|
||||||
|
// EmbeddedBox defines an embedded box
|
||||||
|
type EmbeddedBox struct {
|
||||||
|
Name string // box name
|
||||||
|
Time time.Time // embed time
|
||||||
|
EmbedType int // kind of embedding
|
||||||
|
Files map[string]*EmbeddedFile // ALL embedded files by full path
|
||||||
|
Dirs map[string]*EmbeddedDir // ALL embedded dirs by full path
|
||||||
|
}
|
||||||
|
|
||||||
|
// Link creates the ChildDirs and ChildFiles links in all EmbeddedDir's
|
||||||
|
func (e *EmbeddedBox) Link() {
|
||||||
|
for path, ed := range e.Dirs {
|
||||||
|
fmt.Println(path)
|
||||||
|
ed.ChildDirs = make([]*EmbeddedDir, 0)
|
||||||
|
ed.ChildFiles = make([]*EmbeddedFile, 0)
|
||||||
|
}
|
||||||
|
for path, ed := range e.Dirs {
|
||||||
|
parentDirpath, _ := filepath.Split(path)
|
||||||
|
if strings.HasSuffix(parentDirpath, "/") {
|
||||||
|
parentDirpath = parentDirpath[:len(parentDirpath)-1]
|
||||||
|
}
|
||||||
|
parentDir := e.Dirs[parentDirpath]
|
||||||
|
if parentDir == nil {
|
||||||
|
panic("parentDir `" + parentDirpath + "` is missing in embedded box")
|
||||||
|
}
|
||||||
|
parentDir.ChildDirs = append(parentDir.ChildDirs, ed)
|
||||||
|
}
|
||||||
|
for path, ef := range e.Files {
|
||||||
|
dirpath, _ := filepath.Split(path)
|
||||||
|
if strings.HasSuffix(dirpath, "/") {
|
||||||
|
dirpath = dirpath[:len(dirpath)-1]
|
||||||
|
}
|
||||||
|
dir := e.Dirs[dirpath]
|
||||||
|
if dir == nil {
|
||||||
|
panic("dir `" + dirpath + "` is missing in embedded box")
|
||||||
|
}
|
||||||
|
dir.ChildFiles = append(dir.ChildFiles, ef)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// EmbeddedDir is instanced in the code generated by the rice tool and contains all necicary information about an embedded file
|
||||||
|
type EmbeddedDir struct {
|
||||||
|
Filename string
|
||||||
|
DirModTime time.Time
|
||||||
|
ChildDirs []*EmbeddedDir // direct childs, as returned by virtualDir.Readdir()
|
||||||
|
ChildFiles []*EmbeddedFile // direct childs, as returned by virtualDir.Readdir()
|
||||||
|
}
|
||||||
|
|
||||||
|
// EmbeddedFile is instanced in the code generated by the rice tool and contains all necicary information about an embedded file
|
||||||
|
type EmbeddedFile struct {
|
||||||
|
Filename string // filename
|
||||||
|
FileModTime time.Time
|
||||||
|
Content string
|
||||||
|
}
|
||||||
|
|
||||||
|
// EmbeddedBoxes is a public register of embedded boxes
|
||||||
|
var EmbeddedBoxes = make(map[string]*EmbeddedBox)
|
||||||
|
|
||||||
|
// RegisterEmbeddedBox registers an EmbeddedBox
|
||||||
|
func RegisterEmbeddedBox(name string, box *EmbeddedBox) {
|
||||||
|
if _, exists := EmbeddedBoxes[name]; exists {
|
||||||
|
panic(fmt.Sprintf("EmbeddedBox with name `%s` exists already", name))
|
||||||
|
}
|
||||||
|
EmbeddedBoxes[name] = box
|
||||||
|
}
|
69
vendor/github.com/GeertJohan/go.rice/example/example.go
generated
vendored
Normal file
69
vendor/github.com/GeertJohan/go.rice/example/example.go
generated
vendored
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/GeertJohan/go.rice"
|
||||||
|
"github.com/davecgh/go-spew/spew"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
conf := rice.Config{
|
||||||
|
LocateOrder: []rice.LocateMethod{rice.LocateEmbedded, rice.LocateAppended, rice.LocateFS},
|
||||||
|
}
|
||||||
|
box, err := conf.FindBox("example-files")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("error opening rice.Box: %s\n", err)
|
||||||
|
}
|
||||||
|
// spew.Dump(box)
|
||||||
|
|
||||||
|
contentString, err := box.String("file.txt")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("could not read file contents as string: %s\n", err)
|
||||||
|
}
|
||||||
|
log.Printf("Read some file contents as string:\n%s\n", contentString)
|
||||||
|
|
||||||
|
contentBytes, err := box.Bytes("file.txt")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("could not read file contents as byteSlice: %s\n", err)
|
||||||
|
}
|
||||||
|
log.Printf("Read some file contents as byteSlice:\n%s\n", hex.Dump(contentBytes))
|
||||||
|
|
||||||
|
file, err := box.Open("file.txt")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("could not open file: %s\n", err)
|
||||||
|
}
|
||||||
|
spew.Dump(file)
|
||||||
|
|
||||||
|
// find/create a rice.Box
|
||||||
|
templateBox, err := rice.FindBox("example-templates")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
// get file contents as string
|
||||||
|
templateString, err := templateBox.String("message.tmpl")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
// parse and execute the template
|
||||||
|
tmplMessage, err := template.New("message").Parse(templateString)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
tmplMessage.Execute(os.Stdout, map[string]string{"Message": "Hello, world!"})
|
||||||
|
|
||||||
|
http.Handle("/", http.FileServer(box.HTTPBox()))
|
||||||
|
go func() {
|
||||||
|
fmt.Println("Serving files on :8080, press ctrl-C to exit")
|
||||||
|
err := http.ListenAndServe(":8080", nil)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("error serving files: %v", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
select {}
|
||||||
|
}
|
144
vendor/github.com/GeertJohan/go.rice/file.go
generated
vendored
Normal file
144
vendor/github.com/GeertJohan/go.rice/file.go
generated
vendored
Normal file
@ -0,0 +1,144 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
// File implements the io.Reader, io.Seeker, io.Closer and http.File interfaces
|
||||||
|
type File struct {
|
||||||
|
// File abstracts file methods so the user doesn't see the difference between rice.virtualFile, rice.virtualDir and os.File
|
||||||
|
// TODO: maybe use internal File interface and four implementations: *os.File, appendedFile, virtualFile, virtualDir
|
||||||
|
|
||||||
|
// real file on disk
|
||||||
|
realF *os.File
|
||||||
|
|
||||||
|
// when embedded (go)
|
||||||
|
virtualF *virtualFile
|
||||||
|
virtualD *virtualDir
|
||||||
|
|
||||||
|
// when appended (zip)
|
||||||
|
appendedF *appendedFile
|
||||||
|
appendedFileReader *bytes.Reader
|
||||||
|
// TODO: is appendedFileReader subject of races? Might need a lock here..
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close is like (*os.File).Close()
|
||||||
|
// Visit http://golang.org/pkg/os/#File.Close for more information
|
||||||
|
func (f *File) Close() error {
|
||||||
|
if f.appendedF != nil {
|
||||||
|
if f.appendedFileReader == nil {
|
||||||
|
return errors.New("already closed")
|
||||||
|
}
|
||||||
|
f.appendedFileReader = nil
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if f.virtualF != nil {
|
||||||
|
return f.virtualF.close()
|
||||||
|
}
|
||||||
|
if f.virtualD != nil {
|
||||||
|
return f.virtualD.close()
|
||||||
|
}
|
||||||
|
return f.realF.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stat is like (*os.File).Stat()
|
||||||
|
// Visit http://golang.org/pkg/os/#File.Stat for more information
|
||||||
|
func (f *File) Stat() (os.FileInfo, error) {
|
||||||
|
if f.appendedF != nil {
|
||||||
|
if f.appendedF.dir {
|
||||||
|
return f.appendedF.dirInfo, nil
|
||||||
|
}
|
||||||
|
if f.appendedFileReader == nil {
|
||||||
|
return nil, errors.New("file is closed")
|
||||||
|
}
|
||||||
|
return f.appendedF.zipFile.FileInfo(), nil
|
||||||
|
}
|
||||||
|
if f.virtualF != nil {
|
||||||
|
return f.virtualF.stat()
|
||||||
|
}
|
||||||
|
if f.virtualD != nil {
|
||||||
|
return f.virtualD.stat()
|
||||||
|
}
|
||||||
|
return f.realF.Stat()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Readdir is like (*os.File).Readdir()
|
||||||
|
// Visit http://golang.org/pkg/os/#File.Readdir for more information
|
||||||
|
func (f *File) Readdir(count int) ([]os.FileInfo, error) {
|
||||||
|
if f.appendedF != nil {
|
||||||
|
if f.appendedF.dir {
|
||||||
|
fi := make([]os.FileInfo, 0, len(f.appendedF.children))
|
||||||
|
for _, childAppendedFile := range f.appendedF.children {
|
||||||
|
if childAppendedFile.dir {
|
||||||
|
fi = append(fi, childAppendedFile.dirInfo)
|
||||||
|
} else {
|
||||||
|
fi = append(fi, childAppendedFile.zipFile.FileInfo())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return fi, nil
|
||||||
|
}
|
||||||
|
//++ TODO: is os.ErrInvalid the correct error for Readdir on file?
|
||||||
|
return nil, os.ErrInvalid
|
||||||
|
}
|
||||||
|
if f.virtualF != nil {
|
||||||
|
return f.virtualF.readdir(count)
|
||||||
|
}
|
||||||
|
if f.virtualD != nil {
|
||||||
|
return f.virtualD.readdir(count)
|
||||||
|
}
|
||||||
|
return f.realF.Readdir(count)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read is like (*os.File).Read()
|
||||||
|
// Visit http://golang.org/pkg/os/#File.Read for more information
|
||||||
|
func (f *File) Read(bts []byte) (int, error) {
|
||||||
|
if f.appendedF != nil {
|
||||||
|
if f.appendedFileReader == nil {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "read",
|
||||||
|
Path: filepath.Base(f.appendedF.zipFile.Name),
|
||||||
|
Err: errors.New("file is closed"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if f.appendedF.dir {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "read",
|
||||||
|
Path: filepath.Base(f.appendedF.zipFile.Name),
|
||||||
|
Err: errors.New("is a directory"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return f.appendedFileReader.Read(bts)
|
||||||
|
}
|
||||||
|
if f.virtualF != nil {
|
||||||
|
return f.virtualF.read(bts)
|
||||||
|
}
|
||||||
|
if f.virtualD != nil {
|
||||||
|
return f.virtualD.read(bts)
|
||||||
|
}
|
||||||
|
return f.realF.Read(bts)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Seek is like (*os.File).Seek()
|
||||||
|
// Visit http://golang.org/pkg/os/#File.Seek for more information
|
||||||
|
func (f *File) Seek(offset int64, whence int) (int64, error) {
|
||||||
|
if f.appendedF != nil {
|
||||||
|
if f.appendedFileReader == nil {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "seek",
|
||||||
|
Path: filepath.Base(f.appendedF.zipFile.Name),
|
||||||
|
Err: errors.New("file is closed"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return f.appendedFileReader.Seek(offset, whence)
|
||||||
|
}
|
||||||
|
if f.virtualF != nil {
|
||||||
|
return f.virtualF.seek(offset, whence)
|
||||||
|
}
|
||||||
|
if f.virtualD != nil {
|
||||||
|
return f.virtualD.seek(offset, whence)
|
||||||
|
}
|
||||||
|
return f.realF.Seek(offset, whence)
|
||||||
|
}
|
21
vendor/github.com/GeertJohan/go.rice/http.go
generated
vendored
Normal file
21
vendor/github.com/GeertJohan/go.rice/http.go
generated
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
// HTTPBox implements http.FileSystem which allows the use of Box with a http.FileServer.
|
||||||
|
// e.g.: http.Handle("/", http.FileServer(rice.MustFindBox("http-files").HTTPBox()))
|
||||||
|
type HTTPBox struct {
|
||||||
|
*Box
|
||||||
|
}
|
||||||
|
|
||||||
|
// HTTPBox creates a new HTTPBox from an existing Box
|
||||||
|
func (b *Box) HTTPBox() *HTTPBox {
|
||||||
|
return &HTTPBox{b}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open returns a File using the http.File interface
|
||||||
|
func (hb *HTTPBox) Open(name string) (http.File, error) {
|
||||||
|
return hb.Box.Open(name)
|
||||||
|
}
|
172
vendor/github.com/GeertJohan/go.rice/rice/append.go
generated
vendored
Normal file
172
vendor/github.com/GeertJohan/go.rice/rice/append.go
generated
vendored
Normal file
@ -0,0 +1,172 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"archive/zip"
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/daaku/go.zipexe"
|
||||||
|
)
|
||||||
|
|
||||||
|
func operationAppend(pkgs []*build.Package) {
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
_, err := exec.LookPath("zip")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println("#### WARNING ! ####")
|
||||||
|
fmt.Println("`rice append` is known not to work under windows because the `zip` command is not available. Please let me know if you got this to work (and how).")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// MARKED FOR DELETION
|
||||||
|
// This is actually not required, the append command now has the option --exec required.
|
||||||
|
// // check if package is a command
|
||||||
|
// if !pkg.IsCommand() {
|
||||||
|
// fmt.Println("Error: can not append to non-main package. Please follow instructions at github.com/GeertJohan/go.rice")
|
||||||
|
// os.Exit(1)
|
||||||
|
// }
|
||||||
|
|
||||||
|
// create tmp zipfile
|
||||||
|
tmpZipfileName := filepath.Join(os.TempDir(), fmt.Sprintf("ricebox-%d-%s.zip", time.Now().Unix(), randomString(10)))
|
||||||
|
verbosef("Will create tmp zipfile: %s\n", tmpZipfileName)
|
||||||
|
tmpZipfile, err := os.Create(tmpZipfileName)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error creating tmp zipfile: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
defer func() {
|
||||||
|
tmpZipfile.Close()
|
||||||
|
os.Remove(tmpZipfileName)
|
||||||
|
}()
|
||||||
|
|
||||||
|
// find abs path for binary file
|
||||||
|
binfileName, err := filepath.Abs(flags.Append.Executable)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error finding absolute path for executable to append: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
verbosef("Will append to file: %s\n", binfileName)
|
||||||
|
|
||||||
|
// check that command doesn't already have zip appended
|
||||||
|
if rd, _ := zipexe.Open(binfileName); rd != nil {
|
||||||
|
fmt.Printf("Cannot append to already appended executable. Please remove %s and build a fresh one.\n", binfileName)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// open binfile
|
||||||
|
binfile, err := os.OpenFile(binfileName, os.O_WRONLY, os.ModeAppend)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error: unable to open executable file: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// create zip.Writer
|
||||||
|
zipWriter := zip.NewWriter(tmpZipfile)
|
||||||
|
|
||||||
|
for _, pkg := range pkgs {
|
||||||
|
// find boxes for this command
|
||||||
|
boxMap := findBoxes(pkg)
|
||||||
|
|
||||||
|
// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ?
|
||||||
|
if len(boxMap) == 0 {
|
||||||
|
fmt.Printf("no calls to rice.FindBox() or rice.MustFindBox() found in import path `%s`\n", pkg.ImportPath)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
verbosef("\n")
|
||||||
|
|
||||||
|
for boxname := range boxMap {
|
||||||
|
appendedBoxName := strings.Replace(boxname, `/`, `-`, -1)
|
||||||
|
|
||||||
|
// walk box path's and insert files
|
||||||
|
boxPath := filepath.Clean(filepath.Join(pkg.Dir, boxname))
|
||||||
|
filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error {
|
||||||
|
if info == nil {
|
||||||
|
fmt.Printf("Error: box \"%s\" not found on disk\n", path)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
// create zipFilename
|
||||||
|
zipFileName := filepath.Join(appendedBoxName, strings.TrimPrefix(path, boxPath))
|
||||||
|
// write directories as empty file with comment "dir"
|
||||||
|
if info.IsDir() {
|
||||||
|
_, err := zipWriter.CreateHeader(&zip.FileHeader{
|
||||||
|
Name: zipFileName,
|
||||||
|
Comment: "dir",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error creating dir in tmp zip: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// create zipFileWriter
|
||||||
|
zipFileHeader, err := zip.FileInfoHeader(info)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error creating zip FileHeader: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
zipFileHeader.Name = zipFileName
|
||||||
|
zipFileWriter, err := zipWriter.CreateHeader(zipFileHeader)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error creating file in tmp zip: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
srcFile, err := os.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error opening file to append: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
_, err = io.Copy(zipFileWriter, srcFile)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error copying file contents to zip: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
srcFile.Close()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
err = zipWriter.Close()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error closing tmp zipfile: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = tmpZipfile.Sync()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error syncing tmp zipfile: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
_, err = tmpZipfile.Seek(0, 0)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error seeking tmp zipfile: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
_, err = binfile.Seek(0, 2)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error seeking bin file: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = io.Copy(binfile, tmpZipfile)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error appending zipfile to executable: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
zipA := exec.Command("zip", "-A", binfileName)
|
||||||
|
err = zipA.Run()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("Error setting zip offset: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
33
vendor/github.com/GeertJohan/go.rice/rice/clean.go
generated
vendored
Normal file
33
vendor/github.com/GeertJohan/go.rice/rice/clean.go
generated
vendored
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func operationClean(pkg *build.Package) {
|
||||||
|
filepath.Walk(pkg.Dir, func(filename string, info os.FileInfo, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error walking pkg dir to clean files: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
if info.IsDir() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
verbosef("checking file '%s'\n", filename)
|
||||||
|
if filepath.Base(filename) == "rice-box.go" ||
|
||||||
|
strings.HasSuffix(filename, ".rice-box.go") ||
|
||||||
|
strings.HasSuffix(filename, ".rice-box.syso") {
|
||||||
|
err := os.Remove(filename)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error removing file (%s): %s\n", filename, err)
|
||||||
|
os.Exit(-1)
|
||||||
|
}
|
||||||
|
verbosef("removed file '%s'\n", filename)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
158
vendor/github.com/GeertJohan/go.rice/rice/embed-go.go
generated
vendored
Normal file
158
vendor/github.com/GeertJohan/go.rice/rice/embed-go.go
generated
vendored
Normal file
@ -0,0 +1,158 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"go/format"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const boxFilename = "rice-box.go"
|
||||||
|
|
||||||
|
func operationEmbedGo(pkg *build.Package) {
|
||||||
|
|
||||||
|
boxMap := findBoxes(pkg)
|
||||||
|
|
||||||
|
// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ?
|
||||||
|
if len(boxMap) == 0 {
|
||||||
|
fmt.Println("no calls to rice.FindBox() found")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
verbosef("\n")
|
||||||
|
var boxes []*boxDataType
|
||||||
|
|
||||||
|
for boxname := range boxMap {
|
||||||
|
// find path and filename for this box
|
||||||
|
boxPath := filepath.Join(pkg.Dir, boxname)
|
||||||
|
|
||||||
|
// Check to see if the path for the box is a symbolic link. If so, simply
|
||||||
|
// box what the symbolic link points to. Note: the filepath.Walk function
|
||||||
|
// will NOT follow any nested symbolic links. This only handles the case
|
||||||
|
// where the root of the box is a symbolic link.
|
||||||
|
symPath, serr := os.Readlink(boxPath)
|
||||||
|
if serr == nil {
|
||||||
|
boxPath = symPath
|
||||||
|
}
|
||||||
|
|
||||||
|
// verbose info
|
||||||
|
verbosef("embedding box '%s' to '%s'\n", boxname, boxFilename)
|
||||||
|
|
||||||
|
// read box metadata
|
||||||
|
boxInfo, ierr := os.Stat(boxPath)
|
||||||
|
if ierr != nil {
|
||||||
|
fmt.Printf("Error: unable to access box at %s\n", boxPath)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// create box datastructure (used by template)
|
||||||
|
box := &boxDataType{
|
||||||
|
BoxName: boxname,
|
||||||
|
UnixNow: boxInfo.ModTime().Unix(),
|
||||||
|
Files: make([]*fileDataType, 0),
|
||||||
|
Dirs: make(map[string]*dirDataType),
|
||||||
|
}
|
||||||
|
|
||||||
|
if !boxInfo.IsDir() {
|
||||||
|
fmt.Printf("Error: Box %s must point to a directory but points to %s instead\n",
|
||||||
|
boxname, boxPath)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// fill box datastructure with file data
|
||||||
|
filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error walking box: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
filename := strings.TrimPrefix(path, boxPath)
|
||||||
|
filename = strings.Replace(filename, "\\", "/", -1)
|
||||||
|
filename = strings.TrimPrefix(filename, "/")
|
||||||
|
if info.IsDir() {
|
||||||
|
dirData := &dirDataType{
|
||||||
|
Identifier: "dir" + nextIdentifier(),
|
||||||
|
FileName: filename,
|
||||||
|
ModTime: info.ModTime().Unix(),
|
||||||
|
ChildFiles: make([]*fileDataType, 0),
|
||||||
|
ChildDirs: make([]*dirDataType, 0),
|
||||||
|
}
|
||||||
|
verbosef("\tincludes dir: '%s'\n", dirData.FileName)
|
||||||
|
box.Dirs[dirData.FileName] = dirData
|
||||||
|
|
||||||
|
// add tree entry (skip for root, it'll create a recursion)
|
||||||
|
if dirData.FileName != "" {
|
||||||
|
pathParts := strings.Split(dirData.FileName, "/")
|
||||||
|
parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")]
|
||||||
|
parentDir.ChildDirs = append(parentDir.ChildDirs, dirData)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
fileData := &fileDataType{
|
||||||
|
Identifier: "file" + nextIdentifier(),
|
||||||
|
FileName: filename,
|
||||||
|
ModTime: info.ModTime().Unix(),
|
||||||
|
}
|
||||||
|
verbosef("\tincludes file: '%s'\n", fileData.FileName)
|
||||||
|
fileData.Content, err = ioutil.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error reading file content while walking box: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
box.Files = append(box.Files, fileData)
|
||||||
|
|
||||||
|
// add tree entry
|
||||||
|
pathParts := strings.Split(fileData.FileName, "/")
|
||||||
|
parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")]
|
||||||
|
if parentDir == nil {
|
||||||
|
fmt.Printf("Error: parent of %s is not within the box\n", path)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
parentDir.ChildFiles = append(parentDir.ChildFiles, fileData)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
boxes = append(boxes, box)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
embedSourceUnformated := bytes.NewBuffer(make([]byte, 0))
|
||||||
|
|
||||||
|
// execute template to buffer
|
||||||
|
err := tmplEmbeddedBox.Execute(
|
||||||
|
embedSourceUnformated,
|
||||||
|
embedFileDataType{pkg.Name, boxes},
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error writing embedded box to file (template execute): %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// format the source code
|
||||||
|
embedSource, err := format.Source(embedSourceUnformated.Bytes())
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error formatting embedSource: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// create go file for box
|
||||||
|
boxFile, err := os.Create(filepath.Join(pkg.Dir, boxFilename))
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error creating embedded box file: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
defer boxFile.Close()
|
||||||
|
|
||||||
|
// write source to file
|
||||||
|
_, err = io.Copy(boxFile, bytes.NewBuffer(embedSource))
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("error writing embedSource to file: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
204
vendor/github.com/GeertJohan/go.rice/rice/embed-syso.go
generated
vendored
Normal file
204
vendor/github.com/GeertJohan/go.rice/rice/embed-syso.go
generated
vendored
Normal file
@ -0,0 +1,204 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/gob"
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/GeertJohan/go.rice/embedded"
|
||||||
|
"github.com/akavel/rsrc/coff"
|
||||||
|
)
|
||||||
|
|
||||||
|
type sizedReader struct {
|
||||||
|
*bytes.Reader
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s sizedReader) Size() int64 {
|
||||||
|
return int64(s.Len())
|
||||||
|
}
|
||||||
|
|
||||||
|
var tmplEmbeddedSysoHelper *template.Template
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
var err error
|
||||||
|
tmplEmbeddedSysoHelper, err = template.New("embeddedSysoHelper").Parse(`package {{.Package}}
|
||||||
|
// ############# GENERATED CODE #####################
|
||||||
|
// ## This file was generated by the rice tool.
|
||||||
|
// ## Do not edit unless you know what you're doing.
|
||||||
|
// ##################################################
|
||||||
|
|
||||||
|
// extern char _bricebox_{{.Symname}}[], _ericebox_{{.Symname}};
|
||||||
|
// int get_{{.Symname}}_length() {
|
||||||
|
// return &_ericebox_{{.Symname}} - _bricebox_{{.Symname}};
|
||||||
|
// }
|
||||||
|
import "C"
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/gob"
|
||||||
|
"github.com/GeertJohan/go.rice/embedded"
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
ptr := unsafe.Pointer(&C._bricebox_{{.Symname}})
|
||||||
|
bts := C.GoBytes(ptr, C.get_{{.Symname}}_length())
|
||||||
|
embeddedBox := &embedded.EmbeddedBox{}
|
||||||
|
err := gob.NewDecoder(bytes.NewReader(bts)).Decode(embeddedBox)
|
||||||
|
if err != nil {
|
||||||
|
panic("error decoding embedded box: "+err.Error())
|
||||||
|
}
|
||||||
|
embeddedBox.Link()
|
||||||
|
embedded.RegisterEmbeddedBox(embeddedBox.Name, embeddedBox)
|
||||||
|
}`)
|
||||||
|
if err != nil {
|
||||||
|
panic("could not parse template embeddedSysoHelper: " + err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type embeddedSysoHelperData struct {
|
||||||
|
Package string
|
||||||
|
Symname string
|
||||||
|
}
|
||||||
|
|
||||||
|
func operationEmbedSyso(pkg *build.Package) {
|
||||||
|
|
||||||
|
regexpSynameReplacer := regexp.MustCompile(`[^a-z0-9_]`)
|
||||||
|
|
||||||
|
boxMap := findBoxes(pkg)
|
||||||
|
|
||||||
|
// notify user when no calls to rice.FindBox are made (is this an error and therefore os.Exit(1) ?
|
||||||
|
if len(boxMap) == 0 {
|
||||||
|
fmt.Println("no calls to rice.FindBox() found")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
verbosef("\n")
|
||||||
|
|
||||||
|
for boxname := range boxMap {
|
||||||
|
// find path and filename for this box
|
||||||
|
boxPath := filepath.Join(pkg.Dir, boxname)
|
||||||
|
boxFilename := strings.Replace(boxname, "/", "-", -1)
|
||||||
|
boxFilename = strings.Replace(boxFilename, "..", "back", -1)
|
||||||
|
boxFilename = strings.Replace(boxFilename, ".", "-", -1)
|
||||||
|
|
||||||
|
// verbose info
|
||||||
|
verbosef("embedding box '%s'\n", boxname)
|
||||||
|
verbosef("\tto file %s\n", boxFilename)
|
||||||
|
|
||||||
|
// read box metadata
|
||||||
|
boxInfo, ierr := os.Stat(boxPath)
|
||||||
|
if ierr != nil {
|
||||||
|
fmt.Printf("Error: unable to access box at %s\n", boxPath)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// create box datastructure (used by template)
|
||||||
|
box := &embedded.EmbeddedBox{
|
||||||
|
Name: boxname,
|
||||||
|
Time: boxInfo.ModTime(),
|
||||||
|
EmbedType: embedded.EmbedTypeSyso,
|
||||||
|
Files: make(map[string]*embedded.EmbeddedFile),
|
||||||
|
Dirs: make(map[string]*embedded.EmbeddedDir),
|
||||||
|
}
|
||||||
|
|
||||||
|
// fill box datastructure with file data
|
||||||
|
filepath.Walk(boxPath, func(path string, info os.FileInfo, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error walking box: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
filename := strings.TrimPrefix(path, boxPath)
|
||||||
|
filename = strings.Replace(filename, "\\", "/", -1)
|
||||||
|
filename = strings.TrimPrefix(filename, "/")
|
||||||
|
if info.IsDir() {
|
||||||
|
embeddedDir := &embedded.EmbeddedDir{
|
||||||
|
Filename: filename,
|
||||||
|
DirModTime: info.ModTime(),
|
||||||
|
}
|
||||||
|
verbosef("\tincludes dir: '%s'\n", embeddedDir.Filename)
|
||||||
|
box.Dirs[embeddedDir.Filename] = embeddedDir
|
||||||
|
|
||||||
|
// add tree entry (skip for root, it'll create a recursion)
|
||||||
|
if embeddedDir.Filename != "" {
|
||||||
|
pathParts := strings.Split(embeddedDir.Filename, "/")
|
||||||
|
parentDir := box.Dirs[strings.Join(pathParts[:len(pathParts)-1], "/")]
|
||||||
|
parentDir.ChildDirs = append(parentDir.ChildDirs, embeddedDir)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
embeddedFile := &embedded.EmbeddedFile{
|
||||||
|
Filename: filename,
|
||||||
|
FileModTime: info.ModTime(),
|
||||||
|
Content: "",
|
||||||
|
}
|
||||||
|
verbosef("\tincludes file: '%s'\n", embeddedFile.Filename)
|
||||||
|
contentBytes, err := ioutil.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error reading file content while walking box: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
embeddedFile.Content = string(contentBytes)
|
||||||
|
box.Files[embeddedFile.Filename] = embeddedFile
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
|
// encode embedded box to gob file
|
||||||
|
boxGobBuf := &bytes.Buffer{}
|
||||||
|
err := gob.NewEncoder(boxGobBuf).Encode(box)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error encoding box to gob: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
verbosef("gob-encoded embeddedBox is %d bytes large\n", boxGobBuf.Len())
|
||||||
|
|
||||||
|
// write coff
|
||||||
|
symname := regexpSynameReplacer.ReplaceAllString(boxname, "_")
|
||||||
|
createCoffSyso(boxname, symname, "386", boxGobBuf.Bytes())
|
||||||
|
createCoffSyso(boxname, symname, "amd64", boxGobBuf.Bytes())
|
||||||
|
|
||||||
|
// write go
|
||||||
|
sysoHelperData := embeddedSysoHelperData{
|
||||||
|
Package: pkg.Name,
|
||||||
|
Symname: symname,
|
||||||
|
}
|
||||||
|
fileSysoHelper, err := os.Create(boxFilename + ".rice-box.go")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error creating syso helper: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
err = tmplEmbeddedSysoHelper.Execute(fileSysoHelper, sysoHelperData)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error executing tmplEmbeddedSysoHelper: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func createCoffSyso(boxFilename string, symname string, arch string, data []byte) {
|
||||||
|
boxCoff := coff.NewRDATA()
|
||||||
|
switch arch {
|
||||||
|
case "386":
|
||||||
|
case "amd64":
|
||||||
|
boxCoff.FileHeader.Machine = 0x8664
|
||||||
|
default:
|
||||||
|
panic("invalid arch")
|
||||||
|
}
|
||||||
|
boxCoff.AddData("_bricebox_"+symname, sizedReader{bytes.NewReader(data)})
|
||||||
|
boxCoff.AddData("_ericebox_"+symname, io.NewSectionReader(strings.NewReader("\000\000"), 0, 2)) // TODO: why? copied from rsrc, which copied it from as-generated
|
||||||
|
boxCoff.Freeze()
|
||||||
|
err := writeCoff(boxCoff, boxFilename+"_"+arch+".rice-box.syso")
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error writing %s coff/.syso: %v\n", arch, err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
150
vendor/github.com/GeertJohan/go.rice/rice/find.go
generated
vendored
Normal file
150
vendor/github.com/GeertJohan/go.rice/rice/find.go
generated
vendored
Normal file
@ -0,0 +1,150 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/ast"
|
||||||
|
"go/build"
|
||||||
|
"go/parser"
|
||||||
|
"go/token"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func badArgument(fileset *token.FileSet, p token.Pos) {
|
||||||
|
pos := fileset.Position(p)
|
||||||
|
filename := pos.Filename
|
||||||
|
base, err := os.Getwd()
|
||||||
|
if err == nil {
|
||||||
|
rpath, perr := filepath.Rel(base, pos.Filename)
|
||||||
|
if perr == nil {
|
||||||
|
filename = rpath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
msg := fmt.Sprintf("%s:%d: Error: found call to rice.FindBox, "+
|
||||||
|
"but argument must be a string literal.\n", filename, pos.Line)
|
||||||
|
fmt.Println(msg)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func findBoxes(pkg *build.Package) map[string]bool {
|
||||||
|
// create map of boxes to embed
|
||||||
|
var boxMap = make(map[string]bool)
|
||||||
|
|
||||||
|
// create one list of files for this package
|
||||||
|
filenames := make([]string, 0, len(pkg.GoFiles)+len(pkg.CgoFiles))
|
||||||
|
filenames = append(filenames, pkg.GoFiles...)
|
||||||
|
filenames = append(filenames, pkg.CgoFiles...)
|
||||||
|
|
||||||
|
// loop over files, search for rice.FindBox(..) calls
|
||||||
|
for _, filename := range filenames {
|
||||||
|
// find full filepath
|
||||||
|
fullpath := filepath.Join(pkg.Dir, filename)
|
||||||
|
if strings.HasSuffix(filename, "rice-box.go") {
|
||||||
|
// Ignore *.rice-box.go files
|
||||||
|
verbosef("skipping file %q\n", fullpath)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
verbosef("scanning file %q\n", fullpath)
|
||||||
|
|
||||||
|
fset := token.NewFileSet()
|
||||||
|
f, err := parser.ParseFile(fset, fullpath, nil, 0)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
var riceIsImported bool
|
||||||
|
ricePkgName := "rice"
|
||||||
|
for _, imp := range f.Imports {
|
||||||
|
if strings.HasSuffix(imp.Path.Value, "go.rice\"") {
|
||||||
|
if imp.Name != nil {
|
||||||
|
ricePkgName = imp.Name.Name
|
||||||
|
}
|
||||||
|
riceIsImported = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !riceIsImported {
|
||||||
|
// Rice wasn't imported, so we won't find a box.
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if ricePkgName == "_" {
|
||||||
|
// Rice pkg is unnamed, so we won't find a box.
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Inspect AST, looking for calls to (Must)?FindBox.
|
||||||
|
// First parameter of the func must be a basic literal.
|
||||||
|
// Identifiers won't be resolved.
|
||||||
|
var nextIdentIsBoxFunc bool
|
||||||
|
var nextBasicLitParamIsBoxName bool
|
||||||
|
var boxCall token.Pos
|
||||||
|
var variableToRemember string
|
||||||
|
var validVariablesForBoxes map[string]bool = make(map[string]bool)
|
||||||
|
|
||||||
|
ast.Inspect(f, func(node ast.Node) bool {
|
||||||
|
if node == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
switch x := node.(type) {
|
||||||
|
// this case fixes the var := func() style assignments, not assignments to vars declared separately from the assignment.
|
||||||
|
case *ast.AssignStmt:
|
||||||
|
var assign = node.(*ast.AssignStmt)
|
||||||
|
name, found := assign.Lhs[0].(*ast.Ident)
|
||||||
|
if found {
|
||||||
|
variableToRemember = name.Name
|
||||||
|
composite, first := assign.Rhs[0].(*ast.CompositeLit)
|
||||||
|
if first {
|
||||||
|
riceSelector, second := composite.Type.(*ast.SelectorExpr)
|
||||||
|
|
||||||
|
if second {
|
||||||
|
callCorrect := riceSelector.Sel.Name == "Config"
|
||||||
|
packageName, third := riceSelector.X.(*ast.Ident)
|
||||||
|
|
||||||
|
if third && callCorrect && packageName.Name == ricePkgName {
|
||||||
|
validVariablesForBoxes[name.Name] = true
|
||||||
|
verbosef("\tfound variable, saving to scan for boxes: %q\n", name.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case *ast.Ident:
|
||||||
|
if nextIdentIsBoxFunc || ricePkgName == "." {
|
||||||
|
nextIdentIsBoxFunc = false
|
||||||
|
if x.Name == "FindBox" || x.Name == "MustFindBox" {
|
||||||
|
nextBasicLitParamIsBoxName = true
|
||||||
|
boxCall = x.Pos()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if x.Name == ricePkgName || validVariablesForBoxes[x.Name] {
|
||||||
|
nextIdentIsBoxFunc = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case *ast.BasicLit:
|
||||||
|
if nextBasicLitParamIsBoxName {
|
||||||
|
if x.Kind == token.STRING {
|
||||||
|
nextBasicLitParamIsBoxName = false
|
||||||
|
// trim "" or ``
|
||||||
|
name := x.Value[1 : len(x.Value)-1]
|
||||||
|
boxMap[name] = true
|
||||||
|
verbosef("\tfound box %q\n", name)
|
||||||
|
} else {
|
||||||
|
badArgument(fset, boxCall)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
if nextIdentIsBoxFunc {
|
||||||
|
nextIdentIsBoxFunc = false
|
||||||
|
}
|
||||||
|
if nextBasicLitParamIsBoxName {
|
||||||
|
badArgument(fset, boxCall)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return boxMap
|
||||||
|
}
|
80
vendor/github.com/GeertJohan/go.rice/rice/flags.go
generated
vendored
Normal file
80
vendor/github.com/GeertJohan/go.rice/rice/flags.go
generated
vendored
Normal file
@ -0,0 +1,80 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
goflags "github.com/jessevdk/go-flags" // rename import to `goflags` (file scope) so we can use `var flags` (package scope)
|
||||||
|
)
|
||||||
|
|
||||||
|
// flags
|
||||||
|
var flags struct {
|
||||||
|
Verbose bool `long:"verbose" short:"v" description:"Show verbose debug information"`
|
||||||
|
ImportPaths []string `long:"import-path" short:"i" description:"Import path(s) to use. Using PWD when left empty. Specify multiple times for more import paths to append"`
|
||||||
|
|
||||||
|
Append struct {
|
||||||
|
Executable string `long:"exec" description:"Executable to append" required:"true"`
|
||||||
|
} `command:"append"`
|
||||||
|
|
||||||
|
EmbedGo struct{} `command:"embed-go" alias:"embed"`
|
||||||
|
EmbedSyso struct{} `command:"embed-syso"`
|
||||||
|
Clean struct{} `command:"clean"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// flags parser
|
||||||
|
var flagsParser *goflags.Parser
|
||||||
|
|
||||||
|
// initFlags parses the given flags.
|
||||||
|
// when the user asks for help (-h or --help): the application exists with status 0
|
||||||
|
// when unexpected flags is given: the application exits with status 1
|
||||||
|
func parseArguments() {
|
||||||
|
// create flags parser in global var, for flagsParser.Active.Name (operation)
|
||||||
|
flagsParser = goflags.NewParser(&flags, goflags.Default)
|
||||||
|
|
||||||
|
// parse flags
|
||||||
|
args, err := flagsParser.Parse()
|
||||||
|
if err != nil {
|
||||||
|
// assert the err to be a flags.Error
|
||||||
|
flagError := err.(*goflags.Error)
|
||||||
|
if flagError.Type == goflags.ErrHelp {
|
||||||
|
// user asked for help on flags.
|
||||||
|
// program can exit successfully
|
||||||
|
os.Exit(0)
|
||||||
|
}
|
||||||
|
if flagError.Type == goflags.ErrUnknownFlag {
|
||||||
|
fmt.Println("Use --help to view available options.")
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
if flagError.Type == goflags.ErrRequired {
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
fmt.Printf("Error parsing flags: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// error on left-over arguments
|
||||||
|
if len(args) > 0 {
|
||||||
|
fmt.Printf("Unexpected arguments: %s\nUse --help to view available options.", args)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// default ImportPath to pwd when not set
|
||||||
|
if len(flags.ImportPaths) == 0 {
|
||||||
|
pwd, err := os.Getwd()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error getting pwd: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
verbosef("using pwd as import path\n")
|
||||||
|
// find non-absolute path for this pwd
|
||||||
|
pkg, err := build.ImportDir(pwd, build.FindOnly)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error using current directory as import path: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
flags.ImportPaths = append(flags.ImportPaths, pkg.ImportPath)
|
||||||
|
verbosef("using import paths: %s\n", flags.ImportPaths)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
14
vendor/github.com/GeertJohan/go.rice/rice/identifier.go
generated
vendored
Normal file
14
vendor/github.com/GeertJohan/go.rice/rice/identifier.go
generated
vendored
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/GeertJohan/go.incremental"
|
||||||
|
)
|
||||||
|
|
||||||
|
var identifierCount incremental.Uint64
|
||||||
|
|
||||||
|
func nextIdentifier() string {
|
||||||
|
num := identifierCount.Next()
|
||||||
|
return strconv.FormatUint(num, 36) // 0123456789abcdefghijklmnopqrstuvwxyz
|
||||||
|
}
|
68
vendor/github.com/GeertJohan/go.rice/rice/main.go
generated
vendored
Normal file
68
vendor/github.com/GeertJohan/go.rice/rice/main.go
generated
vendored
Normal file
@ -0,0 +1,68 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/build"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// parser arguments
|
||||||
|
parseArguments()
|
||||||
|
|
||||||
|
// find package for path
|
||||||
|
var pkgs []*build.Package
|
||||||
|
for _, importPath := range flags.ImportPaths {
|
||||||
|
pkg := pkgForPath(importPath)
|
||||||
|
pkgs = append(pkgs, pkg)
|
||||||
|
}
|
||||||
|
|
||||||
|
// switch on the operation to perform
|
||||||
|
switch flagsParser.Active.Name {
|
||||||
|
case "embed", "embed-go":
|
||||||
|
for _, pkg := range pkgs {
|
||||||
|
operationEmbedGo(pkg)
|
||||||
|
}
|
||||||
|
case "embed-syso":
|
||||||
|
log.Println("WARNING: embedding .syso is experimental..")
|
||||||
|
for _, pkg := range pkgs {
|
||||||
|
operationEmbedSyso(pkg)
|
||||||
|
}
|
||||||
|
case "append":
|
||||||
|
operationAppend(pkgs)
|
||||||
|
case "clean":
|
||||||
|
for _, pkg := range pkgs {
|
||||||
|
operationClean(pkg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// all done
|
||||||
|
verbosef("\n")
|
||||||
|
verbosef("rice finished successfully\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
// helper function to get *build.Package for given path
|
||||||
|
func pkgForPath(path string) *build.Package {
|
||||||
|
// get pwd for relative imports
|
||||||
|
pwd, err := os.Getwd()
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error getting pwd (required for relative imports): %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// read full package information
|
||||||
|
pkg, err := build.Import(path, pwd, 0)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error reading package: %s\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
return pkg
|
||||||
|
}
|
||||||
|
|
||||||
|
func verbosef(format string, stuff ...interface{}) {
|
||||||
|
if flags.Verbose {
|
||||||
|
log.Printf(format, stuff...)
|
||||||
|
}
|
||||||
|
}
|
98
vendor/github.com/GeertJohan/go.rice/rice/templates.go
generated
vendored
Normal file
98
vendor/github.com/GeertJohan/go.rice/rice/templates.go
generated
vendored
Normal file
@ -0,0 +1,98 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"text/template"
|
||||||
|
)
|
||||||
|
|
||||||
|
var tmplEmbeddedBox *template.Template
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// parse embedded box template
|
||||||
|
tmplEmbeddedBox, err = template.New("embeddedBox").Parse(`package {{.Package}}
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/GeertJohan/go.rice/embedded"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
{{range .Boxes}}
|
||||||
|
func init() {
|
||||||
|
|
||||||
|
// define files
|
||||||
|
{{range .Files}}{{.Identifier}} := &embedded.EmbeddedFile{
|
||||||
|
Filename: ` + "`" + `{{.FileName}}` + "`" + `,
|
||||||
|
FileModTime: time.Unix({{.ModTime}}, 0),
|
||||||
|
Content: string({{.Content | printf "%q"}}),
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
// define dirs
|
||||||
|
{{range .Dirs}}{{.Identifier}} := &embedded.EmbeddedDir{
|
||||||
|
Filename: ` + "`" + `{{.FileName}}` + "`" + `,
|
||||||
|
DirModTime: time.Unix({{.ModTime}}, 0),
|
||||||
|
ChildFiles: []*embedded.EmbeddedFile{
|
||||||
|
{{range .ChildFiles}}{{.Identifier}}, // {{.FileName}}
|
||||||
|
{{end}}
|
||||||
|
},
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
// link ChildDirs
|
||||||
|
{{range .Dirs}}{{.Identifier}}.ChildDirs = []*embedded.EmbeddedDir{
|
||||||
|
{{range .ChildDirs}}{{.Identifier}}, // {{.FileName}}
|
||||||
|
{{end}}
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
// register embeddedBox
|
||||||
|
embedded.RegisterEmbeddedBox(` + "`" + `{{.BoxName}}` + "`" + `, &embedded.EmbeddedBox{
|
||||||
|
Name: ` + "`" + `{{.BoxName}}` + "`" + `,
|
||||||
|
Time: time.Unix({{.UnixNow}}, 0),
|
||||||
|
Dirs: map[string]*embedded.EmbeddedDir{
|
||||||
|
{{range .Dirs}}"{{.FileName}}": {{.Identifier}},
|
||||||
|
{{end}}
|
||||||
|
},
|
||||||
|
Files: map[string]*embedded.EmbeddedFile{
|
||||||
|
{{range .Files}}"{{.FileName}}": {{.Identifier}},
|
||||||
|
{{end}}
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
{{end}}`)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("error parsing embedded box template: %s\n", err)
|
||||||
|
os.Exit(-1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type embedFileDataType struct {
|
||||||
|
Package string
|
||||||
|
Boxes []*boxDataType
|
||||||
|
}
|
||||||
|
|
||||||
|
type boxDataType struct {
|
||||||
|
BoxName string
|
||||||
|
UnixNow int64
|
||||||
|
Files []*fileDataType
|
||||||
|
Dirs map[string]*dirDataType
|
||||||
|
}
|
||||||
|
|
||||||
|
type fileDataType struct {
|
||||||
|
Identifier string
|
||||||
|
FileName string
|
||||||
|
Content []byte
|
||||||
|
ModTime int64
|
||||||
|
}
|
||||||
|
|
||||||
|
type dirDataType struct {
|
||||||
|
Identifier string
|
||||||
|
FileName string
|
||||||
|
Content []byte
|
||||||
|
ModTime int64
|
||||||
|
ChildDirs []*dirDataType
|
||||||
|
ChildFiles []*fileDataType
|
||||||
|
}
|
22
vendor/github.com/GeertJohan/go.rice/rice/util.go
generated
vendored
Normal file
22
vendor/github.com/GeertJohan/go.rice/rice/util.go
generated
vendored
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/rand"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// randomString generates a pseudo-random alpha-numeric string with given length.
|
||||||
|
func randomString(length int) string {
|
||||||
|
rand.Seed(time.Now().UnixNano())
|
||||||
|
k := make([]rune, length)
|
||||||
|
for i := 0; i < length; i++ {
|
||||||
|
c := rand.Intn(35)
|
||||||
|
if c < 10 {
|
||||||
|
c += 48 // numbers (0-9) (0+48 == 48 == '0', 9+48 == 57 == '9')
|
||||||
|
} else {
|
||||||
|
c += 87 // lower case alphabets (a-z) (10+87 == 97 == 'a', 35+87 == 122 = 'z')
|
||||||
|
}
|
||||||
|
k[i] = rune(c)
|
||||||
|
}
|
||||||
|
return string(k)
|
||||||
|
}
|
42
vendor/github.com/GeertJohan/go.rice/rice/writecoff.go
generated
vendored
Normal file
42
vendor/github.com/GeertJohan/go.rice/rice/writecoff.go
generated
vendored
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"reflect"
|
||||||
|
|
||||||
|
"github.com/akavel/rsrc/binutil"
|
||||||
|
"github.com/akavel/rsrc/coff"
|
||||||
|
)
|
||||||
|
|
||||||
|
// copied from github.com/akavel/rsrc
|
||||||
|
// LICENSE: MIT
|
||||||
|
// Copyright 2013-2014 The rsrc Authors. (https://github.com/akavel/rsrc/blob/master/AUTHORS)
|
||||||
|
func writeCoff(coff *coff.Coff, fnameout string) error {
|
||||||
|
out, err := os.Create(fnameout)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer out.Close()
|
||||||
|
w := binutil.Writer{W: out}
|
||||||
|
|
||||||
|
// write the resulting file to disk
|
||||||
|
binutil.Walk(coff, func(v reflect.Value, path string) error {
|
||||||
|
if binutil.Plain(v.Kind()) {
|
||||||
|
w.WriteLE(v.Interface())
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
vv, ok := v.Interface().(binutil.SizedReader)
|
||||||
|
if ok {
|
||||||
|
w.WriteFromSized(vv)
|
||||||
|
return binutil.WALK_SKIP
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
|
if w.Err != nil {
|
||||||
|
return fmt.Errorf("Error writing output file: %s", w.Err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
19
vendor/github.com/GeertJohan/go.rice/sort.go
generated
vendored
Normal file
19
vendor/github.com/GeertJohan/go.rice/sort.go
generated
vendored
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import "os"
|
||||||
|
|
||||||
|
// SortByName allows an array of os.FileInfo objects
|
||||||
|
// to be easily sorted by filename using sort.Sort(SortByName(array))
|
||||||
|
type SortByName []os.FileInfo
|
||||||
|
|
||||||
|
func (f SortByName) Len() int { return len(f) }
|
||||||
|
func (f SortByName) Less(i, j int) bool { return f[i].Name() < f[j].Name() }
|
||||||
|
func (f SortByName) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
|
||||||
|
|
||||||
|
// SortByModified allows an array of os.FileInfo objects
|
||||||
|
// to be easily sorted by modified date using sort.Sort(SortByModified(array))
|
||||||
|
type SortByModified []os.FileInfo
|
||||||
|
|
||||||
|
func (f SortByModified) Len() int { return len(f) }
|
||||||
|
func (f SortByModified) Less(i, j int) bool { return f[i].ModTime().Unix() > f[j].ModTime().Unix() }
|
||||||
|
func (f SortByModified) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
|
252
vendor/github.com/GeertJohan/go.rice/virtual.go
generated
vendored
Normal file
252
vendor/github.com/GeertJohan/go.rice/virtual.go
generated
vendored
Normal file
@ -0,0 +1,252 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
|
|
||||||
|
"github.com/GeertJohan/go.rice/embedded"
|
||||||
|
)
|
||||||
|
|
||||||
|
//++ TODO: IDEA: merge virtualFile and virtualDir, this decreases work done by rice.File
|
||||||
|
|
||||||
|
// Error indicating some function is not implemented yet (but available to satisfy an interface)
|
||||||
|
var ErrNotImplemented = errors.New("not implemented yet")
|
||||||
|
|
||||||
|
// virtualFile is a 'stateful' virtual file.
|
||||||
|
// virtualFile wraps an *EmbeddedFile for a call to Box.Open() and virtualizes 'read cursor' (offset) and 'closing'.
|
||||||
|
// virtualFile is only internally visible and should be exposed through rice.File
|
||||||
|
type virtualFile struct {
|
||||||
|
*embedded.EmbeddedFile // the actual embedded file, embedded to obtain methods
|
||||||
|
offset int64 // read position on the virtual file
|
||||||
|
closed bool // closed when true
|
||||||
|
}
|
||||||
|
|
||||||
|
// create a new virtualFile for given EmbeddedFile
|
||||||
|
func newVirtualFile(ef *embedded.EmbeddedFile) *virtualFile {
|
||||||
|
vf := &virtualFile{
|
||||||
|
EmbeddedFile: ef,
|
||||||
|
offset: 0,
|
||||||
|
closed: false,
|
||||||
|
}
|
||||||
|
return vf
|
||||||
|
}
|
||||||
|
|
||||||
|
//++ TODO check for nil pointers in all these methods. When so: return os.PathError with Err: os.ErrInvalid
|
||||||
|
|
||||||
|
func (vf *virtualFile) close() error {
|
||||||
|
if vf.closed {
|
||||||
|
return &os.PathError{
|
||||||
|
Op: "close",
|
||||||
|
Path: vf.EmbeddedFile.Filename,
|
||||||
|
Err: errors.New("already closed"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
vf.EmbeddedFile = nil
|
||||||
|
vf.closed = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vf *virtualFile) stat() (os.FileInfo, error) {
|
||||||
|
if vf.closed {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "stat",
|
||||||
|
Path: vf.EmbeddedFile.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return (*embeddedFileInfo)(vf.EmbeddedFile), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vf *virtualFile) readdir(count int) ([]os.FileInfo, error) {
|
||||||
|
if vf.closed {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "readdir",
|
||||||
|
Path: vf.EmbeddedFile.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
//TODO: return proper error for a readdir() call on a file
|
||||||
|
return nil, ErrNotImplemented
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vf *virtualFile) read(bts []byte) (int, error) {
|
||||||
|
if vf.closed {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "read",
|
||||||
|
Path: vf.EmbeddedFile.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
end := vf.offset + int64(len(bts))
|
||||||
|
|
||||||
|
if end >= int64(len(vf.Content)) {
|
||||||
|
// end of file, so return what we have + EOF
|
||||||
|
n := copy(bts, vf.Content[vf.offset:])
|
||||||
|
vf.offset = 0
|
||||||
|
return n, io.EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
n := copy(bts, vf.Content[vf.offset:end])
|
||||||
|
vf.offset += int64(n)
|
||||||
|
return n, nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vf *virtualFile) seek(offset int64, whence int) (int64, error) {
|
||||||
|
if vf.closed {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "seek",
|
||||||
|
Path: vf.EmbeddedFile.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var e error
|
||||||
|
|
||||||
|
//++ TODO: check if this is correct implementation for seek
|
||||||
|
switch whence {
|
||||||
|
case os.SEEK_SET:
|
||||||
|
//++ check if new offset isn't out of bounds, set e when it is, then break out of switch
|
||||||
|
vf.offset = offset
|
||||||
|
case os.SEEK_CUR:
|
||||||
|
//++ check if new offset isn't out of bounds, set e when it is, then break out of switch
|
||||||
|
vf.offset += offset
|
||||||
|
case os.SEEK_END:
|
||||||
|
//++ check if new offset isn't out of bounds, set e when it is, then break out of switch
|
||||||
|
vf.offset = int64(len(vf.EmbeddedFile.Content)) - offset
|
||||||
|
}
|
||||||
|
|
||||||
|
if e != nil {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "seek",
|
||||||
|
Path: vf.Filename,
|
||||||
|
Err: e,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return vf.offset, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// virtualDir is a 'stateful' virtual directory.
|
||||||
|
// virtualDir wraps an *EmbeddedDir for a call to Box.Open() and virtualizes 'closing'.
|
||||||
|
// virtualDir is only internally visible and should be exposed through rice.File
|
||||||
|
type virtualDir struct {
|
||||||
|
*embedded.EmbeddedDir
|
||||||
|
offset int // readdir position on the directory
|
||||||
|
closed bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// create a new virtualDir for given EmbeddedDir
|
||||||
|
func newVirtualDir(ed *embedded.EmbeddedDir) *virtualDir {
|
||||||
|
vd := &virtualDir{
|
||||||
|
EmbeddedDir: ed,
|
||||||
|
offset: 0,
|
||||||
|
closed: false,
|
||||||
|
}
|
||||||
|
return vd
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vd *virtualDir) close() error {
|
||||||
|
//++ TODO: needs sync mutex?
|
||||||
|
if vd.closed {
|
||||||
|
return &os.PathError{
|
||||||
|
Op: "close",
|
||||||
|
Path: vd.EmbeddedDir.Filename,
|
||||||
|
Err: errors.New("already closed"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
vd.closed = true
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vd *virtualDir) stat() (os.FileInfo, error) {
|
||||||
|
if vd.closed {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "stat",
|
||||||
|
Path: vd.EmbeddedDir.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return (*embeddedDirInfo)(vd.EmbeddedDir), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vd *virtualDir) readdir(n int) (fi []os.FileInfo, err error) {
|
||||||
|
|
||||||
|
if vd.closed {
|
||||||
|
return nil, &os.PathError{
|
||||||
|
Op: "readdir",
|
||||||
|
Path: vd.EmbeddedDir.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build up the array of our contents
|
||||||
|
var files []os.FileInfo
|
||||||
|
|
||||||
|
// Add the child directories
|
||||||
|
for _, child := range vd.ChildDirs {
|
||||||
|
child.Filename = filepath.Base(child.Filename)
|
||||||
|
files = append(files, (*embeddedDirInfo)(child))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the child files
|
||||||
|
for _, child := range vd.ChildFiles {
|
||||||
|
child.Filename = filepath.Base(child.Filename)
|
||||||
|
files = append(files, (*embeddedFileInfo)(child))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sort it by filename (lexical order)
|
||||||
|
sort.Sort(SortByName(files))
|
||||||
|
|
||||||
|
// Return all contents if that's what is requested
|
||||||
|
if n <= 0 {
|
||||||
|
vd.offset = 0
|
||||||
|
return files, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// If user has requested past the end of our list
|
||||||
|
// return what we can and send an EOF
|
||||||
|
if vd.offset+n >= len(files) {
|
||||||
|
offset := vd.offset
|
||||||
|
vd.offset = 0
|
||||||
|
return files[offset:], io.EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
offset := vd.offset
|
||||||
|
vd.offset += n
|
||||||
|
return files[offset : offset+n], nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vd *virtualDir) read(bts []byte) (int, error) {
|
||||||
|
if vd.closed {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "read",
|
||||||
|
Path: vd.EmbeddedDir.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "read",
|
||||||
|
Path: vd.EmbeddedDir.Filename,
|
||||||
|
Err: errors.New("is a directory"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (vd *virtualDir) seek(offset int64, whence int) (int64, error) {
|
||||||
|
if vd.closed {
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "seek",
|
||||||
|
Path: vd.EmbeddedDir.Filename,
|
||||||
|
Err: errors.New("bad file descriptor"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0, &os.PathError{
|
||||||
|
Op: "seek",
|
||||||
|
Path: vd.Filename,
|
||||||
|
Err: errors.New("is a directory"),
|
||||||
|
}
|
||||||
|
}
|
122
vendor/github.com/GeertJohan/go.rice/walk.go
generated
vendored
Normal file
122
vendor/github.com/GeertJohan/go.rice/walk.go
generated
vendored
Normal file
@ -0,0 +1,122 @@
|
|||||||
|
package rice
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Walk is like filepath.Walk()
|
||||||
|
// Visit http://golang.org/pkg/path/filepath/#Walk for more information
|
||||||
|
func (b *Box) Walk(path string, walkFn filepath.WalkFunc) error {
|
||||||
|
|
||||||
|
pathFile, err := b.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer pathFile.Close()
|
||||||
|
|
||||||
|
pathInfo, err := pathFile.Stat()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if b.IsAppended() || b.IsEmbedded() {
|
||||||
|
return b.walk(path, pathInfo, walkFn)
|
||||||
|
}
|
||||||
|
|
||||||
|
// We don't have any embedded or appended box so use live filesystem mode
|
||||||
|
return filepath.Walk(b.absolutePath+string(os.PathSeparator)+path, func(path string, info os.FileInfo, err error) error {
|
||||||
|
|
||||||
|
// Strip out the box name from the returned paths
|
||||||
|
path = strings.TrimPrefix(path, b.absolutePath+string(os.PathSeparator))
|
||||||
|
return walkFn(path, info, err)
|
||||||
|
|
||||||
|
})
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// walk recursively descends path.
|
||||||
|
// See walk() in $GOROOT/src/pkg/path/filepath/path.go
|
||||||
|
func (b *Box) walk(path string, info os.FileInfo, walkFn filepath.WalkFunc) error {
|
||||||
|
|
||||||
|
err := walkFn(path, info, nil)
|
||||||
|
if err != nil {
|
||||||
|
if info.IsDir() && err == filepath.SkipDir {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !info.IsDir() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
names, err := b.readDirNames(path)
|
||||||
|
if err != nil {
|
||||||
|
return walkFn(path, info, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, name := range names {
|
||||||
|
|
||||||
|
filename := filepath.Join(path, name)
|
||||||
|
fileObject, err := b.Open(filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer fileObject.Close()
|
||||||
|
|
||||||
|
fileInfo, err := fileObject.Stat()
|
||||||
|
if err != nil {
|
||||||
|
if err := walkFn(filename, fileInfo, err); err != nil && err != filepath.SkipDir {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
err = b.walk(filename, fileInfo, walkFn)
|
||||||
|
if err != nil {
|
||||||
|
if !fileInfo.IsDir() || err != filepath.SkipDir {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// readDirNames reads the directory named by path and returns a sorted list of directory entries.
|
||||||
|
// See readDirNames() in $GOROOT/pkg/path/filepath/path.go
|
||||||
|
func (b *Box) readDirNames(path string) ([]string, error) {
|
||||||
|
|
||||||
|
f, err := b.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
stat, err := f.Stat()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !stat.IsDir() {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
infos, err := f.Readdir(0)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var names []string
|
||||||
|
|
||||||
|
for _, info := range infos {
|
||||||
|
names = append(names, info.Name())
|
||||||
|
}
|
||||||
|
|
||||||
|
sort.Strings(names)
|
||||||
|
return names, nil
|
||||||
|
|
||||||
|
}
|
21
vendor/github.com/daaku/go.zipexe/license
generated
vendored
Normal file
21
vendor/github.com/daaku/go.zipexe/license
generated
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright © 2012-2015 Carlos Castillo
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
||||||
|
this software and associated documentation files (the “Software”), to deal in
|
||||||
|
the Software without restriction, including without limitation the rights to
|
||||||
|
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
|
||||||
|
of the Software, and to permit persons to whom the Software is furnished to do
|
||||||
|
so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
142
vendor/github.com/daaku/go.zipexe/zipexe.go
generated
vendored
Normal file
142
vendor/github.com/daaku/go.zipexe/zipexe.go
generated
vendored
Normal file
@ -0,0 +1,142 @@
|
|||||||
|
// Package zipexe attempts to open an executable binary file as a zip file.
|
||||||
|
package zipexe
|
||||||
|
|
||||||
|
import (
|
||||||
|
"archive/zip"
|
||||||
|
"debug/elf"
|
||||||
|
"debug/macho"
|
||||||
|
"debug/pe"
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Opens a zip file by path.
|
||||||
|
func Open(path string) (*zip.Reader, error) {
|
||||||
|
_, rd, err := OpenCloser(path)
|
||||||
|
return rd, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// OpenCloser is like Open but returns an additional Closer to avoid leaking open files.
|
||||||
|
func OpenCloser(path string) (io.Closer, *zip.Reader, error) {
|
||||||
|
file, err := os.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
finfo, err := file.Stat()
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
zr, err := NewReader(file, finfo.Size())
|
||||||
|
if err != nil {
|
||||||
|
return nil, nil, err
|
||||||
|
}
|
||||||
|
return file, zr, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open a zip file, specially handling various binaries that may have been
|
||||||
|
// augmented with zip data.
|
||||||
|
func NewReader(rda io.ReaderAt, size int64) (*zip.Reader, error) {
|
||||||
|
handlers := []func(io.ReaderAt, int64) (*zip.Reader, error){
|
||||||
|
zip.NewReader,
|
||||||
|
zipExeReaderMacho,
|
||||||
|
zipExeReaderElf,
|
||||||
|
zipExeReaderPe,
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, handler := range handlers {
|
||||||
|
zfile, err := handler(rda, size)
|
||||||
|
if err == nil {
|
||||||
|
return zfile, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, errors.New("Couldn't Open As Executable")
|
||||||
|
}
|
||||||
|
|
||||||
|
// zipExeReaderMacho treats the file as a Mach-O binary
|
||||||
|
// (Mac OS X / Darwin executable) and attempts to find a zip archive.
|
||||||
|
func zipExeReaderMacho(rda io.ReaderAt, size int64) (*zip.Reader, error) {
|
||||||
|
file, err := macho.NewFile(rda)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var max int64
|
||||||
|
for _, load := range file.Loads {
|
||||||
|
seg, ok := load.(*macho.Segment)
|
||||||
|
if ok {
|
||||||
|
// Check if the segment contains a zip file
|
||||||
|
if zfile, err := zip.NewReader(seg, int64(seg.Filesz)); err == nil {
|
||||||
|
return zfile, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Otherwise move end of file pointer
|
||||||
|
end := int64(seg.Offset + seg.Filesz)
|
||||||
|
if end > max {
|
||||||
|
max = end
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// No zip file within binary, try appended to end
|
||||||
|
section := io.NewSectionReader(rda, max, size-max)
|
||||||
|
return zip.NewReader(section, section.Size())
|
||||||
|
}
|
||||||
|
|
||||||
|
// zipExeReaderPe treats the file as a Portable Exectuable binary
|
||||||
|
// (Windows executable) and attempts to find a zip archive.
|
||||||
|
func zipExeReaderPe(rda io.ReaderAt, size int64) (*zip.Reader, error) {
|
||||||
|
file, err := pe.NewFile(rda)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var max int64
|
||||||
|
for _, sec := range file.Sections {
|
||||||
|
// Check if this section has a zip file
|
||||||
|
if zfile, err := zip.NewReader(sec, int64(sec.Size)); err == nil {
|
||||||
|
return zfile, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Otherwise move end of file pointer
|
||||||
|
end := int64(sec.Offset + sec.Size)
|
||||||
|
if end > max {
|
||||||
|
max = end
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// No zip file within binary, try appended to end
|
||||||
|
section := io.NewSectionReader(rda, max, size-max)
|
||||||
|
return zip.NewReader(section, section.Size())
|
||||||
|
}
|
||||||
|
|
||||||
|
// zipExeReaderElf treats the file as a ELF binary
|
||||||
|
// (linux/BSD/etc... executable) and attempts to find a zip archive.
|
||||||
|
func zipExeReaderElf(rda io.ReaderAt, size int64) (*zip.Reader, error) {
|
||||||
|
file, err := elf.NewFile(rda)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var max int64
|
||||||
|
for _, sect := range file.Sections {
|
||||||
|
if sect.Type == elf.SHT_NOBITS {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if this section has a zip file
|
||||||
|
if zfile, err := zip.NewReader(sect, int64(sect.Size)); err == nil {
|
||||||
|
return zfile, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Otherwise move end of file pointer
|
||||||
|
end := int64(sect.Offset + sect.Size)
|
||||||
|
if end > max {
|
||||||
|
max = end
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// No zip file within binary, try appended to end
|
||||||
|
section := io.NewSectionReader(rda, max, size-max)
|
||||||
|
return zip.NewReader(section, section.Size())
|
||||||
|
}
|
8
vendor/github.com/dgrijalva/jwt-go/LICENSE
generated
vendored
Normal file
8
vendor/github.com/dgrijalva/jwt-go/LICENSE
generated
vendored
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
Copyright (c) 2012 Dave Grijalva
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
||||||
|
|
134
vendor/github.com/dgrijalva/jwt-go/claims.go
generated
vendored
Normal file
134
vendor/github.com/dgrijalva/jwt-go/claims.go
generated
vendored
Normal file
@ -0,0 +1,134 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/subtle"
|
||||||
|
"fmt"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// For a type to be a Claims object, it must just have a Valid method that determines
|
||||||
|
// if the token is invalid for any supported reason
|
||||||
|
type Claims interface {
|
||||||
|
Valid() error
|
||||||
|
}
|
||||||
|
|
||||||
|
// Structured version of Claims Section, as referenced at
|
||||||
|
// https://tools.ietf.org/html/rfc7519#section-4.1
|
||||||
|
// See examples for how to use this with your own claim types
|
||||||
|
type StandardClaims struct {
|
||||||
|
Audience string `json:"aud,omitempty"`
|
||||||
|
ExpiresAt int64 `json:"exp,omitempty"`
|
||||||
|
Id string `json:"jti,omitempty"`
|
||||||
|
IssuedAt int64 `json:"iat,omitempty"`
|
||||||
|
Issuer string `json:"iss,omitempty"`
|
||||||
|
NotBefore int64 `json:"nbf,omitempty"`
|
||||||
|
Subject string `json:"sub,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validates time based claims "exp, iat, nbf".
|
||||||
|
// There is no accounting for clock skew.
|
||||||
|
// As well, if any of the above claims are not in the token, it will still
|
||||||
|
// be considered a valid claim.
|
||||||
|
func (c StandardClaims) Valid() error {
|
||||||
|
vErr := new(ValidationError)
|
||||||
|
now := TimeFunc().Unix()
|
||||||
|
|
||||||
|
// The claims below are optional, by default, so if they are set to the
|
||||||
|
// default value in Go, let's not fail the verification for them.
|
||||||
|
if c.VerifyExpiresAt(now, false) == false {
|
||||||
|
delta := time.Unix(now, 0).Sub(time.Unix(c.ExpiresAt, 0))
|
||||||
|
vErr.Inner = fmt.Errorf("token is expired by %v", delta)
|
||||||
|
vErr.Errors |= ValidationErrorExpired
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.VerifyIssuedAt(now, false) == false {
|
||||||
|
vErr.Inner = fmt.Errorf("Token used before issued")
|
||||||
|
vErr.Errors |= ValidationErrorIssuedAt
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.VerifyNotBefore(now, false) == false {
|
||||||
|
vErr.Inner = fmt.Errorf("token is not valid yet")
|
||||||
|
vErr.Errors |= ValidationErrorNotValidYet
|
||||||
|
}
|
||||||
|
|
||||||
|
if vErr.valid() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return vErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the aud claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (c *StandardClaims) VerifyAudience(cmp string, req bool) bool {
|
||||||
|
return verifyAud(c.Audience, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the exp claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (c *StandardClaims) VerifyExpiresAt(cmp int64, req bool) bool {
|
||||||
|
return verifyExp(c.ExpiresAt, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the iat claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (c *StandardClaims) VerifyIssuedAt(cmp int64, req bool) bool {
|
||||||
|
return verifyIat(c.IssuedAt, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the iss claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (c *StandardClaims) VerifyIssuer(cmp string, req bool) bool {
|
||||||
|
return verifyIss(c.Issuer, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the nbf claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (c *StandardClaims) VerifyNotBefore(cmp int64, req bool) bool {
|
||||||
|
return verifyNbf(c.NotBefore, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ----- helpers
|
||||||
|
|
||||||
|
func verifyAud(aud string, cmp string, required bool) bool {
|
||||||
|
if aud == "" {
|
||||||
|
return !required
|
||||||
|
}
|
||||||
|
if subtle.ConstantTimeCompare([]byte(aud), []byte(cmp)) != 0 {
|
||||||
|
return true
|
||||||
|
} else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func verifyExp(exp int64, now int64, required bool) bool {
|
||||||
|
if exp == 0 {
|
||||||
|
return !required
|
||||||
|
}
|
||||||
|
return now <= exp
|
||||||
|
}
|
||||||
|
|
||||||
|
func verifyIat(iat int64, now int64, required bool) bool {
|
||||||
|
if iat == 0 {
|
||||||
|
return !required
|
||||||
|
}
|
||||||
|
return now >= iat
|
||||||
|
}
|
||||||
|
|
||||||
|
func verifyIss(iss string, cmp string, required bool) bool {
|
||||||
|
if iss == "" {
|
||||||
|
return !required
|
||||||
|
}
|
||||||
|
if subtle.ConstantTimeCompare([]byte(iss), []byte(cmp)) != 0 {
|
||||||
|
return true
|
||||||
|
} else {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func verifyNbf(nbf int64, now int64, required bool) bool {
|
||||||
|
if nbf == 0 {
|
||||||
|
return !required
|
||||||
|
}
|
||||||
|
return now >= nbf
|
||||||
|
}
|
282
vendor/github.com/dgrijalva/jwt-go/cmd/jwt/app.go
generated
vendored
Normal file
282
vendor/github.com/dgrijalva/jwt-go/cmd/jwt/app.go
generated
vendored
Normal file
@ -0,0 +1,282 @@
|
|||||||
|
// A useful example app. You can use this to debug your tokens on the command line.
|
||||||
|
// This is also a great place to look at how you might use this library.
|
||||||
|
//
|
||||||
|
// Example usage:
|
||||||
|
// The following will create and sign a token, then verify it and output the original claims.
|
||||||
|
// echo {\"foo\":\"bar\"} | bin/jwt -key test/sample_key -alg RS256 -sign - | bin/jwt -key test/sample_key.pub -verify -
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
jwt "github.com/dgrijalva/jwt-go"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// Options
|
||||||
|
flagAlg = flag.String("alg", "", "signing algorithm identifier")
|
||||||
|
flagKey = flag.String("key", "", "path to key file or '-' to read from stdin")
|
||||||
|
flagCompact = flag.Bool("compact", false, "output compact JSON")
|
||||||
|
flagDebug = flag.Bool("debug", false, "print out all kinds of debug data")
|
||||||
|
flagClaims = make(ArgList)
|
||||||
|
flagHead = make(ArgList)
|
||||||
|
|
||||||
|
// Modes - exactly one of these is required
|
||||||
|
flagSign = flag.String("sign", "", "path to claims object to sign, '-' to read from stdin, or '+' to use only -claim args")
|
||||||
|
flagVerify = flag.String("verify", "", "path to JWT token to verify or '-' to read from stdin")
|
||||||
|
flagShow = flag.String("show", "", "path to JWT file or '-' to read from stdin")
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Plug in Var flags
|
||||||
|
flag.Var(flagClaims, "claim", "add additional claims. may be used more than once")
|
||||||
|
flag.Var(flagHead, "header", "add additional header params. may be used more than once")
|
||||||
|
|
||||||
|
// Usage message if you ask for -help or if you mess up inputs.
|
||||||
|
flag.Usage = func() {
|
||||||
|
fmt.Fprintf(os.Stderr, "Usage of %s:\n", os.Args[0])
|
||||||
|
fmt.Fprintf(os.Stderr, " One of the following flags is required: sign, verify\n")
|
||||||
|
flag.PrintDefaults()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse command line options
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
// Do the thing. If something goes wrong, print error to stderr
|
||||||
|
// and exit with a non-zero status code
|
||||||
|
if err := start(); err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Figure out which thing to do and then do that
|
||||||
|
func start() error {
|
||||||
|
if *flagSign != "" {
|
||||||
|
return signToken()
|
||||||
|
} else if *flagVerify != "" {
|
||||||
|
return verifyToken()
|
||||||
|
} else if *flagShow != "" {
|
||||||
|
return showToken()
|
||||||
|
} else {
|
||||||
|
flag.Usage()
|
||||||
|
return fmt.Errorf("None of the required flags are present. What do you want me to do?")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Helper func: Read input from specified file or stdin
|
||||||
|
func loadData(p string) ([]byte, error) {
|
||||||
|
if p == "" {
|
||||||
|
return nil, fmt.Errorf("No path specified")
|
||||||
|
}
|
||||||
|
|
||||||
|
var rdr io.Reader
|
||||||
|
if p == "-" {
|
||||||
|
rdr = os.Stdin
|
||||||
|
} else if p == "+" {
|
||||||
|
return []byte("{}"), nil
|
||||||
|
} else {
|
||||||
|
if f, err := os.Open(p); err == nil {
|
||||||
|
rdr = f
|
||||||
|
defer f.Close()
|
||||||
|
} else {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ioutil.ReadAll(rdr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Print a json object in accordance with the prophecy (or the command line options)
|
||||||
|
func printJSON(j interface{}) error {
|
||||||
|
var out []byte
|
||||||
|
var err error
|
||||||
|
|
||||||
|
if *flagCompact == false {
|
||||||
|
out, err = json.MarshalIndent(j, "", " ")
|
||||||
|
} else {
|
||||||
|
out, err = json.Marshal(j)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err == nil {
|
||||||
|
fmt.Println(string(out))
|
||||||
|
}
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify a token and output the claims. This is a great example
|
||||||
|
// of how to verify and view a token.
|
||||||
|
func verifyToken() error {
|
||||||
|
// get the token
|
||||||
|
tokData, err := loadData(*flagVerify)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Couldn't read token: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// trim possible whitespace from token
|
||||||
|
tokData = regexp.MustCompile(`\s*$`).ReplaceAll(tokData, []byte{})
|
||||||
|
if *flagDebug {
|
||||||
|
fmt.Fprintf(os.Stderr, "Token len: %v bytes\n", len(tokData))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the token. Load the key from command line option
|
||||||
|
token, err := jwt.Parse(string(tokData), func(t *jwt.Token) (interface{}, error) {
|
||||||
|
data, err := loadData(*flagKey)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if isEs() {
|
||||||
|
return jwt.ParseECPublicKeyFromPEM(data)
|
||||||
|
} else if isRs() {
|
||||||
|
return jwt.ParseRSAPublicKeyFromPEM(data)
|
||||||
|
}
|
||||||
|
return data, nil
|
||||||
|
})
|
||||||
|
|
||||||
|
// Print some debug data
|
||||||
|
if *flagDebug && token != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "Header:\n%v\n", token.Header)
|
||||||
|
fmt.Fprintf(os.Stderr, "Claims:\n%v\n", token.Claims)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Print an error if we can't parse for some reason
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Couldn't parse token: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Is token invalid?
|
||||||
|
if !token.Valid {
|
||||||
|
return fmt.Errorf("Token is invalid")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Print the token details
|
||||||
|
if err := printJSON(token.Claims); err != nil {
|
||||||
|
return fmt.Errorf("Failed to output claims: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create, sign, and output a token. This is a great, simple example of
|
||||||
|
// how to use this library to create and sign a token.
|
||||||
|
func signToken() error {
|
||||||
|
// get the token data from command line arguments
|
||||||
|
tokData, err := loadData(*flagSign)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Couldn't read token: %v", err)
|
||||||
|
} else if *flagDebug {
|
||||||
|
fmt.Fprintf(os.Stderr, "Token: %v bytes", len(tokData))
|
||||||
|
}
|
||||||
|
|
||||||
|
// parse the JSON of the claims
|
||||||
|
var claims jwt.MapClaims
|
||||||
|
if err := json.Unmarshal(tokData, &claims); err != nil {
|
||||||
|
return fmt.Errorf("Couldn't parse claims JSON: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// add command line claims
|
||||||
|
if len(flagClaims) > 0 {
|
||||||
|
for k, v := range flagClaims {
|
||||||
|
claims[k] = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the key
|
||||||
|
var key interface{}
|
||||||
|
key, err = loadData(*flagKey)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Couldn't read key: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the signing alg
|
||||||
|
alg := jwt.GetSigningMethod(*flagAlg)
|
||||||
|
if alg == nil {
|
||||||
|
return fmt.Errorf("Couldn't find signing method: %v", *flagAlg)
|
||||||
|
}
|
||||||
|
|
||||||
|
// create a new token
|
||||||
|
token := jwt.NewWithClaims(alg, claims)
|
||||||
|
|
||||||
|
// add command line headers
|
||||||
|
if len(flagHead) > 0 {
|
||||||
|
for k, v := range flagHead {
|
||||||
|
token.Header[k] = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if isEs() {
|
||||||
|
if k, ok := key.([]byte); !ok {
|
||||||
|
return fmt.Errorf("Couldn't convert key data to key")
|
||||||
|
} else {
|
||||||
|
key, err = jwt.ParseECPrivateKeyFromPEM(k)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if isRs() {
|
||||||
|
if k, ok := key.([]byte); !ok {
|
||||||
|
return fmt.Errorf("Couldn't convert key data to key")
|
||||||
|
} else {
|
||||||
|
key, err = jwt.ParseRSAPrivateKeyFromPEM(k)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if out, err := token.SignedString(key); err == nil {
|
||||||
|
fmt.Println(out)
|
||||||
|
} else {
|
||||||
|
return fmt.Errorf("Error signing token: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// showToken pretty-prints the token on the command line.
|
||||||
|
func showToken() error {
|
||||||
|
// get the token
|
||||||
|
tokData, err := loadData(*flagShow)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("Couldn't read token: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// trim possible whitespace from token
|
||||||
|
tokData = regexp.MustCompile(`\s*$`).ReplaceAll(tokData, []byte{})
|
||||||
|
if *flagDebug {
|
||||||
|
fmt.Fprintf(os.Stderr, "Token len: %v bytes\n", len(tokData))
|
||||||
|
}
|
||||||
|
|
||||||
|
token, err := jwt.Parse(string(tokData), nil)
|
||||||
|
if token == nil {
|
||||||
|
return fmt.Errorf("malformed token: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Print the token details
|
||||||
|
fmt.Println("Header:")
|
||||||
|
if err := printJSON(token.Header); err != nil {
|
||||||
|
return fmt.Errorf("Failed to output header: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println("Claims:")
|
||||||
|
if err := printJSON(token.Claims); err != nil {
|
||||||
|
return fmt.Errorf("Failed to output claims: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func isEs() bool {
|
||||||
|
return strings.HasPrefix(*flagAlg, "ES")
|
||||||
|
}
|
||||||
|
|
||||||
|
func isRs() bool {
|
||||||
|
return strings.HasPrefix(*flagAlg, "RS")
|
||||||
|
}
|
23
vendor/github.com/dgrijalva/jwt-go/cmd/jwt/args.go
generated
vendored
Normal file
23
vendor/github.com/dgrijalva/jwt-go/cmd/jwt/args.go
generated
vendored
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ArgList map[string]string
|
||||||
|
|
||||||
|
func (l ArgList) String() string {
|
||||||
|
data, _ := json.Marshal(l)
|
||||||
|
return string(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l ArgList) Set(arg string) error {
|
||||||
|
parts := strings.SplitN(arg, "=", 2)
|
||||||
|
if len(parts) != 2 {
|
||||||
|
return fmt.Errorf("Invalid argument '%v'. Must use format 'key=value'. %v", arg, parts)
|
||||||
|
}
|
||||||
|
l[parts[0]] = parts[1]
|
||||||
|
return nil
|
||||||
|
}
|
4
vendor/github.com/dgrijalva/jwt-go/doc.go
generated
vendored
Normal file
4
vendor/github.com/dgrijalva/jwt-go/doc.go
generated
vendored
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
// Package jwt is a Go implementation of JSON Web Tokens: http://self-issued.info/docs/draft-jones-json-web-token.html
|
||||||
|
//
|
||||||
|
// See README.md for more info.
|
||||||
|
package jwt
|
147
vendor/github.com/dgrijalva/jwt-go/ecdsa.go
generated
vendored
Normal file
147
vendor/github.com/dgrijalva/jwt-go/ecdsa.go
generated
vendored
Normal file
@ -0,0 +1,147 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto"
|
||||||
|
"crypto/ecdsa"
|
||||||
|
"crypto/rand"
|
||||||
|
"errors"
|
||||||
|
"math/big"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// Sadly this is missing from crypto/ecdsa compared to crypto/rsa
|
||||||
|
ErrECDSAVerification = errors.New("crypto/ecdsa: verification error")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Implements the ECDSA family of signing methods signing methods
|
||||||
|
type SigningMethodECDSA struct {
|
||||||
|
Name string
|
||||||
|
Hash crypto.Hash
|
||||||
|
KeySize int
|
||||||
|
CurveBits int
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specific instances for EC256 and company
|
||||||
|
var (
|
||||||
|
SigningMethodES256 *SigningMethodECDSA
|
||||||
|
SigningMethodES384 *SigningMethodECDSA
|
||||||
|
SigningMethodES512 *SigningMethodECDSA
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// ES256
|
||||||
|
SigningMethodES256 = &SigningMethodECDSA{"ES256", crypto.SHA256, 32, 256}
|
||||||
|
RegisterSigningMethod(SigningMethodES256.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodES256
|
||||||
|
})
|
||||||
|
|
||||||
|
// ES384
|
||||||
|
SigningMethodES384 = &SigningMethodECDSA{"ES384", crypto.SHA384, 48, 384}
|
||||||
|
RegisterSigningMethod(SigningMethodES384.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodES384
|
||||||
|
})
|
||||||
|
|
||||||
|
// ES512
|
||||||
|
SigningMethodES512 = &SigningMethodECDSA{"ES512", crypto.SHA512, 66, 521}
|
||||||
|
RegisterSigningMethod(SigningMethodES512.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodES512
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *SigningMethodECDSA) Alg() string {
|
||||||
|
return m.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Verify method from SigningMethod
|
||||||
|
// For this verify method, key must be an ecdsa.PublicKey struct
|
||||||
|
func (m *SigningMethodECDSA) Verify(signingString, signature string, key interface{}) error {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Decode the signature
|
||||||
|
var sig []byte
|
||||||
|
if sig, err = DecodeSegment(signature); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the key
|
||||||
|
var ecdsaKey *ecdsa.PublicKey
|
||||||
|
switch k := key.(type) {
|
||||||
|
case *ecdsa.PublicKey:
|
||||||
|
ecdsaKey = k
|
||||||
|
default:
|
||||||
|
return ErrInvalidKeyType
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(sig) != 2*m.KeySize {
|
||||||
|
return ErrECDSAVerification
|
||||||
|
}
|
||||||
|
|
||||||
|
r := big.NewInt(0).SetBytes(sig[:m.KeySize])
|
||||||
|
s := big.NewInt(0).SetBytes(sig[m.KeySize:])
|
||||||
|
|
||||||
|
// Create hasher
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return ErrHashUnavailable
|
||||||
|
}
|
||||||
|
hasher := m.Hash.New()
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
// Verify the signature
|
||||||
|
if verifystatus := ecdsa.Verify(ecdsaKey, hasher.Sum(nil), r, s); verifystatus == true {
|
||||||
|
return nil
|
||||||
|
} else {
|
||||||
|
return ErrECDSAVerification
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Sign method from SigningMethod
|
||||||
|
// For this signing method, key must be an ecdsa.PrivateKey struct
|
||||||
|
func (m *SigningMethodECDSA) Sign(signingString string, key interface{}) (string, error) {
|
||||||
|
// Get the key
|
||||||
|
var ecdsaKey *ecdsa.PrivateKey
|
||||||
|
switch k := key.(type) {
|
||||||
|
case *ecdsa.PrivateKey:
|
||||||
|
ecdsaKey = k
|
||||||
|
default:
|
||||||
|
return "", ErrInvalidKeyType
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create the hasher
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return "", ErrHashUnavailable
|
||||||
|
}
|
||||||
|
|
||||||
|
hasher := m.Hash.New()
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
// Sign the string and return r, s
|
||||||
|
if r, s, err := ecdsa.Sign(rand.Reader, ecdsaKey, hasher.Sum(nil)); err == nil {
|
||||||
|
curveBits := ecdsaKey.Curve.Params().BitSize
|
||||||
|
|
||||||
|
if m.CurveBits != curveBits {
|
||||||
|
return "", ErrInvalidKey
|
||||||
|
}
|
||||||
|
|
||||||
|
keyBytes := curveBits / 8
|
||||||
|
if curveBits%8 > 0 {
|
||||||
|
keyBytes += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
// We serialize the outpus (r and s) into big-endian byte arrays and pad
|
||||||
|
// them with zeros on the left to make sure the sizes work out. Both arrays
|
||||||
|
// must be keyBytes long, and the output must be 2*keyBytes long.
|
||||||
|
rBytes := r.Bytes()
|
||||||
|
rBytesPadded := make([]byte, keyBytes)
|
||||||
|
copy(rBytesPadded[keyBytes-len(rBytes):], rBytes)
|
||||||
|
|
||||||
|
sBytes := s.Bytes()
|
||||||
|
sBytesPadded := make([]byte, keyBytes)
|
||||||
|
copy(sBytesPadded[keyBytes-len(sBytes):], sBytes)
|
||||||
|
|
||||||
|
out := append(rBytesPadded, sBytesPadded...)
|
||||||
|
|
||||||
|
return EncodeSegment(out), nil
|
||||||
|
} else {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
67
vendor/github.com/dgrijalva/jwt-go/ecdsa_utils.go
generated
vendored
Normal file
67
vendor/github.com/dgrijalva/jwt-go/ecdsa_utils.go
generated
vendored
Normal file
@ -0,0 +1,67 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/ecdsa"
|
||||||
|
"crypto/x509"
|
||||||
|
"encoding/pem"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrNotECPublicKey = errors.New("Key is not a valid ECDSA public key")
|
||||||
|
ErrNotECPrivateKey = errors.New("Key is not a valid ECDSA private key")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Parse PEM encoded Elliptic Curve Private Key Structure
|
||||||
|
func ParseECPrivateKeyFromPEM(key []byte) (*ecdsa.PrivateKey, error) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Parse PEM block
|
||||||
|
var block *pem.Block
|
||||||
|
if block, _ = pem.Decode(key); block == nil {
|
||||||
|
return nil, ErrKeyMustBePEMEncoded
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the key
|
||||||
|
var parsedKey interface{}
|
||||||
|
if parsedKey, err = x509.ParseECPrivateKey(block.Bytes); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var pkey *ecdsa.PrivateKey
|
||||||
|
var ok bool
|
||||||
|
if pkey, ok = parsedKey.(*ecdsa.PrivateKey); !ok {
|
||||||
|
return nil, ErrNotECPrivateKey
|
||||||
|
}
|
||||||
|
|
||||||
|
return pkey, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse PEM encoded PKCS1 or PKCS8 public key
|
||||||
|
func ParseECPublicKeyFromPEM(key []byte) (*ecdsa.PublicKey, error) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Parse PEM block
|
||||||
|
var block *pem.Block
|
||||||
|
if block, _ = pem.Decode(key); block == nil {
|
||||||
|
return nil, ErrKeyMustBePEMEncoded
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the key
|
||||||
|
var parsedKey interface{}
|
||||||
|
if parsedKey, err = x509.ParsePKIXPublicKey(block.Bytes); err != nil {
|
||||||
|
if cert, err := x509.ParseCertificate(block.Bytes); err == nil {
|
||||||
|
parsedKey = cert.PublicKey
|
||||||
|
} else {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var pkey *ecdsa.PublicKey
|
||||||
|
var ok bool
|
||||||
|
if pkey, ok = parsedKey.(*ecdsa.PublicKey); !ok {
|
||||||
|
return nil, ErrNotECPublicKey
|
||||||
|
}
|
||||||
|
|
||||||
|
return pkey, nil
|
||||||
|
}
|
59
vendor/github.com/dgrijalva/jwt-go/errors.go
generated
vendored
Normal file
59
vendor/github.com/dgrijalva/jwt-go/errors.go
generated
vendored
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Error constants
|
||||||
|
var (
|
||||||
|
ErrInvalidKey = errors.New("key is invalid")
|
||||||
|
ErrInvalidKeyType = errors.New("key is of invalid type")
|
||||||
|
ErrHashUnavailable = errors.New("the requested hash function is unavailable")
|
||||||
|
)
|
||||||
|
|
||||||
|
// The errors that might occur when parsing and validating a token
|
||||||
|
const (
|
||||||
|
ValidationErrorMalformed uint32 = 1 << iota // Token is malformed
|
||||||
|
ValidationErrorUnverifiable // Token could not be verified because of signing problems
|
||||||
|
ValidationErrorSignatureInvalid // Signature validation failed
|
||||||
|
|
||||||
|
// Standard Claim validation errors
|
||||||
|
ValidationErrorAudience // AUD validation failed
|
||||||
|
ValidationErrorExpired // EXP validation failed
|
||||||
|
ValidationErrorIssuedAt // IAT validation failed
|
||||||
|
ValidationErrorIssuer // ISS validation failed
|
||||||
|
ValidationErrorNotValidYet // NBF validation failed
|
||||||
|
ValidationErrorId // JTI validation failed
|
||||||
|
ValidationErrorClaimsInvalid // Generic claims validation error
|
||||||
|
)
|
||||||
|
|
||||||
|
// Helper for constructing a ValidationError with a string error message
|
||||||
|
func NewValidationError(errorText string, errorFlags uint32) *ValidationError {
|
||||||
|
return &ValidationError{
|
||||||
|
text: errorText,
|
||||||
|
Errors: errorFlags,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// The error from Parse if token is not valid
|
||||||
|
type ValidationError struct {
|
||||||
|
Inner error // stores the error returned by external dependencies, i.e.: KeyFunc
|
||||||
|
Errors uint32 // bitfield. see ValidationError... constants
|
||||||
|
text string // errors that do not have a valid error just have text
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validation error is an error type
|
||||||
|
func (e ValidationError) Error() string {
|
||||||
|
if e.Inner != nil {
|
||||||
|
return e.Inner.Error()
|
||||||
|
} else if e.text != "" {
|
||||||
|
return e.text
|
||||||
|
} else {
|
||||||
|
return "token is invalid"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// No errors
|
||||||
|
func (e *ValidationError) valid() bool {
|
||||||
|
return e.Errors == 0
|
||||||
|
}
|
94
vendor/github.com/dgrijalva/jwt-go/hmac.go
generated
vendored
Normal file
94
vendor/github.com/dgrijalva/jwt-go/hmac.go
generated
vendored
Normal file
@ -0,0 +1,94 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto"
|
||||||
|
"crypto/hmac"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Implements the HMAC-SHA family of signing methods signing methods
|
||||||
|
type SigningMethodHMAC struct {
|
||||||
|
Name string
|
||||||
|
Hash crypto.Hash
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specific instances for HS256 and company
|
||||||
|
var (
|
||||||
|
SigningMethodHS256 *SigningMethodHMAC
|
||||||
|
SigningMethodHS384 *SigningMethodHMAC
|
||||||
|
SigningMethodHS512 *SigningMethodHMAC
|
||||||
|
ErrSignatureInvalid = errors.New("signature is invalid")
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// HS256
|
||||||
|
SigningMethodHS256 = &SigningMethodHMAC{"HS256", crypto.SHA256}
|
||||||
|
RegisterSigningMethod(SigningMethodHS256.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodHS256
|
||||||
|
})
|
||||||
|
|
||||||
|
// HS384
|
||||||
|
SigningMethodHS384 = &SigningMethodHMAC{"HS384", crypto.SHA384}
|
||||||
|
RegisterSigningMethod(SigningMethodHS384.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodHS384
|
||||||
|
})
|
||||||
|
|
||||||
|
// HS512
|
||||||
|
SigningMethodHS512 = &SigningMethodHMAC{"HS512", crypto.SHA512}
|
||||||
|
RegisterSigningMethod(SigningMethodHS512.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodHS512
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *SigningMethodHMAC) Alg() string {
|
||||||
|
return m.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify the signature of HSXXX tokens. Returns nil if the signature is valid.
|
||||||
|
func (m *SigningMethodHMAC) Verify(signingString, signature string, key interface{}) error {
|
||||||
|
// Verify the key is the right type
|
||||||
|
keyBytes, ok := key.([]byte)
|
||||||
|
if !ok {
|
||||||
|
return ErrInvalidKeyType
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode signature, for comparison
|
||||||
|
sig, err := DecodeSegment(signature)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Can we use the specified hashing method?
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return ErrHashUnavailable
|
||||||
|
}
|
||||||
|
|
||||||
|
// This signing method is symmetric, so we validate the signature
|
||||||
|
// by reproducing the signature from the signing string and key, then
|
||||||
|
// comparing that against the provided signature.
|
||||||
|
hasher := hmac.New(m.Hash.New, keyBytes)
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
if !hmac.Equal(sig, hasher.Sum(nil)) {
|
||||||
|
return ErrSignatureInvalid
|
||||||
|
}
|
||||||
|
|
||||||
|
// No validation errors. Signature is good.
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Sign method from SigningMethod for this signing method.
|
||||||
|
// Key must be []byte
|
||||||
|
func (m *SigningMethodHMAC) Sign(signingString string, key interface{}) (string, error) {
|
||||||
|
if keyBytes, ok := key.([]byte); ok {
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return "", ErrHashUnavailable
|
||||||
|
}
|
||||||
|
|
||||||
|
hasher := hmac.New(m.Hash.New, keyBytes)
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
return EncodeSegment(hasher.Sum(nil)), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", ErrInvalidKey
|
||||||
|
}
|
94
vendor/github.com/dgrijalva/jwt-go/map_claims.go
generated
vendored
Normal file
94
vendor/github.com/dgrijalva/jwt-go/map_claims.go
generated
vendored
Normal file
@ -0,0 +1,94 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
// "fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Claims type that uses the map[string]interface{} for JSON decoding
|
||||||
|
// This is the default claims type if you don't supply one
|
||||||
|
type MapClaims map[string]interface{}
|
||||||
|
|
||||||
|
// Compares the aud claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (m MapClaims) VerifyAudience(cmp string, req bool) bool {
|
||||||
|
aud, _ := m["aud"].(string)
|
||||||
|
return verifyAud(aud, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the exp claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (m MapClaims) VerifyExpiresAt(cmp int64, req bool) bool {
|
||||||
|
switch exp := m["exp"].(type) {
|
||||||
|
case float64:
|
||||||
|
return verifyExp(int64(exp), cmp, req)
|
||||||
|
case json.Number:
|
||||||
|
v, _ := exp.Int64()
|
||||||
|
return verifyExp(v, cmp, req)
|
||||||
|
}
|
||||||
|
return req == false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the iat claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (m MapClaims) VerifyIssuedAt(cmp int64, req bool) bool {
|
||||||
|
switch iat := m["iat"].(type) {
|
||||||
|
case float64:
|
||||||
|
return verifyIat(int64(iat), cmp, req)
|
||||||
|
case json.Number:
|
||||||
|
v, _ := iat.Int64()
|
||||||
|
return verifyIat(v, cmp, req)
|
||||||
|
}
|
||||||
|
return req == false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the iss claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (m MapClaims) VerifyIssuer(cmp string, req bool) bool {
|
||||||
|
iss, _ := m["iss"].(string)
|
||||||
|
return verifyIss(iss, cmp, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compares the nbf claim against cmp.
|
||||||
|
// If required is false, this method will return true if the value matches or is unset
|
||||||
|
func (m MapClaims) VerifyNotBefore(cmp int64, req bool) bool {
|
||||||
|
switch nbf := m["nbf"].(type) {
|
||||||
|
case float64:
|
||||||
|
return verifyNbf(int64(nbf), cmp, req)
|
||||||
|
case json.Number:
|
||||||
|
v, _ := nbf.Int64()
|
||||||
|
return verifyNbf(v, cmp, req)
|
||||||
|
}
|
||||||
|
return req == false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validates time based claims "exp, iat, nbf".
|
||||||
|
// There is no accounting for clock skew.
|
||||||
|
// As well, if any of the above claims are not in the token, it will still
|
||||||
|
// be considered a valid claim.
|
||||||
|
func (m MapClaims) Valid() error {
|
||||||
|
vErr := new(ValidationError)
|
||||||
|
now := TimeFunc().Unix()
|
||||||
|
|
||||||
|
if m.VerifyExpiresAt(now, false) == false {
|
||||||
|
vErr.Inner = errors.New("Token is expired")
|
||||||
|
vErr.Errors |= ValidationErrorExpired
|
||||||
|
}
|
||||||
|
|
||||||
|
if m.VerifyIssuedAt(now, false) == false {
|
||||||
|
vErr.Inner = errors.New("Token used before issued")
|
||||||
|
vErr.Errors |= ValidationErrorIssuedAt
|
||||||
|
}
|
||||||
|
|
||||||
|
if m.VerifyNotBefore(now, false) == false {
|
||||||
|
vErr.Inner = errors.New("Token is not valid yet")
|
||||||
|
vErr.Errors |= ValidationErrorNotValidYet
|
||||||
|
}
|
||||||
|
|
||||||
|
if vErr.valid() {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return vErr
|
||||||
|
}
|
52
vendor/github.com/dgrijalva/jwt-go/none.go
generated
vendored
Normal file
52
vendor/github.com/dgrijalva/jwt-go/none.go
generated
vendored
Normal file
@ -0,0 +1,52 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
// Implements the none signing method. This is required by the spec
|
||||||
|
// but you probably should never use it.
|
||||||
|
var SigningMethodNone *signingMethodNone
|
||||||
|
|
||||||
|
const UnsafeAllowNoneSignatureType unsafeNoneMagicConstant = "none signing method allowed"
|
||||||
|
|
||||||
|
var NoneSignatureTypeDisallowedError error
|
||||||
|
|
||||||
|
type signingMethodNone struct{}
|
||||||
|
type unsafeNoneMagicConstant string
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
SigningMethodNone = &signingMethodNone{}
|
||||||
|
NoneSignatureTypeDisallowedError = NewValidationError("'none' signature type is not allowed", ValidationErrorSignatureInvalid)
|
||||||
|
|
||||||
|
RegisterSigningMethod(SigningMethodNone.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodNone
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *signingMethodNone) Alg() string {
|
||||||
|
return "none"
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only allow 'none' alg type if UnsafeAllowNoneSignatureType is specified as the key
|
||||||
|
func (m *signingMethodNone) Verify(signingString, signature string, key interface{}) (err error) {
|
||||||
|
// Key must be UnsafeAllowNoneSignatureType to prevent accidentally
|
||||||
|
// accepting 'none' signing method
|
||||||
|
if _, ok := key.(unsafeNoneMagicConstant); !ok {
|
||||||
|
return NoneSignatureTypeDisallowedError
|
||||||
|
}
|
||||||
|
// If signing method is none, signature must be an empty string
|
||||||
|
if signature != "" {
|
||||||
|
return NewValidationError(
|
||||||
|
"'none' signing method with non-empty signature",
|
||||||
|
ValidationErrorSignatureInvalid,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Accept 'none' signing method.
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Only allow 'none' signing if UnsafeAllowNoneSignatureType is specified as the key
|
||||||
|
func (m *signingMethodNone) Sign(signingString string, key interface{}) (string, error) {
|
||||||
|
if _, ok := key.(unsafeNoneMagicConstant); ok {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
return "", NoneSignatureTypeDisallowedError
|
||||||
|
}
|
131
vendor/github.com/dgrijalva/jwt-go/parser.go
generated
vendored
Normal file
131
vendor/github.com/dgrijalva/jwt-go/parser.go
generated
vendored
Normal file
@ -0,0 +1,131 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Parser struct {
|
||||||
|
ValidMethods []string // If populated, only these methods will be considered valid
|
||||||
|
UseJSONNumber bool // Use JSON Number format in JSON decoder
|
||||||
|
SkipClaimsValidation bool // Skip claims validation during token parsing
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse, validate, and return a token.
|
||||||
|
// keyFunc will receive the parsed token and should return the key for validating.
|
||||||
|
// If everything is kosher, err will be nil
|
||||||
|
func (p *Parser) Parse(tokenString string, keyFunc Keyfunc) (*Token, error) {
|
||||||
|
return p.ParseWithClaims(tokenString, MapClaims{}, keyFunc)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Parser) ParseWithClaims(tokenString string, claims Claims, keyFunc Keyfunc) (*Token, error) {
|
||||||
|
parts := strings.Split(tokenString, ".")
|
||||||
|
if len(parts) != 3 {
|
||||||
|
return nil, NewValidationError("token contains an invalid number of segments", ValidationErrorMalformed)
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
token := &Token{Raw: tokenString}
|
||||||
|
|
||||||
|
// parse Header
|
||||||
|
var headerBytes []byte
|
||||||
|
if headerBytes, err = DecodeSegment(parts[0]); err != nil {
|
||||||
|
if strings.HasPrefix(strings.ToLower(tokenString), "bearer ") {
|
||||||
|
return token, NewValidationError("tokenstring should not contain 'bearer '", ValidationErrorMalformed)
|
||||||
|
}
|
||||||
|
return token, &ValidationError{Inner: err, Errors: ValidationErrorMalformed}
|
||||||
|
}
|
||||||
|
if err = json.Unmarshal(headerBytes, &token.Header); err != nil {
|
||||||
|
return token, &ValidationError{Inner: err, Errors: ValidationErrorMalformed}
|
||||||
|
}
|
||||||
|
|
||||||
|
// parse Claims
|
||||||
|
var claimBytes []byte
|
||||||
|
token.Claims = claims
|
||||||
|
|
||||||
|
if claimBytes, err = DecodeSegment(parts[1]); err != nil {
|
||||||
|
return token, &ValidationError{Inner: err, Errors: ValidationErrorMalformed}
|
||||||
|
}
|
||||||
|
dec := json.NewDecoder(bytes.NewBuffer(claimBytes))
|
||||||
|
if p.UseJSONNumber {
|
||||||
|
dec.UseNumber()
|
||||||
|
}
|
||||||
|
// JSON Decode. Special case for map type to avoid weird pointer behavior
|
||||||
|
if c, ok := token.Claims.(MapClaims); ok {
|
||||||
|
err = dec.Decode(&c)
|
||||||
|
} else {
|
||||||
|
err = dec.Decode(&claims)
|
||||||
|
}
|
||||||
|
// Handle decode error
|
||||||
|
if err != nil {
|
||||||
|
return token, &ValidationError{Inner: err, Errors: ValidationErrorMalformed}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Lookup signature method
|
||||||
|
if method, ok := token.Header["alg"].(string); ok {
|
||||||
|
if token.Method = GetSigningMethod(method); token.Method == nil {
|
||||||
|
return token, NewValidationError("signing method (alg) is unavailable.", ValidationErrorUnverifiable)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
return token, NewValidationError("signing method (alg) is unspecified.", ValidationErrorUnverifiable)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify signing method is in the required set
|
||||||
|
if p.ValidMethods != nil {
|
||||||
|
var signingMethodValid = false
|
||||||
|
var alg = token.Method.Alg()
|
||||||
|
for _, m := range p.ValidMethods {
|
||||||
|
if m == alg {
|
||||||
|
signingMethodValid = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !signingMethodValid {
|
||||||
|
// signing method is not in the listed set
|
||||||
|
return token, NewValidationError(fmt.Sprintf("signing method %v is invalid", alg), ValidationErrorSignatureInvalid)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Lookup key
|
||||||
|
var key interface{}
|
||||||
|
if keyFunc == nil {
|
||||||
|
// keyFunc was not provided. short circuiting validation
|
||||||
|
return token, NewValidationError("no Keyfunc was provided.", ValidationErrorUnverifiable)
|
||||||
|
}
|
||||||
|
if key, err = keyFunc(token); err != nil {
|
||||||
|
// keyFunc returned an error
|
||||||
|
return token, &ValidationError{Inner: err, Errors: ValidationErrorUnverifiable}
|
||||||
|
}
|
||||||
|
|
||||||
|
vErr := &ValidationError{}
|
||||||
|
|
||||||
|
// Validate Claims
|
||||||
|
if !p.SkipClaimsValidation {
|
||||||
|
if err := token.Claims.Valid(); err != nil {
|
||||||
|
|
||||||
|
// If the Claims Valid returned an error, check if it is a validation error,
|
||||||
|
// If it was another error type, create a ValidationError with a generic ClaimsInvalid flag set
|
||||||
|
if e, ok := err.(*ValidationError); !ok {
|
||||||
|
vErr = &ValidationError{Inner: err, Errors: ValidationErrorClaimsInvalid}
|
||||||
|
} else {
|
||||||
|
vErr = e
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Perform validation
|
||||||
|
token.Signature = parts[2]
|
||||||
|
if err = token.Method.Verify(strings.Join(parts[0:2], "."), token.Signature, key); err != nil {
|
||||||
|
vErr.Inner = err
|
||||||
|
vErr.Errors |= ValidationErrorSignatureInvalid
|
||||||
|
}
|
||||||
|
|
||||||
|
if vErr.valid() {
|
||||||
|
token.Valid = true
|
||||||
|
return token, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return token, vErr
|
||||||
|
}
|
7
vendor/github.com/dgrijalva/jwt-go/request/doc.go
generated
vendored
Normal file
7
vendor/github.com/dgrijalva/jwt-go/request/doc.go
generated
vendored
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
// Utility package for extracting JWT tokens from
|
||||||
|
// HTTP requests.
|
||||||
|
//
|
||||||
|
// The main function is ParseFromRequest and it's WithClaims variant.
|
||||||
|
// See examples for how to use the various Extractor implementations
|
||||||
|
// or roll your own.
|
||||||
|
package request
|
81
vendor/github.com/dgrijalva/jwt-go/request/extractor.go
generated
vendored
Normal file
81
vendor/github.com/dgrijalva/jwt-go/request/extractor.go
generated
vendored
Normal file
@ -0,0 +1,81 @@
|
|||||||
|
package request
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Errors
|
||||||
|
var (
|
||||||
|
ErrNoTokenInRequest = errors.New("no token present in request")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Interface for extracting a token from an HTTP request.
|
||||||
|
// The ExtractToken method should return a token string or an error.
|
||||||
|
// If no token is present, you must return ErrNoTokenInRequest.
|
||||||
|
type Extractor interface {
|
||||||
|
ExtractToken(*http.Request) (string, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extractor for finding a token in a header. Looks at each specified
|
||||||
|
// header in order until there's a match
|
||||||
|
type HeaderExtractor []string
|
||||||
|
|
||||||
|
func (e HeaderExtractor) ExtractToken(req *http.Request) (string, error) {
|
||||||
|
// loop over header names and return the first one that contains data
|
||||||
|
for _, header := range e {
|
||||||
|
if ah := req.Header.Get(header); ah != "" {
|
||||||
|
return ah, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", ErrNoTokenInRequest
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract token from request arguments. This includes a POSTed form or
|
||||||
|
// GET URL arguments. Argument names are tried in order until there's a match.
|
||||||
|
// This extractor calls `ParseMultipartForm` on the request
|
||||||
|
type ArgumentExtractor []string
|
||||||
|
|
||||||
|
func (e ArgumentExtractor) ExtractToken(req *http.Request) (string, error) {
|
||||||
|
// Make sure form is parsed
|
||||||
|
req.ParseMultipartForm(10e6)
|
||||||
|
|
||||||
|
// loop over arg names and return the first one that contains data
|
||||||
|
for _, arg := range e {
|
||||||
|
if ah := req.Form.Get(arg); ah != "" {
|
||||||
|
return ah, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", ErrNoTokenInRequest
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tries Extractors in order until one returns a token string or an error occurs
|
||||||
|
type MultiExtractor []Extractor
|
||||||
|
|
||||||
|
func (e MultiExtractor) ExtractToken(req *http.Request) (string, error) {
|
||||||
|
// loop over header names and return the first one that contains data
|
||||||
|
for _, extractor := range e {
|
||||||
|
if tok, err := extractor.ExtractToken(req); tok != "" {
|
||||||
|
return tok, nil
|
||||||
|
} else if err != ErrNoTokenInRequest {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", ErrNoTokenInRequest
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wrap an Extractor in this to post-process the value before it's handed off.
|
||||||
|
// See AuthorizationHeaderExtractor for an example
|
||||||
|
type PostExtractionFilter struct {
|
||||||
|
Extractor
|
||||||
|
Filter func(string) (string, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *PostExtractionFilter) ExtractToken(req *http.Request) (string, error) {
|
||||||
|
if tok, err := e.Extractor.ExtractToken(req); tok != "" {
|
||||||
|
return e.Filter(tok)
|
||||||
|
} else {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
28
vendor/github.com/dgrijalva/jwt-go/request/oauth2.go
generated
vendored
Normal file
28
vendor/github.com/dgrijalva/jwt-go/request/oauth2.go
generated
vendored
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
package request
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Strips 'Bearer ' prefix from bearer token string
|
||||||
|
func stripBearerPrefixFromTokenString(tok string) (string, error) {
|
||||||
|
// Should be a bearer token
|
||||||
|
if len(tok) > 6 && strings.ToUpper(tok[0:7]) == "BEARER " {
|
||||||
|
return tok[7:], nil
|
||||||
|
}
|
||||||
|
return tok, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract bearer token from Authorization header
|
||||||
|
// Uses PostExtractionFilter to strip "Bearer " prefix from header
|
||||||
|
var AuthorizationHeaderExtractor = &PostExtractionFilter{
|
||||||
|
HeaderExtractor{"Authorization"},
|
||||||
|
stripBearerPrefixFromTokenString,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extractor for OAuth2 access tokens. Looks in 'Authorization'
|
||||||
|
// header then 'access_token' argument for a token.
|
||||||
|
var OAuth2Extractor = &MultiExtractor{
|
||||||
|
AuthorizationHeaderExtractor,
|
||||||
|
ArgumentExtractor{"access_token"},
|
||||||
|
}
|
24
vendor/github.com/dgrijalva/jwt-go/request/request.go
generated
vendored
Normal file
24
vendor/github.com/dgrijalva/jwt-go/request/request.go
generated
vendored
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
package request
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/dgrijalva/jwt-go"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Extract and parse a JWT token from an HTTP request.
|
||||||
|
// This behaves the same as Parse, but accepts a request and an extractor
|
||||||
|
// instead of a token string. The Extractor interface allows you to define
|
||||||
|
// the logic for extracting a token. Several useful implementations are provided.
|
||||||
|
func ParseFromRequest(req *http.Request, extractor Extractor, keyFunc jwt.Keyfunc) (token *jwt.Token, err error) {
|
||||||
|
return ParseFromRequestWithClaims(req, extractor, jwt.MapClaims{}, keyFunc)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseFromRequest but with custom Claims type
|
||||||
|
func ParseFromRequestWithClaims(req *http.Request, extractor Extractor, claims jwt.Claims, keyFunc jwt.Keyfunc) (token *jwt.Token, err error) {
|
||||||
|
// Extract token from request
|
||||||
|
if tokStr, err := extractor.ExtractToken(req); err == nil {
|
||||||
|
return jwt.ParseWithClaims(tokStr, claims, keyFunc)
|
||||||
|
} else {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
100
vendor/github.com/dgrijalva/jwt-go/rsa.go
generated
vendored
Normal file
100
vendor/github.com/dgrijalva/jwt-go/rsa.go
generated
vendored
Normal file
@ -0,0 +1,100 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto"
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/rsa"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Implements the RSA family of signing methods signing methods
|
||||||
|
type SigningMethodRSA struct {
|
||||||
|
Name string
|
||||||
|
Hash crypto.Hash
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specific instances for RS256 and company
|
||||||
|
var (
|
||||||
|
SigningMethodRS256 *SigningMethodRSA
|
||||||
|
SigningMethodRS384 *SigningMethodRSA
|
||||||
|
SigningMethodRS512 *SigningMethodRSA
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// RS256
|
||||||
|
SigningMethodRS256 = &SigningMethodRSA{"RS256", crypto.SHA256}
|
||||||
|
RegisterSigningMethod(SigningMethodRS256.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodRS256
|
||||||
|
})
|
||||||
|
|
||||||
|
// RS384
|
||||||
|
SigningMethodRS384 = &SigningMethodRSA{"RS384", crypto.SHA384}
|
||||||
|
RegisterSigningMethod(SigningMethodRS384.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodRS384
|
||||||
|
})
|
||||||
|
|
||||||
|
// RS512
|
||||||
|
SigningMethodRS512 = &SigningMethodRSA{"RS512", crypto.SHA512}
|
||||||
|
RegisterSigningMethod(SigningMethodRS512.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodRS512
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *SigningMethodRSA) Alg() string {
|
||||||
|
return m.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Verify method from SigningMethod
|
||||||
|
// For this signing method, must be an rsa.PublicKey structure.
|
||||||
|
func (m *SigningMethodRSA) Verify(signingString, signature string, key interface{}) error {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Decode the signature
|
||||||
|
var sig []byte
|
||||||
|
if sig, err = DecodeSegment(signature); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var rsaKey *rsa.PublicKey
|
||||||
|
var ok bool
|
||||||
|
|
||||||
|
if rsaKey, ok = key.(*rsa.PublicKey); !ok {
|
||||||
|
return ErrInvalidKeyType
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create hasher
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return ErrHashUnavailable
|
||||||
|
}
|
||||||
|
hasher := m.Hash.New()
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
// Verify the signature
|
||||||
|
return rsa.VerifyPKCS1v15(rsaKey, m.Hash, hasher.Sum(nil), sig)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Sign method from SigningMethod
|
||||||
|
// For this signing method, must be an rsa.PrivateKey structure.
|
||||||
|
func (m *SigningMethodRSA) Sign(signingString string, key interface{}) (string, error) {
|
||||||
|
var rsaKey *rsa.PrivateKey
|
||||||
|
var ok bool
|
||||||
|
|
||||||
|
// Validate type of key
|
||||||
|
if rsaKey, ok = key.(*rsa.PrivateKey); !ok {
|
||||||
|
return "", ErrInvalidKey
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create the hasher
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return "", ErrHashUnavailable
|
||||||
|
}
|
||||||
|
|
||||||
|
hasher := m.Hash.New()
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
// Sign the string and return the encoded bytes
|
||||||
|
if sigBytes, err := rsa.SignPKCS1v15(rand.Reader, rsaKey, m.Hash, hasher.Sum(nil)); err == nil {
|
||||||
|
return EncodeSegment(sigBytes), nil
|
||||||
|
} else {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
126
vendor/github.com/dgrijalva/jwt-go/rsa_pss.go
generated
vendored
Normal file
126
vendor/github.com/dgrijalva/jwt-go/rsa_pss.go
generated
vendored
Normal file
@ -0,0 +1,126 @@
|
|||||||
|
// +build go1.4
|
||||||
|
|
||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto"
|
||||||
|
"crypto/rand"
|
||||||
|
"crypto/rsa"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Implements the RSAPSS family of signing methods signing methods
|
||||||
|
type SigningMethodRSAPSS struct {
|
||||||
|
*SigningMethodRSA
|
||||||
|
Options *rsa.PSSOptions
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specific instances for RS/PS and company
|
||||||
|
var (
|
||||||
|
SigningMethodPS256 *SigningMethodRSAPSS
|
||||||
|
SigningMethodPS384 *SigningMethodRSAPSS
|
||||||
|
SigningMethodPS512 *SigningMethodRSAPSS
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// PS256
|
||||||
|
SigningMethodPS256 = &SigningMethodRSAPSS{
|
||||||
|
&SigningMethodRSA{
|
||||||
|
Name: "PS256",
|
||||||
|
Hash: crypto.SHA256,
|
||||||
|
},
|
||||||
|
&rsa.PSSOptions{
|
||||||
|
SaltLength: rsa.PSSSaltLengthAuto,
|
||||||
|
Hash: crypto.SHA256,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
RegisterSigningMethod(SigningMethodPS256.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodPS256
|
||||||
|
})
|
||||||
|
|
||||||
|
// PS384
|
||||||
|
SigningMethodPS384 = &SigningMethodRSAPSS{
|
||||||
|
&SigningMethodRSA{
|
||||||
|
Name: "PS384",
|
||||||
|
Hash: crypto.SHA384,
|
||||||
|
},
|
||||||
|
&rsa.PSSOptions{
|
||||||
|
SaltLength: rsa.PSSSaltLengthAuto,
|
||||||
|
Hash: crypto.SHA384,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
RegisterSigningMethod(SigningMethodPS384.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodPS384
|
||||||
|
})
|
||||||
|
|
||||||
|
// PS512
|
||||||
|
SigningMethodPS512 = &SigningMethodRSAPSS{
|
||||||
|
&SigningMethodRSA{
|
||||||
|
Name: "PS512",
|
||||||
|
Hash: crypto.SHA512,
|
||||||
|
},
|
||||||
|
&rsa.PSSOptions{
|
||||||
|
SaltLength: rsa.PSSSaltLengthAuto,
|
||||||
|
Hash: crypto.SHA512,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
RegisterSigningMethod(SigningMethodPS512.Alg(), func() SigningMethod {
|
||||||
|
return SigningMethodPS512
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Verify method from SigningMethod
|
||||||
|
// For this verify method, key must be an rsa.PublicKey struct
|
||||||
|
func (m *SigningMethodRSAPSS) Verify(signingString, signature string, key interface{}) error {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Decode the signature
|
||||||
|
var sig []byte
|
||||||
|
if sig, err = DecodeSegment(signature); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var rsaKey *rsa.PublicKey
|
||||||
|
switch k := key.(type) {
|
||||||
|
case *rsa.PublicKey:
|
||||||
|
rsaKey = k
|
||||||
|
default:
|
||||||
|
return ErrInvalidKey
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create hasher
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return ErrHashUnavailable
|
||||||
|
}
|
||||||
|
hasher := m.Hash.New()
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
return rsa.VerifyPSS(rsaKey, m.Hash, hasher.Sum(nil), sig, m.Options)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Implements the Sign method from SigningMethod
|
||||||
|
// For this signing method, key must be an rsa.PrivateKey struct
|
||||||
|
func (m *SigningMethodRSAPSS) Sign(signingString string, key interface{}) (string, error) {
|
||||||
|
var rsaKey *rsa.PrivateKey
|
||||||
|
|
||||||
|
switch k := key.(type) {
|
||||||
|
case *rsa.PrivateKey:
|
||||||
|
rsaKey = k
|
||||||
|
default:
|
||||||
|
return "", ErrInvalidKeyType
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create the hasher
|
||||||
|
if !m.Hash.Available() {
|
||||||
|
return "", ErrHashUnavailable
|
||||||
|
}
|
||||||
|
|
||||||
|
hasher := m.Hash.New()
|
||||||
|
hasher.Write([]byte(signingString))
|
||||||
|
|
||||||
|
// Sign the string and return the encoded bytes
|
||||||
|
if sigBytes, err := rsa.SignPSS(rand.Reader, rsaKey, m.Hash, hasher.Sum(nil), m.Options); err == nil {
|
||||||
|
return EncodeSegment(sigBytes), nil
|
||||||
|
} else {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
69
vendor/github.com/dgrijalva/jwt-go/rsa_utils.go
generated
vendored
Normal file
69
vendor/github.com/dgrijalva/jwt-go/rsa_utils.go
generated
vendored
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rsa"
|
||||||
|
"crypto/x509"
|
||||||
|
"encoding/pem"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrKeyMustBePEMEncoded = errors.New("Invalid Key: Key must be PEM encoded PKCS1 or PKCS8 private key")
|
||||||
|
ErrNotRSAPrivateKey = errors.New("Key is not a valid RSA private key")
|
||||||
|
ErrNotRSAPublicKey = errors.New("Key is not a valid RSA public key")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Parse PEM encoded PKCS1 or PKCS8 private key
|
||||||
|
func ParseRSAPrivateKeyFromPEM(key []byte) (*rsa.PrivateKey, error) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Parse PEM block
|
||||||
|
var block *pem.Block
|
||||||
|
if block, _ = pem.Decode(key); block == nil {
|
||||||
|
return nil, ErrKeyMustBePEMEncoded
|
||||||
|
}
|
||||||
|
|
||||||
|
var parsedKey interface{}
|
||||||
|
if parsedKey, err = x509.ParsePKCS1PrivateKey(block.Bytes); err != nil {
|
||||||
|
if parsedKey, err = x509.ParsePKCS8PrivateKey(block.Bytes); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var pkey *rsa.PrivateKey
|
||||||
|
var ok bool
|
||||||
|
if pkey, ok = parsedKey.(*rsa.PrivateKey); !ok {
|
||||||
|
return nil, ErrNotRSAPrivateKey
|
||||||
|
}
|
||||||
|
|
||||||
|
return pkey, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse PEM encoded PKCS1 or PKCS8 public key
|
||||||
|
func ParseRSAPublicKeyFromPEM(key []byte) (*rsa.PublicKey, error) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Parse PEM block
|
||||||
|
var block *pem.Block
|
||||||
|
if block, _ = pem.Decode(key); block == nil {
|
||||||
|
return nil, ErrKeyMustBePEMEncoded
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse the key
|
||||||
|
var parsedKey interface{}
|
||||||
|
if parsedKey, err = x509.ParsePKIXPublicKey(block.Bytes); err != nil {
|
||||||
|
if cert, err := x509.ParseCertificate(block.Bytes); err == nil {
|
||||||
|
parsedKey = cert.PublicKey
|
||||||
|
} else {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
var pkey *rsa.PublicKey
|
||||||
|
var ok bool
|
||||||
|
if pkey, ok = parsedKey.(*rsa.PublicKey); !ok {
|
||||||
|
return nil, ErrNotRSAPublicKey
|
||||||
|
}
|
||||||
|
|
||||||
|
return pkey, nil
|
||||||
|
}
|
35
vendor/github.com/dgrijalva/jwt-go/signing_method.go
generated
vendored
Normal file
35
vendor/github.com/dgrijalva/jwt-go/signing_method.go
generated
vendored
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
var signingMethods = map[string]func() SigningMethod{}
|
||||||
|
var signingMethodLock = new(sync.RWMutex)
|
||||||
|
|
||||||
|
// Implement SigningMethod to add new methods for signing or verifying tokens.
|
||||||
|
type SigningMethod interface {
|
||||||
|
Verify(signingString, signature string, key interface{}) error // Returns nil if signature is valid
|
||||||
|
Sign(signingString string, key interface{}) (string, error) // Returns encoded signature or error
|
||||||
|
Alg() string // returns the alg identifier for this method (example: 'HS256')
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register the "alg" name and a factory function for signing method.
|
||||||
|
// This is typically done during init() in the method's implementation
|
||||||
|
func RegisterSigningMethod(alg string, f func() SigningMethod) {
|
||||||
|
signingMethodLock.Lock()
|
||||||
|
defer signingMethodLock.Unlock()
|
||||||
|
|
||||||
|
signingMethods[alg] = f
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get a signing method from an "alg" string
|
||||||
|
func GetSigningMethod(alg string) (method SigningMethod) {
|
||||||
|
signingMethodLock.RLock()
|
||||||
|
defer signingMethodLock.RUnlock()
|
||||||
|
|
||||||
|
if methodF, ok := signingMethods[alg]; ok {
|
||||||
|
method = methodF()
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
42
vendor/github.com/dgrijalva/jwt-go/test/helpers.go
generated
vendored
Normal file
42
vendor/github.com/dgrijalva/jwt-go/test/helpers.go
generated
vendored
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
package test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rsa"
|
||||||
|
"github.com/dgrijalva/jwt-go"
|
||||||
|
"io/ioutil"
|
||||||
|
)
|
||||||
|
|
||||||
|
func LoadRSAPrivateKeyFromDisk(location string) *rsa.PrivateKey {
|
||||||
|
keyData, e := ioutil.ReadFile(location)
|
||||||
|
if e != nil {
|
||||||
|
panic(e.Error())
|
||||||
|
}
|
||||||
|
key, e := jwt.ParseRSAPrivateKeyFromPEM(keyData)
|
||||||
|
if e != nil {
|
||||||
|
panic(e.Error())
|
||||||
|
}
|
||||||
|
return key
|
||||||
|
}
|
||||||
|
|
||||||
|
func LoadRSAPublicKeyFromDisk(location string) *rsa.PublicKey {
|
||||||
|
keyData, e := ioutil.ReadFile(location)
|
||||||
|
if e != nil {
|
||||||
|
panic(e.Error())
|
||||||
|
}
|
||||||
|
key, e := jwt.ParseRSAPublicKeyFromPEM(keyData)
|
||||||
|
if e != nil {
|
||||||
|
panic(e.Error())
|
||||||
|
}
|
||||||
|
return key
|
||||||
|
}
|
||||||
|
|
||||||
|
func MakeSampleToken(c jwt.Claims, key interface{}) string {
|
||||||
|
token := jwt.NewWithClaims(jwt.SigningMethodRS256, c)
|
||||||
|
s, e := token.SignedString(key)
|
||||||
|
|
||||||
|
if e != nil {
|
||||||
|
panic(e.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return s
|
||||||
|
}
|
108
vendor/github.com/dgrijalva/jwt-go/token.go
generated
vendored
Normal file
108
vendor/github.com/dgrijalva/jwt-go/token.go
generated
vendored
Normal file
@ -0,0 +1,108 @@
|
|||||||
|
package jwt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TimeFunc provides the current time when parsing token to validate "exp" claim (expiration time).
|
||||||
|
// You can override it to use another time value. This is useful for testing or if your
|
||||||
|
// server uses a different time zone than your tokens.
|
||||||
|
var TimeFunc = time.Now
|
||||||
|
|
||||||
|
// Parse methods use this callback function to supply
|
||||||
|
// the key for verification. The function receives the parsed,
|
||||||
|
// but unverified Token. This allows you to use properties in the
|
||||||
|
// Header of the token (such as `kid`) to identify which key to use.
|
||||||
|
type Keyfunc func(*Token) (interface{}, error)
|
||||||
|
|
||||||
|
// A JWT Token. Different fields will be used depending on whether you're
|
||||||
|
// creating or parsing/verifying a token.
|
||||||
|
type Token struct {
|
||||||
|
Raw string // The raw token. Populated when you Parse a token
|
||||||
|
Method SigningMethod // The signing method used or to be used
|
||||||
|
Header map[string]interface{} // The first segment of the token
|
||||||
|
Claims Claims // The second segment of the token
|
||||||
|
Signature string // The third segment of the token. Populated when you Parse a token
|
||||||
|
Valid bool // Is the token valid? Populated when you Parse/Verify a token
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a new Token. Takes a signing method
|
||||||
|
func New(method SigningMethod) *Token {
|
||||||
|
return NewWithClaims(method, MapClaims{})
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewWithClaims(method SigningMethod, claims Claims) *Token {
|
||||||
|
return &Token{
|
||||||
|
Header: map[string]interface{}{
|
||||||
|
"typ": "JWT",
|
||||||
|
"alg": method.Alg(),
|
||||||
|
},
|
||||||
|
Claims: claims,
|
||||||
|
Method: method,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the complete, signed token
|
||||||
|
func (t *Token) SignedString(key interface{}) (string, error) {
|
||||||
|
var sig, sstr string
|
||||||
|
var err error
|
||||||
|
if sstr, err = t.SigningString(); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if sig, err = t.Method.Sign(sstr, key); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return strings.Join([]string{sstr, sig}, "."), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate the signing string. This is the
|
||||||
|
// most expensive part of the whole deal. Unless you
|
||||||
|
// need this for something special, just go straight for
|
||||||
|
// the SignedString.
|
||||||
|
func (t *Token) SigningString() (string, error) {
|
||||||
|
var err error
|
||||||
|
parts := make([]string, 2)
|
||||||
|
for i, _ := range parts {
|
||||||
|
var jsonValue []byte
|
||||||
|
if i == 0 {
|
||||||
|
if jsonValue, err = json.Marshal(t.Header); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if jsonValue, err = json.Marshal(t.Claims); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
parts[i] = EncodeSegment(jsonValue)
|
||||||
|
}
|
||||||
|
return strings.Join(parts, "."), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse, validate, and return a token.
|
||||||
|
// keyFunc will receive the parsed token and should return the key for validating.
|
||||||
|
// If everything is kosher, err will be nil
|
||||||
|
func Parse(tokenString string, keyFunc Keyfunc) (*Token, error) {
|
||||||
|
return new(Parser).Parse(tokenString, keyFunc)
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseWithClaims(tokenString string, claims Claims, keyFunc Keyfunc) (*Token, error) {
|
||||||
|
return new(Parser).ParseWithClaims(tokenString, claims, keyFunc)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Encode JWT specific base64url encoding with padding stripped
|
||||||
|
func EncodeSegment(seg []byte) string {
|
||||||
|
return strings.TrimRight(base64.URLEncoding.EncodeToString(seg), "=")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode JWT specific base64url encoding with padding stripped
|
||||||
|
func DecodeSegment(seg string) ([]byte, error) {
|
||||||
|
if l := len(seg) % 4; l > 0 {
|
||||||
|
seg += strings.Repeat("=", 4-l)
|
||||||
|
}
|
||||||
|
|
||||||
|
return base64.URLEncoding.DecodeString(seg)
|
||||||
|
}
|
21
vendor/github.com/facebookgo/clock/LICENSE
generated
vendored
Normal file
21
vendor/github.com/facebookgo/clock/LICENSE
generated
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright (c) 2014 Ben Johnson
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
363
vendor/github.com/facebookgo/clock/clock.go
generated
vendored
Normal file
363
vendor/github.com/facebookgo/clock/clock.go
generated
vendored
Normal file
@ -0,0 +1,363 @@
|
|||||||
|
package clock
|
||||||
|
|
||||||
|
import (
|
||||||
|
"runtime"
|
||||||
|
"sort"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Clock represents an interface to the functions in the standard library time
|
||||||
|
// package. Two implementations are available in the clock package. The first
|
||||||
|
// is a real-time clock which simply wraps the time package's functions. The
|
||||||
|
// second is a mock clock which will only make forward progress when
|
||||||
|
// programmatically adjusted.
|
||||||
|
type Clock interface {
|
||||||
|
After(d time.Duration) <-chan time.Time
|
||||||
|
AfterFunc(d time.Duration, f func()) *Timer
|
||||||
|
Now() time.Time
|
||||||
|
Sleep(d time.Duration)
|
||||||
|
Tick(d time.Duration) <-chan time.Time
|
||||||
|
Ticker(d time.Duration) *Ticker
|
||||||
|
Timer(d time.Duration) *Timer
|
||||||
|
}
|
||||||
|
|
||||||
|
// New returns an instance of a real-time clock.
|
||||||
|
func New() Clock {
|
||||||
|
return &clock{}
|
||||||
|
}
|
||||||
|
|
||||||
|
// clock implements a real-time clock by simply wrapping the time package functions.
|
||||||
|
type clock struct{}
|
||||||
|
|
||||||
|
func (c *clock) After(d time.Duration) <-chan time.Time { return time.After(d) }
|
||||||
|
|
||||||
|
func (c *clock) AfterFunc(d time.Duration, f func()) *Timer {
|
||||||
|
return &Timer{timer: time.AfterFunc(d, f)}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *clock) Now() time.Time { return time.Now() }
|
||||||
|
|
||||||
|
func (c *clock) Sleep(d time.Duration) { time.Sleep(d) }
|
||||||
|
|
||||||
|
func (c *clock) Tick(d time.Duration) <-chan time.Time { return time.Tick(d) }
|
||||||
|
|
||||||
|
func (c *clock) Ticker(d time.Duration) *Ticker {
|
||||||
|
t := time.NewTicker(d)
|
||||||
|
return &Ticker{C: t.C, ticker: t}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *clock) Timer(d time.Duration) *Timer {
|
||||||
|
t := time.NewTimer(d)
|
||||||
|
return &Timer{C: t.C, timer: t}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mock represents a mock clock that only moves forward programmically.
|
||||||
|
// It can be preferable to a real-time clock when testing time-based functionality.
|
||||||
|
type Mock struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
now time.Time // current time
|
||||||
|
timers clockTimers // tickers & timers
|
||||||
|
|
||||||
|
calls Calls
|
||||||
|
waiting []waiting
|
||||||
|
callsMutex sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewMock returns an instance of a mock clock.
|
||||||
|
// The current time of the mock clock on initialization is the Unix epoch.
|
||||||
|
func NewMock() *Mock {
|
||||||
|
return &Mock{now: time.Unix(0, 0)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add moves the current time of the mock clock forward by the duration.
|
||||||
|
// This should only be called from a single goroutine at a time.
|
||||||
|
func (m *Mock) Add(d time.Duration) {
|
||||||
|
// Calculate the final current time.
|
||||||
|
t := m.now.Add(d)
|
||||||
|
|
||||||
|
// Continue to execute timers until there are no more before the new time.
|
||||||
|
for {
|
||||||
|
if !m.runNextTimer(t) {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure that we end with the new time.
|
||||||
|
m.mu.Lock()
|
||||||
|
m.now = t
|
||||||
|
m.mu.Unlock()
|
||||||
|
|
||||||
|
// Give a small buffer to make sure the other goroutines get handled.
|
||||||
|
gosched()
|
||||||
|
}
|
||||||
|
|
||||||
|
// runNextTimer executes the next timer in chronological order and moves the
|
||||||
|
// current time to the timer's next tick time. The next time is not executed if
|
||||||
|
// it's next time if after the max time. Returns true if a timer is executed.
|
||||||
|
func (m *Mock) runNextTimer(max time.Time) bool {
|
||||||
|
m.mu.Lock()
|
||||||
|
|
||||||
|
// Sort timers by time.
|
||||||
|
sort.Sort(m.timers)
|
||||||
|
|
||||||
|
// If we have no more timers then exit.
|
||||||
|
if len(m.timers) == 0 {
|
||||||
|
m.mu.Unlock()
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Retrieve next timer. Exit if next tick is after new time.
|
||||||
|
t := m.timers[0]
|
||||||
|
if t.Next().After(max) {
|
||||||
|
m.mu.Unlock()
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Move "now" forward and unlock clock.
|
||||||
|
m.now = t.Next()
|
||||||
|
m.mu.Unlock()
|
||||||
|
|
||||||
|
// Execute timer.
|
||||||
|
t.Tick(m.now)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// After waits for the duration to elapse and then sends the current time on the returned channel.
|
||||||
|
func (m *Mock) After(d time.Duration) <-chan time.Time {
|
||||||
|
defer m.inc(&m.calls.After)
|
||||||
|
return m.Timer(d).C
|
||||||
|
}
|
||||||
|
|
||||||
|
// AfterFunc waits for the duration to elapse and then executes a function.
|
||||||
|
// A Timer is returned that can be stopped.
|
||||||
|
func (m *Mock) AfterFunc(d time.Duration, f func()) *Timer {
|
||||||
|
defer m.inc(&m.calls.AfterFunc)
|
||||||
|
t := m.Timer(d)
|
||||||
|
t.C = nil
|
||||||
|
t.fn = f
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now returns the current wall time on the mock clock.
|
||||||
|
func (m *Mock) Now() time.Time {
|
||||||
|
defer m.inc(&m.calls.Now)
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
return m.now
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sleep pauses the goroutine for the given duration on the mock clock.
|
||||||
|
// The clock must be moved forward in a separate goroutine.
|
||||||
|
func (m *Mock) Sleep(d time.Duration) {
|
||||||
|
defer m.inc(&m.calls.Sleep)
|
||||||
|
<-m.After(d)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tick is a convenience function for Ticker().
|
||||||
|
// It will return a ticker channel that cannot be stopped.
|
||||||
|
func (m *Mock) Tick(d time.Duration) <-chan time.Time {
|
||||||
|
defer m.inc(&m.calls.Tick)
|
||||||
|
return m.Ticker(d).C
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ticker creates a new instance of Ticker.
|
||||||
|
func (m *Mock) Ticker(d time.Duration) *Ticker {
|
||||||
|
defer m.inc(&m.calls.Ticker)
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
ch := make(chan time.Time)
|
||||||
|
t := &Ticker{
|
||||||
|
C: ch,
|
||||||
|
c: ch,
|
||||||
|
mock: m,
|
||||||
|
d: d,
|
||||||
|
next: m.now.Add(d),
|
||||||
|
}
|
||||||
|
m.timers = append(m.timers, (*internalTicker)(t))
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
|
||||||
|
// Timer creates a new instance of Timer.
|
||||||
|
func (m *Mock) Timer(d time.Duration) *Timer {
|
||||||
|
defer m.inc(&m.calls.Timer)
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
ch := make(chan time.Time)
|
||||||
|
t := &Timer{
|
||||||
|
C: ch,
|
||||||
|
c: ch,
|
||||||
|
mock: m,
|
||||||
|
next: m.now.Add(d),
|
||||||
|
}
|
||||||
|
m.timers = append(m.timers, (*internalTimer)(t))
|
||||||
|
return t
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mock) removeClockTimer(t clockTimer) {
|
||||||
|
m.mu.Lock()
|
||||||
|
defer m.mu.Unlock()
|
||||||
|
for i, timer := range m.timers {
|
||||||
|
if timer == t {
|
||||||
|
copy(m.timers[i:], m.timers[i+1:])
|
||||||
|
m.timers[len(m.timers)-1] = nil
|
||||||
|
m.timers = m.timers[:len(m.timers)-1]
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sort.Sort(m.timers)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mock) inc(addr *uint32) {
|
||||||
|
m.callsMutex.Lock()
|
||||||
|
defer m.callsMutex.Unlock()
|
||||||
|
*addr++
|
||||||
|
var newWaiting []waiting
|
||||||
|
for _, w := range m.waiting {
|
||||||
|
if m.calls.atLeast(w.expected) {
|
||||||
|
close(w.done)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
newWaiting = append(newWaiting, w)
|
||||||
|
}
|
||||||
|
m.waiting = newWaiting
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait waits for at least the relevant calls before returning. The expected
|
||||||
|
// Calls are always over the lifetime of the Mock. Values in the Calls struct
|
||||||
|
// are used as the minimum number of calls, this allows you to wait for only
|
||||||
|
// the calls you care about.
|
||||||
|
func (m *Mock) Wait(s Calls) {
|
||||||
|
m.callsMutex.Lock()
|
||||||
|
if m.calls.atLeast(s) {
|
||||||
|
m.callsMutex.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
done := make(chan struct{})
|
||||||
|
m.waiting = append(m.waiting, waiting{expected: s, done: done})
|
||||||
|
m.callsMutex.Unlock()
|
||||||
|
<-done
|
||||||
|
}
|
||||||
|
|
||||||
|
// clockTimer represents an object with an associated start time.
|
||||||
|
type clockTimer interface {
|
||||||
|
Next() time.Time
|
||||||
|
Tick(time.Time)
|
||||||
|
}
|
||||||
|
|
||||||
|
// clockTimers represents a list of sortable timers.
|
||||||
|
type clockTimers []clockTimer
|
||||||
|
|
||||||
|
func (a clockTimers) Len() int { return len(a) }
|
||||||
|
func (a clockTimers) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
||||||
|
func (a clockTimers) Less(i, j int) bool { return a[i].Next().Before(a[j].Next()) }
|
||||||
|
|
||||||
|
// Timer represents a single event.
|
||||||
|
// The current time will be sent on C, unless the timer was created by AfterFunc.
|
||||||
|
type Timer struct {
|
||||||
|
C <-chan time.Time
|
||||||
|
c chan time.Time
|
||||||
|
timer *time.Timer // realtime impl, if set
|
||||||
|
next time.Time // next tick time
|
||||||
|
mock *Mock // mock clock, if set
|
||||||
|
fn func() // AfterFunc function, if set
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stop turns off the ticker.
|
||||||
|
func (t *Timer) Stop() {
|
||||||
|
if t.timer != nil {
|
||||||
|
t.timer.Stop()
|
||||||
|
} else {
|
||||||
|
t.mock.removeClockTimer((*internalTimer)(t))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type internalTimer Timer
|
||||||
|
|
||||||
|
func (t *internalTimer) Next() time.Time { return t.next }
|
||||||
|
func (t *internalTimer) Tick(now time.Time) {
|
||||||
|
if t.fn != nil {
|
||||||
|
t.fn()
|
||||||
|
} else {
|
||||||
|
t.c <- now
|
||||||
|
}
|
||||||
|
t.mock.removeClockTimer((*internalTimer)(t))
|
||||||
|
gosched()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ticker holds a channel that receives "ticks" at regular intervals.
|
||||||
|
type Ticker struct {
|
||||||
|
C <-chan time.Time
|
||||||
|
c chan time.Time
|
||||||
|
ticker *time.Ticker // realtime impl, if set
|
||||||
|
next time.Time // next tick time
|
||||||
|
mock *Mock // mock clock, if set
|
||||||
|
d time.Duration // time between ticks
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stop turns off the ticker.
|
||||||
|
func (t *Ticker) Stop() {
|
||||||
|
if t.ticker != nil {
|
||||||
|
t.ticker.Stop()
|
||||||
|
} else {
|
||||||
|
t.mock.removeClockTimer((*internalTicker)(t))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type internalTicker Ticker
|
||||||
|
|
||||||
|
func (t *internalTicker) Next() time.Time { return t.next }
|
||||||
|
func (t *internalTicker) Tick(now time.Time) {
|
||||||
|
select {
|
||||||
|
case t.c <- now:
|
||||||
|
case <-time.After(1 * time.Millisecond):
|
||||||
|
}
|
||||||
|
t.next = now.Add(t.d)
|
||||||
|
gosched()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sleep momentarily so that other goroutines can process.
|
||||||
|
func gosched() { runtime.Gosched() }
|
||||||
|
|
||||||
|
// Calls keeps track of the count of calls for each of the methods on the Clock
|
||||||
|
// interface.
|
||||||
|
type Calls struct {
|
||||||
|
After uint32
|
||||||
|
AfterFunc uint32
|
||||||
|
Now uint32
|
||||||
|
Sleep uint32
|
||||||
|
Tick uint32
|
||||||
|
Ticker uint32
|
||||||
|
Timer uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// atLeast returns true if at least the number of calls in o have been made.
|
||||||
|
func (c Calls) atLeast(o Calls) bool {
|
||||||
|
if c.After < o.After {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if c.AfterFunc < o.AfterFunc {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if c.Now < o.Now {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if c.Sleep < o.Sleep {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if c.Tick < o.Tick {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if c.Ticker < o.Ticker {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if c.Timer < o.Timer {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
type waiting struct {
|
||||||
|
expected Calls
|
||||||
|
done chan struct{}
|
||||||
|
}
|
190
vendor/github.com/facebookgo/grace/gracehttp/http.go
generated
vendored
Normal file
190
vendor/github.com/facebookgo/grace/gracehttp/http.go
generated
vendored
Normal file
@ -0,0 +1,190 @@
|
|||||||
|
// Package gracehttp provides easy to use graceful restart
|
||||||
|
// functionality for HTTP server.
|
||||||
|
package gracehttp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"crypto/tls"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"sync"
|
||||||
|
"syscall"
|
||||||
|
|
||||||
|
"github.com/facebookgo/grace/gracenet"
|
||||||
|
"github.com/facebookgo/httpdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
logger *log.Logger
|
||||||
|
didInherit = os.Getenv("LISTEN_FDS") != ""
|
||||||
|
ppid = os.Getppid()
|
||||||
|
)
|
||||||
|
|
||||||
|
// An app contains one or more servers and associated configuration.
|
||||||
|
type app struct {
|
||||||
|
servers []*http.Server
|
||||||
|
http *httpdown.HTTP
|
||||||
|
net *gracenet.Net
|
||||||
|
listeners []net.Listener
|
||||||
|
sds []httpdown.Server
|
||||||
|
errors chan error
|
||||||
|
}
|
||||||
|
|
||||||
|
func newApp(servers []*http.Server) *app {
|
||||||
|
return &app{
|
||||||
|
servers: servers,
|
||||||
|
http: &httpdown.HTTP{},
|
||||||
|
net: &gracenet.Net{},
|
||||||
|
listeners: make([]net.Listener, 0, len(servers)),
|
||||||
|
sds: make([]httpdown.Server, 0, len(servers)),
|
||||||
|
|
||||||
|
// 2x num servers for possible Close or Stop errors + 1 for possible
|
||||||
|
// StartProcess error.
|
||||||
|
errors: make(chan error, 1+(len(servers)*2)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *app) listen() error {
|
||||||
|
for _, s := range a.servers {
|
||||||
|
// TODO: default addresses
|
||||||
|
l, err := a.net.Listen("tcp", s.Addr)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if s.TLSConfig != nil {
|
||||||
|
l = tls.NewListener(l, s.TLSConfig)
|
||||||
|
}
|
||||||
|
a.listeners = append(a.listeners, l)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *app) serve() {
|
||||||
|
for i, s := range a.servers {
|
||||||
|
a.sds = append(a.sds, a.http.Serve(s, a.listeners[i]))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *app) wait() {
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(len(a.sds) * 2) // Wait & Stop
|
||||||
|
go a.signalHandler(&wg)
|
||||||
|
for _, s := range a.sds {
|
||||||
|
go func(s httpdown.Server) {
|
||||||
|
defer wg.Done()
|
||||||
|
if err := s.Wait(); err != nil {
|
||||||
|
a.errors <- err
|
||||||
|
}
|
||||||
|
}(s)
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *app) term(wg *sync.WaitGroup) {
|
||||||
|
for _, s := range a.sds {
|
||||||
|
go func(s httpdown.Server) {
|
||||||
|
defer wg.Done()
|
||||||
|
if err := s.Stop(); err != nil {
|
||||||
|
a.errors <- err
|
||||||
|
}
|
||||||
|
}(s)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *app) signalHandler(wg *sync.WaitGroup) {
|
||||||
|
ch := make(chan os.Signal, 10)
|
||||||
|
signal.Notify(ch, syscall.SIGINT, syscall.SIGTERM, syscall.SIGUSR2)
|
||||||
|
for {
|
||||||
|
sig := <-ch
|
||||||
|
switch sig {
|
||||||
|
case syscall.SIGINT, syscall.SIGTERM:
|
||||||
|
// this ensures a subsequent INT/TERM will trigger standard go behaviour of
|
||||||
|
// terminating.
|
||||||
|
signal.Stop(ch)
|
||||||
|
a.term(wg)
|
||||||
|
return
|
||||||
|
case syscall.SIGUSR2:
|
||||||
|
// we only return here if there's an error, otherwise the new process
|
||||||
|
// will send us a TERM when it's ready to trigger the actual shutdown.
|
||||||
|
if _, err := a.net.StartProcess(); err != nil {
|
||||||
|
a.errors <- err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Serve will serve the given http.Servers and will monitor for signals
|
||||||
|
// allowing for graceful termination (SIGTERM) or restart (SIGUSR2).
|
||||||
|
func Serve(servers ...*http.Server) error {
|
||||||
|
a := newApp(servers)
|
||||||
|
|
||||||
|
// Acquire Listeners
|
||||||
|
if err := a.listen(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Some useful logging.
|
||||||
|
if logger != nil {
|
||||||
|
if didInherit {
|
||||||
|
if ppid == 1 {
|
||||||
|
logger.Printf("Listening on init activated %s", pprintAddr(a.listeners))
|
||||||
|
} else {
|
||||||
|
const msg = "Graceful handoff of %s with new pid %d and old pid %d"
|
||||||
|
logger.Printf(msg, pprintAddr(a.listeners), os.Getpid(), ppid)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
const msg = "Serving %s with pid %d"
|
||||||
|
logger.Printf(msg, pprintAddr(a.listeners), os.Getpid())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start serving.
|
||||||
|
a.serve()
|
||||||
|
|
||||||
|
// Close the parent if we inherited and it wasn't init that started us.
|
||||||
|
if didInherit && ppid != 1 {
|
||||||
|
if err := syscall.Kill(ppid, syscall.SIGTERM); err != nil {
|
||||||
|
return fmt.Errorf("failed to close parent: %s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
waitdone := make(chan struct{})
|
||||||
|
go func() {
|
||||||
|
defer close(waitdone)
|
||||||
|
a.wait()
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case err := <-a.errors:
|
||||||
|
if err == nil {
|
||||||
|
panic("unexpected nil error")
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
case <-waitdone:
|
||||||
|
if logger != nil {
|
||||||
|
logger.Printf("Exiting pid %d.", os.Getpid())
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Used for pretty printing addresses.
|
||||||
|
func pprintAddr(listeners []net.Listener) []byte {
|
||||||
|
var out bytes.Buffer
|
||||||
|
for i, l := range listeners {
|
||||||
|
if i != 0 {
|
||||||
|
fmt.Fprint(&out, ", ")
|
||||||
|
}
|
||||||
|
fmt.Fprint(&out, l.Addr())
|
||||||
|
}
|
||||||
|
return out.Bytes()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetLogger sets logger to be able to grab some useful logs
|
||||||
|
func SetLogger(l *log.Logger) {
|
||||||
|
logger = l
|
||||||
|
}
|
30
vendor/github.com/facebookgo/grace/gracehttp/license
generated
vendored
Normal file
30
vendor/github.com/facebookgo/grace/gracehttp/license
generated
vendored
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
BSD License
|
||||||
|
|
||||||
|
For grace software
|
||||||
|
|
||||||
|
Copyright (c) 2015, Facebook, Inc. All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without modification,
|
||||||
|
are permitted provided that the following conditions are met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above copyright notice, this
|
||||||
|
list of conditions and the following disclaimer.
|
||||||
|
|
||||||
|
* Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
this list of conditions and the following disclaimer in the documentation
|
||||||
|
and/or other materials provided with the distribution.
|
||||||
|
|
||||||
|
* Neither the name Facebook nor the names of its contributors may be used to
|
||||||
|
endorse or promote products derived from this software without specific
|
||||||
|
prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
||||||
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||||
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
||||||
|
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
30
vendor/github.com/facebookgo/grace/gracenet/license
generated
vendored
Normal file
30
vendor/github.com/facebookgo/grace/gracenet/license
generated
vendored
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
BSD License
|
||||||
|
|
||||||
|
For grace software
|
||||||
|
|
||||||
|
Copyright (c) 2015, Facebook, Inc. All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without modification,
|
||||||
|
are permitted provided that the following conditions are met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above copyright notice, this
|
||||||
|
list of conditions and the following disclaimer.
|
||||||
|
|
||||||
|
* Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
this list of conditions and the following disclaimer in the documentation
|
||||||
|
and/or other materials provided with the distribution.
|
||||||
|
|
||||||
|
* Neither the name Facebook nor the names of its contributors may be used to
|
||||||
|
endorse or promote products derived from this software without specific
|
||||||
|
prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
||||||
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||||
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
||||||
|
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
252
vendor/github.com/facebookgo/grace/gracenet/net.go
generated
vendored
Normal file
252
vendor/github.com/facebookgo/grace/gracenet/net.go
generated
vendored
Normal file
@ -0,0 +1,252 @@
|
|||||||
|
// Package gracenet provides a family of Listen functions that either open a
|
||||||
|
// fresh connection or provide an inherited connection from when the process
|
||||||
|
// was started. The behave like their counterparts in the net package, but
|
||||||
|
// transparently provide support for graceful restarts without dropping
|
||||||
|
// connections. This is provided in a systemd socket activation compatible form
|
||||||
|
// to allow using socket activation.
|
||||||
|
//
|
||||||
|
// BUG: Doesn't handle closing of listeners.
|
||||||
|
package gracenet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Used to indicate a graceful restart in the new process.
|
||||||
|
envCountKey = "LISTEN_FDS"
|
||||||
|
envCountKeyPrefix = envCountKey + "="
|
||||||
|
)
|
||||||
|
|
||||||
|
// In order to keep the working directory the same as when we started we record
|
||||||
|
// it at startup.
|
||||||
|
var originalWD, _ = os.Getwd()
|
||||||
|
|
||||||
|
// Net provides the family of Listen functions and maintains the associated
|
||||||
|
// state. Typically you will have only once instance of Net per application.
|
||||||
|
type Net struct {
|
||||||
|
inherited []net.Listener
|
||||||
|
active []net.Listener
|
||||||
|
mutex sync.Mutex
|
||||||
|
inheritOnce sync.Once
|
||||||
|
|
||||||
|
// used in tests to override the default behavior of starting from fd 3.
|
||||||
|
fdStart int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *Net) inherit() error {
|
||||||
|
var retErr error
|
||||||
|
n.inheritOnce.Do(func() {
|
||||||
|
n.mutex.Lock()
|
||||||
|
defer n.mutex.Unlock()
|
||||||
|
countStr := os.Getenv(envCountKey)
|
||||||
|
if countStr == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
count, err := strconv.Atoi(countStr)
|
||||||
|
if err != nil {
|
||||||
|
retErr = fmt.Errorf("found invalid count value: %s=%s", envCountKey, countStr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// In tests this may be overridden.
|
||||||
|
fdStart := n.fdStart
|
||||||
|
if fdStart == 0 {
|
||||||
|
// In normal operations if we are inheriting, the listeners will begin at
|
||||||
|
// fd 3.
|
||||||
|
fdStart = 3
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := fdStart; i < fdStart+count; i++ {
|
||||||
|
file := os.NewFile(uintptr(i), "listener")
|
||||||
|
l, err := net.FileListener(file)
|
||||||
|
if err != nil {
|
||||||
|
file.Close()
|
||||||
|
retErr = fmt.Errorf("error inheriting socket fd %d: %s", i, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := file.Close(); err != nil {
|
||||||
|
retErr = fmt.Errorf("error closing inherited socket fd %d: %s", i, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
n.inherited = append(n.inherited, l)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return retErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// Listen announces on the local network address laddr. The network net must be
|
||||||
|
// a stream-oriented network: "tcp", "tcp4", "tcp6", "unix" or "unixpacket". It
|
||||||
|
// returns an inherited net.Listener for the matching network and address, or
|
||||||
|
// creates a new one using net.Listen.
|
||||||
|
func (n *Net) Listen(nett, laddr string) (net.Listener, error) {
|
||||||
|
switch nett {
|
||||||
|
default:
|
||||||
|
return nil, net.UnknownNetworkError(nett)
|
||||||
|
case "tcp", "tcp4", "tcp6":
|
||||||
|
addr, err := net.ResolveTCPAddr(nett, laddr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return n.ListenTCP(nett, addr)
|
||||||
|
case "unix", "unixpacket", "invalid_unix_net_for_test":
|
||||||
|
addr, err := net.ResolveUnixAddr(nett, laddr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return n.ListenUnix(nett, addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListenTCP announces on the local network address laddr. The network net must
|
||||||
|
// be: "tcp", "tcp4" or "tcp6". It returns an inherited net.Listener for the
|
||||||
|
// matching network and address, or creates a new one using net.ListenTCP.
|
||||||
|
func (n *Net) ListenTCP(nett string, laddr *net.TCPAddr) (*net.TCPListener, error) {
|
||||||
|
if err := n.inherit(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
n.mutex.Lock()
|
||||||
|
defer n.mutex.Unlock()
|
||||||
|
|
||||||
|
// look for an inherited listener
|
||||||
|
for i, l := range n.inherited {
|
||||||
|
if l == nil { // we nil used inherited listeners
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if isSameAddr(l.Addr(), laddr) {
|
||||||
|
n.inherited[i] = nil
|
||||||
|
n.active = append(n.active, l)
|
||||||
|
return l.(*net.TCPListener), nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// make a fresh listener
|
||||||
|
l, err := net.ListenTCP(nett, laddr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
n.active = append(n.active, l)
|
||||||
|
return l, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListenUnix announces on the local network address laddr. The network net
|
||||||
|
// must be a: "unix" or "unixpacket". It returns an inherited net.Listener for
|
||||||
|
// the matching network and address, or creates a new one using net.ListenUnix.
|
||||||
|
func (n *Net) ListenUnix(nett string, laddr *net.UnixAddr) (*net.UnixListener, error) {
|
||||||
|
if err := n.inherit(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
n.mutex.Lock()
|
||||||
|
defer n.mutex.Unlock()
|
||||||
|
|
||||||
|
// look for an inherited listener
|
||||||
|
for i, l := range n.inherited {
|
||||||
|
if l == nil { // we nil used inherited listeners
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if isSameAddr(l.Addr(), laddr) {
|
||||||
|
n.inherited[i] = nil
|
||||||
|
n.active = append(n.active, l)
|
||||||
|
return l.(*net.UnixListener), nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// make a fresh listener
|
||||||
|
l, err := net.ListenUnix(nett, laddr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
n.active = append(n.active, l)
|
||||||
|
return l, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// activeListeners returns a snapshot copy of the active listeners.
|
||||||
|
func (n *Net) activeListeners() ([]net.Listener, error) {
|
||||||
|
n.mutex.Lock()
|
||||||
|
defer n.mutex.Unlock()
|
||||||
|
ls := make([]net.Listener, len(n.active))
|
||||||
|
copy(ls, n.active)
|
||||||
|
return ls, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func isSameAddr(a1, a2 net.Addr) bool {
|
||||||
|
if a1.Network() != a2.Network() {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
a1s := a1.String()
|
||||||
|
a2s := a2.String()
|
||||||
|
if a1s == a2s {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// This allows for ipv6 vs ipv4 local addresses to compare as equal. This
|
||||||
|
// scenario is common when listening on localhost.
|
||||||
|
const ipv6prefix = "[::]"
|
||||||
|
a1s = strings.TrimPrefix(a1s, ipv6prefix)
|
||||||
|
a2s = strings.TrimPrefix(a2s, ipv6prefix)
|
||||||
|
const ipv4prefix = "0.0.0.0"
|
||||||
|
a1s = strings.TrimPrefix(a1s, ipv4prefix)
|
||||||
|
a2s = strings.TrimPrefix(a2s, ipv4prefix)
|
||||||
|
return a1s == a2s
|
||||||
|
}
|
||||||
|
|
||||||
|
// StartProcess starts a new process passing it the active listeners. It
|
||||||
|
// doesn't fork, but starts a new process using the same environment and
|
||||||
|
// arguments as when it was originally started. This allows for a newly
|
||||||
|
// deployed binary to be started. It returns the pid of the newly started
|
||||||
|
// process when successful.
|
||||||
|
func (n *Net) StartProcess() (int, error) {
|
||||||
|
listeners, err := n.activeListeners()
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Extract the fds from the listeners.
|
||||||
|
files := make([]*os.File, len(listeners))
|
||||||
|
for i, l := range listeners {
|
||||||
|
files[i], err = l.(filer).File()
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
defer files[i].Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use the original binary location. This works with symlinks such that if
|
||||||
|
// the file it points to has been changed we will use the updated symlink.
|
||||||
|
argv0, err := exec.LookPath(os.Args[0])
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pass on the environment and replace the old count key with the new one.
|
||||||
|
var env []string
|
||||||
|
for _, v := range os.Environ() {
|
||||||
|
if !strings.HasPrefix(v, envCountKeyPrefix) {
|
||||||
|
env = append(env, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
env = append(env, fmt.Sprintf("%s%d", envCountKeyPrefix, len(listeners)))
|
||||||
|
|
||||||
|
allFiles := append([]*os.File{os.Stdin, os.Stdout, os.Stderr}, files...)
|
||||||
|
process, err := os.StartProcess(argv0, os.Args, &os.ProcAttr{
|
||||||
|
Dir: originalWD,
|
||||||
|
Env: env,
|
||||||
|
Files: allFiles,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return process.Pid, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type filer interface {
|
||||||
|
File() (*os.File, error)
|
||||||
|
}
|
376
vendor/github.com/facebookgo/httpdown/httpdown.go
generated
vendored
Normal file
376
vendor/github.com/facebookgo/httpdown/httpdown.go
generated
vendored
Normal file
@ -0,0 +1,376 @@
|
|||||||
|
// Package httpdown provides http.ConnState enabled graceful termination of
|
||||||
|
// http.Server.
|
||||||
|
package httpdown
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/tls"
|
||||||
|
"fmt"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/signal"
|
||||||
|
"sync"
|
||||||
|
"syscall"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/facebookgo/clock"
|
||||||
|
"github.com/facebookgo/stats"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
defaultStopTimeout = time.Minute
|
||||||
|
defaultKillTimeout = time.Minute
|
||||||
|
)
|
||||||
|
|
||||||
|
// A Server allows encapsulates the process of accepting new connections and
|
||||||
|
// serving them, and gracefully shutting down the listener without dropping
|
||||||
|
// active connections.
|
||||||
|
type Server interface {
|
||||||
|
// Wait waits for the serving loop to finish. This will happen when Stop is
|
||||||
|
// called, at which point it returns no error, or if there is an error in the
|
||||||
|
// serving loop. You must call Wait after calling Serve or ListenAndServe.
|
||||||
|
Wait() error
|
||||||
|
|
||||||
|
// Stop stops the listener. It will block until all connections have been
|
||||||
|
// closed.
|
||||||
|
Stop() error
|
||||||
|
}
|
||||||
|
|
||||||
|
// HTTP defines the configuration for serving a http.Server. Multiple calls to
|
||||||
|
// Serve or ListenAndServe can be made on the same HTTP instance. The default
|
||||||
|
// timeouts of 1 minute each result in a maximum of 2 minutes before a Stop()
|
||||||
|
// returns.
|
||||||
|
type HTTP struct {
|
||||||
|
// StopTimeout is the duration before we begin force closing connections.
|
||||||
|
// Defaults to 1 minute.
|
||||||
|
StopTimeout time.Duration
|
||||||
|
|
||||||
|
// KillTimeout is the duration before which we completely give up and abort
|
||||||
|
// even though we still have connected clients. This is useful when a large
|
||||||
|
// number of client connections exist and closing them can take a long time.
|
||||||
|
// Note, this is in addition to the StopTimeout. Defaults to 1 minute.
|
||||||
|
KillTimeout time.Duration
|
||||||
|
|
||||||
|
// Stats is optional. If provided, it will be used to record various metrics.
|
||||||
|
Stats stats.Client
|
||||||
|
|
||||||
|
// Clock allows for testing timing related functionality. Do not specify this
|
||||||
|
// in production code.
|
||||||
|
Clock clock.Clock
|
||||||
|
}
|
||||||
|
|
||||||
|
// Serve provides the low-level API which is useful if you're creating your own
|
||||||
|
// net.Listener.
|
||||||
|
func (h HTTP) Serve(s *http.Server, l net.Listener) Server {
|
||||||
|
stopTimeout := h.StopTimeout
|
||||||
|
if stopTimeout == 0 {
|
||||||
|
stopTimeout = defaultStopTimeout
|
||||||
|
}
|
||||||
|
killTimeout := h.KillTimeout
|
||||||
|
if killTimeout == 0 {
|
||||||
|
killTimeout = defaultKillTimeout
|
||||||
|
}
|
||||||
|
klock := h.Clock
|
||||||
|
if klock == nil {
|
||||||
|
klock = clock.New()
|
||||||
|
}
|
||||||
|
|
||||||
|
ss := &server{
|
||||||
|
stopTimeout: stopTimeout,
|
||||||
|
killTimeout: killTimeout,
|
||||||
|
stats: h.Stats,
|
||||||
|
clock: klock,
|
||||||
|
oldConnState: s.ConnState,
|
||||||
|
listener: l,
|
||||||
|
server: s,
|
||||||
|
serveDone: make(chan struct{}),
|
||||||
|
serveErr: make(chan error, 1),
|
||||||
|
new: make(chan net.Conn),
|
||||||
|
active: make(chan net.Conn),
|
||||||
|
idle: make(chan net.Conn),
|
||||||
|
closed: make(chan net.Conn),
|
||||||
|
stop: make(chan chan struct{}),
|
||||||
|
kill: make(chan chan struct{}),
|
||||||
|
}
|
||||||
|
s.ConnState = ss.connState
|
||||||
|
go ss.manage()
|
||||||
|
go ss.serve()
|
||||||
|
return ss
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListenAndServe returns a Server for the given http.Server. It is equivalent
|
||||||
|
// to ListenAndServe from the standard library, but returns immediately.
|
||||||
|
// Requests will be accepted in a background goroutine. If the http.Server has
|
||||||
|
// a non-nil TLSConfig, a TLS enabled listener will be setup.
|
||||||
|
func (h HTTP) ListenAndServe(s *http.Server) (Server, error) {
|
||||||
|
addr := s.Addr
|
||||||
|
if addr == "" {
|
||||||
|
if s.TLSConfig == nil {
|
||||||
|
addr = ":http"
|
||||||
|
} else {
|
||||||
|
addr = ":https"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
l, err := net.Listen("tcp", addr)
|
||||||
|
if err != nil {
|
||||||
|
stats.BumpSum(h.Stats, "listen.error", 1)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if s.TLSConfig != nil {
|
||||||
|
l = tls.NewListener(l, s.TLSConfig)
|
||||||
|
}
|
||||||
|
return h.Serve(s, l), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// server manages the serving process and allows for gracefully stopping it.
|
||||||
|
type server struct {
|
||||||
|
stopTimeout time.Duration
|
||||||
|
killTimeout time.Duration
|
||||||
|
stats stats.Client
|
||||||
|
clock clock.Clock
|
||||||
|
|
||||||
|
oldConnState func(net.Conn, http.ConnState)
|
||||||
|
server *http.Server
|
||||||
|
serveDone chan struct{}
|
||||||
|
serveErr chan error
|
||||||
|
listener net.Listener
|
||||||
|
|
||||||
|
new chan net.Conn
|
||||||
|
active chan net.Conn
|
||||||
|
idle chan net.Conn
|
||||||
|
closed chan net.Conn
|
||||||
|
stop chan chan struct{}
|
||||||
|
kill chan chan struct{}
|
||||||
|
|
||||||
|
stopOnce sync.Once
|
||||||
|
stopErr error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *server) connState(c net.Conn, cs http.ConnState) {
|
||||||
|
if s.oldConnState != nil {
|
||||||
|
s.oldConnState(c, cs)
|
||||||
|
}
|
||||||
|
|
||||||
|
switch cs {
|
||||||
|
case http.StateNew:
|
||||||
|
s.new <- c
|
||||||
|
case http.StateActive:
|
||||||
|
s.active <- c
|
||||||
|
case http.StateIdle:
|
||||||
|
s.idle <- c
|
||||||
|
case http.StateHijacked, http.StateClosed:
|
||||||
|
s.closed <- c
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *server) manage() {
|
||||||
|
defer func() {
|
||||||
|
close(s.new)
|
||||||
|
close(s.active)
|
||||||
|
close(s.idle)
|
||||||
|
close(s.closed)
|
||||||
|
close(s.stop)
|
||||||
|
close(s.kill)
|
||||||
|
}()
|
||||||
|
|
||||||
|
var stopDone chan struct{}
|
||||||
|
|
||||||
|
conns := map[net.Conn]http.ConnState{}
|
||||||
|
var countNew, countActive, countIdle float64
|
||||||
|
|
||||||
|
// decConn decrements the count associated with the current state of the
|
||||||
|
// given connection.
|
||||||
|
decConn := func(c net.Conn) {
|
||||||
|
switch conns[c] {
|
||||||
|
default:
|
||||||
|
panic(fmt.Errorf("unknown existing connection: %s", c))
|
||||||
|
case http.StateNew:
|
||||||
|
countNew--
|
||||||
|
case http.StateActive:
|
||||||
|
countActive--
|
||||||
|
case http.StateIdle:
|
||||||
|
countIdle--
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// setup a ticker to report various values every minute. if we don't have a
|
||||||
|
// Stats implementation provided, we Stop it so it never ticks.
|
||||||
|
statsTicker := s.clock.Ticker(time.Minute)
|
||||||
|
if s.stats == nil {
|
||||||
|
statsTicker.Stop()
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-statsTicker.C:
|
||||||
|
// we'll only get here when s.stats is not nil
|
||||||
|
s.stats.BumpAvg("http-state.new", countNew)
|
||||||
|
s.stats.BumpAvg("http-state.active", countActive)
|
||||||
|
s.stats.BumpAvg("http-state.idle", countIdle)
|
||||||
|
s.stats.BumpAvg("http-state.total", countNew+countActive+countIdle)
|
||||||
|
case c := <-s.new:
|
||||||
|
conns[c] = http.StateNew
|
||||||
|
countNew++
|
||||||
|
case c := <-s.active:
|
||||||
|
decConn(c)
|
||||||
|
countActive++
|
||||||
|
|
||||||
|
conns[c] = http.StateActive
|
||||||
|
case c := <-s.idle:
|
||||||
|
decConn(c)
|
||||||
|
countIdle++
|
||||||
|
|
||||||
|
conns[c] = http.StateIdle
|
||||||
|
|
||||||
|
// if we're already stopping, close it
|
||||||
|
if stopDone != nil {
|
||||||
|
c.Close()
|
||||||
|
}
|
||||||
|
case c := <-s.closed:
|
||||||
|
stats.BumpSum(s.stats, "conn.closed", 1)
|
||||||
|
decConn(c)
|
||||||
|
delete(conns, c)
|
||||||
|
|
||||||
|
// if we're waiting to stop and are all empty, we just closed the last
|
||||||
|
// connection and we're done.
|
||||||
|
if stopDone != nil && len(conns) == 0 {
|
||||||
|
close(stopDone)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
case stopDone = <-s.stop:
|
||||||
|
// if we're already all empty, we're already done
|
||||||
|
if len(conns) == 0 {
|
||||||
|
close(stopDone)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// close current idle connections right away
|
||||||
|
for c, cs := range conns {
|
||||||
|
if cs == http.StateIdle {
|
||||||
|
c.Close()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// continue the loop and wait for all the ConnState updates which will
|
||||||
|
// eventually close(stopDone) and return from this goroutine.
|
||||||
|
|
||||||
|
case killDone := <-s.kill:
|
||||||
|
// force close all connections
|
||||||
|
stats.BumpSum(s.stats, "kill.conn.count", float64(len(conns)))
|
||||||
|
for c := range conns {
|
||||||
|
c.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// don't block the kill.
|
||||||
|
close(killDone)
|
||||||
|
|
||||||
|
// continue the loop and we wait for all the ConnState updates and will
|
||||||
|
// return from this goroutine when we're all done. otherwise we'll try to
|
||||||
|
// send those ConnState updates on closed channels.
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *server) serve() {
|
||||||
|
stats.BumpSum(s.stats, "serve", 1)
|
||||||
|
s.serveErr <- s.server.Serve(s.listener)
|
||||||
|
close(s.serveDone)
|
||||||
|
close(s.serveErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *server) Wait() error {
|
||||||
|
if err := <-s.serveErr; !isUseOfClosedError(err) {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *server) Stop() error {
|
||||||
|
s.stopOnce.Do(func() {
|
||||||
|
defer stats.BumpTime(s.stats, "stop.time").End()
|
||||||
|
stats.BumpSum(s.stats, "stop", 1)
|
||||||
|
|
||||||
|
// first disable keep-alive for new connections
|
||||||
|
s.server.SetKeepAlivesEnabled(false)
|
||||||
|
|
||||||
|
// then close the listener so new connections can't connect come thru
|
||||||
|
closeErr := s.listener.Close()
|
||||||
|
<-s.serveDone
|
||||||
|
|
||||||
|
// then trigger the background goroutine to stop and wait for it
|
||||||
|
stopDone := make(chan struct{})
|
||||||
|
s.stop <- stopDone
|
||||||
|
|
||||||
|
// wait for stop
|
||||||
|
select {
|
||||||
|
case <-stopDone:
|
||||||
|
case <-s.clock.After(s.stopTimeout):
|
||||||
|
defer stats.BumpTime(s.stats, "kill.time").End()
|
||||||
|
stats.BumpSum(s.stats, "kill", 1)
|
||||||
|
|
||||||
|
// stop timed out, wait for kill
|
||||||
|
killDone := make(chan struct{})
|
||||||
|
s.kill <- killDone
|
||||||
|
select {
|
||||||
|
case <-killDone:
|
||||||
|
case <-s.clock.After(s.killTimeout):
|
||||||
|
// kill timed out, give up
|
||||||
|
stats.BumpSum(s.stats, "kill.timeout", 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if closeErr != nil && !isUseOfClosedError(closeErr) {
|
||||||
|
stats.BumpSum(s.stats, "listener.close.error", 1)
|
||||||
|
s.stopErr = closeErr
|
||||||
|
}
|
||||||
|
})
|
||||||
|
return s.stopErr
|
||||||
|
}
|
||||||
|
|
||||||
|
func isUseOfClosedError(err error) bool {
|
||||||
|
if err == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if opErr, ok := err.(*net.OpError); ok {
|
||||||
|
err = opErr.Err
|
||||||
|
}
|
||||||
|
return err.Error() == "use of closed network connection"
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListenAndServe is a convenience function to serve and wait for a SIGTERM
|
||||||
|
// or SIGINT before shutting down.
|
||||||
|
func ListenAndServe(s *http.Server, hd *HTTP) error {
|
||||||
|
if hd == nil {
|
||||||
|
hd = &HTTP{}
|
||||||
|
}
|
||||||
|
hs, err := hd.ListenAndServe(s)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
waiterr := make(chan error, 1)
|
||||||
|
go func() {
|
||||||
|
defer close(waiterr)
|
||||||
|
waiterr <- hs.Wait()
|
||||||
|
}()
|
||||||
|
|
||||||
|
signals := make(chan os.Signal, 10)
|
||||||
|
signal.Notify(signals, syscall.SIGTERM, syscall.SIGINT)
|
||||||
|
|
||||||
|
select {
|
||||||
|
case err := <-waiterr:
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
case <-signals:
|
||||||
|
signal.Stop(signals)
|
||||||
|
if err := hs.Stop(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := <-waiterr; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
43
vendor/github.com/facebookgo/httpdown/httpdown_example/main.go
generated
vendored
Normal file
43
vendor/github.com/facebookgo/httpdown/httpdown_example/main.go
generated
vendored
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"flag"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/facebookgo/httpdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
func handler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
duration, err := time.ParseDuration(r.FormValue("duration"))
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, err.Error(), 400)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Fprintf(w, "going to sleep %s with pid %d\n", duration, os.Getpid())
|
||||||
|
w.(http.Flusher).Flush()
|
||||||
|
time.Sleep(duration)
|
||||||
|
fmt.Fprintf(w, "slept %s with pid %d\n", duration, os.Getpid())
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
server := &http.Server{
|
||||||
|
Addr: "127.0.0.1:8080",
|
||||||
|
Handler: http.HandlerFunc(handler),
|
||||||
|
}
|
||||||
|
hd := &httpdown.HTTP{
|
||||||
|
StopTimeout: 10 * time.Second,
|
||||||
|
KillTimeout: 1 * time.Second,
|
||||||
|
}
|
||||||
|
|
||||||
|
flag.StringVar(&server.Addr, "addr", server.Addr, "http address")
|
||||||
|
flag.DurationVar(&hd.StopTimeout, "stop-timeout", hd.StopTimeout, "stop timeout")
|
||||||
|
flag.DurationVar(&hd.KillTimeout, "kill-timeout", hd.KillTimeout, "kill timeout")
|
||||||
|
flag.Parse()
|
||||||
|
|
||||||
|
if err := httpdown.ListenAndServe(server, hd); err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
30
vendor/github.com/facebookgo/httpdown/license
generated
vendored
Normal file
30
vendor/github.com/facebookgo/httpdown/license
generated
vendored
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
BSD License
|
||||||
|
|
||||||
|
For httpdown software
|
||||||
|
|
||||||
|
Copyright (c) 2015, Facebook, Inc. All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without modification,
|
||||||
|
are permitted provided that the following conditions are met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above copyright notice, this
|
||||||
|
list of conditions and the following disclaimer.
|
||||||
|
|
||||||
|
* Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
this list of conditions and the following disclaimer in the documentation
|
||||||
|
and/or other materials provided with the distribution.
|
||||||
|
|
||||||
|
* Neither the name Facebook nor the names of its contributors may be used to
|
||||||
|
endorse or promote products derived from this software without specific
|
||||||
|
prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
||||||
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||||
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
||||||
|
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
35
vendor/github.com/facebookgo/stats/aggregation.go
generated
vendored
Normal file
35
vendor/github.com/facebookgo/stats/aggregation.go
generated
vendored
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package stats
|
||||||
|
|
||||||
|
import "sort"
|
||||||
|
|
||||||
|
// Average returns the average value
|
||||||
|
func Average(values []float64) float64 {
|
||||||
|
if len(values) == 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
var val float64
|
||||||
|
for _, point := range values {
|
||||||
|
val += point
|
||||||
|
}
|
||||||
|
return val / float64(len(values))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sum returns the sum of all the given values
|
||||||
|
func Sum(values []float64) float64 {
|
||||||
|
var val float64
|
||||||
|
for _, point := range values {
|
||||||
|
val += point
|
||||||
|
}
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
// Percentiles returns a map containing the asked for percentiles
|
||||||
|
func Percentiles(values []float64, percentiles map[string]float64) map[string]float64 {
|
||||||
|
sort.Float64s(values)
|
||||||
|
results := map[string]float64{}
|
||||||
|
for label, p := range percentiles {
|
||||||
|
results[label] = values[int(float64(len(values))*p)]
|
||||||
|
}
|
||||||
|
return results
|
||||||
|
}
|
112
vendor/github.com/facebookgo/stats/counter.go
generated
vendored
Normal file
112
vendor/github.com/facebookgo/stats/counter.go
generated
vendored
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
package stats
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
// Type is the type of aggregation of apply
|
||||||
|
type Type int
|
||||||
|
|
||||||
|
const (
|
||||||
|
AggregateAvg Type = iota
|
||||||
|
AggregateSum
|
||||||
|
AggregateHistogram
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// HistogramPercentiles is used to determine which percentiles to return for
|
||||||
|
// SimpleCounter.Aggregate
|
||||||
|
HistogramPercentiles = map[string]float64{
|
||||||
|
"p50": 0.5,
|
||||||
|
"p95": 0.95,
|
||||||
|
"p99": 0.99,
|
||||||
|
}
|
||||||
|
|
||||||
|
// MinSamplesForPercentiles is used by SimpleCounter.Aggregate to determine
|
||||||
|
// what the minimum number of samples is required for percentile analysis
|
||||||
|
MinSamplesForPercentiles = 10
|
||||||
|
)
|
||||||
|
|
||||||
|
// Aggregates can be used to merge counters together. This is not goroutine safe
|
||||||
|
type Aggregates map[string]Counter
|
||||||
|
|
||||||
|
// Add adds the counter for aggregation. This is not goroutine safe
|
||||||
|
func (a Aggregates) Add(c Counter) error {
|
||||||
|
key := c.FullKey()
|
||||||
|
if counter, ok := a[key]; ok {
|
||||||
|
if counter.GetType() != c.GetType() {
|
||||||
|
return fmt.Errorf("stats: mismatched aggregation type for: %s", key)
|
||||||
|
}
|
||||||
|
counter.AddValues(c.GetValues()...)
|
||||||
|
} else {
|
||||||
|
a[key] = c
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Counter is the interface used by Aggregates to merge counters together
|
||||||
|
type Counter interface {
|
||||||
|
// FullKey is used to uniquely identify the counter
|
||||||
|
FullKey() string
|
||||||
|
|
||||||
|
// AddValues adds values for aggregation
|
||||||
|
AddValues(...float64)
|
||||||
|
|
||||||
|
// GetValues returns the values for aggregation
|
||||||
|
GetValues() []float64
|
||||||
|
|
||||||
|
// GetType returns the type of aggregation to apply
|
||||||
|
GetType() Type
|
||||||
|
}
|
||||||
|
|
||||||
|
// SimpleCounter is a basic implementation of the Counter interface
|
||||||
|
type SimpleCounter struct {
|
||||||
|
Key string
|
||||||
|
Values []float64
|
||||||
|
Type Type
|
||||||
|
}
|
||||||
|
|
||||||
|
// FullKey is part of the Counter interace
|
||||||
|
func (s *SimpleCounter) FullKey() string {
|
||||||
|
return s.Key
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetValues is part of the Counter interface
|
||||||
|
func (s *SimpleCounter) GetValues() []float64 {
|
||||||
|
return s.Values
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddValues is part of the Counter interface
|
||||||
|
func (s *SimpleCounter) AddValues(vs ...float64) {
|
||||||
|
s.Values = append(s.Values, vs...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetType is part of the Counter interface
|
||||||
|
func (s *SimpleCounter) GetType() Type {
|
||||||
|
return s.Type
|
||||||
|
}
|
||||||
|
|
||||||
|
// Aggregate aggregates the provided values appropriately, returning a map
|
||||||
|
// from key to value. If AggregateHistogram is specified, the map will contain
|
||||||
|
// the relevant percentiles as specified by HistogramPercentiles
|
||||||
|
func (s *SimpleCounter) Aggregate() map[string]float64 {
|
||||||
|
switch s.Type {
|
||||||
|
case AggregateAvg:
|
||||||
|
return map[string]float64{
|
||||||
|
s.Key: Average(s.Values),
|
||||||
|
}
|
||||||
|
case AggregateSum:
|
||||||
|
return map[string]float64{
|
||||||
|
s.Key: Sum(s.Values),
|
||||||
|
}
|
||||||
|
case AggregateHistogram:
|
||||||
|
histogram := map[string]float64{
|
||||||
|
s.Key: Average(s.Values),
|
||||||
|
}
|
||||||
|
if len(s.Values) > MinSamplesForPercentiles {
|
||||||
|
for k, v := range Percentiles(s.Values, HistogramPercentiles) {
|
||||||
|
histogram[fmt.Sprintf("%s.%s", s.Key, k)] = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return histogram
|
||||||
|
}
|
||||||
|
panic("stats: unsupported aggregation type")
|
||||||
|
}
|
30
vendor/github.com/facebookgo/stats/license
generated
vendored
Normal file
30
vendor/github.com/facebookgo/stats/license
generated
vendored
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
BSD License
|
||||||
|
|
||||||
|
For stats software
|
||||||
|
|
||||||
|
Copyright (c) 2015, Facebook, Inc. All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without modification,
|
||||||
|
are permitted provided that the following conditions are met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above copyright notice, this
|
||||||
|
list of conditions and the following disclaimer.
|
||||||
|
|
||||||
|
* Redistributions in binary form must reproduce the above copyright notice,
|
||||||
|
this list of conditions and the following disclaimer in the documentation
|
||||||
|
and/or other materials provided with the distribution.
|
||||||
|
|
||||||
|
* Neither the name Facebook nor the names of its contributors may be used to
|
||||||
|
endorse or promote products derived from this software without specific
|
||||||
|
prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||||
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||||
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||||
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
|
||||||
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||||
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||||
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
||||||
|
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||||
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
166
vendor/github.com/facebookgo/stats/stats.go
generated
vendored
Normal file
166
vendor/github.com/facebookgo/stats/stats.go
generated
vendored
Normal file
@ -0,0 +1,166 @@
|
|||||||
|
// Package stats defines a lightweight interface for collecting statistics. It
|
||||||
|
// doesn't provide an implementation, just the shared interface.
|
||||||
|
package stats
|
||||||
|
|
||||||
|
// Client provides methods to collection statistics.
|
||||||
|
type Client interface {
|
||||||
|
// BumpAvg bumps the average for the given key.
|
||||||
|
BumpAvg(key string, val float64)
|
||||||
|
|
||||||
|
// BumpSum bumps the sum for the given key.
|
||||||
|
BumpSum(key string, val float64)
|
||||||
|
|
||||||
|
// BumpHistogram bumps the histogram for the given key.
|
||||||
|
BumpHistogram(key string, val float64)
|
||||||
|
|
||||||
|
// BumpTime is a special version of BumpHistogram which is specialized for
|
||||||
|
// timers. Calling it starts the timer, and it returns a value on which End()
|
||||||
|
// can be called to indicate finishing the timer. A convenient way of
|
||||||
|
// recording the duration of a function is calling it like such at the top of
|
||||||
|
// the function:
|
||||||
|
//
|
||||||
|
// defer s.BumpTime("my.function").End()
|
||||||
|
BumpTime(key string) interface {
|
||||||
|
End()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// PrefixClient adds multiple keys for the same value, with each prefix
|
||||||
|
// added to the key and calls the underlying client.
|
||||||
|
func PrefixClient(prefixes []string, client Client) Client {
|
||||||
|
return &prefixClient{
|
||||||
|
Prefixes: prefixes,
|
||||||
|
Client: client,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type prefixClient struct {
|
||||||
|
Prefixes []string
|
||||||
|
Client Client
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *prefixClient) BumpAvg(key string, val float64) {
|
||||||
|
for _, prefix := range p.Prefixes {
|
||||||
|
p.Client.BumpAvg(prefix+key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *prefixClient) BumpSum(key string, val float64) {
|
||||||
|
for _, prefix := range p.Prefixes {
|
||||||
|
p.Client.BumpSum(prefix+key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *prefixClient) BumpHistogram(key string, val float64) {
|
||||||
|
for _, prefix := range p.Prefixes {
|
||||||
|
p.Client.BumpHistogram(prefix+key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *prefixClient) BumpTime(key string) interface {
|
||||||
|
End()
|
||||||
|
} {
|
||||||
|
var m multiEnder
|
||||||
|
for _, prefix := range p.Prefixes {
|
||||||
|
m = append(m, p.Client.BumpTime(prefix+key))
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
// multiEnder combines many enders together.
|
||||||
|
type multiEnder []interface {
|
||||||
|
End()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m multiEnder) End() {
|
||||||
|
for _, e := range m {
|
||||||
|
e.End()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// HookClient is useful for testing. It provides optional hooks for each
|
||||||
|
// expected method in the interface, which if provided will be called. If a
|
||||||
|
// hook is not provided, it will be ignored.
|
||||||
|
type HookClient struct {
|
||||||
|
BumpAvgHook func(key string, val float64)
|
||||||
|
BumpSumHook func(key string, val float64)
|
||||||
|
BumpHistogramHook func(key string, val float64)
|
||||||
|
BumpTimeHook func(key string) interface {
|
||||||
|
End()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpAvg will call BumpAvgHook if defined.
|
||||||
|
func (c *HookClient) BumpAvg(key string, val float64) {
|
||||||
|
if c.BumpAvgHook != nil {
|
||||||
|
c.BumpAvgHook(key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpSum will call BumpSumHook if defined.
|
||||||
|
func (c *HookClient) BumpSum(key string, val float64) {
|
||||||
|
if c.BumpSumHook != nil {
|
||||||
|
c.BumpSumHook(key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpHistogram will call BumpHistogramHook if defined.
|
||||||
|
func (c *HookClient) BumpHistogram(key string, val float64) {
|
||||||
|
if c.BumpHistogramHook != nil {
|
||||||
|
c.BumpHistogramHook(key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpTime will call BumpTimeHook if defined.
|
||||||
|
func (c *HookClient) BumpTime(key string) interface {
|
||||||
|
End()
|
||||||
|
} {
|
||||||
|
if c.BumpTimeHook != nil {
|
||||||
|
return c.BumpTimeHook(key)
|
||||||
|
}
|
||||||
|
return NoOpEnd
|
||||||
|
}
|
||||||
|
|
||||||
|
type noOpEnd struct{}
|
||||||
|
|
||||||
|
func (n noOpEnd) End() {}
|
||||||
|
|
||||||
|
// NoOpEnd provides a dummy value for use in tests as valid return value for
|
||||||
|
// BumpTime().
|
||||||
|
var NoOpEnd = noOpEnd{}
|
||||||
|
|
||||||
|
// BumpAvg calls BumpAvg on the Client if it isn't nil. This is useful when a
|
||||||
|
// component has an optional stats.Client.
|
||||||
|
func BumpAvg(c Client, key string, val float64) {
|
||||||
|
if c != nil {
|
||||||
|
c.BumpAvg(key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpSum calls BumpSum on the Client if it isn't nil. This is useful when a
|
||||||
|
// component has an optional stats.Client.
|
||||||
|
func BumpSum(c Client, key string, val float64) {
|
||||||
|
if c != nil {
|
||||||
|
c.BumpSum(key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpHistogram calls BumpHistogram on the Client if it isn't nil. This is
|
||||||
|
// useful when a component has an optional stats.Client.
|
||||||
|
func BumpHistogram(c Client, key string, val float64) {
|
||||||
|
if c != nil {
|
||||||
|
c.BumpHistogram(key, val)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BumpTime calls BumpTime on the Client if it isn't nil. If the Client is nil
|
||||||
|
// it still returns a valid return value which will be a no-op. This is useful
|
||||||
|
// when a component has an optional stats.Client.
|
||||||
|
func BumpTime(c Client, key string) interface {
|
||||||
|
End()
|
||||||
|
} {
|
||||||
|
if c != nil {
|
||||||
|
return c.BumpTime(key)
|
||||||
|
}
|
||||||
|
return NoOpEnd
|
||||||
|
}
|
17
vendor/github.com/facebookgo/stats/stopper.go
generated
vendored
Normal file
17
vendor/github.com/facebookgo/stats/stopper.go
generated
vendored
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
package stats
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
// Stopper calls Client.BumpSum and Client.BumpHistogram when End'ed
|
||||||
|
type Stopper struct {
|
||||||
|
Key string
|
||||||
|
Start time.Time
|
||||||
|
Client Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// End the Stopper
|
||||||
|
func (s *Stopper) End() {
|
||||||
|
since := time.Since(s.Start).Seconds() * 1000.0
|
||||||
|
s.Client.BumpSum(s.Key+".total", since)
|
||||||
|
s.Client.BumpHistogram(s.Key, since)
|
||||||
|
}
|
27
vendor/github.com/google/go-querystring/query/LICENSE
generated
vendored
Normal file
27
vendor/github.com/google/go-querystring/query/LICENSE
generated
vendored
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
Copyright (c) 2013 Google. All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without
|
||||||
|
modification, are permitted provided that the following conditions are
|
||||||
|
met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above copyright
|
||||||
|
notice, this list of conditions and the following disclaimer.
|
||||||
|
* Redistributions in binary form must reproduce the above
|
||||||
|
copyright notice, this list of conditions and the following disclaimer
|
||||||
|
in the documentation and/or other materials provided with the
|
||||||
|
distribution.
|
||||||
|
* Neither the name of Google Inc. nor the names of its
|
||||||
|
contributors may be used to endorse or promote products derived from
|
||||||
|
this software without specific prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||||
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||||
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||||
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||||
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||||
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||||
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||||
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
320
vendor/github.com/google/go-querystring/query/encode.go
generated
vendored
Normal file
320
vendor/github.com/google/go-querystring/query/encode.go
generated
vendored
Normal file
@ -0,0 +1,320 @@
|
|||||||
|
// Copyright 2013 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.
|
||||||
|
|
||||||
|
// Package query implements encoding of structs into URL query parameters.
|
||||||
|
//
|
||||||
|
// As a simple example:
|
||||||
|
//
|
||||||
|
// type Options struct {
|
||||||
|
// Query string `url:"q"`
|
||||||
|
// ShowAll bool `url:"all"`
|
||||||
|
// Page int `url:"page"`
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// opt := Options{ "foo", true, 2 }
|
||||||
|
// v, _ := query.Values(opt)
|
||||||
|
// fmt.Print(v.Encode()) // will output: "q=foo&all=true&page=2"
|
||||||
|
//
|
||||||
|
// The exact mapping between Go values and url.Values is described in the
|
||||||
|
// documentation for the Values() function.
|
||||||
|
package query
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
"reflect"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var timeType = reflect.TypeOf(time.Time{})
|
||||||
|
|
||||||
|
var encoderType = reflect.TypeOf(new(Encoder)).Elem()
|
||||||
|
|
||||||
|
// Encoder is an interface implemented by any type that wishes to encode
|
||||||
|
// itself into URL values in a non-standard way.
|
||||||
|
type Encoder interface {
|
||||||
|
EncodeValues(key string, v *url.Values) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// Values returns the url.Values encoding of v.
|
||||||
|
//
|
||||||
|
// Values expects to be passed a struct, and traverses it recursively using the
|
||||||
|
// following encoding rules.
|
||||||
|
//
|
||||||
|
// Each exported struct field is encoded as a URL parameter unless
|
||||||
|
//
|
||||||
|
// - the field's tag is "-", or
|
||||||
|
// - the field is empty and its tag specifies the "omitempty" option
|
||||||
|
//
|
||||||
|
// The empty values are false, 0, any nil pointer or interface value, any array
|
||||||
|
// slice, map, or string of length zero, and any time.Time that returns true
|
||||||
|
// for IsZero().
|
||||||
|
//
|
||||||
|
// The URL parameter name defaults to the struct field name but can be
|
||||||
|
// specified in the struct field's tag value. The "url" key in the struct
|
||||||
|
// field's tag value is the key name, followed by an optional comma and
|
||||||
|
// options. For example:
|
||||||
|
//
|
||||||
|
// // Field is ignored by this package.
|
||||||
|
// Field int `url:"-"`
|
||||||
|
//
|
||||||
|
// // Field appears as URL parameter "myName".
|
||||||
|
// Field int `url:"myName"`
|
||||||
|
//
|
||||||
|
// // Field appears as URL parameter "myName" and the field is omitted if
|
||||||
|
// // its value is empty
|
||||||
|
// Field int `url:"myName,omitempty"`
|
||||||
|
//
|
||||||
|
// // Field appears as URL parameter "Field" (the default), but the field
|
||||||
|
// // is skipped if empty. Note the leading comma.
|
||||||
|
// Field int `url:",omitempty"`
|
||||||
|
//
|
||||||
|
// For encoding individual field values, the following type-dependent rules
|
||||||
|
// apply:
|
||||||
|
//
|
||||||
|
// Boolean values default to encoding as the strings "true" or "false".
|
||||||
|
// Including the "int" option signals that the field should be encoded as the
|
||||||
|
// strings "1" or "0".
|
||||||
|
//
|
||||||
|
// time.Time values default to encoding as RFC3339 timestamps. Including the
|
||||||
|
// "unix" option signals that the field should be encoded as a Unix time (see
|
||||||
|
// time.Unix())
|
||||||
|
//
|
||||||
|
// Slice and Array values default to encoding as multiple URL values of the
|
||||||
|
// same name. Including the "comma" option signals that the field should be
|
||||||
|
// encoded as a single comma-delimited value. Including the "space" option
|
||||||
|
// similarly encodes the value as a single space-delimited string. Including
|
||||||
|
// the "semicolon" option will encode the value as a semicolon-delimited string.
|
||||||
|
// Including the "brackets" option signals that the multiple URL values should
|
||||||
|
// have "[]" appended to the value name. "numbered" will append a number to
|
||||||
|
// the end of each incidence of the value name, example:
|
||||||
|
// name0=value0&name1=value1, etc.
|
||||||
|
//
|
||||||
|
// Anonymous struct fields are usually encoded as if their inner exported
|
||||||
|
// fields were fields in the outer struct, subject to the standard Go
|
||||||
|
// visibility rules. An anonymous struct field with a name given in its URL
|
||||||
|
// tag is treated as having that name, rather than being anonymous.
|
||||||
|
//
|
||||||
|
// Non-nil pointer values are encoded as the value pointed to.
|
||||||
|
//
|
||||||
|
// Nested structs are encoded including parent fields in value names for
|
||||||
|
// scoping. e.g:
|
||||||
|
//
|
||||||
|
// "user[name]=acme&user[addr][postcode]=1234&user[addr][city]=SFO"
|
||||||
|
//
|
||||||
|
// All other values are encoded using their default string representation.
|
||||||
|
//
|
||||||
|
// Multiple fields that encode to the same URL parameter name will be included
|
||||||
|
// as multiple URL values of the same name.
|
||||||
|
func Values(v interface{}) (url.Values, error) {
|
||||||
|
values := make(url.Values)
|
||||||
|
val := reflect.ValueOf(v)
|
||||||
|
for val.Kind() == reflect.Ptr {
|
||||||
|
if val.IsNil() {
|
||||||
|
return values, nil
|
||||||
|
}
|
||||||
|
val = val.Elem()
|
||||||
|
}
|
||||||
|
|
||||||
|
if v == nil {
|
||||||
|
return values, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if val.Kind() != reflect.Struct {
|
||||||
|
return nil, fmt.Errorf("query: Values() expects struct input. Got %v", val.Kind())
|
||||||
|
}
|
||||||
|
|
||||||
|
err := reflectValue(values, val, "")
|
||||||
|
return values, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// reflectValue populates the values parameter from the struct fields in val.
|
||||||
|
// Embedded structs are followed recursively (using the rules defined in the
|
||||||
|
// Values function documentation) breadth-first.
|
||||||
|
func reflectValue(values url.Values, val reflect.Value, scope string) error {
|
||||||
|
var embedded []reflect.Value
|
||||||
|
|
||||||
|
typ := val.Type()
|
||||||
|
for i := 0; i < typ.NumField(); i++ {
|
||||||
|
sf := typ.Field(i)
|
||||||
|
if sf.PkgPath != "" && !sf.Anonymous { // unexported
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
sv := val.Field(i)
|
||||||
|
tag := sf.Tag.Get("url")
|
||||||
|
if tag == "-" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
name, opts := parseTag(tag)
|
||||||
|
if name == "" {
|
||||||
|
if sf.Anonymous && sv.Kind() == reflect.Struct {
|
||||||
|
// save embedded struct for later processing
|
||||||
|
embedded = append(embedded, sv)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
name = sf.Name
|
||||||
|
}
|
||||||
|
|
||||||
|
if scope != "" {
|
||||||
|
name = scope + "[" + name + "]"
|
||||||
|
}
|
||||||
|
|
||||||
|
if opts.Contains("omitempty") && isEmptyValue(sv) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if sv.Type().Implements(encoderType) {
|
||||||
|
if !reflect.Indirect(sv).IsValid() {
|
||||||
|
sv = reflect.New(sv.Type().Elem())
|
||||||
|
}
|
||||||
|
|
||||||
|
m := sv.Interface().(Encoder)
|
||||||
|
if err := m.EncodeValues(name, &values); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if sv.Kind() == reflect.Slice || sv.Kind() == reflect.Array {
|
||||||
|
var del byte
|
||||||
|
if opts.Contains("comma") {
|
||||||
|
del = ','
|
||||||
|
} else if opts.Contains("space") {
|
||||||
|
del = ' '
|
||||||
|
} else if opts.Contains("semicolon") {
|
||||||
|
del = ';'
|
||||||
|
} else if opts.Contains("brackets") {
|
||||||
|
name = name + "[]"
|
||||||
|
}
|
||||||
|
|
||||||
|
if del != 0 {
|
||||||
|
s := new(bytes.Buffer)
|
||||||
|
first := true
|
||||||
|
for i := 0; i < sv.Len(); i++ {
|
||||||
|
if first {
|
||||||
|
first = false
|
||||||
|
} else {
|
||||||
|
s.WriteByte(del)
|
||||||
|
}
|
||||||
|
s.WriteString(valueString(sv.Index(i), opts))
|
||||||
|
}
|
||||||
|
values.Add(name, s.String())
|
||||||
|
} else {
|
||||||
|
for i := 0; i < sv.Len(); i++ {
|
||||||
|
k := name
|
||||||
|
if opts.Contains("numbered") {
|
||||||
|
k = fmt.Sprintf("%s%d", name, i)
|
||||||
|
}
|
||||||
|
values.Add(k, valueString(sv.Index(i), opts))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if sv.Type() == timeType {
|
||||||
|
values.Add(name, valueString(sv, opts))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
for sv.Kind() == reflect.Ptr {
|
||||||
|
if sv.IsNil() {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
sv = sv.Elem()
|
||||||
|
}
|
||||||
|
|
||||||
|
if sv.Kind() == reflect.Struct {
|
||||||
|
reflectValue(values, sv, name)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
values.Add(name, valueString(sv, opts))
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, f := range embedded {
|
||||||
|
if err := reflectValue(values, f, scope); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// valueString returns the string representation of a value.
|
||||||
|
func valueString(v reflect.Value, opts tagOptions) string {
|
||||||
|
for v.Kind() == reflect.Ptr {
|
||||||
|
if v.IsNil() {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
v = v.Elem()
|
||||||
|
}
|
||||||
|
|
||||||
|
if v.Kind() == reflect.Bool && opts.Contains("int") {
|
||||||
|
if v.Bool() {
|
||||||
|
return "1"
|
||||||
|
}
|
||||||
|
return "0"
|
||||||
|
}
|
||||||
|
|
||||||
|
if v.Type() == timeType {
|
||||||
|
t := v.Interface().(time.Time)
|
||||||
|
if opts.Contains("unix") {
|
||||||
|
return strconv.FormatInt(t.Unix(), 10)
|
||||||
|
}
|
||||||
|
return t.Format(time.RFC3339)
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprint(v.Interface())
|
||||||
|
}
|
||||||
|
|
||||||
|
// isEmptyValue checks if a value should be considered empty for the purposes
|
||||||
|
// of omitting fields with the "omitempty" option.
|
||||||
|
func isEmptyValue(v reflect.Value) bool {
|
||||||
|
switch v.Kind() {
|
||||||
|
case reflect.Array, reflect.Map, reflect.Slice, reflect.String:
|
||||||
|
return v.Len() == 0
|
||||||
|
case reflect.Bool:
|
||||||
|
return !v.Bool()
|
||||||
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
|
||||||
|
return v.Int() == 0
|
||||||
|
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
|
||||||
|
return v.Uint() == 0
|
||||||
|
case reflect.Float32, reflect.Float64:
|
||||||
|
return v.Float() == 0
|
||||||
|
case reflect.Interface, reflect.Ptr:
|
||||||
|
return v.IsNil()
|
||||||
|
}
|
||||||
|
|
||||||
|
if v.Type() == timeType {
|
||||||
|
return v.Interface().(time.Time).IsZero()
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// tagOptions is the string following a comma in a struct field's "url" tag, or
|
||||||
|
// the empty string. It does not include the leading comma.
|
||||||
|
type tagOptions []string
|
||||||
|
|
||||||
|
// parseTag splits a struct field's url tag into its name and comma-separated
|
||||||
|
// options.
|
||||||
|
func parseTag(tag string) (string, tagOptions) {
|
||||||
|
s := strings.Split(tag, ",")
|
||||||
|
return s[0], s[1:]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Contains checks whether the tagOptions contains the specified option.
|
||||||
|
func (o tagOptions) Contains(option string) bool {
|
||||||
|
for _, s := range o {
|
||||||
|
if s == option {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
27
vendor/github.com/kardianos/osext/LICENSE
generated
vendored
Normal file
27
vendor/github.com/kardianos/osext/LICENSE
generated
vendored
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
Copyright (c) 2012 The Go Authors. All rights reserved.
|
||||||
|
|
||||||
|
Redistribution and use in source and binary forms, with or without
|
||||||
|
modification, are permitted provided that the following conditions are
|
||||||
|
met:
|
||||||
|
|
||||||
|
* Redistributions of source code must retain the above copyright
|
||||||
|
notice, this list of conditions and the following disclaimer.
|
||||||
|
* Redistributions in binary form must reproduce the above
|
||||||
|
copyright notice, this list of conditions and the following disclaimer
|
||||||
|
in the documentation and/or other materials provided with the
|
||||||
|
distribution.
|
||||||
|
* Neither the name of Google Inc. nor the names of its
|
||||||
|
contributors may be used to endorse or promote products derived from
|
||||||
|
this software without specific prior written permission.
|
||||||
|
|
||||||
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||||
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||||
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||||
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
||||||
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||||
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||||
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||||
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||||
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||||
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||||
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
33
vendor/github.com/kardianos/osext/osext.go
generated
vendored
Normal file
33
vendor/github.com/kardianos/osext/osext.go
generated
vendored
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
// Copyright 2012 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.
|
||||||
|
|
||||||
|
// Extensions to the standard "os" package.
|
||||||
|
package osext // import "github.com/kardianos/osext"
|
||||||
|
|
||||||
|
import "path/filepath"
|
||||||
|
|
||||||
|
var cx, ce = executableClean()
|
||||||
|
|
||||||
|
func executableClean() (string, error) {
|
||||||
|
p, err := executable()
|
||||||
|
return filepath.Clean(p), err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Executable returns an absolute path that can be used to
|
||||||
|
// re-invoke the current program.
|
||||||
|
// It may not be valid after the current program exits.
|
||||||
|
func Executable() (string, error) {
|
||||||
|
return cx, ce
|
||||||
|
}
|
||||||
|
|
||||||
|
// Returns same path as Executable, returns just the folder
|
||||||
|
// path. Excludes the executable name and any trailing slash.
|
||||||
|
func ExecutableFolder() (string, error) {
|
||||||
|
p, err := Executable()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return filepath.Dir(p), nil
|
||||||
|
}
|
9
vendor/github.com/kardianos/osext/osext_go18.go
generated
vendored
Normal file
9
vendor/github.com/kardianos/osext/osext_go18.go
generated
vendored
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
//+build go1.8
|
||||||
|
|
||||||
|
package osext
|
||||||
|
|
||||||
|
import "os"
|
||||||
|
|
||||||
|
func executable() (string, error) {
|
||||||
|
return os.Executable()
|
||||||
|
}
|
22
vendor/github.com/kardianos/osext/osext_plan9.go
generated
vendored
Normal file
22
vendor/github.com/kardianos/osext/osext_plan9.go
generated
vendored
Normal file
@ -0,0 +1,22 @@
|
|||||||
|
// Copyright 2012 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.
|
||||||
|
|
||||||
|
//+build !go1.8
|
||||||
|
|
||||||
|
package osext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"syscall"
|
||||||
|
)
|
||||||
|
|
||||||
|
func executable() (string, error) {
|
||||||
|
f, err := os.Open("/proc/" + strconv.Itoa(os.Getpid()) + "/text")
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
return syscall.Fd2path(int(f.Fd()))
|
||||||
|
}
|
36
vendor/github.com/kardianos/osext/osext_procfs.go
generated
vendored
Normal file
36
vendor/github.com/kardianos/osext/osext_procfs.go
generated
vendored
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
// Copyright 2012 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.
|
||||||
|
|
||||||
|
// +build !go1.8,linux !go1.8,netbsd !go1.8,solaris !go1.8,dragonfly
|
||||||
|
|
||||||
|
package osext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func executable() (string, error) {
|
||||||
|
switch runtime.GOOS {
|
||||||
|
case "linux":
|
||||||
|
const deletedTag = " (deleted)"
|
||||||
|
execpath, err := os.Readlink("/proc/self/exe")
|
||||||
|
if err != nil {
|
||||||
|
return execpath, err
|
||||||
|
}
|
||||||
|
execpath = strings.TrimSuffix(execpath, deletedTag)
|
||||||
|
execpath = strings.TrimPrefix(execpath, deletedTag)
|
||||||
|
return execpath, nil
|
||||||
|
case "netbsd":
|
||||||
|
return os.Readlink("/proc/curproc/exe")
|
||||||
|
case "dragonfly":
|
||||||
|
return os.Readlink("/proc/curproc/file")
|
||||||
|
case "solaris":
|
||||||
|
return os.Readlink(fmt.Sprintf("/proc/%d/path/a.out", os.Getpid()))
|
||||||
|
}
|
||||||
|
return "", errors.New("ExecPath not implemented for " + runtime.GOOS)
|
||||||
|
}
|
126
vendor/github.com/kardianos/osext/osext_sysctl.go
generated
vendored
Normal file
126
vendor/github.com/kardianos/osext/osext_sysctl.go
generated
vendored
Normal file
@ -0,0 +1,126 @@
|
|||||||
|
// Copyright 2012 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.
|
||||||
|
|
||||||
|
// +build !go1.8,darwin !go1.8,freebsd !go1.8,openbsd
|
||||||
|
|
||||||
|
package osext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
"syscall"
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
var initCwd, initCwdErr = os.Getwd()
|
||||||
|
|
||||||
|
func executable() (string, error) {
|
||||||
|
var mib [4]int32
|
||||||
|
switch runtime.GOOS {
|
||||||
|
case "freebsd":
|
||||||
|
mib = [4]int32{1 /* CTL_KERN */, 14 /* KERN_PROC */, 12 /* KERN_PROC_PATHNAME */, -1}
|
||||||
|
case "darwin":
|
||||||
|
mib = [4]int32{1 /* CTL_KERN */, 38 /* KERN_PROCARGS */, int32(os.Getpid()), -1}
|
||||||
|
case "openbsd":
|
||||||
|
mib = [4]int32{1 /* CTL_KERN */, 55 /* KERN_PROC_ARGS */, int32(os.Getpid()), 1 /* KERN_PROC_ARGV */}
|
||||||
|
}
|
||||||
|
|
||||||
|
n := uintptr(0)
|
||||||
|
// Get length.
|
||||||
|
_, _, errNum := syscall.Syscall6(syscall.SYS___SYSCTL, uintptr(unsafe.Pointer(&mib[0])), 4, 0, uintptr(unsafe.Pointer(&n)), 0, 0)
|
||||||
|
if errNum != 0 {
|
||||||
|
return "", errNum
|
||||||
|
}
|
||||||
|
if n == 0 { // This shouldn't happen.
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
buf := make([]byte, n)
|
||||||
|
_, _, errNum = syscall.Syscall6(syscall.SYS___SYSCTL, uintptr(unsafe.Pointer(&mib[0])), 4, uintptr(unsafe.Pointer(&buf[0])), uintptr(unsafe.Pointer(&n)), 0, 0)
|
||||||
|
if errNum != 0 {
|
||||||
|
return "", errNum
|
||||||
|
}
|
||||||
|
if n == 0 { // This shouldn't happen.
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
var execPath string
|
||||||
|
switch runtime.GOOS {
|
||||||
|
case "openbsd":
|
||||||
|
// buf now contains **argv, with pointers to each of the C-style
|
||||||
|
// NULL terminated arguments.
|
||||||
|
var args []string
|
||||||
|
argv := uintptr(unsafe.Pointer(&buf[0]))
|
||||||
|
Loop:
|
||||||
|
for {
|
||||||
|
argp := *(**[1 << 20]byte)(unsafe.Pointer(argv))
|
||||||
|
if argp == nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
for i := 0; uintptr(i) < n; i++ {
|
||||||
|
// we don't want the full arguments list
|
||||||
|
if string(argp[i]) == " " {
|
||||||
|
break Loop
|
||||||
|
}
|
||||||
|
if argp[i] != 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
args = append(args, string(argp[:i]))
|
||||||
|
n -= uintptr(i)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
if n < unsafe.Sizeof(argv) {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
argv += unsafe.Sizeof(argv)
|
||||||
|
n -= unsafe.Sizeof(argv)
|
||||||
|
}
|
||||||
|
execPath = args[0]
|
||||||
|
// There is no canonical way to get an executable path on
|
||||||
|
// OpenBSD, so check PATH in case we are called directly
|
||||||
|
if execPath[0] != '/' && execPath[0] != '.' {
|
||||||
|
execIsInPath, err := exec.LookPath(execPath)
|
||||||
|
if err == nil {
|
||||||
|
execPath = execIsInPath
|
||||||
|
}
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
for i, v := range buf {
|
||||||
|
if v == 0 {
|
||||||
|
buf = buf[:i]
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
execPath = string(buf)
|
||||||
|
}
|
||||||
|
|
||||||
|
var err error
|
||||||
|
// execPath will not be empty due to above checks.
|
||||||
|
// Try to get the absolute path if the execPath is not rooted.
|
||||||
|
if execPath[0] != '/' {
|
||||||
|
execPath, err = getAbs(execPath)
|
||||||
|
if err != nil {
|
||||||
|
return execPath, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// For darwin KERN_PROCARGS may return the path to a symlink rather than the
|
||||||
|
// actual executable.
|
||||||
|
if runtime.GOOS == "darwin" {
|
||||||
|
if execPath, err = filepath.EvalSymlinks(execPath); err != nil {
|
||||||
|
return execPath, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return execPath, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getAbs(execPath string) (string, error) {
|
||||||
|
if initCwdErr != nil {
|
||||||
|
return execPath, initCwdErr
|
||||||
|
}
|
||||||
|
// The execPath may begin with a "../" or a "./" so clean it first.
|
||||||
|
// Join the two paths, trailing and starting slashes undetermined, so use
|
||||||
|
// the generic Join function.
|
||||||
|
return filepath.Join(initCwd, filepath.Clean(execPath)), nil
|
||||||
|
}
|
36
vendor/github.com/kardianos/osext/osext_windows.go
generated
vendored
Normal file
36
vendor/github.com/kardianos/osext/osext_windows.go
generated
vendored
Normal file
@ -0,0 +1,36 @@
|
|||||||
|
// Copyright 2012 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.
|
||||||
|
|
||||||
|
//+build !go1.8
|
||||||
|
|
||||||
|
package osext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"syscall"
|
||||||
|
"unicode/utf16"
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
kernel = syscall.MustLoadDLL("kernel32.dll")
|
||||||
|
getModuleFileNameProc = kernel.MustFindProc("GetModuleFileNameW")
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetModuleFileName() with hModule = NULL
|
||||||
|
func executable() (exePath string, err error) {
|
||||||
|
return getModuleFileName()
|
||||||
|
}
|
||||||
|
|
||||||
|
func getModuleFileName() (string, error) {
|
||||||
|
var n uint32
|
||||||
|
b := make([]uint16, syscall.MAX_PATH)
|
||||||
|
size := uint32(len(b))
|
||||||
|
|
||||||
|
r0, _, e1 := getModuleFileNameProc.Call(0, uintptr(unsafe.Pointer(&b[0])), uintptr(size))
|
||||||
|
n = uint32(r0)
|
||||||
|
if n == 0 {
|
||||||
|
return "", e1
|
||||||
|
}
|
||||||
|
return string(utf16.Decode(b[0:n])), nil
|
||||||
|
}
|
21
vendor/github.com/labstack/echo/LICENSE
generated
vendored
Normal file
21
vendor/github.com/labstack/echo/LICENSE
generated
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright (c) 2017 LabStack
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in all
|
||||||
|
copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
SOFTWARE.
|
259
vendor/github.com/labstack/echo/bind.go
generated
vendored
Normal file
259
vendor/github.com/labstack/echo/bind.go
generated
vendored
Normal file
@ -0,0 +1,259 @@
|
|||||||
|
package echo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"encoding/xml"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"reflect"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// Binder is the interface that wraps the Bind method.
|
||||||
|
Binder interface {
|
||||||
|
Bind(i interface{}, c Context) error
|
||||||
|
}
|
||||||
|
|
||||||
|
// DefaultBinder is the default implementation of the Binder interface.
|
||||||
|
DefaultBinder struct{}
|
||||||
|
|
||||||
|
// BindUnmarshaler is the interface used to wrap the UnmarshalParam method.
|
||||||
|
BindUnmarshaler interface {
|
||||||
|
// UnmarshalParam decodes and assigns a value from an form or query param.
|
||||||
|
UnmarshalParam(param string) error
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// Bind implements the `Binder#Bind` function.
|
||||||
|
func (b *DefaultBinder) Bind(i interface{}, c Context) (err error) {
|
||||||
|
req := c.Request()
|
||||||
|
if req.Method == GET {
|
||||||
|
if err = b.bindData(i, c.QueryParams(), "query"); err != nil {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, err.Error())
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ctype := req.Header.Get(HeaderContentType)
|
||||||
|
if req.ContentLength == 0 {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, "Request body can't be empty")
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case strings.HasPrefix(ctype, MIMEApplicationJSON):
|
||||||
|
if err = json.NewDecoder(req.Body).Decode(i); err != nil {
|
||||||
|
if ute, ok := err.(*json.UnmarshalTypeError); ok {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unmarshal type error: expected=%v, got=%v, offset=%v", ute.Type, ute.Value, ute.Offset))
|
||||||
|
} else if se, ok := err.(*json.SyntaxError); ok {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Syntax error: offset=%v, error=%v", se.Offset, se.Error()))
|
||||||
|
} else {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case strings.HasPrefix(ctype, MIMEApplicationXML):
|
||||||
|
if err = xml.NewDecoder(req.Body).Decode(i); err != nil {
|
||||||
|
if ute, ok := err.(*xml.UnsupportedTypeError); ok {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Unsupported type error: type=%v, error=%v", ute.Type, ute.Error()))
|
||||||
|
} else if se, ok := err.(*xml.SyntaxError); ok {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, fmt.Sprintf("Syntax error: line=%v, error=%v", se.Line, se.Error()))
|
||||||
|
} else {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, err.Error())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case strings.HasPrefix(ctype, MIMEApplicationForm), strings.HasPrefix(ctype, MIMEMultipartForm):
|
||||||
|
params, err := c.FormParams()
|
||||||
|
if err != nil {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, err.Error())
|
||||||
|
}
|
||||||
|
if err = b.bindData(i, params, "form"); err != nil {
|
||||||
|
return NewHTTPError(http.StatusBadRequest, err.Error())
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
return ErrUnsupportedMediaType
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *DefaultBinder) bindData(ptr interface{}, data map[string][]string, tag string) error {
|
||||||
|
typ := reflect.TypeOf(ptr).Elem()
|
||||||
|
val := reflect.ValueOf(ptr).Elem()
|
||||||
|
|
||||||
|
if typ.Kind() != reflect.Struct {
|
||||||
|
return errors.New("Binding element must be a struct")
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < typ.NumField(); i++ {
|
||||||
|
typeField := typ.Field(i)
|
||||||
|
structField := val.Field(i)
|
||||||
|
if !structField.CanSet() {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
structFieldKind := structField.Kind()
|
||||||
|
inputFieldName := typeField.Tag.Get(tag)
|
||||||
|
|
||||||
|
if inputFieldName == "" {
|
||||||
|
inputFieldName = typeField.Name
|
||||||
|
// If tag is nil, we inspect if the field is a struct.
|
||||||
|
if _, ok := bindUnmarshaler(structField); !ok && structFieldKind == reflect.Struct {
|
||||||
|
err := b.bindData(structField.Addr().Interface(), data, tag)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
inputValue, exists := data[inputFieldName]
|
||||||
|
if !exists {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Call this first, in case we're dealing with an alias to an array type
|
||||||
|
if ok, err := unmarshalField(typeField.Type.Kind(), inputValue[0], structField); ok {
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
numElems := len(inputValue)
|
||||||
|
if structFieldKind == reflect.Slice && numElems > 0 {
|
||||||
|
sliceOf := structField.Type().Elem().Kind()
|
||||||
|
slice := reflect.MakeSlice(structField.Type(), numElems, numElems)
|
||||||
|
for j := 0; j < numElems; j++ {
|
||||||
|
if err := setWithProperType(sliceOf, inputValue[j], slice.Index(j)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
val.Field(i).Set(slice)
|
||||||
|
} else {
|
||||||
|
if err := setWithProperType(typeField.Type.Kind(), inputValue[0], structField); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func setWithProperType(valueKind reflect.Kind, val string, structField reflect.Value) error {
|
||||||
|
// But also call it here, in case we're dealing with an array of BindUnmarshalers
|
||||||
|
if ok, err := unmarshalField(valueKind, val, structField); ok {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch valueKind {
|
||||||
|
case reflect.Int:
|
||||||
|
return setIntField(val, 0, structField)
|
||||||
|
case reflect.Int8:
|
||||||
|
return setIntField(val, 8, structField)
|
||||||
|
case reflect.Int16:
|
||||||
|
return setIntField(val, 16, structField)
|
||||||
|
case reflect.Int32:
|
||||||
|
return setIntField(val, 32, structField)
|
||||||
|
case reflect.Int64:
|
||||||
|
return setIntField(val, 64, structField)
|
||||||
|
case reflect.Uint:
|
||||||
|
return setUintField(val, 0, structField)
|
||||||
|
case reflect.Uint8:
|
||||||
|
return setUintField(val, 8, structField)
|
||||||
|
case reflect.Uint16:
|
||||||
|
return setUintField(val, 16, structField)
|
||||||
|
case reflect.Uint32:
|
||||||
|
return setUintField(val, 32, structField)
|
||||||
|
case reflect.Uint64:
|
||||||
|
return setUintField(val, 64, structField)
|
||||||
|
case reflect.Bool:
|
||||||
|
return setBoolField(val, structField)
|
||||||
|
case reflect.Float32:
|
||||||
|
return setFloatField(val, 32, structField)
|
||||||
|
case reflect.Float64:
|
||||||
|
return setFloatField(val, 64, structField)
|
||||||
|
case reflect.String:
|
||||||
|
structField.SetString(val)
|
||||||
|
default:
|
||||||
|
return errors.New("unknown type")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func unmarshalField(valueKind reflect.Kind, val string, field reflect.Value) (bool, error) {
|
||||||
|
switch valueKind {
|
||||||
|
case reflect.Ptr:
|
||||||
|
return unmarshalFieldPtr(val, field)
|
||||||
|
default:
|
||||||
|
return unmarshalFieldNonPtr(val, field)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// bindUnmarshaler attempts to unmarshal a reflect.Value into a BindUnmarshaler
|
||||||
|
func bindUnmarshaler(field reflect.Value) (BindUnmarshaler, bool) {
|
||||||
|
ptr := reflect.New(field.Type())
|
||||||
|
if ptr.CanInterface() {
|
||||||
|
iface := ptr.Interface()
|
||||||
|
if unmarshaler, ok := iface.(BindUnmarshaler); ok {
|
||||||
|
return unmarshaler, ok
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
|
func unmarshalFieldNonPtr(value string, field reflect.Value) (bool, error) {
|
||||||
|
if unmarshaler, ok := bindUnmarshaler(field); ok {
|
||||||
|
err := unmarshaler.UnmarshalParam(value)
|
||||||
|
field.Set(reflect.ValueOf(unmarshaler).Elem())
|
||||||
|
return true, err
|
||||||
|
}
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func unmarshalFieldPtr(value string, field reflect.Value) (bool, error) {
|
||||||
|
if field.IsNil() {
|
||||||
|
// Initialize the pointer to a nil value
|
||||||
|
field.Set(reflect.New(field.Type().Elem()))
|
||||||
|
}
|
||||||
|
return unmarshalFieldNonPtr(value, field.Elem())
|
||||||
|
}
|
||||||
|
|
||||||
|
func setIntField(value string, bitSize int, field reflect.Value) error {
|
||||||
|
if value == "" {
|
||||||
|
value = "0"
|
||||||
|
}
|
||||||
|
intVal, err := strconv.ParseInt(value, 10, bitSize)
|
||||||
|
if err == nil {
|
||||||
|
field.SetInt(intVal)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func setUintField(value string, bitSize int, field reflect.Value) error {
|
||||||
|
if value == "" {
|
||||||
|
value = "0"
|
||||||
|
}
|
||||||
|
uintVal, err := strconv.ParseUint(value, 10, bitSize)
|
||||||
|
if err == nil {
|
||||||
|
field.SetUint(uintVal)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func setBoolField(value string, field reflect.Value) error {
|
||||||
|
if value == "" {
|
||||||
|
value = "false"
|
||||||
|
}
|
||||||
|
boolVal, err := strconv.ParseBool(value)
|
||||||
|
if err == nil {
|
||||||
|
field.SetBool(boolVal)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
func setFloatField(value string, bitSize int, field reflect.Value) error {
|
||||||
|
if value == "" {
|
||||||
|
value = "0.0"
|
||||||
|
}
|
||||||
|
floatVal, err := strconv.ParseFloat(value, bitSize)
|
||||||
|
if err == nil {
|
||||||
|
field.SetFloat(floatVal)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
551
vendor/github.com/labstack/echo/context.go
generated
vendored
Normal file
551
vendor/github.com/labstack/echo/context.go
generated
vendored
Normal file
@ -0,0 +1,551 @@
|
|||||||
|
package echo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"encoding/xml"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"mime/multipart"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// Context represents the context of the current HTTP request. It holds request and
|
||||||
|
// response objects, path, path parameters, data and registered handler.
|
||||||
|
Context interface {
|
||||||
|
// Request returns `*http.Request`.
|
||||||
|
Request() *http.Request
|
||||||
|
|
||||||
|
// SetRequest sets `*http.Request`.
|
||||||
|
SetRequest(r *http.Request)
|
||||||
|
|
||||||
|
// Response returns `*Response`.
|
||||||
|
Response() *Response
|
||||||
|
|
||||||
|
// IsTLS returns true if HTTP connection is TLS otherwise false.
|
||||||
|
IsTLS() bool
|
||||||
|
|
||||||
|
// Scheme returns the HTTP protocol scheme, `http` or `https`.
|
||||||
|
Scheme() string
|
||||||
|
|
||||||
|
// RealIP returns the client's network address based on `X-Forwarded-For`
|
||||||
|
// or `X-Real-IP` request header.
|
||||||
|
RealIP() string
|
||||||
|
|
||||||
|
// Path returns the registered path for the handler.
|
||||||
|
Path() string
|
||||||
|
|
||||||
|
// SetPath sets the registered path for the handler.
|
||||||
|
SetPath(p string)
|
||||||
|
|
||||||
|
// Param returns path parameter by name.
|
||||||
|
Param(name string) string
|
||||||
|
|
||||||
|
// ParamNames returns path parameter names.
|
||||||
|
ParamNames() []string
|
||||||
|
|
||||||
|
// SetParamNames sets path parameter names.
|
||||||
|
SetParamNames(names ...string)
|
||||||
|
|
||||||
|
// ParamValues returns path parameter values.
|
||||||
|
ParamValues() []string
|
||||||
|
|
||||||
|
// SetParamValues sets path parameter values.
|
||||||
|
SetParamValues(values ...string)
|
||||||
|
|
||||||
|
// QueryParam returns the query param for the provided name.
|
||||||
|
QueryParam(name string) string
|
||||||
|
|
||||||
|
// QueryParams returns the query parameters as `url.Values`.
|
||||||
|
QueryParams() url.Values
|
||||||
|
|
||||||
|
// QueryString returns the URL query string.
|
||||||
|
QueryString() string
|
||||||
|
|
||||||
|
// FormValue returns the form field value for the provided name.
|
||||||
|
FormValue(name string) string
|
||||||
|
|
||||||
|
// FormParams returns the form parameters as `url.Values`.
|
||||||
|
FormParams() (url.Values, error)
|
||||||
|
|
||||||
|
// FormFile returns the multipart form file for the provided name.
|
||||||
|
FormFile(name string) (*multipart.FileHeader, error)
|
||||||
|
|
||||||
|
// MultipartForm returns the multipart form.
|
||||||
|
MultipartForm() (*multipart.Form, error)
|
||||||
|
|
||||||
|
// Cookie returns the named cookie provided in the request.
|
||||||
|
Cookie(name string) (*http.Cookie, error)
|
||||||
|
|
||||||
|
// SetCookie adds a `Set-Cookie` header in HTTP response.
|
||||||
|
SetCookie(cookie *http.Cookie)
|
||||||
|
|
||||||
|
// Cookies returns the HTTP cookies sent with the request.
|
||||||
|
Cookies() []*http.Cookie
|
||||||
|
|
||||||
|
// Get retrieves data from the context.
|
||||||
|
Get(key string) interface{}
|
||||||
|
|
||||||
|
// Set saves data in the context.
|
||||||
|
Set(key string, val interface{})
|
||||||
|
|
||||||
|
// Bind binds the request body into provided type `i`. The default binder
|
||||||
|
// does it based on Content-Type header.
|
||||||
|
Bind(i interface{}) error
|
||||||
|
|
||||||
|
// Validate validates provided `i`. It is usually called after `Context#Bind()`.
|
||||||
|
// Validator must be registered using `Echo#Validator`.
|
||||||
|
Validate(i interface{}) error
|
||||||
|
|
||||||
|
// Render renders a template with data and sends a text/html response with status
|
||||||
|
// code. Renderer must be registered using `Echo.Renderer`.
|
||||||
|
Render(code int, name string, data interface{}) error
|
||||||
|
|
||||||
|
// HTML sends an HTTP response with status code.
|
||||||
|
HTML(code int, html string) error
|
||||||
|
|
||||||
|
// HTMLBlob sends an HTTP blob response with status code.
|
||||||
|
HTMLBlob(code int, b []byte) error
|
||||||
|
|
||||||
|
// String sends a string response with status code.
|
||||||
|
String(code int, s string) error
|
||||||
|
|
||||||
|
// JSON sends a JSON response with status code.
|
||||||
|
JSON(code int, i interface{}) error
|
||||||
|
|
||||||
|
// JSONPretty sends a pretty-print JSON with status code.
|
||||||
|
JSONPretty(code int, i interface{}, indent string) error
|
||||||
|
|
||||||
|
// JSONBlob sends a JSON blob response with status code.
|
||||||
|
JSONBlob(code int, b []byte) error
|
||||||
|
|
||||||
|
// JSONP sends a JSONP response with status code. It uses `callback` to construct
|
||||||
|
// the JSONP payload.
|
||||||
|
JSONP(code int, callback string, i interface{}) error
|
||||||
|
|
||||||
|
// JSONPBlob sends a JSONP blob response with status code. It uses `callback`
|
||||||
|
// to construct the JSONP payload.
|
||||||
|
JSONPBlob(code int, callback string, b []byte) error
|
||||||
|
|
||||||
|
// XML sends an XML response with status code.
|
||||||
|
XML(code int, i interface{}) error
|
||||||
|
|
||||||
|
// XMLPretty sends a pretty-print XML with status code.
|
||||||
|
XMLPretty(code int, i interface{}, indent string) error
|
||||||
|
|
||||||
|
// XMLBlob sends an XML blob response with status code.
|
||||||
|
XMLBlob(code int, b []byte) error
|
||||||
|
|
||||||
|
// Blob sends a blob response with status code and content type.
|
||||||
|
Blob(code int, contentType string, b []byte) error
|
||||||
|
|
||||||
|
// Stream sends a streaming response with status code and content type.
|
||||||
|
Stream(code int, contentType string, r io.Reader) error
|
||||||
|
|
||||||
|
// File sends a response with the content of the file.
|
||||||
|
File(file string) error
|
||||||
|
|
||||||
|
// Attachment sends a response as attachment, prompting client to save the
|
||||||
|
// file.
|
||||||
|
Attachment(file string, name string) error
|
||||||
|
|
||||||
|
// Inline sends a response as inline, opening the file in the browser.
|
||||||
|
Inline(file string, name string) error
|
||||||
|
|
||||||
|
// NoContent sends a response with no body and a status code.
|
||||||
|
NoContent(code int) error
|
||||||
|
|
||||||
|
// Redirect redirects the request to a provided URL with status code.
|
||||||
|
Redirect(code int, url string) error
|
||||||
|
|
||||||
|
// Error invokes the registered HTTP error handler. Generally used by middleware.
|
||||||
|
Error(err error)
|
||||||
|
|
||||||
|
// Handler returns the matched handler by router.
|
||||||
|
Handler() HandlerFunc
|
||||||
|
|
||||||
|
// SetHandler sets the matched handler by router.
|
||||||
|
SetHandler(h HandlerFunc)
|
||||||
|
|
||||||
|
// Logger returns the `Logger` instance.
|
||||||
|
Logger() Logger
|
||||||
|
|
||||||
|
// Echo returns the `Echo` instance.
|
||||||
|
Echo() *Echo
|
||||||
|
|
||||||
|
// Reset resets the context after request completes. It must be called along
|
||||||
|
// with `Echo#AcquireContext()` and `Echo#ReleaseContext()`.
|
||||||
|
// See `Echo#ServeHTTP()`
|
||||||
|
Reset(r *http.Request, w http.ResponseWriter)
|
||||||
|
}
|
||||||
|
|
||||||
|
context struct {
|
||||||
|
request *http.Request
|
||||||
|
response *Response
|
||||||
|
path string
|
||||||
|
pnames []string
|
||||||
|
pvalues []string
|
||||||
|
query url.Values
|
||||||
|
handler HandlerFunc
|
||||||
|
store Map
|
||||||
|
echo *Echo
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
defaultMemory = 32 << 20 // 32 MB
|
||||||
|
indexPage = "index.html"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (c *context) Request() *http.Request {
|
||||||
|
return c.request
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) SetRequest(r *http.Request) {
|
||||||
|
c.request = r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Response() *Response {
|
||||||
|
return c.response
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) IsTLS() bool {
|
||||||
|
return c.request.TLS != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Scheme() string {
|
||||||
|
// Can't use `r.Request.URL.Scheme`
|
||||||
|
// See: https://groups.google.com/forum/#!topic/golang-nuts/pMUkBlQBDF0
|
||||||
|
if c.IsTLS() {
|
||||||
|
return "https"
|
||||||
|
}
|
||||||
|
return "http"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) RealIP() string {
|
||||||
|
ra := c.request.RemoteAddr
|
||||||
|
if ip := c.request.Header.Get(HeaderXForwardedFor); ip != "" {
|
||||||
|
ra = ip
|
||||||
|
} else if ip := c.request.Header.Get(HeaderXRealIP); ip != "" {
|
||||||
|
ra = ip
|
||||||
|
} else {
|
||||||
|
ra, _, _ = net.SplitHostPort(ra)
|
||||||
|
}
|
||||||
|
return ra
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Path() string {
|
||||||
|
return c.path
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) SetPath(p string) {
|
||||||
|
c.path = p
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Param(name string) string {
|
||||||
|
for i, n := range c.pnames {
|
||||||
|
if i < len(c.pvalues) {
|
||||||
|
if n == name {
|
||||||
|
return c.pvalues[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Param name with aliases
|
||||||
|
for _, p := range strings.Split(n, ",") {
|
||||||
|
if p == name {
|
||||||
|
return c.pvalues[i]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) ParamNames() []string {
|
||||||
|
return c.pnames
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) SetParamNames(names ...string) {
|
||||||
|
c.pnames = names
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) ParamValues() []string {
|
||||||
|
return c.pvalues
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) SetParamValues(values ...string) {
|
||||||
|
c.pvalues = values
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) QueryParam(name string) string {
|
||||||
|
if c.query == nil {
|
||||||
|
c.query = c.request.URL.Query()
|
||||||
|
}
|
||||||
|
return c.query.Get(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) QueryParams() url.Values {
|
||||||
|
if c.query == nil {
|
||||||
|
c.query = c.request.URL.Query()
|
||||||
|
}
|
||||||
|
return c.query
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) QueryString() string {
|
||||||
|
return c.request.URL.RawQuery
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) FormValue(name string) string {
|
||||||
|
return c.request.FormValue(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) FormParams() (url.Values, error) {
|
||||||
|
if strings.HasPrefix(c.request.Header.Get(HeaderContentType), MIMEMultipartForm) {
|
||||||
|
if err := c.request.ParseMultipartForm(defaultMemory); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if err := c.request.ParseForm(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return c.request.Form, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) FormFile(name string) (*multipart.FileHeader, error) {
|
||||||
|
_, fh, err := c.request.FormFile(name)
|
||||||
|
return fh, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) MultipartForm() (*multipart.Form, error) {
|
||||||
|
err := c.request.ParseMultipartForm(defaultMemory)
|
||||||
|
return c.request.MultipartForm, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Cookie(name string) (*http.Cookie, error) {
|
||||||
|
return c.request.Cookie(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) SetCookie(cookie *http.Cookie) {
|
||||||
|
http.SetCookie(c.Response(), cookie)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Cookies() []*http.Cookie {
|
||||||
|
return c.request.Cookies()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Get(key string) interface{} {
|
||||||
|
return c.store[key]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Set(key string, val interface{}) {
|
||||||
|
if c.store == nil {
|
||||||
|
c.store = make(Map)
|
||||||
|
}
|
||||||
|
c.store[key] = val
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Bind(i interface{}) error {
|
||||||
|
return c.echo.Binder.Bind(i, c)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Validate(i interface{}) error {
|
||||||
|
if c.echo.Validator == nil {
|
||||||
|
return ErrValidatorNotRegistered
|
||||||
|
}
|
||||||
|
return c.echo.Validator.Validate(i)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Render(code int, name string, data interface{}) (err error) {
|
||||||
|
if c.echo.Renderer == nil {
|
||||||
|
return ErrRendererNotRegistered
|
||||||
|
}
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
if err = c.echo.Renderer.Render(buf, name, data, c); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.HTMLBlob(code, buf.Bytes())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) HTML(code int, html string) (err error) {
|
||||||
|
return c.HTMLBlob(code, []byte(html))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) HTMLBlob(code int, b []byte) (err error) {
|
||||||
|
return c.Blob(code, MIMETextHTMLCharsetUTF8, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) String(code int, s string) (err error) {
|
||||||
|
return c.Blob(code, MIMETextPlainCharsetUTF8, []byte(s))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) JSON(code int, i interface{}) (err error) {
|
||||||
|
if c.echo.Debug {
|
||||||
|
return c.JSONPretty(code, i, " ")
|
||||||
|
}
|
||||||
|
b, err := json.Marshal(i)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.JSONBlob(code, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) JSONPretty(code int, i interface{}, indent string) (err error) {
|
||||||
|
b, err := json.MarshalIndent(i, "", indent)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.JSONBlob(code, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) JSONBlob(code int, b []byte) (err error) {
|
||||||
|
return c.Blob(code, MIMEApplicationJSONCharsetUTF8, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) JSONP(code int, callback string, i interface{}) (err error) {
|
||||||
|
b, err := json.Marshal(i)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.JSONPBlob(code, callback, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) JSONPBlob(code int, callback string, b []byte) (err error) {
|
||||||
|
c.response.Header().Set(HeaderContentType, MIMEApplicationJavaScriptCharsetUTF8)
|
||||||
|
c.response.WriteHeader(code)
|
||||||
|
if _, err = c.response.Write([]byte(callback + "(")); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if _, err = c.response.Write(b); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_, err = c.response.Write([]byte(");"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) XML(code int, i interface{}) (err error) {
|
||||||
|
if c.echo.Debug {
|
||||||
|
return c.XMLPretty(code, i, " ")
|
||||||
|
}
|
||||||
|
b, err := xml.Marshal(i)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.XMLBlob(code, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) XMLPretty(code int, i interface{}, indent string) (err error) {
|
||||||
|
b, err := xml.MarshalIndent(i, "", indent)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.XMLBlob(code, b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) XMLBlob(code int, b []byte) (err error) {
|
||||||
|
c.response.Header().Set(HeaderContentType, MIMEApplicationXMLCharsetUTF8)
|
||||||
|
c.response.WriteHeader(code)
|
||||||
|
if _, err = c.response.Write([]byte(xml.Header)); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
_, err = c.response.Write(b)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Blob(code int, contentType string, b []byte) (err error) {
|
||||||
|
c.response.Header().Set(HeaderContentType, contentType)
|
||||||
|
c.response.WriteHeader(code)
|
||||||
|
_, err = c.response.Write(b)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Stream(code int, contentType string, r io.Reader) (err error) {
|
||||||
|
c.response.Header().Set(HeaderContentType, contentType)
|
||||||
|
c.response.WriteHeader(code)
|
||||||
|
_, err = io.Copy(c.response, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) File(file string) error {
|
||||||
|
f, err := os.Open(file)
|
||||||
|
if err != nil {
|
||||||
|
return ErrNotFound
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
fi, _ := f.Stat()
|
||||||
|
if fi.IsDir() {
|
||||||
|
file = filepath.Join(file, indexPage)
|
||||||
|
f, err = os.Open(file)
|
||||||
|
if err != nil {
|
||||||
|
return ErrNotFound
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
if fi, err = f.Stat(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
http.ServeContent(c.Response(), c.Request(), fi.Name(), fi.ModTime(), f)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Attachment(file, name string) (err error) {
|
||||||
|
return c.contentDisposition(file, name, "attachment")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Inline(file, name string) (err error) {
|
||||||
|
return c.contentDisposition(file, name, "inline")
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) contentDisposition(file, name, dispositionType string) (err error) {
|
||||||
|
c.response.Header().Set(HeaderContentDisposition, fmt.Sprintf("%s; filename=%s", dispositionType, name))
|
||||||
|
c.File(file)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) NoContent(code int) error {
|
||||||
|
c.response.WriteHeader(code)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Redirect(code int, url string) error {
|
||||||
|
if code < http.StatusMultipleChoices || code > http.StatusTemporaryRedirect {
|
||||||
|
return ErrInvalidRedirectCode
|
||||||
|
}
|
||||||
|
c.response.Header().Set(HeaderLocation, url)
|
||||||
|
c.response.WriteHeader(code)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Error(err error) {
|
||||||
|
c.echo.HTTPErrorHandler(err, c)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Echo() *Echo {
|
||||||
|
return c.echo
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Handler() HandlerFunc {
|
||||||
|
return c.handler
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) SetHandler(h HandlerFunc) {
|
||||||
|
c.handler = h
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Logger() Logger {
|
||||||
|
return c.echo.Logger
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *context) Reset(r *http.Request, w http.ResponseWriter) {
|
||||||
|
c.request = r
|
||||||
|
c.response.reset(w)
|
||||||
|
c.query = nil
|
||||||
|
c.handler = NotFoundHandler
|
||||||
|
c.store = nil
|
||||||
|
}
|
26
vendor/github.com/labstack/echo/cookbook/auto-tls/server.go
generated
vendored
Normal file
26
vendor/github.com/labstack/echo/cookbook/auto-tls/server.go
generated
vendored
Normal file
@ -0,0 +1,26 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"golang.org/x/crypto/acme/autocert"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
// e.AutoTLSManager.HostPolicy = autocert.HostWhitelist("<DOMAIN>")
|
||||||
|
// Cache certificates
|
||||||
|
e.AutoTLSManager.Cache = autocert.DirCache("/var/www/.cache")
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.GET("/", func(c echo.Context) error {
|
||||||
|
return c.HTML(http.StatusOK, `
|
||||||
|
<h1>Welcome to Echo!</h1>
|
||||||
|
<h3>TLS certificates automatically installed from Let's Encrypt :)</h3>
|
||||||
|
`)
|
||||||
|
})
|
||||||
|
e.Logger.Fatal(e.StartAutoTLS(":443"))
|
||||||
|
}
|
38
vendor/github.com/labstack/echo/cookbook/cors/server.go
generated
vendored
Normal file
38
vendor/github.com/labstack/echo/cookbook/cors/server.go
generated
vendored
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
users = []string{"Joe", "Veer", "Zion"}
|
||||||
|
)
|
||||||
|
|
||||||
|
func getUsers(c echo.Context) error {
|
||||||
|
return c.JSON(http.StatusOK, users)
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
// CORS default
|
||||||
|
// Allows requests from any origin wth GET, HEAD, PUT, POST or DELETE method.
|
||||||
|
// e.Use(middleware.CORS())
|
||||||
|
|
||||||
|
// CORS restricted
|
||||||
|
// Allows requests from any `https://labstack.com` or `https://labstack.net` origin
|
||||||
|
// wth GET, PUT, POST or DELETE method.
|
||||||
|
e.Use(middleware.CORSWithConfig(middleware.CORSConfig{
|
||||||
|
AllowOrigins: []string{"https://labstack.com", "https://labstack.net"},
|
||||||
|
AllowMethods: []string{echo.GET, echo.PUT, echo.POST, echo.DELETE},
|
||||||
|
}))
|
||||||
|
|
||||||
|
e.GET("/api/users", getUsers)
|
||||||
|
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
75
vendor/github.com/labstack/echo/cookbook/crud/server.go
generated
vendored
Normal file
75
vendor/github.com/labstack/echo/cookbook/crud/server.go
generated
vendored
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
user struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
users = map[int]*user{}
|
||||||
|
seq = 1
|
||||||
|
)
|
||||||
|
|
||||||
|
//----------
|
||||||
|
// Handlers
|
||||||
|
//----------
|
||||||
|
|
||||||
|
func createUser(c echo.Context) error {
|
||||||
|
u := &user{
|
||||||
|
ID: seq,
|
||||||
|
}
|
||||||
|
if err := c.Bind(u); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
users[u.ID] = u
|
||||||
|
seq++
|
||||||
|
return c.JSON(http.StatusCreated, u)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getUser(c echo.Context) error {
|
||||||
|
id, _ := strconv.Atoi(c.Param("id"))
|
||||||
|
return c.JSON(http.StatusOK, users[id])
|
||||||
|
}
|
||||||
|
|
||||||
|
func updateUser(c echo.Context) error {
|
||||||
|
u := new(user)
|
||||||
|
if err := c.Bind(u); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
id, _ := strconv.Atoi(c.Param("id"))
|
||||||
|
users[id].Name = u.Name
|
||||||
|
return c.JSON(http.StatusOK, users[id])
|
||||||
|
}
|
||||||
|
|
||||||
|
func deleteUser(c echo.Context) error {
|
||||||
|
id, _ := strconv.Atoi(c.Param("id"))
|
||||||
|
delete(users, id)
|
||||||
|
return c.NoContent(http.StatusNoContent)
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
// Middleware
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
// Routes
|
||||||
|
e.POST("/users", createUser)
|
||||||
|
e.GET("/users/:id", getUser)
|
||||||
|
e.PUT("/users/:id", updateUser)
|
||||||
|
e.DELETE("/users/:id", deleteUser)
|
||||||
|
|
||||||
|
// Start server
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
21
vendor/github.com/labstack/echo/cookbook/embed-resources/server.go
generated
vendored
Normal file
21
vendor/github.com/labstack/echo/cookbook/embed-resources/server.go
generated
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
rice "github.com/GeertJohan/go.rice"
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
// the file server for rice. "app" is the folder where the files come from.
|
||||||
|
assetHandler := http.FileServer(rice.MustFindBox("app").HTTPBox())
|
||||||
|
// serves the index.html from rice
|
||||||
|
e.GET("/", echo.WrapHandler(assetHandler))
|
||||||
|
|
||||||
|
// servers other static files
|
||||||
|
e.GET("/static/*", echo.WrapHandler(http.StripPrefix("/static/", assetHandler)))
|
||||||
|
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
65
vendor/github.com/labstack/echo/cookbook/file-upload/multiple/server.go
generated
vendored
Normal file
65
vendor/github.com/labstack/echo/cookbook/file-upload/multiple/server.go
generated
vendored
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func upload(c echo.Context) error {
|
||||||
|
// Read form fields
|
||||||
|
name := c.FormValue("name")
|
||||||
|
email := c.FormValue("email")
|
||||||
|
|
||||||
|
//------------
|
||||||
|
// Read files
|
||||||
|
//------------
|
||||||
|
|
||||||
|
// Multipart form
|
||||||
|
form, err := c.MultipartForm()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
files := form.File["files"]
|
||||||
|
|
||||||
|
for _, file := range files {
|
||||||
|
// Source
|
||||||
|
src, err := file.Open()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer src.Close()
|
||||||
|
|
||||||
|
// Destination
|
||||||
|
dst, err := os.Create(file.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer dst.Close()
|
||||||
|
|
||||||
|
// Copy
|
||||||
|
if _, err = io.Copy(dst, src); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.HTML(http.StatusOK, fmt.Sprintf("<p>Uploaded successfully %d files with fields name=%s and email=%s.</p>", len(files), name, email))
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
e.Static("/", "public")
|
||||||
|
e.POST("/upload", upload)
|
||||||
|
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
59
vendor/github.com/labstack/echo/cookbook/file-upload/single/server.go
generated
vendored
Normal file
59
vendor/github.com/labstack/echo/cookbook/file-upload/single/server.go
generated
vendored
Normal file
@ -0,0 +1,59 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func upload(c echo.Context) error {
|
||||||
|
// Read form fields
|
||||||
|
name := c.FormValue("name")
|
||||||
|
email := c.FormValue("email")
|
||||||
|
|
||||||
|
//-----------
|
||||||
|
// Read file
|
||||||
|
//-----------
|
||||||
|
|
||||||
|
// Source
|
||||||
|
file, err := c.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
src, err := file.Open()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer src.Close()
|
||||||
|
|
||||||
|
// Destination
|
||||||
|
dst, err := os.Create(file.Filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer dst.Close()
|
||||||
|
|
||||||
|
// Copy
|
||||||
|
if _, err = io.Copy(dst, src); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.HTML(http.StatusOK, fmt.Sprintf("<p>File %s uploaded successfully with fields name=%s and email=%s.</p>", file.Filename, name, email))
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
e.Static("/", "public")
|
||||||
|
e.POST("/upload", upload)
|
||||||
|
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
17
vendor/github.com/labstack/echo/cookbook/google-app-engine/app-engine.go
generated
vendored
Normal file
17
vendor/github.com/labstack/echo/cookbook/google-app-engine/app-engine.go
generated
vendored
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
// +build appengine
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func createMux() *echo.Echo {
|
||||||
|
e := echo.New()
|
||||||
|
// note: we don't need to provide the middleware or static handlers, that's taken care of by the platform
|
||||||
|
// app engine has it's own "main" wrapper - we just need to hook echo into the default handler
|
||||||
|
http.Handle("/", e)
|
||||||
|
return e
|
||||||
|
}
|
25
vendor/github.com/labstack/echo/cookbook/google-app-engine/app-managed.go
generated
vendored
Normal file
25
vendor/github.com/labstack/echo/cookbook/google-app-engine/app-managed.go
generated
vendored
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
// +build appenginevm
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"google.golang.org/appengine"
|
||||||
|
)
|
||||||
|
|
||||||
|
func createMux() *echo.Echo {
|
||||||
|
e := echo.New()
|
||||||
|
// note: we don't need to provide the middleware or static handlers
|
||||||
|
// for the appengine vm version - that's taken care of by the platform
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// the appengine package provides a convenient method to handle the health-check requests
|
||||||
|
// and also run the app on the correct port. We just need to add Echo to the default handler
|
||||||
|
e := echo.New(":8080")
|
||||||
|
http.Handle("/", e)
|
||||||
|
appengine.Main()
|
||||||
|
}
|
24
vendor/github.com/labstack/echo/cookbook/google-app-engine/app-standalone.go
generated
vendored
Normal file
24
vendor/github.com/labstack/echo/cookbook/google-app-engine/app-standalone.go
generated
vendored
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
// +build !appengine,!appenginevm
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func createMux() *echo.Echo {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Gzip())
|
||||||
|
|
||||||
|
e.Static("/", "public")
|
||||||
|
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e.Logger.Fatal(e.Start(":8080"))
|
||||||
|
}
|
4
vendor/github.com/labstack/echo/cookbook/google-app-engine/app.go
generated
vendored
Normal file
4
vendor/github.com/labstack/echo/cookbook/google-app-engine/app.go
generated
vendored
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
// reference our echo instance and create it early
|
||||||
|
var e = createMux()
|
54
vendor/github.com/labstack/echo/cookbook/google-app-engine/users.go
generated
vendored
Normal file
54
vendor/github.com/labstack/echo/cookbook/google-app-engine/users.go
generated
vendored
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
user struct {
|
||||||
|
ID string `json:"id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
users map[string]user
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
users = map[string]user{
|
||||||
|
"1": user{
|
||||||
|
ID: "1",
|
||||||
|
Name: "Wreck-It Ralph",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// hook into the echo instance to create an endpoint group
|
||||||
|
// and add specific middleware to it plus handlers
|
||||||
|
g := e.Group("/users")
|
||||||
|
g.Use(middleware.CORS())
|
||||||
|
|
||||||
|
g.POST("", createUser)
|
||||||
|
g.GET("", getUsers)
|
||||||
|
g.GET("/:id", getUser)
|
||||||
|
}
|
||||||
|
|
||||||
|
func createUser(c echo.Context) error {
|
||||||
|
u := new(user)
|
||||||
|
if err := c.Bind(u); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
users[u.ID] = *u
|
||||||
|
return c.JSON(http.StatusCreated, u)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getUsers(c echo.Context) error {
|
||||||
|
return c.JSON(http.StatusOK, users)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getUser(c echo.Context) error {
|
||||||
|
return c.JSON(http.StatusOK, users[c.Param("id")])
|
||||||
|
}
|
31
vendor/github.com/labstack/echo/cookbook/google-app-engine/welcome.go
generated
vendored
Normal file
31
vendor/github.com/labstack/echo/cookbook/google-app-engine/welcome.go
generated
vendored
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"html/template"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
Template struct {
|
||||||
|
templates *template.Template
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
t := &Template{
|
||||||
|
templates: template.Must(template.ParseFiles("templates/welcome.html")),
|
||||||
|
}
|
||||||
|
e.Renderer = t
|
||||||
|
e.GET("/welcome", welcome)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *Template) Render(w io.Writer, name string, data interface{}, c echo.Context) error {
|
||||||
|
return t.templates.ExecuteTemplate(w, name, data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func welcome(c echo.Context) error {
|
||||||
|
return c.Render(http.StatusOK, "welcome", "Joe")
|
||||||
|
}
|
20
vendor/github.com/labstack/echo/cookbook/graceful-shutdown/grace/server.go
generated
vendored
Normal file
20
vendor/github.com/labstack/echo/cookbook/graceful-shutdown/grace/server.go
generated
vendored
Normal file
@ -0,0 +1,20 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/facebookgo/grace/gracehttp"
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Setup
|
||||||
|
e := echo.New()
|
||||||
|
e.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Six sick bricks tick")
|
||||||
|
})
|
||||||
|
e.Server.Addr = ":1323"
|
||||||
|
|
||||||
|
// Serve it like a boss
|
||||||
|
e.Logger.Fatal(gracehttp.Serve(e.Server))
|
||||||
|
}
|
21
vendor/github.com/labstack/echo/cookbook/graceful-shutdown/graceful/server.go
generated
vendored
Normal file
21
vendor/github.com/labstack/echo/cookbook/graceful-shutdown/graceful/server.go
generated
vendored
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/tylerb/graceful"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Setup
|
||||||
|
e := echo.New()
|
||||||
|
e.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Sue sews rose on slow joe crows nose")
|
||||||
|
})
|
||||||
|
e.Server.Addr = ":1323"
|
||||||
|
|
||||||
|
// Serve it like a boss
|
||||||
|
graceful.ListenAndServe(e.Server, 5*time.Second)
|
||||||
|
}
|
25
vendor/github.com/labstack/echo/cookbook/hello-world/server.go
generated
vendored
Normal file
25
vendor/github.com/labstack/echo/cookbook/hello-world/server.go
generated
vendored
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Echo instance
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
// Middleware
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
// Route => handler
|
||||||
|
e.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Hello, World!\n")
|
||||||
|
})
|
||||||
|
|
||||||
|
// Start server
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
42
vendor/github.com/labstack/echo/cookbook/http2/server.go
generated
vendored
Normal file
42
vendor/github.com/labstack/echo/cookbook/http2/server.go
generated
vendored
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
func request(c echo.Context) error {
|
||||||
|
req := c.Request()
|
||||||
|
format := "<pre><strong>Request Information</strong>\n\n<code>Protocol: %s\nHost: %s\nRemote Address: %s\nMethod: %s\nPath: %s\n</code></pre>"
|
||||||
|
return c.HTML(http.StatusOK, fmt.Sprintf(format, req.Proto, req.Host, req.RemoteAddr, req.Method, req.URL.Path))
|
||||||
|
}
|
||||||
|
|
||||||
|
func stream(c echo.Context) error {
|
||||||
|
res := c.Response()
|
||||||
|
gone := res.CloseNotify()
|
||||||
|
res.Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
|
||||||
|
res.WriteHeader(http.StatusOK)
|
||||||
|
ticker := time.NewTicker(1 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
fmt.Fprint(res, "<pre><strong>Clock Stream</strong>\n\n<code>")
|
||||||
|
for {
|
||||||
|
fmt.Fprintf(res, "%v\n", time.Now())
|
||||||
|
res.Flush()
|
||||||
|
select {
|
||||||
|
case <-ticker.C:
|
||||||
|
case <-gone:
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
e.GET("/request", request)
|
||||||
|
e.GET("/stream", stream)
|
||||||
|
e.Logger.Fatal(e.StartTLS(":1323", "cert.pem", "key.pem"))
|
||||||
|
}
|
35
vendor/github.com/labstack/echo/cookbook/jsonp/server.go
generated
vendored
Normal file
35
vendor/github.com/labstack/echo/cookbook/jsonp/server.go
generated
vendored
Normal file
@ -0,0 +1,35 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"math/rand"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
e.Static("/", "public")
|
||||||
|
|
||||||
|
// JSONP
|
||||||
|
e.GET("/jsonp", func(c echo.Context) error {
|
||||||
|
callback := c.QueryParam("callback")
|
||||||
|
var content struct {
|
||||||
|
Response string `json:"response"`
|
||||||
|
Timestamp time.Time `json:"timestamp"`
|
||||||
|
Random int `json:"random"`
|
||||||
|
}
|
||||||
|
content.Response = "Sent via JSONP"
|
||||||
|
content.Timestamp = time.Now().UTC()
|
||||||
|
content.Random = rand.Intn(1000)
|
||||||
|
return c.JSONP(http.StatusOK, callback, &content)
|
||||||
|
})
|
||||||
|
|
||||||
|
// Start server
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
86
vendor/github.com/labstack/echo/cookbook/jwt/custom-claims/server.go
generated
vendored
Normal file
86
vendor/github.com/labstack/echo/cookbook/jwt/custom-claims/server.go
generated
vendored
Normal file
@ -0,0 +1,86 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
jwt "github.com/dgrijalva/jwt-go"
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
// jwtCustomClaims are custom claims extending default ones.
|
||||||
|
type jwtCustomClaims struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Admin bool `json:"admin"`
|
||||||
|
jwt.StandardClaims
|
||||||
|
}
|
||||||
|
|
||||||
|
func login(c echo.Context) error {
|
||||||
|
username := c.FormValue("username")
|
||||||
|
password := c.FormValue("password")
|
||||||
|
|
||||||
|
if username == "jon" && password == "shhh!" {
|
||||||
|
|
||||||
|
// Set custom claims
|
||||||
|
claims := &jwtCustomClaims{
|
||||||
|
"Jon Snow",
|
||||||
|
true,
|
||||||
|
jwt.StandardClaims{
|
||||||
|
ExpiresAt: time.Now().Add(time.Hour * 72).Unix(),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create token with claims
|
||||||
|
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
|
||||||
|
|
||||||
|
// Generate encoded token and send it as response.
|
||||||
|
t, err := token.SignedString([]byte("secret"))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return c.JSON(http.StatusOK, echo.Map{
|
||||||
|
"token": t,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return echo.ErrUnauthorized
|
||||||
|
}
|
||||||
|
|
||||||
|
func accessible(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Accessible")
|
||||||
|
}
|
||||||
|
|
||||||
|
func restricted(c echo.Context) error {
|
||||||
|
user := c.Get("user").(*jwt.Token)
|
||||||
|
claims := user.Claims.(*jwtCustomClaims)
|
||||||
|
name := claims.Name
|
||||||
|
return c.String(http.StatusOK, "Welcome "+name+"!")
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
// Middleware
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
// Login route
|
||||||
|
e.POST("/login", login)
|
||||||
|
|
||||||
|
// Unauthenticated route
|
||||||
|
e.GET("/", accessible)
|
||||||
|
|
||||||
|
// Restricted group
|
||||||
|
r := e.Group("/restricted")
|
||||||
|
|
||||||
|
// Configure middleware with the custom claims type
|
||||||
|
config := middleware.JWTConfig{
|
||||||
|
Claims: &jwtCustomClaims{},
|
||||||
|
SigningKey: []byte("secret"),
|
||||||
|
}
|
||||||
|
r.Use(middleware.JWTWithConfig(config))
|
||||||
|
r.GET("", restricted)
|
||||||
|
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
69
vendor/github.com/labstack/echo/cookbook/jwt/map-claims/server.go
generated
vendored
Normal file
69
vendor/github.com/labstack/echo/cookbook/jwt/map-claims/server.go
generated
vendored
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
jwt "github.com/dgrijalva/jwt-go"
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
func login(c echo.Context) error {
|
||||||
|
username := c.FormValue("username")
|
||||||
|
password := c.FormValue("password")
|
||||||
|
|
||||||
|
if username == "jon" && password == "shhh!" {
|
||||||
|
// Create token
|
||||||
|
token := jwt.New(jwt.SigningMethodHS256)
|
||||||
|
|
||||||
|
// Set claims
|
||||||
|
claims := token.Claims.(jwt.MapClaims)
|
||||||
|
claims["name"] = "Jon Snow"
|
||||||
|
claims["admin"] = true
|
||||||
|
claims["exp"] = time.Now().Add(time.Hour * 72).Unix()
|
||||||
|
|
||||||
|
// Generate encoded token and send it as response.
|
||||||
|
t, err := token.SignedString([]byte("secret"))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return c.JSON(http.StatusOK, map[string]string{
|
||||||
|
"token": t,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return echo.ErrUnauthorized
|
||||||
|
}
|
||||||
|
|
||||||
|
func accessible(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Accessible")
|
||||||
|
}
|
||||||
|
|
||||||
|
func restricted(c echo.Context) error {
|
||||||
|
user := c.Get("user").(*jwt.Token)
|
||||||
|
claims := user.Claims.(jwt.MapClaims)
|
||||||
|
name := claims["name"].(string)
|
||||||
|
return c.String(http.StatusOK, "Welcome "+name+"!")
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
// Middleware
|
||||||
|
e.Use(middleware.Logger())
|
||||||
|
e.Use(middleware.Recover())
|
||||||
|
|
||||||
|
// Login route
|
||||||
|
e.POST("/login", login)
|
||||||
|
|
||||||
|
// Unauthenticated route
|
||||||
|
e.GET("/", accessible)
|
||||||
|
|
||||||
|
// Restricted group
|
||||||
|
r := e.Group("/restricted")
|
||||||
|
r.Use(middleware.JWT([]byte("secret")))
|
||||||
|
r.GET("", restricted)
|
||||||
|
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
82
vendor/github.com/labstack/echo/cookbook/middleware/server.go
generated
vendored
Normal file
82
vendor/github.com/labstack/echo/cookbook/middleware/server.go
generated
vendored
Normal file
@ -0,0 +1,82 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
Stats struct {
|
||||||
|
Uptime time.Time `json:"uptime"`
|
||||||
|
RequestCount uint64 `json:"requestCount"`
|
||||||
|
Statuses map[string]int `json:"statuses"`
|
||||||
|
mutex sync.RWMutex
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewStats() *Stats {
|
||||||
|
return &Stats{
|
||||||
|
Uptime: time.Now(),
|
||||||
|
Statuses: make(map[string]int),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Process is the middleware function.
|
||||||
|
func (s *Stats) Process(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) error {
|
||||||
|
if err := next(c); err != nil {
|
||||||
|
c.Error(err)
|
||||||
|
}
|
||||||
|
s.mutex.Lock()
|
||||||
|
defer s.mutex.Unlock()
|
||||||
|
s.RequestCount++
|
||||||
|
status := strconv.Itoa(c.Response().Status)
|
||||||
|
s.Statuses[status]++
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle is the endpoint to get stats.
|
||||||
|
func (s *Stats) Handle(c echo.Context) error {
|
||||||
|
s.mutex.RLock()
|
||||||
|
defer s.mutex.RUnlock()
|
||||||
|
return c.JSON(http.StatusOK, s)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ServerHeader middleware adds a `Server` header to the response.
|
||||||
|
func ServerHeader(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) error {
|
||||||
|
c.Response().Header().Set(echo.HeaderServer, "Echo/3.0")
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
|
||||||
|
// Debug mode
|
||||||
|
e.Debug = true
|
||||||
|
|
||||||
|
//-------------------
|
||||||
|
// Custom middleware
|
||||||
|
//-------------------
|
||||||
|
// Stats
|
||||||
|
s := NewStats()
|
||||||
|
e.Use(s.Process)
|
||||||
|
e.GET("/stats", s.Handle) // Endpoint to get stats
|
||||||
|
|
||||||
|
// Server header
|
||||||
|
e.Use(ServerHeader)
|
||||||
|
|
||||||
|
// Handler
|
||||||
|
e.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Hello, World!")
|
||||||
|
})
|
||||||
|
|
||||||
|
// Start server
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
45
vendor/github.com/labstack/echo/cookbook/streaming-response/server.go
generated
vendored
Normal file
45
vendor/github.com/labstack/echo/cookbook/streaming-response/server.go
generated
vendored
Normal file
@ -0,0 +1,45 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"encoding/json"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
Geolocation struct {
|
||||||
|
Altitude float64
|
||||||
|
Latitude float64
|
||||||
|
Longitude float64
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
locations = []Geolocation{
|
||||||
|
{-97, 37.819929, -122.478255},
|
||||||
|
{1899, 39.096849, -120.032351},
|
||||||
|
{2619, 37.865101, -119.538329},
|
||||||
|
{42, 33.812092, -117.918974},
|
||||||
|
{15, 37.77493, -122.419416},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
e := echo.New()
|
||||||
|
e.GET("/", func(c echo.Context) error {
|
||||||
|
c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
||||||
|
c.Response().WriteHeader(http.StatusOK)
|
||||||
|
for _, l := range locations {
|
||||||
|
if err := json.NewEncoder(c.Response()).Encode(l); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
c.Response().Flush()
|
||||||
|
time.Sleep(1 * time.Second)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
78
vendor/github.com/labstack/echo/cookbook/subdomains/server.go
generated
vendored
Normal file
78
vendor/github.com/labstack/echo/cookbook/subdomains/server.go
generated
vendored
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/middleware"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
Host struct {
|
||||||
|
Echo *echo.Echo
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// Hosts
|
||||||
|
hosts := make(map[string]*Host)
|
||||||
|
|
||||||
|
//-----
|
||||||
|
// API
|
||||||
|
//-----
|
||||||
|
|
||||||
|
api := echo.New()
|
||||||
|
api.Use(middleware.Logger())
|
||||||
|
api.Use(middleware.Recover())
|
||||||
|
|
||||||
|
hosts["api.localhost:1323"] = &Host{api}
|
||||||
|
|
||||||
|
api.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "API")
|
||||||
|
})
|
||||||
|
|
||||||
|
//------
|
||||||
|
// Blog
|
||||||
|
//------
|
||||||
|
|
||||||
|
blog := echo.New()
|
||||||
|
blog.Use(middleware.Logger())
|
||||||
|
blog.Use(middleware.Recover())
|
||||||
|
|
||||||
|
hosts["blog.localhost:1323"] = &Host{blog}
|
||||||
|
|
||||||
|
blog.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Blog")
|
||||||
|
})
|
||||||
|
|
||||||
|
//---------
|
||||||
|
// Website
|
||||||
|
//---------
|
||||||
|
|
||||||
|
site := echo.New()
|
||||||
|
site.Use(middleware.Logger())
|
||||||
|
site.Use(middleware.Recover())
|
||||||
|
|
||||||
|
hosts["localhost:1323"] = &Host{site}
|
||||||
|
|
||||||
|
site.GET("/", func(c echo.Context) error {
|
||||||
|
return c.String(http.StatusOK, "Website")
|
||||||
|
})
|
||||||
|
|
||||||
|
// Server
|
||||||
|
e := echo.New()
|
||||||
|
e.Any("/*", func(c echo.Context) (err error) {
|
||||||
|
req := c.Request()
|
||||||
|
res := c.Response()
|
||||||
|
host := hosts[req.Host]
|
||||||
|
|
||||||
|
if host == nil {
|
||||||
|
err = echo.ErrNotFound
|
||||||
|
} else {
|
||||||
|
host.Echo.ServeHTTP(res, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
})
|
||||||
|
e.Logger.Fatal(e.Start(":1323"))
|
||||||
|
}
|
14
vendor/github.com/labstack/echo/cookbook/twitter/handler/handler.go
generated
vendored
Normal file
14
vendor/github.com/labstack/echo/cookbook/twitter/handler/handler.go
generated
vendored
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import mgo "gopkg.in/mgo.v2"
|
||||||
|
|
||||||
|
type (
|
||||||
|
Handler struct {
|
||||||
|
DB *mgo.Session
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Key (Should come from somewhere else).
|
||||||
|
Key = "secret"
|
||||||
|
)
|
73
vendor/github.com/labstack/echo/cookbook/twitter/handler/post.go
generated
vendored
Normal file
73
vendor/github.com/labstack/echo/cookbook/twitter/handler/post.go
generated
vendored
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/cookbook/twitter/model"
|
||||||
|
mgo "gopkg.in/mgo.v2"
|
||||||
|
"gopkg.in/mgo.v2/bson"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *Handler) CreatePost(c echo.Context) (err error) {
|
||||||
|
u := &model.User{
|
||||||
|
ID: bson.ObjectIdHex(userIDFromToken(c)),
|
||||||
|
}
|
||||||
|
p := &model.Post{
|
||||||
|
ID: bson.NewObjectId(),
|
||||||
|
From: u.ID.Hex(),
|
||||||
|
}
|
||||||
|
if err = c.Bind(p); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validation
|
||||||
|
if p.To == "" || p.Message == "" {
|
||||||
|
return &echo.HTTPError{Code: http.StatusBadRequest, Message: "invalid to or message fields"}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find user from database
|
||||||
|
db := h.DB.Clone()
|
||||||
|
defer db.Close()
|
||||||
|
if err = db.DB("twitter").C("users").FindId(u.ID).One(u); err != nil {
|
||||||
|
if err == mgo.ErrNotFound {
|
||||||
|
return echo.ErrNotFound
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save post in database
|
||||||
|
if err = db.DB("twitter").C("posts").Insert(p); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return c.JSON(http.StatusCreated, p)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) FetchPost(c echo.Context) (err error) {
|
||||||
|
userID := userIDFromToken(c)
|
||||||
|
page, _ := strconv.Atoi(c.QueryParam("page"))
|
||||||
|
limit, _ := strconv.Atoi(c.QueryParam("limit"))
|
||||||
|
|
||||||
|
// Defaults
|
||||||
|
if page == 0 {
|
||||||
|
page = 1
|
||||||
|
}
|
||||||
|
if limit == 0 {
|
||||||
|
limit = 100
|
||||||
|
}
|
||||||
|
|
||||||
|
// Retrieve posts from database
|
||||||
|
posts := []*model.Post{}
|
||||||
|
db := h.DB.Clone()
|
||||||
|
if err = db.DB("twitter").C("posts").
|
||||||
|
Find(bson.M{"to": userID}).
|
||||||
|
Skip((page - 1) * limit).
|
||||||
|
Limit(limit).
|
||||||
|
All(&posts); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
|
||||||
|
return c.JSON(http.StatusOK, posts)
|
||||||
|
}
|
97
vendor/github.com/labstack/echo/cookbook/twitter/handler/user.go
generated
vendored
Normal file
97
vendor/github.com/labstack/echo/cookbook/twitter/handler/user.go
generated
vendored
Normal file
@ -0,0 +1,97 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
jwt "github.com/dgrijalva/jwt-go"
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
"github.com/labstack/echo/cookbook/twitter/model"
|
||||||
|
mgo "gopkg.in/mgo.v2"
|
||||||
|
"gopkg.in/mgo.v2/bson"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *Handler) Signup(c echo.Context) (err error) {
|
||||||
|
// Bind
|
||||||
|
u := &model.User{ID: bson.NewObjectId()}
|
||||||
|
if err = c.Bind(u); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate
|
||||||
|
if u.Email == "" || u.Password == "" {
|
||||||
|
return &echo.HTTPError{Code: http.StatusBadRequest, Message: "invalid email or password"}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save user
|
||||||
|
db := h.DB.Clone()
|
||||||
|
defer db.Close()
|
||||||
|
if err = db.DB("twitter").C("users").Insert(u); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.JSON(http.StatusCreated, u)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) Login(c echo.Context) (err error) {
|
||||||
|
// Bind
|
||||||
|
u := new(model.User)
|
||||||
|
if err = c.Bind(u); err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find user
|
||||||
|
db := h.DB.Clone()
|
||||||
|
defer db.Close()
|
||||||
|
if err = db.DB("twitter").C("users").
|
||||||
|
Find(bson.M{"email": u.Email, "password": u.Password}).One(u); err != nil {
|
||||||
|
if err == mgo.ErrNotFound {
|
||||||
|
return &echo.HTTPError{Code: http.StatusUnauthorized, Message: "invalid email or password"}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
//-----
|
||||||
|
// JWT
|
||||||
|
//-----
|
||||||
|
|
||||||
|
// Create token
|
||||||
|
token := jwt.New(jwt.SigningMethodHS256)
|
||||||
|
|
||||||
|
// Set claims
|
||||||
|
claims := token.Claims.(jwt.MapClaims)
|
||||||
|
claims["id"] = u.ID
|
||||||
|
claims["exp"] = time.Now().Add(time.Hour * 72).Unix()
|
||||||
|
|
||||||
|
// Generate encoded token and send it as response
|
||||||
|
u.Token, err = token.SignedString([]byte(Key))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
u.Password = "" // Don't send password
|
||||||
|
return c.JSON(http.StatusOK, u)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) Follow(c echo.Context) (err error) {
|
||||||
|
userID := userIDFromToken(c)
|
||||||
|
id := c.Param("id")
|
||||||
|
|
||||||
|
// Add a follower to user
|
||||||
|
db := h.DB.Clone()
|
||||||
|
defer db.Close()
|
||||||
|
if err = db.DB("twitter").C("users").
|
||||||
|
UpdateId(bson.ObjectIdHex(id), bson.M{"$addToSet": bson.M{"followers": userID}}); err != nil {
|
||||||
|
if err == mgo.ErrNotFound {
|
||||||
|
return echo.ErrNotFound
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func userIDFromToken(c echo.Context) string {
|
||||||
|
user := c.Get("user").(*jwt.Token)
|
||||||
|
claims := user.Claims.(jwt.MapClaims)
|
||||||
|
return claims["id"].(string)
|
||||||
|
}
|
12
vendor/github.com/labstack/echo/cookbook/twitter/model/post.go
generated
vendored
Normal file
12
vendor/github.com/labstack/echo/cookbook/twitter/model/post.go
generated
vendored
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import "gopkg.in/mgo.v2/bson"
|
||||||
|
|
||||||
|
type (
|
||||||
|
Post struct {
|
||||||
|
ID bson.ObjectId `json:"id" bson:"_id,omitempty"`
|
||||||
|
To string `json:"to" bson:"to"`
|
||||||
|
From string `json:"from" bson:"from"`
|
||||||
|
Message string `json:"message" bson:"message"`
|
||||||
|
}
|
||||||
|
)
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user