mirror of
https://github.com/gomods/athens
synced 2026-02-03 11:00:32 +00:00
d26b99d41c
* Upgrade buffalo * Switch to go modules everywhere * Fixes from buffalo fix * Add missing modules from module list * Update vendored modules in /vendor * Stop using vendor directory for tests * Check go.mod and go.sum files on verify * Upgrade Buffalo from v0.13.0 to v0.13.1 * Fix test for new Buffalo Allow for new Buffalo code * Add test for endpoint with trailing slash
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package buffalo
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
"text/tabwriter"
|
|
|
|
"github.com/gobuffalo/x/randx"
|
|
"github.com/markbates/grift/grift"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
// Grifts decorates the app with tasks
|
|
func Grifts(app *App) {
|
|
routesGrift(app)
|
|
middlewareGrift(app)
|
|
secretGrift()
|
|
}
|
|
|
|
func secretGrift() {
|
|
grift.Desc("secret", "Generate a cryptographically secure secret key")
|
|
grift.Add("secret", func(c *grift.Context) error {
|
|
bb := []byte{}
|
|
for i := 0; i < 4; i++ {
|
|
b := []byte(randx.String(64))
|
|
b, err := bcrypt.GenerateFromPassword(b, bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return errors.WithStack(err)
|
|
}
|
|
bb = append(bb, b...)
|
|
}
|
|
rx := regexp.MustCompile(`(\W+)`)
|
|
bb = rx.ReplaceAll(bb, []byte(""))
|
|
s := randx.String(6) + string(bb)
|
|
fmt.Println(s[:127])
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func middlewareGrift(a *App) {
|
|
grift.Desc("middleware", "Prints out your middleware stack")
|
|
grift.Add("middleware", func(c *grift.Context) error {
|
|
printMiddleware(a)
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func printMiddleware(a *App) {
|
|
fmt.Printf("-> %s\n", a.Name)
|
|
fmt.Printf("%v\n", a.Middleware.String())
|
|
for _, x := range a.children {
|
|
printMiddleware(x)
|
|
}
|
|
}
|
|
|
|
func routesGrift(a *App) {
|
|
grift.Desc("routes", "Print out all defined routes")
|
|
grift.Add("routes", func(c *grift.Context) error {
|
|
routes := a.Routes()
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', tabwriter.Debug)
|
|
fmt.Fprintln(w, "METHOD\t PATH\t ALIASES\t NAME\t HANDLER")
|
|
fmt.Fprintln(w, "------\t ----\t -------\t ----\t -------")
|
|
for _, r := range routes {
|
|
fmt.Fprintf(w, "%s\t %s\t %s\t %s\t %s\n", r.Method, r.Path, strings.Join(r.Aliases, " "), r.PathName, r.HandlerName)
|
|
}
|
|
w.Flush()
|
|
return nil
|
|
})
|
|
}
|