mirror/cowyo
1
0
Fork 0
mirror of https://github.com/schollz/cowyo.git synced 2023-08-10 21:13:00 +03:00
Code Issues Projects Releases Wiki Activity

Added debug mode, Fix #55

Browse Source
This commit is contained in:
Zack Scholl 2016-09-09 15:39:45 -04:00
parent c10c7b16e2
commit f6ee202630
3 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
main.go
View File

@ -38,6 +38,7 @@ var RuntimeArgs struct {
ForceWss bool
DumpDataset string
RestoreDataset string
Debug bool
}
var VersionNum string
@ -56,6 +57,7 @@ func main() {
flag.StringVar(&RuntimeArgs.ServerKey, "key", "", "location of SSL key")
flag.StringVar(&RuntimeArgs.WikiName, "w", "cowyo", "custom name for wiki")
flag.BoolVar(&RuntimeArgs.ForceWss, "e", false, "force encrypted sockets (use if using Caddy auto HTTPS)")
flag.BoolVar(&RuntimeArgs.Debug, "d", false, "debugging mode")
flag.StringVar(&RuntimeArgs.DumpDataset, "dump", "", "directory to dump all data to")
flag.StringVar(&RuntimeArgs.RestoreDataset, "restore", "", "directory to restore all data from")
flag.CommandLine.Usage = func() {
@ -76,6 +78,13 @@ Options:`)
}
flag.Parse()
// Set the log level
if RuntimeArgs.Debug == false {
logger.Level(2)
} else {
logger.Level(0)
}
if len(RuntimeArgs.DumpDataset) > 0 {
fmt.Println("Dumping data to '" + RuntimeArgs.DumpDataset + "' folder...")
dumpEverything(RuntimeArgs.DumpDataset)
@ -84,7 +93,9 @@ Options:`)
RuntimeArgs.ExternalIP = flag.Arg(0)
if RuntimeArgs.ExternalIP == "" {
logger.Debug("Getting external ip...")
RuntimeArgs.ExternalIP = GetLocalIP() + RuntimeArgs.Port
logger.Debug("Using ip: %s and port %s", GetLocalIP(), RuntimeArgs.Port)
}
RuntimeArgs.SourcePath = cwd

6
routes.go
View File

@ -235,9 +235,6 @@ func getRecentlyEdited(title string, c *gin.Context) []string {
recentlyEdited = title
} else {
editedThings = strings.Split(v.(string), "|||")
fmt.Println(editedThings)
fmt.Println(v.(string))
fmt.Println(title)
if !stringInSlice(title, editedThings) {
recentlyEdited = v.(string) + "|||" + title
} else {
@ -618,7 +615,7 @@ func dumpEverything(folderpath string) {
defer Close()
err := os.MkdirAll(folderpath, 0777)
if err != nil {
fmt.Println("Already exists")
fmt.Printf("%s folder already exists.", folderpath)
}
db.View(func(tx *bolt.Tx) error {
// Assume bucket exists and has keys
@ -627,7 +624,6 @@ func dumpEverything(folderpath string) {
for k, _ := c.First(); k != nil; k, _ = c.Next() {
var p WikiData
p.load(string(k))
fmt.Println(string(k), len(p.CurrentText))
if len(p.CurrentText) > 0 {
ioutil.WriteFile(path.Join(folderpath, string(k)), []byte(p.CurrentText), 0644)
}

6
utils.go
View File

@ -11,6 +11,7 @@ import (
"strings"
"time"
"github.com/jcelliott/lumber"
"github.com/sergi/go-diff/diffmatchpatch"
)
@ -23,12 +24,15 @@ type versionsInfo struct {
VersionNum int
}
var logger *lumber.ConsoleLogger
func init() {
rand.Seed(time.Now().Unix())
animalsText, _ := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "static/text/animals"))
animals = strings.Split(string(animalsText), ",")
adjectivesText, _ := ioutil.ReadFile(path.Join(RuntimeArgs.SourcePath, "static/text/adjectives"))
adjectives = strings.Split(string(adjectivesText), "\n")
logger = lumber.NewConsoleLogger(lumber.INFO)
}
func randomAnimal() string {
@ -225,7 +229,7 @@ func GetLocalIP() string {
for _, address := range addrs {
// check the address type and if it is not a loopback the display it
if ipnet, ok := address.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
if ipnet.IP.To4() != nil && (strings.Contains(ipnet.IP.String(), "192.168.1") || strings.Contains(ipnet.IP.String(), "192.168")) {
if ipnet.IP.To4() != nil {
return ipnet.IP.String()
}
}