mirror of
https://github.com/coredns/coredns.git
synced 2025-11-27 06:04:03 -05:00
@@ -1,60 +0,0 @@
|
|||||||
// Package storage abstracts away where middleware can store assests (zones, keys, etc).
|
|
||||||
package storage
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
"path/filepath"
|
|
||||||
"runtime"
|
|
||||||
)
|
|
||||||
|
|
||||||
// dir wraps an http.Dir that restrict file access to a specific directory tree, see http.Dir's documentation
|
|
||||||
// for methods for accessing files.
|
|
||||||
type dir http.Dir
|
|
||||||
|
|
||||||
// CoreDir is the directory where middleware can store assets, like zone files after a zone transfer
|
|
||||||
// or public and private keys or anything else a middleware might need. The convention is to place
|
|
||||||
// assets in a subdirectory named after the zone prefixed with "D", to prevent the root zone become a hidden directory.
|
|
||||||
//
|
|
||||||
// Dexample.org/Kexample.org<something>.key
|
|
||||||
//
|
|
||||||
// Note that subzone(s) under example.org are places in the own directory under CoreDir:
|
|
||||||
//
|
|
||||||
// Dexample.org/...
|
|
||||||
// Db.example.org/...
|
|
||||||
//
|
|
||||||
// CoreDir will default to "$HOME/.coredns" on Unix, but it's location can be overridden with the COREDNSPATH
|
|
||||||
// environment variable.
|
|
||||||
var CoreDir = dir(fsPath())
|
|
||||||
|
|
||||||
func (d dir) Zone(z string) dir {
|
|
||||||
if z != "." && z[len(z)-2] == '.' {
|
|
||||||
return dir(path.Join(string(d), "D"+z[:len(z)-1]))
|
|
||||||
}
|
|
||||||
return dir(path.Join(string(d), "D"+z))
|
|
||||||
}
|
|
||||||
|
|
||||||
// fsPath returns the path to the directory where the application may store data.
|
|
||||||
// If COREDNSPATH env variable. is set, that value is used. Otherwise, the path is
|
|
||||||
// the result of evaluating "$HOME/.coredns".
|
|
||||||
func fsPath() string {
|
|
||||||
if corePath := os.Getenv("COREDNSPATH"); corePath != "" {
|
|
||||||
return corePath
|
|
||||||
}
|
|
||||||
return filepath.Join(userHomeDir(), ".coredns")
|
|
||||||
}
|
|
||||||
|
|
||||||
// userHomeDir returns the user's home directory according to environment variables.
|
|
||||||
//
|
|
||||||
// Credit: http://stackoverflow.com/a/7922977/1048862
|
|
||||||
func userHomeDir() string {
|
|
||||||
if runtime.GOOS == "windows" {
|
|
||||||
home := os.Getenv("HOMEDRIVE") + os.Getenv("HOMEPATH")
|
|
||||||
if home == "" {
|
|
||||||
home = os.Getenv("USERPROFILE")
|
|
||||||
}
|
|
||||||
return home
|
|
||||||
}
|
|
||||||
return os.Getenv("HOME")
|
|
||||||
}
|
|
||||||
@@ -1,42 +0,0 @@
|
|||||||
package storage
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
"path"
|
|
||||||
"strings"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestFsPath(t *testing.T) {
|
|
||||||
if actual := fsPath(); !strings.HasSuffix(actual, ".coredns") {
|
|
||||||
t.Errorf("Expected path to be a .coredns folder, got: %v", actual)
|
|
||||||
}
|
|
||||||
|
|
||||||
os.Setenv("COREDNSPATH", "testpath")
|
|
||||||
defer os.Setenv("COREDNSPATH", "")
|
|
||||||
if actual, expected := fsPath(), "testpath"; actual != expected {
|
|
||||||
t.Errorf("Expected path to be %v, got: %v", expected, actual)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestZone(t *testing.T) {
|
|
||||||
for _, ts := range []string{"example.org.", "example.org"} {
|
|
||||||
d := CoreDir.Zone(ts)
|
|
||||||
actual := path.Base(string(d))
|
|
||||||
expected := "D" + ts
|
|
||||||
if actual != expected {
|
|
||||||
t.Errorf("Expected path to be %v, got %v", actual, expected)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestZoneRoot(t *testing.T) {
|
|
||||||
for _, ts := range []string{"."} {
|
|
||||||
d := CoreDir.Zone(ts)
|
|
||||||
actual := path.Base(string(d))
|
|
||||||
expected := "D" + ts
|
|
||||||
if actual != expected {
|
|
||||||
t.Errorf("Expected path to be %v, got %v", actual, expected)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user