filebrowser/cmd/filemanager/main.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
}