From 72473d51710c65283e6ad4e76050be32a5d5720d Mon Sep 17 00:00:00 2001 From: Grégoire Duchêne Date: Sun, 12 Apr 2020 12:40:47 +0100 Subject: Put both entry points in the same directory --- cmd/gcp/main.go | 81 --------------------------------------------------------- 1 file changed, 81 deletions(-) delete mode 100644 cmd/gcp/main.go (limited to 'cmd/gcp') diff --git a/cmd/gcp/main.go b/cmd/gcp/main.go deleted file mode 100644 index c726408..0000000 --- a/cmd/gcp/main.go +++ /dev/null @@ -1,81 +0,0 @@ -// 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" - "log" - "net" - "net/http" - "os" - "os/signal" - "path" - "syscall" - "time" - - "go.awhk.org/go-import-redirect/internal" -) - -func redirect(resp http.ResponseWriter, req *http.Request) { - if req.Method != "GET" { - resp.Header()["Allow"] = []string{"GET"} - resp.WriteHeader(http.StatusMethodNotAllowed) - return - } - - pkg := path.Join(req.Host, req.URL.Path) - resp.Header()["Content-Type"] = []string{"text/html; charset=utf-8"} - if v, ok := req.URL.Query()["go-get"]; ok && len(v) > 0 && v[0] == "1" { - resp.WriteHeader(http.StatusOK) - } else { - resp.Header()["Location"] = []string{"https://godoc.org/" + pkg} - resp.WriteHeader(http.StatusFound) - } - resp.Write([]byte(internal.GetBody(pkg))) -} - -func main() { - mux := http.NewServeMux() - mux.HandleFunc("/", redirect) - srv := http.Server{Handler: mux} - - done := make(chan os.Signal, 1) - signal.Notify(done, os.Interrupt, syscall.SIGINT, syscall.SIGTERM) - - go func() { - var ( - addr = os.Getenv("ADDR") - l net.Listener - err error - ) - if addr != "" && addr[0] == '/' { - if l, err = net.Listen("unix", addr); err != nil { - log.Fatalln("net.Listen:", err) - } - // We do not do any authorization anyway, so 0666 makes sense here. - if err = os.Chmod(addr, 0666); err != nil { - log.Fatalln("os.Chmod:", err) - } - } else { - if addr == "" { - addr = ":8080" - } - if l, err = net.Listen("tcp", addr); err != nil { - log.Fatalln("net.Listen:", err) - } - } - if err = srv.Serve(l); err != nil && err != http.ErrServerClosed { - log.Fatalln("server.ListenAndServe:", err) - } - }() - - <-done - ctx, cancel := context.WithTimeout(context.Background(), time.Second) - defer cancel() - if err := srv.Shutdown(ctx); err != nil && err != http.ErrServerClosed { - log.Fatalln("server.Shutdown:", err) - } -} -- cgit v1.2.3-70-g09d2