mirror of
https://github.com/nicksherron/bashhub-server.git
synced 2025-01-01 04:41:45 +08:00
128 lines
3.3 KiB
Go
128 lines
3.3 KiB
Go
/*
|
|
*
|
|
* Copyright © 2020 nicksherron <nsherron90@gmail.com>
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*/
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/nicksherron/bashhub-server/internal"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var cfgFile string
|
|
|
|
// rootCmd represents the base command when called without any subcommands
|
|
var (
|
|
noWarning bool
|
|
rootCmd = &cobra.Command{
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
cmd.Flags().Parse(args)
|
|
if !noWarning {
|
|
checkBhEnv()
|
|
}
|
|
startupMessage()
|
|
internal.Run()
|
|
},
|
|
}
|
|
)
|
|
|
|
// Execute runs root command
|
|
func Execute() {
|
|
if err := rootCmd.Execute(); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
cobra.OnInitialize()
|
|
rootCmd.PersistentFlags().StringVar(&internal.LogFile, "log", "", `Set filepath for HTTP log. "" logs to stderr`)
|
|
rootCmd.PersistentFlags().StringVar(&internal.DbPath, "db", dbPath(), "DB location (sqlite or postgres)")
|
|
rootCmd.PersistentFlags().StringVarP(&internal.Addr, "addr", "a", listenAddr(), "Ip and port to listen and serve on")
|
|
rootCmd.PersistentFlags().BoolVarP(&noWarning, "warning-disable", "w", false, `Disable BH_URL env variable startup warning`)
|
|
|
|
}
|
|
|
|
// StartupMessage prints startup banner
|
|
func startupMessage() {
|
|
banner := fmt.Sprintf(`
|
|
_ _ _ _
|
|
| | | | | | | | version: %v
|
|
| |__ __ _ ___| |__ | |__ _ _| | address: %v
|
|
| '_ \ / _' / __| '_ \| '_ \| | | | '_ \
|
|
| |_) | (_| \__ \ | | | | | | |_| | |_) |
|
|
|_.__/ \__,_|___/_| |_|_| |_|\__,_|_.__/
|
|
___ ___ _ ____ _____ _ __
|
|
/ __|/ _ \ '__\ \ / / _ \ '__|
|
|
\__ \ __/ | \ V / __/ |
|
|
|___/\___|_| \_/ \___|_|
|
|
|
|
`, Version, internal.Addr)
|
|
color.HiGreen(banner)
|
|
fmt.Print("\n")
|
|
log.Printf("Listening and serving HTTP on %v", internal.Addr)
|
|
fmt.Print("\n")
|
|
}
|
|
|
|
func listenAddr() string {
|
|
var a string
|
|
if os.Getenv("BH_SERVER_URL") != "" {
|
|
a = os.Getenv("BH_SERVER_URL")
|
|
return a
|
|
}
|
|
a = "http://0.0.0.0:8080"
|
|
return a
|
|
|
|
}
|
|
|
|
func dbPath() string {
|
|
dbFile := "data.db"
|
|
f := filepath.Join(appDir(), dbFile)
|
|
return f
|
|
}
|
|
|
|
func appDir() string {
|
|
cfgDir, err := os.UserConfigDir()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
ch := filepath.Join(cfgDir, "bashhub-server")
|
|
err = os.MkdirAll(ch, 0755)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
return ch
|
|
}
|
|
func checkBhEnv() {
|
|
bhURL := os.Getenv("BH_URL")
|
|
if strings.Contains(bhURL, "https://bashhub.com") {
|
|
msg := fmt.Sprintf(`
|
|
WARNING: BH_URL is set to https://bashhub.com on this machine
|
|
If you will be running bashhub-client locally be sure to add
|
|
export BH_URL=%v to your .bashrc or .zshrc`, internal.Addr)
|
|
fmt.Println(msg, "\n")
|
|
}
|
|
}
|