mirror of
https://github.com/golang/go
synced 2024-10-01 08:28:43 -06:00
bd17c084df
This change replaces the env package with a new golangorgenv package.
The previous env package existed primarily to configure the godoc
instance that was running golang.org. By now, the golang.org website
has been factored out to x/website, which has its own env package,
but ends up still using this env package indirectly via x/tools/godoc.
The goal of this change is to make env available for other services
that run on subdomains of golang.org, so they can continue to safely
rely on the x/tools/playground, which will be modified in the next
commit to also use the new golangorgenv.
The golangorgenv package replaces the IsProd function with a more
specific one. Start using it in packages x/tools/{,cmd}/godoc. Also,
re-arrange the order of checks to give the host suffix check higher
priority than the environment variable check. This way, if the
environment variable isn't set, the host suffix check gets a chance
to run.
When getting the value of "X-AppEngine-Country" header, use its
canonical format "X-Appengine-Country" to avoid an allocation.
This does not change behavior.
Updates golang/go#30486
Change-Id: I97b47211a45ca0351f31fcb4fa6d408a4b0c4c7c
Reviewed-on: https://go-review.googlesource.com/c/tools/+/165459
Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
// Copyright 2018 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// Package golangorgenv provides environment information for programs running at
|
|
// golang.org and its subdomains.
|
|
package golangorgenv
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
var (
|
|
checkCountry = boolEnv("GOLANGORG_CHECK_COUNTRY")
|
|
enforceHosts = boolEnv("GOLANGORG_ENFORCE_HOSTS")
|
|
)
|
|
|
|
// CheckCountry reports whether country restrictions should be enforced.
|
|
func CheckCountry() bool {
|
|
return checkCountry
|
|
}
|
|
|
|
// EnforceHosts reports whether host filtering should be enforced.
|
|
func EnforceHosts() bool {
|
|
return enforceHosts
|
|
}
|
|
|
|
func boolEnv(key string) bool {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
// TODO(dmitshur): In the future, consider detecting if running in App Engine,
|
|
// and if so, making the environment variables mandatory rather than optional.
|
|
return false
|
|
}
|
|
b, err := strconv.ParseBool(v)
|
|
if err != nil {
|
|
log.Fatalf("environment variable %s (%q) must be a boolean", key, v)
|
|
}
|
|
return b
|
|
}
|