Skip to content

Commit

Permalink
Add an introduction page as the home page (#1945)
Browse files Browse the repository at this point in the history
A default homepage is baked into the server that uses the request host address, or in HTTP 2 the authority. This includes ports. It also checks for schema. The values are used to indicate to users how to configure their go env

Of course, this won't work on all installations - especially enterprise ones. For that, we've introduced ATHENS_HOME_TEMPLATE_PATH as an environment variable along with HomeTemplatePath in the config. This value defaults to /var/lib/athens/home.html but can be configured to any location that Athens can reliably read from. This is a Go HTML template so it should use Go HTML template formatting and logic.
  • Loading branch information
matt0x6F authored Apr 20, 2024
1 parent 8d9b767 commit 359c119
Show file tree
Hide file tree
Showing 7 changed files with 291 additions and 31 deletions.
2 changes: 1 addition & 1 deletion cmd/proxy/actions/app_proxy.go
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ func addProxyRoutes(
l *log.Logger,
c *config.Config,
) error {
r.HandleFunc("/", proxyHomeHandler)
r.HandleFunc("/", proxyHomeHandler(c))
r.HandleFunc("/healthz", healthHandler)
r.HandleFunc("/readyz", getReadinessHandler(s))
r.HandleFunc("/version", versionHandler)
Expand Down
44 changes: 33 additions & 11 deletions cmd/proxy/actions/app_proxy_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ import (
"net/http/httptest"
"strings"
"testing"
"text/template"

"github.com/gomods/athens/pkg/build"
"github.com/gomods/athens/pkg/config"
Expand All @@ -21,7 +22,7 @@ type routeTest struct {
method string
path string
body string
test func(t *testing.T, resp *http.Response)
test func(t *testing.T, req *http.Request, resp *http.Response)
}

func TestProxyRoutes(t *testing.T) {
Expand All @@ -40,22 +41,43 @@ func TestProxyRoutes(t *testing.T) {
baseURL := "https://athens.azurefd.net" + c.PathPrefix

testCases := []routeTest{
{"GET", "/", "", func(t *testing.T, resp *http.Response) {
{"GET", "/", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
body, err := io.ReadAll(resp.Body)
require.NoError(t, err)
assert.Equal(t, `"Welcome to The Athens Proxy"`, string(body))
tmp, err := template.New("home").Parse(homepage)
assert.NoError(t, err)

var templateData = make(map[string]string)

templateData["Host"] = req.Host

if !strings.HasPrefix(templateData["Host"], "http://") && !strings.HasPrefix(templateData["Host"], "https://") {
if req.TLS != nil {
templateData["Host"] = "https://" + templateData["Host"]
} else {
templateData["Host"] = "http://" + templateData["Host"]
}
}

templateData["NoSumPatterns"] = strings.Join(c.NoSumPatterns, ",")

var expected strings.Builder
err = tmp.ExecuteTemplate(&expected, "home", templateData)
require.NoError(t, err)

assert.Equal(t, expected.String(), string(body))
}},
{"GET", "/badz", "", func(t *testing.T, resp *http.Response) {
{"GET", "/badz", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
}},
{"GET", "/healthz", "", func(t *testing.T, resp *http.Response) {
{"GET", "/healthz", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
}},
{"GET", "/readyz", "", func(t *testing.T, resp *http.Response) {
{"GET", "/readyz", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
}},
{"GET", "/version", "", func(t *testing.T, resp *http.Response) {
{"GET", "/version", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
details := build.Details{}
err := json.NewDecoder(resp.Body).Decode(&details)
Expand All @@ -64,13 +86,13 @@ func TestProxyRoutes(t *testing.T) {
}},

// Default sumdb is sum.golang.org
{"GET", "/sumdb/sum.golang.org/supported", "", func(t *testing.T, resp *http.Response) {
{"GET", "/sumdb/sum.golang.org/supported", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusOK, resp.StatusCode)
}},
{"GET", "/sumdb/sum.rust-lang.org/supported", "", func(t *testing.T, resp *http.Response) {
{"GET", "/sumdb/sum.rust-lang.org/supported", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusNotFound, resp.StatusCode)
}},
{"GET", "/sumdb/sum.golang.org/lookup/github.com/gomods/athens", "", func(t *testing.T, resp *http.Response) {
{"GET", "/sumdb/sum.golang.org/lookup/github.com/gomods/athens", "", func(t *testing.T, req *http.Request, resp *http.Response) {
assert.Equal(t, http.StatusForbidden, resp.StatusCode)
}},
}
Expand All @@ -84,7 +106,7 @@ func TestProxyRoutes(t *testing.T) {
t.Run(req.RequestURI, func(t *testing.T) {
w := httptest.NewRecorder()
r.ServeHTTP(w, req)
tc.test(t, w.Result())
tc.test(t, req, w.Result())
})
}

Expand Down
132 changes: 130 additions & 2 deletions cmd/proxy/actions/home.go
Original file line number Diff line number Diff line change
@@ -1,9 +1,137 @@
package actions

import (
"errors"
"html/template"
"net/http"
"os"
"strings"

"github.com/gomods/athens/pkg/config"
"github.com/gomods/athens/pkg/log"
)

func proxyHomeHandler(w http.ResponseWriter, r *http.Request) {
_, _ = w.Write([]byte(`"Welcome to The Athens Proxy"`))
const homepage = `<!DOCTYPE html>
<html>
<head>
<title>Athens</title>
<style>
body {
font-family: Arial, sans-serif;
margin: 20px;
}
pre {
background-color: #f4f4f4;
padding: 5px;
border-radius: 5px;
width: fit-content;
padding: 10px;
}
code {
background-color: #f4f4f4;
padding: 5px;
border-radius: 5px;
}
</style>
</head>
<body>
<h1>Welcome to Athens</h1>
<h2>Configuring your client</h2>
<pre>GOPROXY={{ .Host }},direct</pre>
{{ if .NoSumPatterns }}
<h3>Excluding checksum database</h3>
<p>Use the following GONOSUM environment variable to exclude checksum database:</p>
<pre>GONOSUM={{ .NoSumPatterns }}</pre>
{{ end }}
<h2>How to use the Athens API</h2>
<p>Use the <a href="/catalog">catalog</a> endpoint to get a list of all modules in the proxy</p>
<h3>List of versions</h3>
<p>This endpoint returns a list of versions that Athens knows about for <code>acidburn/htp</code>:</p>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/list</pre>
<h3>Version info</h3>
<p>This endpoint returns information about a specific version of a module:</p>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/v1.0.0.info</pre>
<p>This returns JSON with information about v1.0.0. It looks like this:
<pre>{
"Name": "v1.0.0",
"Short": "v1.0.0",
"Version": "v1.0.0",
"Time": "1972-07-18T12:34:56Z"
}</pre>
<h3>go.mod file</h3>
<p>This endpoint returns the go.mod file for a specific version of a module:</p>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/v1.0.0.mod</pre>
<p>This returns the go.mod file for version v1.0.0. If {{ .Host }}/github.com/acidburn/htp version v1.0.0 has no dependencies, the response body would look like this:</p>
<pre>module github.com/acidburn/htp</pre>
<h3>Module sources</h3>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/v1.0.0.zip</pre>
<p>This is what it sounds like — it sends back a zip file with the source code for the module in version v1.0.0.</p>
<h3>Latest</h3>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@latest</pre>
<p>This endpoint returns the latest version of the module. If the version does not exist it should retrieve the hash of latest commit.</p>
</body>
</html>
`

func proxyHomeHandler(c *config.Config) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
lggr := log.EntryFromContext(r.Context())

templateData := make(map[string]string)

templateContents := homepage

// load the template from the file system if it exists, otherwise revert to default
rawTemplateFileContents, err := os.ReadFile(c.HomeTemplatePath)
if err != nil {
if !errors.Is(err, os.ErrNotExist) {
// this is some other error, log it and revert to default
lggr.SystemErr(err)
}
} else {
templateContents = string(rawTemplateFileContents)
}

// This should be correct in most cases. If it is not, users can supply their own template
templateData["Host"] = r.Host

// if the host does not have a scheme, add one based on the request
if !strings.HasPrefix(templateData["Host"], "http://") && !strings.HasPrefix(templateData["Host"], "https://") {
if r.TLS != nil {
templateData["Host"] = "https://" + templateData["Host"]
} else {
templateData["Host"] = "http://" + templateData["Host"]
}
}

templateData["NoSumPatterns"] = strings.Join(c.NoSumPatterns, ",")

tmp, err := template.New("home").Parse(templateContents)
if err != nil {
lggr.SystemErr(err)
w.WriteHeader(http.StatusInternalServerError)
}

w.Header().Add("Content-Type", "text/html")
w.WriteHeader(http.StatusOK)

err = tmp.ExecuteTemplate(w, "home", templateData)
if err != nil {
lggr.SystemErr(err)
w.WriteHeader(http.StatusInternalServerError)
}
}
}
4 changes: 4 additions & 0 deletions config.dev.toml
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,10 @@ BasicAuthUser = ""
# Env override: BASIC_AUTH_PASS
BasicAuthPass = ""

# A path on disk to a Go HTML template to be used on the homepage
# Env override: ATHENS_HOME_TEMPLATE_PATH
HomeTemplatePath = "/var/lib/athens/home.html"

# Set to true to force an SSL redirect
# Env override: PROXY_FORCE_SSL
ForceSSL = false
Expand Down
101 changes: 101 additions & 0 deletions docs/content/configuration/home-template.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,101 @@
---
title: Home template configuration
description: How to customize the home template
weight: 8
---

As of v0.14.0 Athens ships with a default, minimal HTML home page that advises users on how to connect to the proxy. It factors in whether `GoNoSumPatterns` is configured, and attempts
to build configuration for `GO_PROXY`. It relies on the users request Host header (on HTTP 1.1) or the Authority header (on HTTP 2) as well as whether the request was over TLS to advise
on configuring `GO_PROXY`. Lastly, the homepage provides a quick guide on how users can leverage the Athens API.

Of course, not all instructions will be this simple. Some installations may be reachable at different addresses in CI than for desktop users. In this case, and others where the default
home page does not make sense it is possible to override the template.

Do so by configuring `HomeTemplatePath` via the config or `ATHENS_HOME_TEMPLATE_PATH` to a location on disk with a Go HTML template or placing a template file at `/var/lib/athens/home.html`.

Athens automatically injects the following variables in templates:

| Setting | Source |
| :------ | :----- |
| `Host` | Built from the request Host (HTTP1) or Authority (HTTP2) header and presence of TLS. Includes ports. |
| `NoSumPatterns` | Comes directly from the configuration. |

Using these values is done by wrapping them in bracers with a dot prepended. Example: `{{ .Host }}`

For more advanced formatting read more about [Go HTML templates](https://pkg.go.dev/html/template).

```html
<!DOCTYPE html>
<html>
<head>
<title>Athens</title>
<style>
body {
font-family: Arial, sans-serif;
margin: 20px;
}
pre {
background-color: #f4f4f4;
padding: 5px;
border-radius: 5px;
width: fit-content;
padding: 10px;
}
code {
background-color: #f4f4f4;
padding: 5px;
border-radius: 5px;
}
</style>
</head>
<body>

<h1>Welcome to Athens</h1>

<h2>Configuring your client</h2>
<pre>GOPROXY={{ .Host }},direct</pre>
{{ if .NoSumPatterns }}
<h3>Excluding checksum database</h3>
<p>Use the following GONOSUM environment variable to exclude checksum database:</p>
<pre>GONOSUM={{ .NoSumPatterns }}</pre>
{{ end }}

<h2>How to use the Athens API</h2>
<p>Use the <a href="/catalog">catalog</a> endpoint to get a list of all modules in the proxy</p>

<h3>List of versions</h3>
<p>This endpoint returns a list of versions that Athens knows about for <code>acidburn/htp</code>:</p>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/list</pre>

<h3>Version info</h3>
<p>This endpoint returns information about a specific version of a module:</p>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/v1.0.0.info</pre>
<p>This returns JSON with information about v1.0.0. It looks like this:
<pre>{
"Name": "v1.0.0",
"Short": "v1.0.0",
"Version": "v1.0.0",
"Time": "1972-07-18T12:34:56Z"
}</pre>

<h3>go.mod file</h3>
<p>This endpoint returns the go.mod file for a specific version of a module:</p>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/v1.0.0.mod</pre>
<p>This returns the go.mod file for version v1.0.0. If {{ .Host }}/github.com/acidburn/htp version v1.0.0 has no dependencies, the response body would look like this:</p>
<pre>module github.com/acidburn/htp</pre>

<h3>Module sources</h3>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@v/v1.0.0.zip</pre>
<p>This is what it sounds like — it sends back a zip file with the source code for the module in version v1.0.0.</p>

<h3>Latest</h3>
<pre>GET {{ .Host }}/github.com/acidburn/htp/@latest</pre>
<p>This endpoint returns the latest version of the module. If the version does not exist it should retrieve the hash of latest commit.</p>

</body>
</html>
```
2 changes: 2 additions & 0 deletions pkg/config/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ type Config struct {
UnixSocket string `envconfig:"ATHENS_UNIX_SOCKET"`
BasicAuthUser string `envconfig:"BASIC_AUTH_USER"`
BasicAuthPass string `envconfig:"BASIC_AUTH_PASS"`
HomeTemplatePath string `envconfig:"ATHENS_HOME_TEMPLATE_PATH"`
ForceSSL bool `envconfig:"PROXY_FORCE_SSL"`
ValidatorHook string `envconfig:"ATHENS_PROXY_VALIDATOR"`
PathPrefix string `envconfig:"ATHENS_PATH_PREFIX"`
Expand Down Expand Up @@ -157,6 +158,7 @@ func defaultConfig() *Config {
PprofPort: ":3001",
StatsExporter: "prometheus",
TimeoutConf: TimeoutConf{Timeout: 300},
HomeTemplatePath: "/var/lib/athens/home.html",
StorageType: "memory",
Port: ":3000",
SingleFlightType: "memory",
Expand Down
Loading

0 comments on commit 359c119

Please sign in to comment.