135 lines
3.2 KiB
Go
135 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
lumberjack "gopkg.in/natefinch/lumberjack.v2"
|
|
|
|
"github.com/hacdias/filemanager"
|
|
"github.com/hacdias/fileutils"
|
|
)
|
|
|
|
// confFile contains the configuration file for this File Manager instance.
|
|
// If the user chooses to use a configuration file, the flags will be ignored.
|
|
type confFile struct {
|
|
Database string `json:"database"`
|
|
Scope string `json:"scope"`
|
|
Address string `json:"address"`
|
|
Commands []string `json:"commands"`
|
|
Port int `json:"port"`
|
|
Logger string `json:"log"`
|
|
AllowCommands bool `json:"allowCommands"`
|
|
AllowEdit bool `json:"allowEdit"`
|
|
AllowNew bool `json:"allowNew"`
|
|
}
|
|
|
|
var (
|
|
addr string
|
|
config string
|
|
database string
|
|
scope string
|
|
commands string
|
|
port string
|
|
logfile string
|
|
allowCommands bool
|
|
allowEdit bool
|
|
allowNew bool
|
|
)
|
|
|
|
func init() {
|
|
flag.StringVar(&config, "config", "", "JSON configuration file")
|
|
flag.StringVar(&port, "port", "0", "HTTP Port (default is random)")
|
|
flag.StringVar(&addr, "address", "", "Address to listen to (default is all of them)")
|
|
flag.StringVar(&database, "database", "./filemanager.db", "Database path")
|
|
flag.StringVar(&scope, "scope", ".", "Default scope for new users")
|
|
flag.StringVar(&commands, "commands", "git svn hg", "Space separated commands available for new users")
|
|
flag.StringVar(&logfile, "log", "stdout", "Logger to log the errors.")
|
|
flag.BoolVar(&allowCommands, "allow-commands", true, "Default allow commands option")
|
|
flag.BoolVar(&allowEdit, "allow-edit", true, "Default allow edit option")
|
|
flag.BoolVar(&allowNew, "allow-new", true, "Default allow new option")
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
// Set up process log before anything bad happens
|
|
switch logfile {
|
|
case "stdout":
|
|
log.SetOutput(os.Stdout)
|
|
case "stderr":
|
|
log.SetOutput(os.Stderr)
|
|
case "":
|
|
log.SetOutput(ioutil.Discard)
|
|
default:
|
|
log.SetOutput(&lumberjack.Logger{
|
|
Filename: logfile,
|
|
MaxSize: 100,
|
|
MaxAge: 14,
|
|
MaxBackups: 10,
|
|
})
|
|
}
|
|
|
|
if config != "" {
|
|
loadConfig()
|
|
}
|
|
|
|
fm, err := filemanager.New(database, filemanager.User{
|
|
AllowCommands: allowCommands,
|
|
AllowEdit: allowEdit,
|
|
AllowNew: allowNew,
|
|
Commands: strings.Split(strings.TrimSpace(commands), " "),
|
|
Rules: []*filemanager.Rule{},
|
|
CSS: "",
|
|
FileSystem: fileutils.Dir(scope),
|
|
})
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fm.SetBaseURL("/")
|
|
fm.SetPrefixURL("/")
|
|
|
|
listener, err := net.Listen("tcp", addr+":"+port)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println("Listening on", listener.Addr().String())
|
|
if err := http.Serve(listener, fm); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func loadConfig() {
|
|
file, err := ioutil.ReadFile(config)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var conf *confFile
|
|
err = json.Unmarshal(file, &conf)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
database = conf.Database
|
|
scope = conf.Scope
|
|
addr = conf.Address
|
|
logfile = conf.Logger
|
|
commands = strings.Join(conf.Commands, " ")
|
|
port = strconv.Itoa(conf.Port)
|
|
allowNew = conf.AllowNew
|
|
allowEdit = conf.AllowEdit
|
|
allowCommands = conf.AllowCommands
|
|
}
|