46 lines
865 B
Go
46 lines
865 B
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
|
|
"github.com/asdine/storm"
|
|
"github.com/filebrowser/filebrowser/bolt"
|
|
"github.com/filebrowser/filebrowser/lib"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func checkErr(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func mustGetString(cmd *cobra.Command, flag string) string {
|
|
s, err := cmd.Flags().GetString(flag)
|
|
checkErr(err)
|
|
return s
|
|
}
|
|
|
|
func mustGetBool(cmd *cobra.Command, flag string) bool {
|
|
b, err := cmd.Flags().GetBool(flag)
|
|
checkErr(err)
|
|
return b
|
|
}
|
|
|
|
func getDB() *storm.DB {
|
|
if _, err := os.Stat(databasePath); err != nil {
|
|
panic(errors.New(databasePath + " does not exist. Please run 'filebrowser init' first."))
|
|
}
|
|
|
|
db, err := storm.Open(databasePath)
|
|
checkErr(err)
|
|
return db
|
|
}
|
|
|
|
func getFileBrowser(db *storm.DB) *lib.FileBrowser {
|
|
fb, err := lib.NewFileBrowser(&bolt.Backend{DB: db})
|
|
checkErr(err)
|
|
return fb
|
|
}
|