chore: style and gofmt (#622)

Former-commit-id: 669963fd1d51d7e2d32520ffb0c1ecd5e4224c78 [formerly 8ef320b07699c842ae0325e4665d570fa4399481] [formerly d171df23332dff1d33f00a7407f5dc3f6dde1ef7 [formerly d0fd97d943]]
Former-commit-id: dc9a921b032e1acee0845186e7a009badbe768c5 [formerly 0c8a0f739bd1d984ca0db2b0623da5fdb813f4e0]
Former-commit-id: 3005cd15261952387fde1adf13770e2c1ee52749
This commit is contained in:
1138-4EB 2019-01-08 19:40:14 +01:00 committed by GitHub
parent c50a9b29bd
commit 9a54abfe62
6 changed files with 47 additions and 32 deletions

View File

@ -2,7 +2,6 @@ package cmd
import ( import (
"crypto/tls" "crypto/tls"
"errors"
"io/ioutil" "io/ioutil"
"log" "log"
"net" "net"
@ -92,8 +91,8 @@ func mustGetStringViperFlag(flags *pflag.FlagSet, key string) string {
var rootCmd = &cobra.Command{ var rootCmd = &cobra.Command{
Use: "filebrowser", Use: "filebrowser",
Short: "A stylish web-based file browser",
Version: version.Version, Version: version.Version,
Short: "A stylish web-based file browser",
Long: `File Browser CLI lets you create the database to use with File Browser, Long: `File Browser CLI lets you create the database to use with File Browser,
manage your users and all the configurations without acessing the manage your users and all the configurations without acessing the
web interface. web interface.
@ -130,6 +129,8 @@ Also, if the database path doesn't exist, File Browser will enter into
the quick setup mode and a new database will be bootstraped and a new the quick setup mode and a new database will be bootstraped and a new
user created with the credentials from options "username" and "password".`, user created with the credentials from options "username" and "password".`,
Run: python(func(cmd *cobra.Command, args []string, d pythonData) { Run: python(func(cmd *cobra.Command, args []string, d pythonData) {
log.Println(cfgFile)
if !d.hadDB { if !d.hadDB {
quickSetup(cmd.Flags(), d) quickSetup(cmd.Flags(), d)
} }
@ -141,22 +142,23 @@ user created with the credentials from options "username" and "password".`,
checkErr(err) checkErr(err)
server.Root = root server.Root = root
handler, err := fbhttp.NewHandler(d.store, server) adr := server.Address + ":" + server.Port
checkErr(err)
var listener net.Listener var listener net.Listener
if server.TLSKey != "" && server.TLSCert != "" { if server.TLSKey != "" && server.TLSCert != "" {
cer, err := tls.LoadX509KeyPair(server.TLSCert, server.TLSKey) cer, err := tls.LoadX509KeyPair(server.TLSCert, server.TLSKey)
checkErr(err) checkErr(err)
config := &tls.Config{Certificates: []tls.Certificate{cer}} listener, err = tls.Listen("tcp", adr, &tls.Config{Certificates: []tls.Certificate{cer}})
listener, err = tls.Listen("tcp", server.Address+":"+server.Port, config)
checkErr(err) checkErr(err)
} else { } else {
listener, err = net.Listen("tcp", server.Address+":"+server.Port) listener, err = net.Listen("tcp", adr)
checkErr(err) checkErr(err)
} }
handler, err := fbhttp.NewHandler(d.store, server)
checkErr(err)
log.Println("Listening on", listener.Addr().String()) log.Println("Listening on", listener.Addr().String())
if err := http.Serve(listener, handler); err != nil { if err := http.Serve(listener, handler); err != nil {
log.Fatal(err) log.Fatal(err)
@ -267,7 +269,7 @@ func quickSetup(flags *pflag.FlagSet, d pythonData) {
} }
if username == "" || password == "" { if username == "" || password == "" {
checkErr(errors.New("username and password cannot be empty during quick setup")) log.Fatal("username and password cannot be empty during quick setup")
} }
user := &users.User{ user := &users.User{
@ -303,7 +305,9 @@ func initConfig() {
if _, ok := err.(v.ConfigParseError); ok { if _, ok := err.(v.ConfigParseError); ok {
panic(err) panic(err)
} }
// TODO: log.Println("No config file provided") cfgFile = "No config file used"
} else {
cfgFile = "Using config file: " + v.ConfigFileUsed()
} }
// else TODO: log.Println("Using config file:", v.ConfigFileUsed())
} }

View File

@ -43,14 +43,14 @@ func runRules(st *storage.Storage, cmd *cobra.Command, users func(*users.User),
return return
} }
settings, err := st.Settings.Get() s, err := st.Settings.Get()
checkErr(err) checkErr(err)
if global != nil { if global != nil {
global(settings) global(s)
} }
printRules(settings.Rules, id) printRules(s.Rules, id)
} }
func getUserIdentifier(flags *pflag.FlagSet) interface{} { func getUserIdentifier(flags *pflag.FlagSet) interface{} {

View File

@ -26,8 +26,10 @@ options you want to change.`,
password := mustGetString(flags, "password") password := mustGetString(flags, "password")
newUsername := mustGetString(flags, "username") newUsername := mustGetString(flags, "username")
var err error var (
var user *users.User err error
user *users.User
)
if id != 0 { if id != 0 {
user, err = d.store.Users.Get("", id) user, err = d.store.Users.Get("", id)

View File

@ -7,7 +7,7 @@ import (
"strings" "strings"
"time" "time"
"github.com/dgrijalva/jwt-go" jwt "github.com/dgrijalva/jwt-go"
"github.com/dgrijalva/jwt-go/request" "github.com/dgrijalva/jwt-go/request"
"github.com/filebrowser/filebrowser/v2/errors" "github.com/filebrowser/filebrowser/v2/errors"
"github.com/filebrowser/filebrowser/v2/users" "github.com/filebrowser/filebrowser/v2/users"

View File

@ -9,7 +9,7 @@ import (
"strings" "strings"
"text/template" "text/template"
"github.com/GeertJohan/go.rice" rice "github.com/GeertJohan/go.rice"
"github.com/filebrowser/filebrowser/v2/auth" "github.com/filebrowser/filebrowser/v2/auth"
"github.com/filebrowser/filebrowser/v2/settings" "github.com/filebrowser/filebrowser/v2/settings"
"github.com/filebrowser/filebrowser/v2/storage" "github.com/filebrowser/filebrowser/v2/storage"

View File

@ -13,8 +13,8 @@ import (
"github.com/asdine/storm" "github.com/asdine/storm"
"github.com/filebrowser/filebrowser/v2/settings" "github.com/filebrowser/filebrowser/v2/settings"
"github.com/filebrowser/filebrowser/v2/storage" "github.com/filebrowser/filebrowser/v2/storage"
"github.com/pelletier/go-toml" toml "github.com/pelletier/go-toml"
"gopkg.in/yaml.v2" yaml "gopkg.in/yaml.v2"
) )
type oldDefs struct { type oldDefs struct {
@ -33,7 +33,7 @@ type oldAuth struct {
} }
type oldConf struct { type oldConf struct {
Port string `json:"port" yaml:"port" toml:"port"` Port string `json:"port" yaml:"port" toml:"port"`
BaseURL string `json:"baseURL" yaml:"baseURL" toml:"baseURL"` BaseURL string `json:"baseURL" yaml:"baseURL" toml:"baseURL"`
Log string `json:"log" yaml:"log" toml:"log"` Log string `json:"log" yaml:"log" toml:"log"`
Address string `json:"address" yaml:"address" toml:"address"` Address string `json:"address" yaml:"address" toml:"address"`
@ -62,14 +62,14 @@ var defaults = &oldConf{
}, },
} }
func importConf(db *storm.DB, path string, sto *storage.Storage) error { func readConf(path string) (*oldConf, error) {
cfg := &oldConf{} cfg := &oldConf{}
if path != "" { if path != "" {
ext := filepath.Ext(path) ext := filepath.Ext(path)
fd, err := os.Open(path) fd, err := os.Open(path)
if err != nil { if err != nil {
return err return nil, err
} }
defer fd.Close() defer fd.Close()
@ -81,23 +81,32 @@ func importConf(db *storm.DB, path string, sto *storage.Storage) error {
case ".yaml", ".yml": case ".yaml", ".yml":
err = yaml.NewDecoder(fd).Decode(cfg) err = yaml.NewDecoder(fd).Decode(cfg)
default: default:
return errors.New("unsupported config extension " + ext) return nil, errors.New("unsupported config extension " + ext)
} }
if err != nil { if err != nil {
return err return nil, err
} }
} else { } else {
cfg = defaults cfg = defaults
path, err := filepath.Abs(".") path, err := filepath.Abs(".")
if err != nil { if err != nil {
return err return nil, err
} }
cfg.Defaults.Scope = path cfg.Defaults.Scope = path
} }
return cfg, nil
}
func importConf(db *storm.DB, path string, sto *storage.Storage) error {
cfg, err := readConf(path)
if err != nil {
return err
}
commands := map[string][]string{} commands := map[string][]string{}
err := db.Get("config", "commands", &commands) err = db.Get("config", "commands", &commands)
if err != nil { if err != nil {
return err return err
} }
@ -109,8 +118,8 @@ func importConf(db *storm.DB, path string, sto *storage.Storage) error {
} }
s := &settings.Settings{ s := &settings.Settings{
Key: key, Key: key,
Signup: false, Signup: false,
Defaults: settings.UserDefaults{ Defaults: settings.UserDefaults{
Scope: cfg.Defaults.Scope, Scope: cfg.Defaults.Scope,
Commands: cfg.Defaults.Commands, Commands: cfg.Defaults.Commands,
@ -130,10 +139,10 @@ func importConf(db *storm.DB, path string, sto *storage.Storage) error {
} }
server := &settings.Server{ server := &settings.Server{
BaseURL : cfg.BaseURL, BaseURL: cfg.BaseURL,
Port : cfg.Port, Port: cfg.Port,
Address : cfg.Address, Address: cfg.Address,
Log : cfg.Log, Log: cfg.Log,
} }
var auther auth.Auther var auther auth.Auther