mirror of https://github.com/coder/coder.git
130 lines
2.7 KiB
Go
130 lines
2.7 KiB
Go
package config
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/kirsle/configdir"
|
|
"golang.org/x/xerrors"
|
|
)
|
|
|
|
const (
|
|
FlagName = "global-config"
|
|
)
|
|
|
|
// Root represents the configuration directory.
|
|
type Root string
|
|
|
|
// mustNotBeEmpty prevents us from accidentally writing configuration to the
|
|
// current directory. This is primarily valuable in development, where we may
|
|
// accidentally use an empty root.
|
|
func (r Root) mustNotEmpty() {
|
|
if r == "" {
|
|
panic("config root must not be empty")
|
|
}
|
|
}
|
|
|
|
func (r Root) Session() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(string(r), "session"))
|
|
}
|
|
|
|
// ReplicaID is a unique identifier for the Coder server.
|
|
func (r Root) ReplicaID() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(string(r), "replica_id"))
|
|
}
|
|
|
|
func (r Root) URL() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(string(r), "url"))
|
|
}
|
|
|
|
func (r Root) Organization() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(string(r), "organization"))
|
|
}
|
|
|
|
func (r Root) DotfilesURL() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(string(r), "dotfilesurl"))
|
|
}
|
|
|
|
func (r Root) PostgresPath() string {
|
|
r.mustNotEmpty()
|
|
return filepath.Join(string(r), "postgres")
|
|
}
|
|
|
|
func (r Root) PostgresPassword() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(r.PostgresPath(), "password"))
|
|
}
|
|
|
|
func (r Root) PostgresPort() File {
|
|
r.mustNotEmpty()
|
|
return File(filepath.Join(r.PostgresPath(), "port"))
|
|
}
|
|
|
|
// File provides convenience methods for interacting with *os.File.
|
|
type File string
|
|
|
|
// Delete deletes the file.
|
|
func (f File) Delete() error {
|
|
if f == "" {
|
|
return xerrors.Errorf("empty file path")
|
|
}
|
|
return os.Remove(string(f))
|
|
}
|
|
|
|
// Write writes the string to the file.
|
|
func (f File) Write(s string) error {
|
|
if f == "" {
|
|
return xerrors.Errorf("empty file path")
|
|
}
|
|
return write(string(f), 0o600, []byte(s))
|
|
}
|
|
|
|
// Read reads the file to a string.
|
|
func (f File) Read() (string, error) {
|
|
if f == "" {
|
|
return "", xerrors.Errorf("empty file path")
|
|
}
|
|
byt, err := read(string(f))
|
|
return string(byt), err
|
|
}
|
|
|
|
// open opens a file in the configuration directory,
|
|
// creating all intermediate directories.
|
|
func open(path string, flag int, mode os.FileMode) (*os.File, error) {
|
|
err := os.MkdirAll(filepath.Dir(path), 0o750)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return os.OpenFile(path, flag, mode)
|
|
}
|
|
|
|
func write(path string, mode os.FileMode, dat []byte) error {
|
|
fi, err := open(path, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, mode)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer fi.Close()
|
|
_, err = fi.Write(dat)
|
|
return err
|
|
}
|
|
|
|
func read(path string) ([]byte, error) {
|
|
fi, err := open(path, os.O_RDONLY, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer fi.Close()
|
|
return io.ReadAll(fi)
|
|
}
|
|
|
|
func DefaultDir() string {
|
|
return configdir.LocalConfig("coderv2")
|
|
}
|