2021-07-19 04:58:41 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2021-07-19 04:59:02 +00:00
|
|
|
"fmt"
|
2021-07-19 04:58:41 +00:00
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
2021-08-02 05:09:26 +00:00
|
|
|
"strings"
|
|
|
|
|
|
|
|
"github.com/BurntSushi/toml"
|
2021-07-19 04:58:41 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type Config struct {
|
2021-07-26 02:41:31 +00:00
|
|
|
Port int
|
|
|
|
Hostname string
|
|
|
|
RootDir string
|
|
|
|
UserDirEnable bool
|
|
|
|
UserDir string
|
|
|
|
DirlistEnable bool
|
|
|
|
DirlistReverse bool
|
|
|
|
DirlistSort string
|
|
|
|
DirlistTitles bool
|
2021-08-02 04:45:48 +00:00
|
|
|
CGIPaths []string
|
2021-08-02 12:28:52 +00:00
|
|
|
UserCGIEnable bool
|
2021-07-19 04:58:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
var defaultConf = &Config{
|
2021-07-26 02:41:31 +00:00
|
|
|
Port: 300,
|
|
|
|
Hostname: "localhost",
|
|
|
|
RootDir: "/var/spartan/",
|
|
|
|
DirlistEnable: true,
|
|
|
|
DirlistReverse: false,
|
|
|
|
DirlistSort: "name",
|
|
|
|
DirlistTitles: true,
|
|
|
|
UserDirEnable: true,
|
|
|
|
UserDir: "public_spartan",
|
2021-08-02 04:45:48 +00:00
|
|
|
CGIPaths: []string{"cgi/"},
|
2021-08-02 12:28:52 +00:00
|
|
|
UserCGIEnable: false, // Turned off by default because scripts are run by server user as of now
|
2021-07-19 04:58:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func LoadConfig(path string) (*Config, error) {
|
|
|
|
var err error
|
|
|
|
var conf Config
|
|
|
|
// Defaults
|
|
|
|
conf = *defaultConf
|
|
|
|
|
|
|
|
_, err = os.Stat(path)
|
|
|
|
if os.IsNotExist(err) {
|
|
|
|
fmt.Println(path, "does not exist, using default configuration values")
|
|
|
|
return &conf, nil
|
|
|
|
}
|
|
|
|
f, err := os.Open(path)
|
|
|
|
if err == nil {
|
|
|
|
defer f.Close()
|
|
|
|
contents, err := ioutil.ReadAll(f)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if _, err = toml.Decode(string(contents), &conf); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-08-02 05:09:26 +00:00
|
|
|
// Config validation
|
2021-07-19 04:58:41 +00:00
|
|
|
if conf.DirlistSort != "name" && conf.DirlistSort != "time" && conf.DirlistSort != "size" {
|
|
|
|
fmt.Println("Warning: DirlistSort config option is not one of name/time/size, defaulting to name.")
|
|
|
|
conf.DirlistSort = "name"
|
|
|
|
}
|
2021-08-02 05:09:26 +00:00
|
|
|
// Strip trailing '/' so /~user to /~user/ redirects can work
|
|
|
|
conf.UserDir = strings.TrimRight(conf.UserDir, "/")
|
2021-07-19 04:58:41 +00:00
|
|
|
|
|
|
|
return &conf, nil
|
|
|
|
}
|