From 286b6bbf425e204219dc4df27b06f25d1b525889 Mon Sep 17 00:00:00 2001 From: Grégoire Duchêne Date: Thu, 22 Aug 2019 23:20:54 +0100 Subject: First version --- main.go | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 55 insertions(+) create mode 100644 main.go (limited to 'main.go') diff --git a/main.go b/main.go new file mode 100644 index 0000000..97a4b55 --- /dev/null +++ b/main.go @@ -0,0 +1,55 @@ +// Copyright (c) 2019, Grégoire Duchêne +// +// Use of this source code is governed by the ISC license that can be +// found in the LICENSE file. + +package main + +import ( + "context" + "fmt" + "github.com/aws/aws-lambda-go/events" + "github.com/aws/aws-lambda-go/lambda" + "os" + "path" + "strings" +) + +func getDest(dest, repo string) string { + dest = strings.TrimRight(dest, "/") + return fmt.Sprintf("%s/%s", dest, repo) +} + +func getRepo(pkg, prefix string) string { + prefix = strings.TrimRight(prefix, "/") + path := strings.TrimLeft(strings.TrimPrefix(pkg, prefix), "/") + return strings.Split(path, "/")[0] +} + +func redirect(ctx context.Context, req events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) { + var ( + pkg = path.Join(req.Headers["Host"], req.Path) + dest = getDest(os.Getenv("DEST"), getRepo(pkg, os.Getenv("PREFIX"))) + doc = fmt.Sprintf("https://godoc.org/%s", pkg) + vcs = os.Getenv("VCS") + body = fmt.Sprintf(` + +go-import-redirect +`, pkg, vcs, dest) + resp = events.APIGatewayProxyResponse{ + Body: body, + Headers: map[string]string{"Content-Type": "text/html; charset=utf-8"}, + } + ) + if v, ok := req.QueryStringParameters["go-get"]; ok && v == "1" { + resp.StatusCode = 200 + } else { + resp.Headers["Location"] = doc + resp.StatusCode = 302 + } + return resp, nil +} + +func main() { + lambda.Start(redirect) +} -- cgit v1.2.3-70-g09d2