gomod easter egg init
This commit is contained in:
parent
8ae5522f68
commit
048e87dfa8
3 changed files with 149 additions and 0 deletions
1
internal/gomod/go.test.mod
Normal file
1
internal/gomod/go.test.mod
Normal file
|
|
@ -0,0 +1 @@
|
|||
WTFIsThis_0.0.0-20190101000000-000000000000
|
||||
19
internal/gomod/htmlTemplate.go
Normal file
19
internal/gomod/htmlTemplate.go
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
package gomod
|
||||
|
||||
import "text/template"
|
||||
|
||||
type goGetHtmlTemplateCtx struct {
|
||||
ModulePath string
|
||||
ProxyBase string
|
||||
}
|
||||
|
||||
var goGetHtmlTemplate = template.Must(template.New("tpl").Parse(`<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="utf-8">
|
||||
<meta name="go-import" content="{{.ModulePath}} mod {{.ProxyBase}}">
|
||||
</head>
|
||||
<body>
|
||||
go get {{.ModulePath}}
|
||||
</body>
|
||||
</html>`))
|
||||
129
internal/gomod/resolve.go
Normal file
129
internal/gomod/resolve.go
Normal file
|
|
@ -0,0 +1,129 @@
|
|||
package gomod
|
||||
|
||||
import (
|
||||
"archive/zip"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"net/url"
|
||||
"path"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/eternal-flame-AD/yoake/internal/echoerror"
|
||||
"github.com/labstack/echo/v4"
|
||||
"golang.org/x/mod/module"
|
||||
)
|
||||
|
||||
type Info struct {
|
||||
Version string // version string
|
||||
Time time.Time // commit time
|
||||
}
|
||||
|
||||
const backendProxy = "https://proxy.golang.org/"
|
||||
|
||||
func random143String() string {
|
||||
randItems := []string{
|
||||
"143",
|
||||
"143.",
|
||||
"4.3",
|
||||
"143",
|
||||
".143",
|
||||
"1.43",
|
||||
"1.4.3",
|
||||
"14.3",
|
||||
"omo",
|
||||
"om.o",
|
||||
"o.m.o",
|
||||
"o.mo",
|
||||
}
|
||||
|
||||
var ret strings.Builder
|
||||
ret.WriteString("v1.4.3-")
|
||||
for i := 0; i < 10; i++ {
|
||||
rand.Intn(len(randItems))
|
||||
ret.WriteString(randItems[rand.Intn(len(randItems))])
|
||||
}
|
||||
ret.WriteString("+incompatible")
|
||||
return ret.String()
|
||||
}
|
||||
|
||||
func resolveModule(c echo.Context, modPathUnesc string, pRequest string) error {
|
||||
modPath, err := module.UnescapePath(modPathUnesc)
|
||||
if err != nil {
|
||||
return echoerror.NewHttp(400, fmt.Errorf("invalid module path: %w", err))
|
||||
}
|
||||
if !strings.HasPrefix(modPath[strings.IndexRune(modPath, '/')+1:], "test-") {
|
||||
return echoerror.NewHttp(400, fmt.Errorf("please prefix your module path with 'test-'"))
|
||||
}
|
||||
if err != nil {
|
||||
return echoerror.NewHttp(400, fmt.Errorf("invalid module path: %v", err))
|
||||
}
|
||||
if pRequest == "list" {
|
||||
return c.String(143, random143String()+"\n")
|
||||
}
|
||||
ext := path.Ext(pRequest)
|
||||
version := strings.TrimSuffix(pRequest, ext)
|
||||
if err := module.Check(modPath, version); err != nil {
|
||||
return echoerror.NewHttp(400, fmt.Errorf("invalid version: %v", err))
|
||||
}
|
||||
switch ext {
|
||||
case ".info":
|
||||
return c.JSON(143, Info{
|
||||
Version: version,
|
||||
Time: time.Now().Add(-time.Hour),
|
||||
})
|
||||
case ".mod":
|
||||
return c.String(143, strings.Repeat("\n", 143-1)+"Welcome.to.white.space.\n")
|
||||
case ".zip":
|
||||
zipFile := zip.NewWriter(c.Response().Writer)
|
||||
mainGo, err := zipFile.Create(fmt.Sprintf("%s@%s/main.go", modPath, version))
|
||||
if err != nil {
|
||||
zipFile.Close()
|
||||
}
|
||||
_, err = mainGo.Write([]byte(("package main\n\nimport \"fmt\"\n\nfunc main() {\n\n}\n")))
|
||||
if err != nil {
|
||||
zipFile.Close()
|
||||
return echoerror.NewHttp(500, fmt.Errorf("failed to write main.go: %v", err))
|
||||
}
|
||||
c.Response().Status = 143
|
||||
return zipFile.Close()
|
||||
}
|
||||
return c.Redirect(302, backendProxy+modPathUnesc+"/@v/"+pRequest)
|
||||
}
|
||||
|
||||
func Register(baseURI string, baseG *echo.Group) (gogetMiddleware echo.MiddlewareFunc) {
|
||||
baseG.GET("*", func(c echo.Context) error {
|
||||
fullURI := c.Request().RequestURI
|
||||
if !strings.HasPrefix(fullURI, baseURI) {
|
||||
return echo.ErrNotFound
|
||||
}
|
||||
fullURI = strings.TrimPrefix(fullURI, baseURI)
|
||||
fullURIUnEscaped, err := url.PathUnescape(fullURI)
|
||||
if err != nil {
|
||||
return echoerror.NewHttp(400, fmt.Errorf("invalid URI: %v", err))
|
||||
}
|
||||
if strings.Contains(fullURIUnEscaped, "/@v/") {
|
||||
split := strings.SplitN(fullURIUnEscaped, "/@v/", 2)
|
||||
return resolveModule(c, strings.TrimPrefix(split[0], "/"), split[1])
|
||||
}
|
||||
return echo.ErrNotFound
|
||||
})
|
||||
|
||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||
return func(c echo.Context) error {
|
||||
if c.Request().Method == "GET" && c.QueryParam("go-get") == "1" {
|
||||
unescapedPath, err := url.PathUnescape(c.Request().URL.Path)
|
||||
if err != nil {
|
||||
return echoerror.NewHttp(400, fmt.Errorf("invalid module path: %v", err))
|
||||
}
|
||||
ctx := goGetHtmlTemplateCtx{
|
||||
ModulePath: c.Request().Host + unescapedPath,
|
||||
ProxyBase: c.Scheme() + "://" + c.Request().Host + baseURI,
|
||||
}
|
||||
|
||||
return goGetHtmlTemplate.Execute(c.Response().Writer, ctx)
|
||||
}
|
||||
return next(c)
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue